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

#include "vtkObjectFactory.h"
#include "vtkSMDoubleVectorProperty.h"
#include "vtkSMProxyProperty.h"
#include "vtkSMSourceProxy.h"
#include "vtkSMTimeKeeper.h"
#include "vtkSMTrace.h"

#include <algorithm>

vtkStandardNewMacro(vtkSMTimeKeeperProxy);
//----------------------------------------------------------------------------
vtkSMTimeKeeperProxy::vtkSMTimeKeeperProxy() = default;

//----------------------------------------------------------------------------
vtkSMTimeKeeperProxy::~vtkSMTimeKeeperProxy() = default;

//----------------------------------------------------------------------------
void vtkSMTimeKeeperProxy::CreateVTKObjects()
{
  if (this->ObjectsCreated)
  {
    return;
  }
  this->Superclass::CreateVTKObjects();
  if (this->ObjectsCreated)
  {
    vtkSMTimeKeeper* tk = vtkSMTimeKeeper::SafeDownCast(this->GetClientSideObject());
    if (tk)
    {
      tk->SetTimestepValuesProperty(this->GetProperty("TimestepValues"));
      tk->SetTimeRangeProperty(this->GetProperty("TimeRange"));
      tk->SetTimeLabelProperty(this->GetProperty("TimeLabel"));
    }
  }
}

//----------------------------------------------------------------------------
bool vtkSMTimeKeeperProxy::AddTimeSource(vtkSMProxy* proxy, bool suppress_input)
{
  vtkSMProxyProperty* pp = vtkSMProxyProperty::SafeDownCast(this->GetProperty("TimeSources"));
  if (!proxy || pp->IsProxyAdded(proxy))
  {
    return false;
  }

  // Suppress produces for proxy
  vtkSMProxyProperty* stspp =
    vtkSMProxyProperty::SafeDownCast(this->GetProperty("SuppressedTimeSources"));
  for (unsigned int cc = 0, max = proxy->GetNumberOfProducers(); suppress_input && cc < max; cc++)
  {
    vtkSMProxy* producer = proxy->GetProducerProxy(cc);
    producer = producer ? vtkSMSourceProxy::SafeDownCast(producer->GetTrueParentProxy()) : nullptr;
    if (producer && !stspp->IsProxyAdded(producer))
    {
      stspp->AddProxy(producer);
    }
  }
  // add the requested proxy.
  pp->AddProxy(proxy);
  this->UpdateVTKObjects();
  return true;
}

//----------------------------------------------------------------------------
bool vtkSMTimeKeeperProxy::RemoveTimeSource(vtkSMProxy* proxy, bool unsuppress_input)
{
  vtkSMProxyProperty* pp = vtkSMProxyProperty::SafeDownCast(this->GetProperty("TimeSources"));
  if (!proxy || !pp->IsProxyAdded(proxy))
  {
    return false;
  }

  // Remove producer suppressions.
  vtkSMProxyProperty* stspp =
    vtkSMProxyProperty::SafeDownCast(this->GetProperty("SuppressedTimeSources"));
  for (unsigned int cc = 0, max = proxy->GetNumberOfProducers(); unsuppress_input && cc < max; cc++)
  {
    vtkSMProxy* producer = proxy->GetProducerProxy(cc);
    producer = producer ? vtkSMSourceProxy::SafeDownCast(producer->GetTrueParentProxy()) : nullptr;
    if (producer)
    {
      stspp->RemoveProxy(producer);
    }
  }

  // remove the requested proxy.
  pp->RemoveProxy(proxy);
  this->UpdateVTKObjects();
  return true;
}

//----------------------------------------------------------------------------
bool vtkSMTimeKeeperProxy::IsTimeSourceTracked(vtkSMProxy* proxy)
{
  vtkSMProxyProperty* pp = vtkSMProxyProperty::SafeDownCast(this->GetProperty("TimeSources"));
  vtkSMProxyProperty* stspp =
    vtkSMProxyProperty::SafeDownCast(this->GetProperty("SuppressedTimeSources"));
  return (proxy && pp->IsProxyAdded(proxy) && !stspp->IsProxyAdded(proxy));
}

//----------------------------------------------------------------------------
bool vtkSMTimeKeeperProxy::SetSuppressTimeSource(vtkSMProxy* proxy, bool suppress)
{
  proxy = vtkSMSourceProxy::SafeDownCast(proxy);
  if (proxy)
  {
    SM_SCOPED_TRACE(CallMethod)
      .arg(this)
      .arg("SetSuppressTimeSource")
      .arg(proxy)
      .arg(suppress)
      .arg("comment", (suppress ? "ignore time" : "don't ignore time"));

    vtkSMProxyProperty* pp =
      vtkSMProxyProperty::SafeDownCast(this->GetProperty("SuppressedTimeSources"));
    if (suppress && pp->IsProxyAdded(proxy) == false)
    {
      pp->AddProxy(proxy);
    }
    else if (!suppress && pp->IsProxyAdded(proxy))
    {
      pp->RemoveProxy(proxy);
    }
    this->UpdateVTKObjects();
    return true;
  }
  return false;
}

//----------------------------------------------------------------------------
double vtkSMTimeKeeperProxy::GetLowerBoundTimeStep(double value)
{
  vtkSMDoubleVectorProperty* dvp =
    vtkSMDoubleVectorProperty::SafeDownCast(this->GetProperty("TimestepValues"));
  if (dvp && dvp->GetNumberOfElements() > 0)
  {
    // we shirk the range so that if value > the last timestep, this simply
    // returns the last timestep's time.
    return *(std::lower_bound(
      dvp->GetElements(), dvp->GetElements() + dvp->GetNumberOfElements() - 1, value));
  }
  return value;
}

//----------------------------------------------------------------------------
int vtkSMTimeKeeperProxy::GetLowerBoundTimeStepIndex(double value)
{
  vtkSMDoubleVectorProperty* dvp =
    vtkSMDoubleVectorProperty::SafeDownCast(this->GetProperty("TimestepValues"));
  if (dvp && dvp->GetNumberOfElements() > 0)
  {
    // we shirk the range so that if value > the last timestep, this simply
    // returns the last timestep's index.
    const double* first = dvp->GetElements();
    const double* last = dvp->GetElements() + dvp->GetNumberOfElements() - 1;
    const double* iter = std::lower_bound(first, last, value);
    return static_cast<int>(iter - first);
  }
  return 0;
}

//----------------------------------------------------------------------------
void vtkSMTimeKeeperProxy::UpdateTimeInformation()
{
  if (vtkSMTimeKeeper* tk = vtkSMTimeKeeper::SafeDownCast(this->GetClientSideObject()))
  {
    tk->UpdateTimeInformation();
  }
}

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