// Created on: 2011-10-14
// Created by: Roman KOZLOV
// Copyright (c) 2011-2014 OPEN CASCADE SAS
//
// This file is part of Open CASCADE Technology software library.
//
// This library is free software; you can redistribute it and/or modify it under
// the terms of the GNU Lesser General Public License version 2.1 as published
// by the Free Software Foundation, with special exception defined in the file
// OCCT_LGPL_EXCEPTION.txt. Consult the file LICENSE_LGPL_21.txt included in OCCT
// distribution for complete text of the license and disclaimer of any warranty.
//
// Alternatively, this file may be used under the terms of Open CASCADE
// commercial license or contractual agreement.

// VIS includes
#include <IVtkTools_ShapeDataSource.hxx>
#include <IVtkOCC_ShapeMesher.hxx>
#include <IVtkTools_ShapeObject.hxx>

// prevent disabling some MSVC warning messages by VTK headers
#ifdef _MSC_VER
  #pragma warning(push)
#endif
#include <vtkObjectFactory.h>
#include <vtkCellData.h>
#include <vtkIdTypeArray.h>
#include <vtkInformation.h>
#include <vtkPoints.h>
#include <vtkPolyData.h>
#include <vtkTransform.h>
#include <vtkTransformPolyDataFilter.h>
#ifdef _MSC_VER
  #pragma warning(pop)
#endif

vtkStandardNewMacro(IVtkTools_ShapeDataSource)

  //=================================================================================================

  IVtkTools_ShapeDataSource::IVtkTools_ShapeDataSource()
    : myPolyData(new IVtkVTK_ShapeData()),
      myIsFastTransformMode(false),
      myIsTransformOnly(false)
{
  this->SetNumberOfInputPorts(0);
  this->SetNumberOfOutputPorts(1);
}

//=================================================================================================

IVtkTools_ShapeDataSource::~IVtkTools_ShapeDataSource() = default;

//=================================================================================================

void IVtkTools_ShapeDataSource::SetShape(const IVtkOCC_Shape::Handle& theOccShape)
{
  myIsTransformOnly = myIsFastTransformMode && !myOccShape.IsNull()
                      && theOccShape->GetShape().IsPartner(myOccShape->GetShape());
  myOccShape = theOccShape;
  this->Modified();
}

//=================================================================================================

int IVtkTools_ShapeDataSource::RequestData(vtkInformation*        vtkNotUsed(theRequest),
                                           vtkInformationVector** vtkNotUsed(theInputVector),
                                           vtkInformationVector*  theOutputVector)
{
  vtkSmartPointer<vtkPolyData> aPolyData = vtkPolyData::GetData(theOutputVector);
  if (aPolyData.GetPointer() == nullptr)
  {
    return 1;
  }

  aPolyData->Allocate();
  vtkSmartPointer<vtkPoints> aPts = vtkSmartPointer<vtkPoints>::New();
  aPolyData->SetPoints(aPts);

  vtkSmartPointer<vtkPolyData> aTransformedData;
  TopoDS_Shape                 aShape    = myOccShape->GetShape();
  const TopLoc_Location        aShapeLoc = aShape.Location();
  if (myIsTransformOnly)
  {
    vtkSmartPointer<vtkPolyData> aPrevData = myPolyData->getVtkPolyData();
    if (!aShapeLoc.IsIdentity())
    {
      aTransformedData = this->transform(aPrevData, aShapeLoc);
    }
    else
    {
      aTransformedData = aPrevData;
    }
  }
  else
  {
    IVtkOCC_Shape::Handle aShapeWrapperCopy = myOccShape;
    if (myIsFastTransformMode && !aShapeLoc.IsIdentity())
    {
      // Reset location before meshing
      aShape.Location(TopLoc_Location());
      aShapeWrapperCopy = new IVtkOCC_Shape(aShape);
      aShapeWrapperCopy->SetAttributes(myOccShape->Attributes());
      aShapeWrapperCopy->SetId(myOccShape->GetId());
    }

    myPolyData                          = new IVtkVTK_ShapeData();
    IVtkOCC_ShapeMesher::Handle aMesher = new IVtkOCC_ShapeMesher();
    aMesher->Build(aShapeWrapperCopy, myPolyData);
    vtkSmartPointer<vtkPolyData> aMeshData = myPolyData->getVtkPolyData();
    if (myIsFastTransformMode && !aShapeLoc.IsIdentity())
    {
      aTransformedData = this->transform(aMeshData, aShapeLoc);
    }
    else
    {
      aTransformedData = aMeshData;
    }
  }

  aPolyData->CopyStructure(aTransformedData);  // Copy points and cells
  aPolyData->CopyAttributes(aTransformedData); // Copy data arrays (sub-shapes IDs)

  // We store the OccShape instance in a IVtkTools_ShapeObject
  // wrapper in vtkInformation object of vtkDataObject, then pass it
  // to the actors through pipelines, so selection logic can access
  // OccShape easily given the actor instance.
  IVtkTools_ShapeObject::SetShapeSource(this, aPolyData);
  aPolyData->GetAttributes(vtkDataObject::CELL)->SetPedigreeIds(SubShapeIDs());
  return 1;
}

//=================================================================================================

vtkSmartPointer<vtkIdTypeArray> IVtkTools_ShapeDataSource::SubShapeIDs()
{
  vtkSmartPointer<vtkDataArray> anArr =
    GetOutput()->GetCellData()->GetArray(IVtkVTK_ShapeData::ARRNAME_SUBSHAPE_IDS());
  return vtkSmartPointer<vtkIdTypeArray>(vtkIdTypeArray::SafeDownCast(anArr.GetPointer()));
}

//=================================================================================================

IVtk_IdType IVtkTools_ShapeDataSource::GetId() const
{
  return myOccShape.IsNull() ? -1 : myOccShape->GetId();
}

//=================================================================================================

bool IVtkTools_ShapeDataSource::Contains(const IVtkOCC_Shape::Handle& theShape) const
{
  return myOccShape == theShape;
}

//=================================================================================================

vtkSmartPointer<vtkPolyData> IVtkTools_ShapeDataSource::transform(vtkPolyData*   theSource,
                                                                  const gp_Trsf& theTrsf) const
{
  vtkSmartPointer<vtkPolyData> aResult = vtkSmartPointer<vtkPolyData>::New();
  aResult->Allocate();
  vtkSmartPointer<vtkPoints> aPts = vtkSmartPointer<vtkPoints>::New();
  aResult->SetPoints(aPts);

  vtkSmartPointer<vtkTransform> aTransform = vtkSmartPointer<vtkTransform>::New();
  vtkSmartPointer<vtkMatrix4x4> aMx        = vtkSmartPointer<vtkMatrix4x4>::New();
  for (int aRow = 0; aRow < 3; ++aRow)
  {
    for (int aCol = 0; aCol < 4; ++aCol)
    {
      aMx->SetElement(aRow, aCol, theTrsf.Value(aRow + 1, aCol + 1));
    }
  }

  aTransform->SetMatrix(aMx);
  vtkSmartPointer<vtkTransformPolyDataFilter> aTrsfFilter =
    vtkSmartPointer<vtkTransformPolyDataFilter>::New();

  aTrsfFilter->SetTransform(aTransform);
  aTrsfFilter->SetInputData(theSource);
  aTrsfFilter->Update();

  vtkSmartPointer<vtkPolyData> aTransformed = aTrsfFilter->GetOutput();
  aResult->CopyStructure(aTransformed);  // Copy points and cells
  aResult->CopyAttributes(aTransformed); // Copy data arrays (sub-shapes ids)

  return aResult;
}