]> Creatis software - cpPlugins.git/blobdiff - appli/ImageMPR/ImageMPR.cxx
update: undo/redo for image to image filters
[cpPlugins.git] / appli / ImageMPR / ImageMPR.cxx
index 24cfd389fe26e4882db063a79f1c76ded0e89b39..402baff2c1b1c4a074f6da2af00f34b50fe3692b 100644 (file)
@@ -1,10 +1,10 @@
 #include "ImageMPR.h"
+#include "MementoState.h"
 #include "ui_ImageMPR.h"
 
-#include <cpPlugins/Interface/ParametersQtDialog.h>
-
 #include <vtkProperty.h>
 #include <vtkRenderWindow.h>
+#include <vtkMetaImageReader.h>
 
 #include <QFileDialog>
 #include <QMessageBox>
@@ -25,7 +25,12 @@ ImageMPR::ImageMPR( QWidget* parent )
     m_UI( new Ui::ImageMPR ),
     m_ImageReaderClass( "" ),
     m_ImageWriterClass( "" ),
-    m_InputImage( NULL )
+    m_MeshReaderClass( "" ),
+    m_MeshWriterClass( "" ),
+    m_MeshCutterClass( "" ),
+    m_Image( NULL ),
+       m_state(0),
+       m_max_state(0)
 {
   this->m_UI->setupUi( this );
 
@@ -55,11 +60,24 @@ ImageMPR::ImageMPR( QWidget* parent )
     this->m_UI->actionOpenInputPolyData, SIGNAL( triggered( ) ),
     this, SLOT( _triggered_actionOpenInputPolyData( ) )
     );
+  QObject::connect(
+       this->m_UI->actionUndo, SIGNAL(triggered()),
+       this, SLOT(_triggered_actionUndo())
+       );
+  QObject::connect(
+       this->m_UI->actionRedo, SIGNAL(triggered()),
+       this, SLOT(_triggered_actionRedo())
+       );
 
   // Start: load all disponible plugins
   this->_LoadPlugins(
     std::string( PLUGIN_PREFIX ) +
-    std::string( "cpPlugins." ) +
+    std::string( "cpPluginsIO." ) +
+    std::string( PLUGIN_EXT )
+    );
+  this->_LoadPlugins(
+    std::string( PLUGIN_PREFIX ) +
+    std::string( "cpPluginsBasicFilters." ) +
     std::string( PLUGIN_EXT )
     );
 }
@@ -86,17 +104,13 @@ _LoadPlugins( const std::string& filename )
   this->m_ImageWriterClass = "";
   this->m_MeshReaderClass = "";
   this->m_MeshWriterClass = "";
-  this->m_ImageToImageFilters.clear( );
-  this->m_ImageToMeshFilters.clear( );
+  this->m_MeshCutterClass = "";
+  this->m_UI->MenuImageToImage->clear( );
+  this->m_UI->MenuImageToMesh->clear( );
 
-  this->m_Plugins.UnloadAll( );
   if( !( this->m_Plugins.Load( filename ) ) )
-  {
-    this->m_Plugins.UnloadAll( );
     return( false );
 
-  } // fi
-
   typedef TPluginsInterface::TClasses _TClasses;
   _TClasses::const_iterator cIt = this->m_Plugins.GetClasses( ).begin( );
   for( ; cIt != this->m_Plugins.GetClasses( ).end( ); ++cIt )
@@ -113,9 +127,13 @@ _LoadPlugins( const std::string& filename )
       this->m_MeshReaderClass = name;
     else if( category == "MeshWriter" )
       this->m_MeshWriterClass = name;
