]> Creatis software - cpPlugins.git/blobdiff - lib/cpPlugins/Plugins/BasicFilters/MacheteFilter.cxx
...
[cpPlugins.git] / lib / cpPlugins / Plugins / BasicFilters / MacheteFilter.cxx
index 3eb60061e0f18d4ba7e0e8dcb19fc2e143361dd2..6877a3d3e2ee65134f72dca8f3d90c9906ad56f8 100644 (file)
 #include <cpPlugins/Interface/Mesh.h>
 
 #include <cpExtensions/Interaction/ImageInteractorStyle.h>
+#include <cpExtensions/DataStructures/InfinitePlaneSpatialObject.h>
+#include <cpExtensions/Algorithms/SpatialObjectMaskImageFilter.h>
 
 #include <itkPlaneSpatialObject.h>
 #include <itkPoint.h>
-#include <itkSpatialObjectToImageFilter.h>
 #include <itkVector.h>
 
 #include <vtkInteractorStyleSwitch.h>
 #include <vtkPlaneWidget.h>
 #include <vtkRenderWindowInteractor.h>
 
+#ifdef cpPlugins_Interface_QT4
+#include <QDialogButtonBox>
+
+// -------------------------------------------------------------------------
+cpPlugins::BasicFilters::MacheteFilter_Dialog::
+MacheteFilter_Dialog(
+  QWidget* parent, MacheteFilter* filter, Qt::WindowFlags f
+  )
+  : QDialog( parent, f | Qt::WindowStaysOnTopHint ),
+    m_Filter( filter )
+{
+  this->m_Title = new QLabel( this );
+  this->m_Title->setText( "Execute machete filter" );
+
+  this->m_MainLayout = new QGridLayout( this );
+  this->m_ToolsLayout = new QVBoxLayout( );
+  this->m_ToolsLayout->addWidget( this->m_Title );
+  this->m_MainLayout->addLayout( this->m_ToolsLayout, 0, 0, 1, 1 );
+
+  // Add buttons
+  QDialogButtonBox* bb = new QDialogButtonBox(
+    QDialogButtonBox::Cancel | QDialogButtonBox::Ok
+    );
+  QObject::connect( bb, SIGNAL( accepted( ) ), this, SLOT( accept( ) ) );
+  QObject::connect( bb, SIGNAL( rejected( ) ), this, SLOT( reject( ) ) );
+  this->m_ToolsLayout->addWidget( bb );
+}
+
+// -------------------------------------------------------------------------
+cpPlugins::BasicFilters::MacheteFilter_Dialog::
+~MacheteFilter_Dialog( )
+{
+  delete this->m_Title;
+  delete this->m_ToolsLayout;
+  delete this->m_MainLayout;
+}
+
+// -------------------------------------------------------------------------
+void cpPlugins::BasicFilters::MacheteFilter_Dialog::
+accept( )
+{
+  // Get interactive widget
+  if( this->m_Filter == NULL )
+    return;
+  vtkPlaneWidget* wdg = this->m_Filter->m_PlaneWidget;
+  if( wdg == NULL )
+    return;
+
+  // Get/Set plane parameters
+  double center[ 3 ], normal[ 3 ];
+  wdg->GetCenter( center );
+  wdg->GetNormal( normal );
+
+  /* TODO
+     this->m_Filter->GetParameters( )->SetPoint( "PlaneCenter", 3, center );
+     this->m_Filter->GetParameters( )->SetVector( "PlaneNormal", 3, normal );
+  */
+
+  // Update filter
+  /* TODO
+     auto plugins = this->m_Filter->GetPlugins( );
+     if( plugins != NULL )
+     {
+     auto app = plugins->GetApplication( );
+     if( app != NULL )
+     app->UpdateActualFilter( );
+
+     } // fi
+  */
+}
+
+// -------------------------------------------------------------------------
+void cpPlugins::BasicFilters::MacheteFilter_Dialog::
+reject( )
+{
+  /*
+  auto plugins = this->m_Filter->GetPlugins( );
+  if( plugins != NULL )
+    plugins->DeactivateFilter( );
+  */
+  this->Superclass::reject( );
+}
+
+#endif // cpPlugins_Interface_QT4
+
 // -------------------------------------------------------------------------
 cpPlugins::BasicFilters::MacheteFilter::
 DialogResult cpPlugins::BasicFilters::MacheteFilter::
 ExecConfigurationDialog( QWidget* parent )
 {
+#ifdef cpPlugins_Interface_QT4
+
   typedef cpExtensions::Interaction::ImageInteractorStyle _TImageStyle;
 
   // Choose a valid 3D interactor
   vtkRenderWindowInteractor* iren = NULL;
+  /* TODO
   auto iIt = this->m_Interactors.begin( );
   for( ; iIt != this->m_Interactors.end( ) && iren == NULL; ++iIt )
   {
@@ -35,13 +124,13 @@ ExecConfigurationDialog( QWidget* parent )
       iren = *iIt;
     
   } // rof
+  */
   if( iren == NULL )
-    return( Self::DialogResult_Cancel );
-
+    return( false );
+  
   // Get bounding box
   double bbox[ 6 ];
-  cpPlugins::Interface::Image* image =
-    this->GetInput< cpPlugins::Interface::Image >( "Input" );
+  auto image = this->GetInputData< cpPlugins::Interface::Image >( "Input" );
   bool input_found = false;
   if( image != NULL )
   {
@@ -49,8 +138,7 @@ ExecConfigurationDialog( QWidget* parent )
     input_found = true;
 
   } // fi
-  cpPlugins::Interface::Mesh* mesh =
-    this->GetInput< cpPlugins::Interface::Mesh >( "Input" );
+  auto mesh = this->GetInputData< cpPlugins::Interface::Mesh >( "Input" );
   if( mesh != NULL )
   {
     mesh->GetVTK< vtkPolyData >( )->GetBounds( bbox );
@@ -58,7 +146,7 @@ ExecConfigurationDialog( QWidget* parent )
 
   } // fi
   if( !input_found )
-    return( Self::DialogResult_Cancel );
+    return( false );
 
   // Create plane widget
   if( this->m_PlaneWidget != NULL )
@@ -91,7 +179,13 @@ ExecConfigurationDialog( QWidget* parent )
   this->m_PlaneWidget->PlaceWidget( );
   this->m_PlaneWidget->On( );
 
-  return( Self::DialogResult_Modal );
+  this->m_Dialog = new MacheteFilter_Dialog( NULL, this );
+  this->m_Dialog->show( );
+
+  return( true );
+#else // cpPlugins_Interface_QT4
+  return( false );
+#endif // cpPlugins_Interface_QT4
 }
 
 // -------------------------------------------------------------------------
