// SPDX-FileCopyrightText: Copyright (c) Kitware Inc.
// SPDX-License-Identifier: BSD-3-Clause
#include "vtkSMDeserializerProtobuf.h"

#include "vtkObjectFactory.h"
#include "vtkPVXMLElement.h"
#include "vtkSMMessage.h"
#include "vtkSMProxy.h"
#include "vtkSMProxyLocator.h"
#include "vtkSMProxyManager.h"
#include "vtkSMSession.h"
#include "vtkSMStateLocator.h"
#include "vtkSmartPointer.h"

#define BEFORE_LOAD(proxy)                                                                         \
  if (session && session->IsProcessingRemoteNotification())                                        \
  {                                                                                                \
    (proxy)->EnableLocalPushOnly();                                                                \
  }

#define AFTER_LOAD(proxy)                                                                          \
  if (session && session->IsProcessingRemoteNotification())                                        \
  {                                                                                                \
    (proxy)->DisableLocalPushOnly();                                                               \
  }

//----------------------------------------------------------------------------
vtkStandardNewMacro(vtkSMDeserializerProtobuf);
vtkCxxSetObjectMacro(vtkSMDeserializerProtobuf, StateLocator, vtkSMStateLocator);
//----------------------------------------------------------------------------
vtkSMDeserializerProtobuf::vtkSMDeserializerProtobuf()
{
  this->StateLocator = nullptr;
}

//----------------------------------------------------------------------------
vtkSMDeserializerProtobuf::~vtkSMDeserializerProtobuf()
{
  this->SetStateLocator(nullptr);
}

//----------------------------------------------------------------------------
vtkSMProxy* vtkSMDeserializerProtobuf::NewProxy(vtkTypeUInt32 id, vtkSMProxyLocator* locator)
{
  vtkSMSession* session = this->GetSession();
  // Make sure that the requested proxy does not already exist
  assert("SMDeserializer should not create a proxy if that proxy exist" &&
    (session == nullptr || session->GetRemoteObject(id) == nullptr));

  vtkSMMessage msg;
  // First extract state for the requested proxy
  // Did not find the proxy, start the creation procedure
  if (!this->StateLocator || !this->StateLocator->FindState(id, &msg))
  {
    return nullptr;
  }

  // ** isn't this unused code especially after the assert(..) above? Commenting
  // out as a result. ***

  //// Try to NOT create a new proxy if already exist and just load the state
  // vtkSMProxy* proxy =
  //    vtkSMProxy::SafeDownCast(this->Session->GetRemoteObject(id));
  // if(proxy)
  //  {
  //  if(this->StateLocator->FindState(id, &msg))
  //    {
  //    BEFORE_LOAD(proxy)
  //    proxy->LoadState(&msg, locator);
  //    proxy->UpdateVTKObjects();
  //    AFTER_LOAD(proxy)
  //    }
  //  return proxy;
  //  }

  const char* group = msg.GetExtension(ProxyState::xml_group).c_str();
  const char* type = msg.GetExtension(ProxyState::xml_name).c_str();
  const char* subProxyName = (msg.HasExtension(ProxyState::xml_sub_proxy_name))
    ? msg.GetExtension(ProxyState::xml_sub_proxy_name).c_str()
    : nullptr;

  if (!type)
  {
    vtkErrorMacro("Could not create proxy from element, missing 'type'.");
    return nullptr;
  }

  // Create Proxy based on its XML definition
  vtkSMProxy* proxy = this->CreateProxy(group, type, subProxyName);
  if (!proxy)
  {
    vtkErrorMacro("Could not create a proxy of group: "
      << (group ? group : "(null)") << " type: " << type
      << " subProxyName: " << (subProxyName ? subProxyName : "(null)"));
    return nullptr;
  }

  // Load the state of the proxy now
  BEFORE_LOAD(proxy)
  proxy->LoadState(&msg, locator);
  proxy->UpdateVTKObjects();
  AFTER_LOAD(proxy)

  return proxy;
}

//----------------------------------------------------------------------------
void vtkSMDeserializerProtobuf::PrintSelf(ostream& os, vtkIndent indent)
{
  this->Superclass::PrintSelf(os, indent);
}
