...

[cpPlugins.git] / lib / cpPlugins / Interface / Mesh.cxx

diff --git a/lib/cpPlugins/Interface/Mesh.cxx b/lib/cpPlugins/Interface/Mesh.cxx
index d56d1929aebeb81719640b70f20d80460975c775..29a3b499ec1eefd790e7d352cbcced5e8a54eb38 100644 (file)
--- a/lib/cpPlugins/Interface/Mesh.cxx
+++ b/lib/cpPlugins/Interface/Mesh.cxx
@@ -9,7 +9,22 @@
 
 // -------------------------------------------------------------------------
 void cpPlugins::Interface::Mesh::
-SetVTK( vtkObject* object )
+SetITK( itk::LightObject* o )
+{
+  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::
+SetVTK( vtkObjectBase* o )
 {
   typedef itk::Mesh< double, 3 >      _TMesh;
   typedef _TMesh::CellType            _TCell;
@@ -18,7 +33,7 @@ SetVTK( vtkObject* object )
   typedef itk::TriangleCell< _TCell > _TTriangle;
   typedef itk::PolygonCell< _TCell >  _TPolygon;
 
-  vtkPolyData* mesh = dynamic_cast< vtkPolyData* >( object );
+  vtkPolyData* mesh = dynamic_cast< vtkPolyData* >( o );
   if( mesh == NULL )
   {
     this->m_ITKObject = NULL;