@@ -100,18 +194,14 @@ MacheteFilter( )
   : Superclass( ),
     m_PlaneWidget( NULL )
 {
-  this->_AddInput( "Input" );
-  this->_MakeOutput< cpPlugins::Interface::DataObject >( "Output" );
+  this->_AddInput( "Input", true );
+  this->_AddOutput< cpPlugins::Interface::DataObject >( "PositiveOutput" );
+  this->_AddOutput< cpPlugins::Interface::DataObject >( "NegativeOutput" );
 
-  itk::Point< double, 3 > center;
-  itk::Vector< double, 3 > normal;
-
-  center.Fill( double( 0 ) );
-  normal.Fill( double( 0 ) );
-  normal[ 0 ] = double( 1 );
-
-  this->m_Parameters->ConfigureAsPoint( "PlaneCenter", 3, center );
-  this->m_Parameters->ConfigureAsVector( "PlaneNormal", 3, normal );
+  /*
+  this->m_Parameters->ConfigureAsPoint( "PlaneCenter" );
+  this->m_Parameters->ConfigureAsVector( "PlaneNormal" );
+  */
 }
 
 // -------------------------------------------------------------------------
@@ -126,12 +216,10 @@ cpPlugins::BasicFilters::MacheteFilter::
 std::string cpPlugins::BasicFilters::MacheteFilter::
 _GenerateData( )
 {
-  cpPlugins::Interface::Image* image =
-    this->GetInput< cpPlugins::Interface::Image >( "Input" );
+  auto image = this->GetInputData< cpPlugins::Interface::Image >( "Input" );
   if( image != NULL )
     return( this->_FromImage( image ) );
-  cpPlugins::Interface::Mesh* mesh =
-    this->GetInput< cpPlugins::Interface::Mesh >( "Input" );
+  auto mesh = this->GetInputData< cpPlugins::Interface::Mesh >( "Input" );
   if( mesh == NULL )
     return( this->_FromMesh( mesh ) );
   return( "MacheteFilter: No valid input." );
@@ -156,32 +244,67 @@ _FromMesh( cpPlugins::Interface::Mesh* mesh )
   return( "" );
 }
 
