occt/src/IVtkTools/IVtkTools_ShapeDataSource.cxx

// 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 <IVtkOCC_ShapeMesher.hxx>
#include <IVtkTools_ShapeDataSource.hxx>
#include <IVtkTools_ShapeObject.hxx>

// VTK includes
#include <vtkCellArray.h>
#include <vtkCellData.h>
#include <vtkDoubleArray.h>
#include <vtkIdTypeArray.h>
#include <vtkInformation.h>
#include <vtkObjectFactory.h> 
#include <vtkPoints.h>
#include <vtkPolyData.h>
#include <vtkStreamingDemandDrivenPipeline.h>
#include <vtkTransform.h>
#include <vtkTransformPolyDataFilter.h>

vtkStandardNewMacro(IVtkTools_ShapeDataSource);

//================================================================
// Function : Constructor
// Purpose  : 
//================================================================
IVtkTools_ShapeDataSource::IVtkTools_ShapeDataSource()
: myPolyData (new IVtkVTK_ShapeData),
  myIsFastTransformMode (Standard_False),
  myIsTransformOnly (Standard_False)
{
  this->SetNumberOfInputPorts (0);
}

//================================================================
// Function : Destructor
// Purpose  : 
//================================================================
IVtkTools_ShapeDataSource::~IVtkTools_ShapeDataSource()
{ }

//================================================================
// Function : SetShape
// Purpose  : 
//================================================================
void IVtkTools_ShapeDataSource::SetShape (const IVtkOCC_Shape::Handle& theOccShape)
{
  if (myIsFastTransformMode && !myOccShape.IsNull() &&
      theOccShape->GetShape().IsPartner (myOccShape->GetShape() ) )
  {
    myIsTransformOnly = Standard_True;
  }
  else
  {
    myIsTransformOnly = Standard_False;
  }

  myOccShape = theOccShape;
  this->Modified();
}

//================================================================
// Function : GetShape
// Purpose  : 
//================================================================
IVtkOCC_Shape::Handle IVtkTools_ShapeDataSource::GetShape()
{
  return myOccShape;
}

//================================================================
// Function : RequestData
// Purpose  : 
//================================================================
int IVtkTools_ShapeDataSource::RequestData (vtkInformation* theRequest,
                                            vtkInformationVector** theInputVector,
                                            vtkInformationVector* theOutputVector)
{
  vtkPolyData* aPolyData = vtkPolyData::GetData (theOutputVector);
  aPolyData->Allocate();
  vtkPoints* aPts = vtkPoints::New();
  aPolyData->SetPoints (aPts);
  aPts->Delete();

  vtkSmartPointer<vtkPolyData> aTransformedData;
  TopoDS_Shape aShape = myOccShape->GetShape();
  TopLoc_Location aShapeLoc = aShape.Location();

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

    myPolyData = new IVtkVTK_ShapeData;
    IVtkOCC_ShapeMesher::Handle aMesher = new IVtkOCC_ShapeMesher;
    aMesher->Build (aShapeWrapperCopy, myPolyData);
    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 Superclass::RequestData (theRequest, theInputVector, theOutputVector);
}

//================================================================
// Function : SubShapeIDs
// Purpose  : 
//================================================================
vtkSmartPointer<vtkIdTypeArray> IVtkTools_ShapeDataSource::SubShapeIDs()
{
  vtkDataArray* arr = GetOutput()->GetCellData()->GetArray(IVtkVTK_ShapeData::ARRNAME_SUBSHAPE_IDS);
  return vtkSmartPointer<vtkIdTypeArray>( vtkIdTypeArray::SafeDownCast(arr) );
}

//================================================================
// Function : GetId
// Purpose  : 
//================================================================
IVtk_IdType IVtkTools_ShapeDataSource::GetId() const
{
  return myOccShape ? myOccShape->GetId() : -1;
}

//================================================================
// Function : Contains
// Purpose  : 
//================================================================
Standard_Boolean IVtkTools_ShapeDataSource::Contains (const IVtkOCC_Shape::Handle& shape) const
{
  return ((myOccShape == shape) ? Standard_True : Standard_False);
}

//================================================================
// Function : transform
// Purpose  : 
//================================================================
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 (Standard_Integer aRow = 0; aRow < 3; ++aRow)
    for (Standard_Integer 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();

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

  return aResult;
}