#include <cpPlugins/Plugins/MarchingCubes.h>
#include <cpPlugins/Interface/Image.h>
#include <cpPlugins/Interface/Mesh.h>

#include <vtkImageData.h>
#include <vtkMarchingCubes.h>
#include <vtkMarchingSquares.h>

// -------------------------------------------------------------------------
cpPlugins::Plugins::MarchingCubes::
MarchingCubes( )
  : Superclass( ),
    m_Algorithm( NULL )
{
  this->m_ClassName = "cpPlugins::MarchingCubes";
  this->m_ClassCategory = "ImageToMeshFilter";

  this->SetNumberOfInputs( 1 );
  this->SetNumberOfOutputs( 1 );
  this->_MakeOutput< cpPlugins::Interface::Mesh >( 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( )
{
  // Get input
  cpPlugins::Interface::Image* image =
    this->GetInput< cpPlugins::Interface::Image >( 0 );
  if( image == NULL )
    return( "MarchingCubes: Input data is not a valid image." );
  vtkImageData* vtk_image = image->GetVTKImageData( );
  if( vtk_image == NULL )
    return( "MarchingCubes: Input does not have a valid VTK conversion." );

  if( this->m_Algorithm != NULL )
    this->m_Algorithm->Delete( );

  std::vector< double > values;
  this->m_Parameters.GetValueAsRealList( values, "Thresholds" );
  if( vtk_image->GetDataDimension( ) == 2 )
  {
    vtkMarchingSquares* ms = vtkMarchingSquares::New( );
    ms->SetInputData( vtk_image );
    for( unsigned int i = 0; i < values.size( ); ++i )
      ms->SetValue( i, values[ i ] );
    this->m_Algorithm = ms;
  }
  else if( vtk_image->GetDataDimension( ) == 3 )
  {
    vtkMarchingCubes* mc = vtkMarchingCubes::New( );
    mc->SetInputData( vtk_image );
    for( unsigned int i = 0; i < values.size( ); ++i )
      mc->SetValue( i, values[ i ] );
    this->m_Algorithm = mc;
  }
  else
    return( "MarchingCubes: Input data does not have a valid dimension." );

  // Execute filter
  this->m_Algorithm->Update( );
  cpPlugins::Interface::Mesh* out =
    this->GetOutput< cpPlugins::Interface::Mesh >( 0 );
  out->SetVTKMesh( this->m_Algorithm->GetOutput( ) );
  return( "" );
}

// eof - $RCSfile$