-#include <itkImageFileWriter.h>
-
 // -------------------------------------------------------------------------
 template< class I >
 std::string cpPlugins::BasicFilters::MacheteFilter::
 _RealImage( itk::DataObject* dobj )
 {
-  typedef typename I::PixelType                         _TPixel;
-  typedef itk::PlaneSpatialObject< I::ImageDimension >  _TPlane;
-  typedef itk::SpatialObjectToImageFilter< _TPlane, I > _TMaskFilter;
+  typedef
+    cpExtensions::DataStructures::
+    InfinitePlaneSpatialObject< I::ImageDimension > _TPlane;
+  typedef
+    cpExtensions::Algorithms::
+    SpatialObjectMaskImageFilter< I, I > _TFilter;
+  typedef cpPlugins::Interface::DataObject _TObj;
+  typedef cpPlugins::Interface::Image      _TImage;
+  typedef typename _TPlane::PointType      _TPoint;
+  typedef typename _TPlane::VectorType     _TVector;
+  typedef typename I::PixelType            _TPixel;
 
   I* image = dynamic_cast< I* >( dobj );
 
-  typename _TMaskFilter::Pointer mask = _TMaskFilter::New( );
-  mask->SetDirection( image->GetDirection( ) );
-  mask->SetOrigin( image->GetOrigin( ) );
-  mask->SetSize( image->GetRequestedRegion( ).GetSize( ) );
-  mask->SetSpacing( image->GetSpacing( ) );
-  mask->SetInsideValue( _TPixel( 1 ) );
-  mask->SetOutsideValue( _TPixel( 0 ) );
-
-  typename itk::ImageFileWriter< I >::Pointer w = 
-    itk::ImageFileWriter< I >::New( );
-  w->SetInput( mask->GetOutput( ) );
-  w->SetFileName( "mask.mhd" );
-  w->Update( );
+  /*
+  _TPoint c = this->m_Parameters->GetPoint< _TPoint >(
+    "PlaneCenter", I::ImageDimension
+    );
+  _TVector n = this->m_Parameters->GetVector< _TVector >(
+    "PlaneNormal", I::ImageDimension
+    );
+  */
+  _TPoint c;
+  _TVector n;
+
+  typename _TPlane::Pointer plane = _TPlane::New( );
+  plane->SetCenter( c );
+  plane->SetNormal( n );
+
+  // Configure filter
+  _TFilter* filter = this->_CreateITK< _TFilter >( );
+  filter->SetInput( image );
+  filter->SetSpatialObject( plane );
+  filter->SetOutsideValue( _TPixel( 0 ) );
+  filter->Update( );
+
+  // Connect outputs (and correct their types and names)
+  _TImage* pos_out = this->GetOutputData< _TImage >( "PositiveOutput" );
+  if( pos_out == NULL )
+  {
+    this->_AddOutput< _TImage >( "PositiveOutput" );
+    pos_out = this->GetOutputData< _TImage >( "PositiveOutput" );
+
+  } // fi
+  _TImage* neg_out = this->GetOutputData< _TImage >( "NegativeOutput" );
+  if( neg_out == NULL )
+  {
+    this->_AddOutput< _TImage >( "NegativeOutput" );
+    neg_out = this->GetOutputData< _TImage >( "NegativeOutput" );
+
+  } // fi
+
+  // Assign outputs
+  pos_out->SetITK< I >( filter->GetPositiveOutput( ) );
+  neg_out->SetITK< I >( filter->GetNegativeOutput( ) );
+  return( "" );
 }
 
 // eof - $RCSfile$