#include <cpPlugins/Interface/Image.h>
#include <cpPlugins/Interface/Mesh.h>
-#include <itkImageFileReader.h>
-
-#define ITK_MANUAL_INSTANTIATION
-#include <itkImage.h>
-#include <itkRGBPixel.h>
-
-// -------------------------------------------------------------------------
-std::string cpPlugins::Plugins::MarchingCubes::
-GetClassName( ) const
-{
- return( "cpPlugins::Plugins::MarchingCubes" );
-}
+#include <vtkImageData.h>
+#include <vtkMarchingCubes.h>
+#include <vtkMarchingSquares.h>
// -------------------------------------------------------------------------
cpPlugins::Plugins::MarchingCubes::
MarchingCubes( )
- : Superclass( )
+ : Superclass( ),
+ m_Algorithm( NULL )
{
+ this->m_ClassName = "cpPlugins::MarchingCubes";
+
this->SetNumberOfInputs( 1 );
this->SetNumberOfOutputs( 1 );
this->_MakeOutput< cpPlugins::Interface::Mesh >( 0 );
- /* TODO
- this->m_DefaultParameters[ "FileName" ] =
- TParameter( "string", "no_file_name" );
- this->m_DefaultParameters[ "PixelType" ] = TParameter( "type", "uchar" );
- this->m_DefaultParameters[ "ImageDimension" ] = TParameter( "int", "2" );
- this->m_DefaultParameters[ "IsColorImage" ] = TParameter( "bool", "0" );
- */
+ using namespace cpPlugins::Interface;
+ this->m_DefaultParameters.Configure( Parameters::RealList, "Thresholds" );
+ this->m_Parameters = this->m_DefaultParameters;
}
// -------------------------------------------------------------------------
cpPlugins::Plugins::MarchingCubes::
~MarchingCubes( )
{
+ if( this->m_Algorithm != NULL )
+ this->m_Algorithm->Delete( );
}
// -------------------------------------------------------------------------
std::string cpPlugins::Plugins::MarchingCubes::
_GenerateData( )
{
- return( "" );
- /* TODO
- TParameters::const_iterator dIt;
-
- // Get image dimension
- dIt = this->m_Parameters.find( "ImageDimension" );
- if( dIt == this->m_Parameters.end( ) )
- dIt = this->m_DefaultParameters.find( "ImageDimension" );
-
- std::string r = "cpPlugins::Plugins::MarchingCubes: itk::Image dimension not supported.";
- if ( dIt->second.second == "1" ) r = this->_GD0< 1 >( );
- else if( dIt->second.second == "2" ) r = this->_GD0< 2 >( );
- else if( dIt->second.second == "3" ) r = this->_GD0< 3 >( );
- else if( dIt->second.second == "4" ) r = this->_GD0< 4 >( );
-
- return( r );
- */
-}
-
-// -------------------------------------------------------------------------
-template< unsigned int D >
-std::string cpPlugins::Plugins::MarchingCubes::
-_GD0( )
-{
- return( "" );
- /* TODO
- TParameters::const_iterator tIt, cIt;
-
- // Get image pixel type
- tIt = this->m_Parameters.find( "PixelType" );
- if( tIt == this->m_Parameters.end( ) )
- tIt = this->m_DefaultParameters.find( "PixelType" );
- cIt = this->m_Parameters.find( "IsColorImage" );
- if( cIt == this->m_Parameters.end( ) )
- cIt = this->m_DefaultParameters.find( "IsColorImage" );
-
- std::string r = "cpPlugins::Plugins::MarchingCubes: itk::Image pixel type not supported";
- if( cIt->second.second == "0" )
+ // Get input
+ cpPlugins::Interface::Image* input =
+ dynamic_cast< cpPlugins::Interface::Image* >(
+ this->m_Inputs[ 0 ].GetPointer( )
+ );
+ if( input == NULL )
+ return( "MarchingCubes: Input data is not a valid image." );
+ vtkImageData* vtk_input =
+ dynamic_cast< vtkImageData* >( input->GetVTKDataObject( ) );
+ if( vtk_input == NULL )
+ return( "MarchingCubes: Input does not have a valid conversion to VTK." );
+
+ if( this->m_Algorithm != NULL )
+ this->m_Algorithm->Delete( );
+
+ std::vector< double > values;
+ this->m_Parameters.GetValueAsRealList( values, "Thresholds" );
+ if( vtk_input->GetDataDimension( ) == 2 )
{
- if( tIt->second.second == "char" )
- r = this->_GD1< char, D >( );
- else if( tIt->second.second == "short" )
- r = this->_GD1< short, D >( );
- else if( tIt->second.second == "int" )
- r = this->_GD1< int, D >( );
- else if( tIt->second.second == "long" )
- r = this->_GD1< long, D >( );
- else if( tIt->second.second == "uchar" )
- r = this->_GD1< unsigned char, D >( );
- else if( tIt->second.second == "ushort" )
- r = this->_GD1< unsigned short, D >( );
- else if( tIt->second.second == "uint" )
- r = this->_GD1< unsigned int, D >( );
- else if( tIt->second.second == "ulong" )
- r = this->_GD1< unsigned long, D >( );
- else if( tIt->second.second == "float" )
- r = this->_GD1< float, D >( );
- else if( tIt->second.second == "double" )
- r = this->_GD1< double, D >( );
+ vtkMarchingSquares* ms = vtkMarchingSquares::New( );
+ ms->SetInputData( vtk_input );
+ for( unsigned int i = 0; i < values.size( ); ++i )
+ ms->SetValue( i, values[ i ] );
+ this->m_Algorithm = ms;
}
- else if( cIt->second.second == "1" )
- {
- if( tIt->second.second == "char" )
- r = this->_GD1< itk::RGBPixel< char >, D >( );
- else if( tIt->second.second == "short" )
- r = this->_GD1< itk::RGBPixel< short >, D >( );
- else if( tIt->second.second == "int" )
- r = this->_GD1< itk::RGBPixel< int >, D >( );
- else if( tIt->second.second == "long" )
- r = this->_GD1< itk::RGBPixel< long >, D >( );
- else if( tIt->second.second == "uchar" )
- r = this->_GD1< itk::RGBPixel< unsigned char >, D >( );
- else if( tIt->second.second == "ushort" )
- r = this->_GD1< itk::RGBPixel< unsigned short >, D >( );
- else if( tIt->second.second == "uint" )
- r = this->_GD1< itk::RGBPixel< unsigned int >, D >( );
- else if( tIt->second.second == "ulong" )
- r = this->_GD1< itk::RGBPixel< unsigned long >, D >( );
- else if( tIt->second.second == "float" )
- r = this->_GD1< itk::RGBPixel< float >, D >( );
- else if( tIt->second.second == "double" )
- r = this->_GD1< itk::RGBPixel< double >, D >( );
- } // fi
- return( r );
-*/
-}
-
-// -------------------------------------------------------------------------
-template< class P, unsigned int D >
-std::string cpPlugins::Plugins::MarchingCubes::
-_GD1( )
-{
- return( "" );
- /* TODO
- TParameters::const_iterator fIt;
-
- // Get filename
- fIt = this->m_Parameters.find( "FileName" );
- if( fIt == this->m_Parameters.end( ) )
- fIt = this->m_DefaultParameters.find( "FileName" );
-
- typedef itk::Image< P, D > _TImage;
- typedef itk::ImageFileReader< _TImage > _TReader;
-
- _TReader* reader =
- dynamic_cast< _TReader* >( this->m_Reader.GetPointer( ) );
- if( reader == NULL )
- {
- this->m_Reader = _TReader::New( );
- reader = dynamic_cast< _TReader* >( this->m_Reader.GetPointer( ) );
-
- } // fi
- reader->SetFileName( fIt->second.second );
- try
+ else if( vtk_input->GetDataDimension( ) == 3 )
{
- reader->Update( );
+ vtkMarchingCubes* mc = vtkMarchingCubes::New( );
+ mc->SetInputData( vtk_input );
+ for( unsigned int i = 0; i < values.size( ); ++i )
+ mc->SetValue( i, values[ i ] );
+ this->m_Algorithm = mc;
}
- catch( itk::ExceptionObject& err )
- {
- return( err.GetDescription( ) );
+ else
+ return( "MarchingCubes: Input data does not have a valid dimension." );
- } // yrt
- this->_SetOutput( 0, reader->GetOutput( ) );
+ // Execute filter
+ this->m_Algorithm->Update( );
+ this->m_Outputs[ 0 ]->SetVTKDataObject( this->m_Algorithm->GetOutput( ) );
return( "" );
-*/
}
// eof - $RCSfile$
git://git.creatis.insa-lyon.fr / cpPlugins.git / blobdiff