// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-License-Identifier: BSD-3-Clause
#include "vtkArrowSource.h"

#include "vtkAppendPolyData.h"
#include "vtkConeSource.h"
#include "vtkCylinderSource.h"
#include "vtkInformation.h"
#include "vtkInformationVector.h"
#include "vtkObjectFactory.h"
#include "vtkPolyData.h"
#include "vtkStreamingDemandDrivenPipeline.h"
#include "vtkTransform.h"
#include "vtkTransformFilter.h"

VTK_ABI_NAMESPACE_BEGIN
vtkStandardNewMacro(vtkArrowSource);

//------------------------------------------------------------------------------
vtkArrowSource::vtkArrowSource()
{
  this->TipResolution = 6;
  this->TipRadius = 0.1;
  this->TipLength = 0.35;
  this->ShaftResolution = 6;
  this->ShaftRadius = 0.03;
  this->Invert = false;
  this->ArrowOrigin = ArrowOrigins::Default;

  this->SetNumberOfInputPorts(0);
}

//------------------------------------------------------------------------------
int vtkArrowSource::RequestInformation(
  vtkInformation* request, vtkInformationVector** inputVector, vtkInformationVector* outputVector)
{
  outputVector->GetInformationObject(0)->Set(CAN_HANDLE_PIECE_REQUEST(), 1);
  return Superclass::RequestInformation(request, inputVector, outputVector);
}

//------------------------------------------------------------------------------
int vtkArrowSource::RequestData(vtkInformation* vtkNotUsed(request),
  vtkInformationVector** vtkNotUsed(inputVector), vtkInformationVector* outputVector)
{
  // get the info object
  vtkInformation* outInfo = outputVector->GetInformationObject(0);

  // get the output
  vtkPolyData* output = vtkPolyData::SafeDownCast(outInfo->Get(vtkDataObject::DATA_OBJECT()));

  int piece, numPieces;
  vtkCylinderSource* cyl = vtkCylinderSource::New();
  vtkTransform* trans0 = vtkTransform::New();
  vtkTransformFilter* tf0 = vtkTransformFilter::New();
  vtkConeSource* cone = vtkConeSource::New();
  vtkTransform* trans1 = vtkTransform::New();
  vtkTransform* trans2 = vtkTransform::New();
  vtkTransformFilter* tf1 = vtkTransformFilter::New();
  vtkTransformFilter* tf2 = vtkTransformFilter::New();
  vtkAppendPolyData* append = vtkAppendPolyData::New();

  piece = outInfo->Get(vtkStreamingDemandDrivenPipeline::UPDATE_PIECE_NUMBER());
  numPieces = outInfo->Get(vtkStreamingDemandDrivenPipeline::UPDATE_NUMBER_OF_PIECES());
  outInfo->Get(vtkStreamingDemandDrivenPipeline::UPDATE_NUMBER_OF_GHOST_LEVELS());

  cyl->SetResolution(this->ShaftResolution);
  cyl->SetRadius(this->ShaftRadius);
  cyl->SetHeight(1.0 - this->TipLength);
  cyl->SetCenter(0, (1.0 - this->TipLength) * 0.5, 0.0);
  cyl->CappingOn();

  trans0->RotateZ(-90.0);
  tf0->SetTransform(trans0);
  tf0->SetInputConnection(cyl->GetOutputPort());

  cone->SetResolution(this->TipResolution);
  cone->SetHeight(this->TipLength);
  cone->SetRadius(this->TipRadius);

  trans1->Translate(1.0 - this->TipLength * 0.5, 0.0, 0.0);
  tf1->SetTransform(trans1);
  tf1->SetInputConnection(cone->GetOutputPort());

  append->AddInputConnection(tf0->GetOutputPort());
  append->AddInputConnection(tf1->GetOutputPort());

  // used only when this->Invert is true.
  trans2->Translate(1, 0, 0);
  trans2->Scale(-1, 1, 1);
  tf2->SetTransform(trans2);
  tf2->SetInputConnection(append->GetOutputPort());

  // used only when this->ArrowOrigin is Center (we aim to orient and scale from the centre).
  vtkTransform* trans3 = vtkTransform::New();
  vtkTransformFilter* tf3 = vtkTransformFilter::New();
  trans3->Translate(-0.5, 0.0, 0.0);
  tf3->SetTransform(trans3);

  if (piece == 0 && numPieces > 0)
  {
    if (this->Invert)
    {
      if (this->ArrowOrigin == ArrowOrigins::Center)
      {
        tf3->SetInputConnection(tf2->GetOutputPort());
        tf3->Update();
        output->ShallowCopy(tf3->GetOutput());
      }
      else
      {
        tf2->Update();
        output->ShallowCopy(tf2->GetOutput());
      }
    }
    else
    {
      if (this->ArrowOrigin == ArrowOrigins::Center)
      {
        tf3->SetInputConnection(append->GetOutputPort());
        tf3->Update();
        output->ShallowCopy(tf3->GetOutput());
      }
      else
      {
        append->Update();
        output->ShallowCopy(append->GetOutput());
      }
    }
  }

  cone->Delete();
  trans0->Delete();
  tf0->Delete();
  cyl->Delete();
  trans1->Delete();
  tf1->Delete();
  append->Delete();
  tf2->Delete();
  trans2->Delete();
  tf3->Delete();
  trans3->Delete();

  return 1;
}

//------------------------------------------------------------------------------
std::string vtkArrowSource::GetArrowOriginAsString() const
{
  switch (this->ArrowOrigin)
  {
    case ArrowOrigins::Default:
      return "Default";
    case ArrowOrigins::Center:
      return "Center";
  }
  return "Invalid";
}

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

  os << indent << "TipResolution: " << this->TipResolution << "\n";
  os << indent << "TipRadius: " << this->TipRadius << "\n";
  os << indent << "TipLength: " << this->TipLength << "\n";

  os << indent << "ShaftResolution: " << this->ShaftResolution << "\n";
  os << indent << "ShaftRadius: " << this->ShaftRadius << "\n";

  os << indent << "Invert: " << this->Invert << "\n";
  os << indent << "Arrow Origin: " << this->GetArrowOriginAsString() << endl;
}
VTK_ABI_NAMESPACE_END
