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

#include "vtkPVXMLElement.h"
#include "vtkSMMessage.h"
#include <cassert>

vtkStandardNewMacro(vtkSIDoubleMapProperty);
//----------------------------------------------------------------------------
vtkSIDoubleMapProperty::vtkSIDoubleMapProperty()
{
  this->CleanCommand = nullptr;
}

//----------------------------------------------------------------------------
vtkSIDoubleMapProperty::~vtkSIDoubleMapProperty()
{
  this->SetCleanCommand(nullptr);
}

//----------------------------------------------------------------------------
bool vtkSIDoubleMapProperty::Push(vtkSMMessage* message, int offset)
{
  if (this->InformationOnly || !this->Command)
  {
    return true;
  }

  vtkClientServerStream stream;
  vtkObjectBase* object = this->GetVTKObject();

  if (this->CleanCommand)
  {
    stream << vtkClientServerStream::Invoke << object << this->CleanCommand
           << vtkClientServerStream::End;
  }

  assert(message->ExtensionSize(ProxyState::property) > offset);

  const ProxyState_Property* prop = &message->GetExtension(ProxyState::property, offset);
  assert(strcmp(prop->name().c_str(), this->GetXMLName()) == 0);

  // Save to cache when pulled for collaboration
  this->SaveValueToCache(message, offset);

  const Variant* variant = &prop->value();

  std::vector<double> values(variant->float64_size());
  for (size_t i = 0; i < values.size(); i++)
  {
    values[i] = variant->float64(static_cast<int>(i));
  }

  int size = variant->idtype_size();
  for (int i = 0; i < size; i++)
  {
    vtkIdType id = variant->idtype(i);

    stream << vtkClientServerStream::Invoke << object << this->Command << id
           << vtkClientServerStream::InsertArray(
                &values[i * this->NumberOfComponents], this->NumberOfComponents)
           << vtkClientServerStream::End;
  }

  return this->ProcessMessage(stream);
}

//---------------------------------------------------------------------------
bool vtkSIDoubleMapProperty::ReadXMLAttributes(vtkSIProxy* parent, vtkPVXMLElement* element)
{
  if (!this->Superclass::ReadXMLAttributes(parent, element))
  {
    return false;
  }

  int number_of_components = 0;
  if (element->GetScalarAttribute("number_of_components", &number_of_components))
  {
    this->NumberOfComponents = number_of_components;
  }

  this->SetCleanCommand(element->GetAttribute("clean_command"));

  return true;
}

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