#include <cpPlugins/Interface/Mesh.h>
+#include <itkMesh.h>
+#include <itkLineCell.h>
+#include <itkTriangleCell.h>
+#include <itkPolygonCell.h>
+
#include <vtkPolyData.h>
// -------------------------------------------------------------------------
void cpPlugins::Interface::Mesh::
-SetVTK( vtkObject* mesh )
+SetITK( itk::LightObject* o )
{
- if( dynamic_cast< vtkPolyData* >( mesh ) != NULL )
- this->m_VTKObject = mesh;
- else
- this->m_VTKObject = NULL;
- this->m_Mapper = NULL;
- this->m_Actor = NULL;
- this->Modified( );
+ this->Superclass::SetITK( o );
+ this->_ITK_2_VTK< itk::Mesh< float, 2 > >( o );
+ this->_ITK_2_VTK< itk::Mesh< double, 2 > >( o );
+ this->_ITK_2_VTK< itk::Mesh< float, 3 > >( o );
+ this->_ITK_2_VTK< itk::Mesh< double, 3 > >( o );
+ this->_ITK_2_VTK< itk::QuadEdgeMesh< float, 2 > >( o );
+ this->_ITK_2_VTK< itk::QuadEdgeMesh< double, 2 > >( o );
+ this->_ITK_2_VTK< itk::QuadEdgeMesh< float, 3 > >( o );
+ this->_ITK_2_VTK< itk::QuadEdgeMesh< double, 3 > >( o );
}
// -------------------------------------------------------------------------
void cpPlugins::Interface::Mesh::
-CreateVTKActor( )
+SetVTK( vtkObjectBase* o )
{
- vtkPolyData* pd = this->GetVTK< vtkPolyData >( );
- if(
- pd != NULL &&
- (
- this->m_Mapper.GetPointer( ) == NULL ||
- this->m_Actor.GetPointer( ) == NULL
- )
- )
+ typedef itk::Mesh< double, 3 > _TMesh;
+ typedef _TMesh::CellType _TCell;
+ typedef _TCell::CellAutoPointer _TCellAutoPointer;
+ typedef itk::LineCell< _TCell > _TLine;
+ typedef itk::TriangleCell< _TCell > _TTriangle;
+ typedef itk::PolygonCell< _TCell > _TPolygon;
+
+ vtkPolyData* mesh = dynamic_cast< vtkPolyData* >( o );
+ if( mesh == NULL )
{
- this->m_Mapper = vtkSmartPointer< vtkPolyDataMapper >::New( );
- this->m_Actor = vtkSmartPointer< vtkActor >::New( );
- this->m_Mapper->SetInputData( pd );
- this->m_Actor->SetMapper( this->m_Mapper );
+ this->m_ITKObject = NULL;
this->Modified( );
+ return;
} // fi
-}
-// -------------------------------------------------------------------------
-vtkActor* cpPlugins::Interface::Mesh::
-GetVTKActor( )
-{
- return( this->m_Actor );
-}
+ if( this->m_VTKObject.GetPointer( ) != mesh )
+ {
+ this->m_VTKObject = mesh;
-// -------------------------------------------------------------------------
-const vtkActor* cpPlugins::Interface::Mesh::
-GetVTKActor( ) const
-{
- return( this->m_Actor );
-}
+ // Copy points
+ _TMesh::Pointer imesh = _TMesh::New( );
+ double point[ 3 ];
+ for( long i = 0; i < mesh->GetNumberOfPoints( ); ++i )
+ {
+ mesh->GetPoint( i, point );
+ _TMesh::PointType ipoint;
+ ipoint[ 0 ] = point[ 0 ];
+ ipoint[ 1 ] = point[ 1 ];
+ ipoint[ 2 ] = point[ 2 ];
+ imesh->SetPoint( i, ipoint );
+
+ } // rof
+
+ // Copy cells
+ for( long i = 0; i < mesh->GetNumberOfCells( ); ++i )
+ {
+ auto cell = mesh->GetCell( i );
+ long nPoints = cell->GetNumberOfPoints( );
+ _TCellAutoPointer icell;
+ if( nPoints == 2 )
+ {
+ icell.TakeOwnership( new _TLine );
+ icell->SetPointId( 0, cell->GetPointId( 0 ) );
+ icell->SetPointId( 1, cell->GetPointId( 1 ) );
+ }
+ else if( nPoints == 3 )
+ {
+ icell.TakeOwnership( new _TTriangle );
+ icell->SetPointId( 0, cell->GetPointId( 0 ) );
+ icell->SetPointId( 1, cell->GetPointId( 1 ) );
+ icell->SetPointId( 2, cell->GetPointId( 2 ) );
+ }
+ else if( nPoints > 3 )
+ {
+ _TPolygon* polygon = new _TPolygon( );
+ for( long j = 0; j < nPoints; ++j )
+ polygon->AddPointId( cell->GetPointId( j ) );
+ icell.TakeOwnership( polygon );
+
+ } // fi
+ imesh->SetCell( imesh->GetNumberOfCells( ), icell );
+ } // rof
+
+ this->m_ITKObject = imesh;
+ this->Modified( );
+
+ } // fi
+}
// -------------------------------------------------------------------------
cpPlugins::Interface::Mesh::
Mesh( )
- : Superclass( ),
- m_Mapper( NULL ),
- m_Actor( NULL )
+ : Superclass( )
{
- this->m_ClassName = "cpPlugins::Interface::Mesh";
- this->m_ClassCategory = "BasicObject";
}
// -------------------------------------------------------------------------