+    else if( category == "MeshToMeshFilter" )
+    {
+      if( name.find_last_of( "Cutter" ) != std::string::npos )
+        this->m_MeshCutterClass = name;
+    }
     else if( category == "ImageToImageFilter" )
     {
-      this->m_ImageToImageFilters.insert( name );
       QAction* action =
         this->m_UI->MenuImageToImage->addAction( QString( name.c_str( ) ) );
       QObject::connect(
@@ -125,7 +143,6 @@ _LoadPlugins( const std::string& filename )
     }
     else if( category == "ImageToMeshFilter" )
     {
-      this->m_ImageToMeshFilters.insert( name );
       QAction* action =
         this->m_UI->MenuImageToMesh->addAction( QString( name.c_str( ) ) );
       QObject::connect(
@@ -142,6 +159,94 @@ _LoadPlugins( const std::string& filename )
   return( true );
 }
 
+// -------------------------------------------------------------------------
+std::string ImageMPR::
+_LoadImage( TPluginImage::Pointer& image )
+{
+  std::string ret = "";
+  image = NULL;
+
+  // Get a reader from loaded plugins
+  TPluginFilter::Pointer reader =
+    this->m_Plugins.CreateProcessObject( this->m_ImageReaderClass );
+  if( reader.IsNotNull( ) )
+  {
+    if( reader->ExecConfigurationDialog( this ) )
+    {
+      // Block application
+      QApplication::setOverrideCursor( Qt::WaitCursor );
+      this->setEnabled( false );
+
+      // Execute and get error message, if any
+      ret = reader->Update( );
+
+      // Assign fresh image, if any
+      if( ret == "" )
+      {
+        image = reader->GetOutput< TPluginImage >( 0 );
+        reader->DisconnectOutputs( );
+
+      } // fi
+
+      // Unblock application
+      QApplication::restoreOverrideCursor( );
+      this->setEnabled( true );
+
+    } // fi
+  }
+  else
+    ret = "No suitable reader object found in loaded plugins.";
+  
+  return( ret );
+}
+
+// -------------------------------------------------------------------------
+std::string ImageMPR::
+_ConfigureMeshActors( )
+{
+  if( this->m_Mesh.IsNull( ) )
+    return( "Valid mesh not found." );
+
+  this->m_Mesh->CreateVTKActor( );
+  vtkActor* vtk_actor = this->m_Mesh->GetVTKActor( );
+  if( vtk_actor != NULL )
+  {
+    this->m_MPRObjects->Get3DRenderer( )->AddActor( vtk_actor );
+    this->m_MPRObjects->Render( 4 );
+
+  } // fi
+
+  TMPRObjects::TMPRActors* mprActors = this->m_MPRObjects->GetMPRActors( );
+
+  std::string err = "";
+  for( unsigned int i = 0; i < 3; ++i )
+  {
+    this->m_Cutters[ i ] = this->m_Plugins.CreateProcessObject( this->m_MeshCutterClass );
+    this->m_Planes[ i ] = TPluginImplicitFunction::New( );
+    this->m_Planes[ i ]->SetFunction( mprActors->GetSliceActors( i )->GetPlaneFunction( ) );
+    this->m_Cutters[ i ]->SetInput( 0, this->m_Mesh );
+    this->m_Cutters[ i ]->SetInput( 1, this->m_Planes[ i ] );
+    std::string lerr = this->m_Cutters[ i ]->Update( );
+    if( lerr == "" )
+    {
+      this->m_Cutters[ i ]->GetOutput< TPluginMesh >( 0 )->CreateVTKActor( );
+      vtkActor* actor = this->m_Cutters[ i ]->GetOutput< TPluginMesh >( 0 )->GetVTKActor( );
+      mprActors->GetSliceActors( i )->AddActor( this->m_Cutters[ i ]->GetVTK< vtkAlgorithm >( ), actor );
+      if( i == 0 )
+        this->m_MPRObjects->GetXRenderer( )->AddActor( actor );
+      else if( i == 1 )
+        this->m_MPRObjects->GetYRenderer( )->AddActor( actor );
+      else if( i == 2 )
+        this->m_MPRObjects->GetZRenderer( )->AddActor( actor );
+
+    } // fi
+    err += lerr;
+
+  } // rof
+  this->m_MPRObjects->RenderAll( );
+  return( err );
+}
+
 // -------------------------------------------------------------------------
 void ImageMPR::
 _triggered_actionOpenPlugins( )
@@ -168,60 +273,27 @@ _triggered_actionOpenPlugins( )
 void ImageMPR::
 _triggered_actionOpenInputImage( )
 {
-  // Show dialog and check if it was accepted
-  QFileDialog dialog( this );
-  dialog.setFileMode( QFileDialog::ExistingFiles );
-  dialog.setDirectory( tr( "." ) );
-  dialog.setNameFilter(
-    tr( "Medical image files (*.mhd *.bin *.dcm);;All files (*)" )
-    );
-  dialog.setDefaultSuffix( tr( "mhd" ) );
-  if( !( dialog.exec( ) ) )
-    return;
-  
-  this->m_InputImage = NULL;
-
-  // Get a reader from plugins
-  TPluginFilter::Pointer reader =
-    this->m_Plugins.CreateProcessObject( this->m_ImageReaderClass );
-
-  // Configure reader
-  TParameters reader_params = reader->GetDefaultParameters( );
-  QStringList q_fnames = dialog.selectedFiles( );
-  QStringList::const_iterator qIt = q_fnames.begin( );
-  for( ; qIt != q_fnames.end( ); ++qIt )
-    reader_params.AddValueToStringList( "FileNames", qIt->toStdString( ) );
-  reader->SetParameters( reader_params );
-
-  // Execute and get error message, if any
-  QApplication::setOverrideCursor( Qt::WaitCursor );
-  this->setEnabled( false );
-  std::string err = reader->Update( );
-  QApplication::restoreOverrideCursor( );
-  this->setEnabled( true );
-
-  // Assign fresh image, if any
+  // Read image
+  std::string err = this->_LoadImage( this->m_Image );
   if( err == "" )
   {
-    this->m_InputImage = reader->GetOutput< TPluginImage >( 0 );
-    reader->DisconnectOutputs( );
-    if( this->m_InputImage.IsNotNull( ) )
+    vtkImageData* vtk_id = this->m_Image->GetVTK< vtkImageData >( );
+    if( vtk_id != NULL )
     {
-      vtkImageData* vtk_id = this->m_InputImage->GetVTKImageData( );
-      if( vtk_id != NULL )
-      {
-        this->m_MPRObjects->SetImage( vtk_id );
-        this->m_MPRObjects->ResetCameras( );
-        this->m_MPRObjects->RenderAll( );
-      }
-      else
-        QMessageBox::critical(
-          this,
-          tr( "Error message" ),
-          tr( "Read image does not have a valid VTK converter." )
-          );
+      this->m_MPRObjects->SetImage( vtk_id );
+      this->m_MPRObjects->ActivateInteractors( );
+      this->m_MPRObjects->ResetCameras( );
+      this->m_MPRObjects->RenderAll( );
 
-    } // fi
+         MementoState(m_state, this->m_Image);  
+         this->m_state++;
+    }
+    else
+      QMessageBox::critical(
+        this,
+        tr( "Error message" ),
+        tr( "Read image does not have a valid VTK converter." )
+        );
   }
   else
     QMessageBox::critical(
@@ -235,62 +307,33 @@ _triggered_actionOpenInputImage( )
 void ImageMPR::
 _triggered_actionOpenSegmentation( )
 {
-  // Show dialog and check if it was accepted
-  QFileDialog dialog( this );
-  dialog.setFileMode( QFileDialog::ExistingFiles );
-  dialog.setDirectory( tr( "." ) );
-  dialog.setNameFilter(
-    tr( "Medical image files (*.mhd *.bin *.dcm);;All files (*)" )
-    );
-  dialog.setDefaultSuffix( tr( "mhd" ) );
-  if( !( dialog.exec( ) ) )
+  if( this->m_Image.IsNull( ) )
+  {
+    QMessageBox::critical(
+      this,
+      tr( "Error message" ),
+      tr( "Before reading a segmentation, first load a raw image." )
+      );
     return;
-  
-  this->m_InputImage = NULL;
-
-  // Get a reader from plugins
-  TPluginFilter::Pointer reader =
-    this->m_Plugins.CreateProcessObject( this->m_ImageReaderClass );
-
-  // Configure reader
-  TParameters reader_params = reader->GetDefaultParameters( );
-  QStringList q_fnames = dialog.selectedFiles( );
-  QStringList::const_iterator qIt = q_fnames.begin( );
-  for( ; qIt != q_fnames.end( ); ++qIt )
-    reader_params.AddValueToStringList( "FileNames", qIt->toStdString( ) );
-  reader->SetParameters( reader_params );
 
-  // Execute and get error message, if any
-  QApplication::setOverrideCursor( Qt::WaitCursor );
-  this->setEnabled( false );
-  std::string err = reader->Update( );
-  QApplication::restoreOverrideCursor( );
-  this->setEnabled( true );
+  } // fi
 
-  // Assign fresh image, if any
+  // Read image
+  std::string err = this->_LoadImage( this->m_Segmentation );
   if( err == "" )
   {
-    this->m_InputSegmentation = reader->GetOutput< TPluginImage >( 0 );
-    reader->DisconnectOutputs( );
-    if( this->m_InputSegmentation.IsNotNull( ) )
+    vtkImageData* vtk_id = this->m_Segmentation->GetVTK< vtkImageData >( );
+    if( vtk_id != NULL )
     {
-      vtkImageData* vtk_id = this->m_InputSegmentation->GetVTKImageData( );
-      if( vtk_id != NULL )
-      {
-        /*
-          this->m_MPRObjects->SetImage( vtk_id );
-          this->m_MPRObjects->ResetCameras( );
-          this->m_MPRObjects->RenderAll( );
-        */
-      }
-      else
-        QMessageBox::critical(
-          this,
-          tr( "Error message" ),
-          tr( "Read image does not have a valid VTK converter." )
-          );
-
-    } // fi
+      this->m_MPRObjects->AddAuxiliaryImage( vtk_id );
+      this->m_MPRObjects->RenderAll( );
+    }
+    else
+      QMessageBox::critical(
+        this,
+        tr( "Error message" ),
+        tr( "Read image does not have a valid VTK converter." )
+        );
   }
   else
     QMessageBox::critical(
@@ -304,57 +347,42 @@ _triggered_actionOpenSegmentation( )
 void ImageMPR::
 _triggered_actionOpenInputPolyData( )
 {
-  // Show dialog and check if it was accepted
-  QFileDialog dialog( this );
-  dialog.setFileMode( QFileDialog::ExistingFile );
-  dialog.setDirectory( tr( "." ) );
-  dialog.setNameFilter(
-    tr( "Mesh files (*.vtk *.obj);;All files (*)" )
-    );
-  dialog.setDefaultSuffix( tr( "vtk" ) );
-  if( !( dialog.exec( ) ) )
-    return;
-  
-  this->m_InputMesh = NULL;
+  this->m_Mesh = NULL;
 
   // Get a reader from plugins
   TPluginFilter::Pointer reader =
     this->m_Plugins.CreateProcessObject( this->m_MeshReaderClass );
 
-  // Configure reader
-  TParameters reader_params = reader->GetDefaultParameters( );
-  QStringList q_fnames = dialog.selectedFiles( );
-  QStringList::const_iterator qIt = q_fnames.begin( );
-  for( ; qIt != q_fnames.end( ); ++qIt )
-    reader_params.SetValueAsString( "FileName", qIt->toStdString( ) );
-  reader->SetParameters( reader_params );
-
-  // Execute and get error message, if any
-  QApplication::setOverrideCursor( Qt::WaitCursor );
-  this->setEnabled( false );
-  std::string err = reader->Update( );
-  QApplication::restoreOverrideCursor( );
-  this->setEnabled( true );
-
-
-  // Assign fresh image, if any
-  if( err == "" )
+  if( reader.IsNotNull( ) )
   {
-    this->m_InputMesh = reader->GetOutput< TPluginMesh >( 0 );
-    reader->DisconnectOutputs( );
-    if( this->m_InputMesh.IsNotNull( ) )
+    // Configure reader
+    if( reader->ExecConfigurationDialog( this ) )
     {
-      vtkActor* vtk_actor = this->m_InputMesh->GetVTKActor( );
-      if( vtk_actor != NULL )
+      // Execute and get error message, if any
+      QApplication::setOverrideCursor( Qt::WaitCursor );
+      this->setEnabled( false );
+      std::string err = reader->Update( );
+      QApplication::restoreOverrideCursor( );
+      this->setEnabled( true );
+
+      // Assign fresh mesh, if any
+      if( err == "" )
       {
-        this->m_MPRObjects->Get3DRenderer( )->AddActor( vtk_actor );
-        this->m_MPRObjects->Render( 4 );
+        this->m_Mesh = reader->GetOutput< TPluginMesh >( 0 );
+        reader->DisconnectOutputs( );
+        err = this->_ConfigureMeshActors( );
+        if( err != "" )
+          QMessageBox::critical(
+            this,
+            tr( "Error message" ),
+            tr( err.c_str( ) )
+            );
       }
       else
         QMessageBox::critical(
           this,
-          tr( "Error message" ),
-          tr( "Read mesh does not have a valid vtkActor." )
+          tr( "Error reading mesh" ),
+          tr( err.c_str( ) )
           );
 
     } // fi
@@ -362,8 +390,8 @@ _triggered_actionOpenInputPolyData( )
   else
     QMessageBox::critical(
       this,
-      tr( "Error reading mesh" ),
-      tr( err.c_str( ) )
+      tr( "Error reading single mesh" ),
+      tr( "No suitable mesh reader found in loaded plugins." )
       );
 }
 
@@ -371,7 +399,7 @@ _triggered_actionOpenInputPolyData( )
 void ImageMPR::
 _triggered_actionImageToImage( )
 {
-  if( this->m_InputImage.IsNull( ) )
+  if( this->m_Image.IsNull( ) )
     return;
 
   // Get filter name
@@ -390,7 +418,7 @@ _triggered_actionImageToImage( )
   // Execute filter
   QApplication::setOverrideCursor( Qt::WaitCursor );
   this->setEnabled( false );
-  filter->SetInput( 0, this->m_InputImage );
+  filter->SetInput( 0, this->m_Image );
   std::string err = filter->Update( );
   QApplication::restoreOverrideCursor( );
   this->setEnabled( true );
@@ -400,9 +428,20 @@ _triggered_actionImageToImage( )
   {
     TPluginImage* result = filter->GetOutput< TPluginImage >( 0 );
     result->DisconnectPipeline( );
-    this->m_InputImage = result;
-    if( this->m_InputImage.IsNotNull( ) )
-      this->m_MPRObjects->SetImage( this->m_InputImage->GetVTKImageData( ) );
+    this->m_Image = result;
+    if( this->m_Image.IsNotNull( ) )
+      this->m_MPRObjects->SetImage(
+        this->m_Image->GetVTK< vtkImageData >( )
+
+               );
+       
+       
+       MementoState(this->m_state, this->m_Image);
+       this->m_state++;
+       if (this->m_state > this->m_max_state)
+       {
+               this->m_max_state = this->m_state;
+       }
   }
   else
     QMessageBox::critical(
@@ -416,7 +455,7 @@ _triggered_actionImageToImage( )
 void ImageMPR::
 _triggered_actionImageToMesh( )
 {
-  if( this->m_InputImage.IsNull( ) )
+  if( this->m_Image.IsNull( ) )
     return;
 
   // Get filter name
@@ -435,7 +474,7 @@ _triggered_actionImageToMesh( )
   // Execute filter
   QApplication::setOverrideCursor( Qt::WaitCursor );
   this->setEnabled( false );
-  filter->SetInput( 0, this->m_InputImage );
+  filter->SetInput( 0, this->m_Image );
   std::string err = filter->Update( );
   QApplication::restoreOverrideCursor( );
   this->setEnabled( true );
@@ -445,10 +484,13 @@ _triggered_actionImageToMesh( )
   {
     TPluginMesh* result = filter->GetOutput< TPluginMesh >( 0 );
     result->DisconnectPipeline( );
-    this->m_InputMesh = result;
-    if( this->m_InputMesh.IsNotNull( ) )
-      this->m_MPRObjects->Get3DRenderer( )->AddActor(
-        this->m_InputMesh->GetVTKActor( )
+    this->m_Mesh = result;
+    err = this->_ConfigureMeshActors( );
+    if( err != "" )
+      QMessageBox::critical(
+        this,
+        tr( "Error message" ),
+        tr( err.c_str( ) )
         );
   }
   else
@@ -459,4 +501,50 @@ _triggered_actionImageToMesh( )
       );
 }
 
+// -------------------------------------------------------------------------
+void ImageMPR::
+_triggered_actionUndo()
+{
+       MementoState memento = MementoState();
+    
+       if (this->m_state>1)
+       {
+               this->m_state--;
+               this->m_MPRObjects->SetImage(
+                       memento.getMemento(this->m_state)->GetOutput()
+                       );
+       } else
+       {
+               QMessageBox::warning(
+                       this,
+                       tr("message"),
+                       tr("No history to undo")
+                       );
+       }
+
+}
+
+// -------------------------------------------------------------------------
+void ImageMPR::
+_triggered_actionRedo()
+{
+       MementoState memento = MementoState();
+               if (this->m_state + 1 <= m_max_state)
+               {
+                       this->m_state++;
+                       this->m_MPRObjects->SetImage(
+                               memento.getMemento(this->m_state)->GetOutput()
+                               );
+               } else
+               {
+                       QMessageBox::warning(
+                               this,
+                               tr("message"),
+                               tr("No history to redo")
+                               );
+               }
+       
+}
+       
+
 // eof - $RCSfile$