X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=appli%2FImageMPR%2FImageMPR.cxx;h=69970b7ccd8b6c25e7592d25e61c2475a5b2b21c;hb=3633aade338a13bc83642e99e6d61b6499e4b3af;hp=a83a5d6b7898c3651f289ccd03eaa3aeb98ec877;hpb=584e1bc8ecb0568639e1843f616408e28d4c6257;p=cpPlugins.git

diff --git a/appli/ImageMPR/ImageMPR.cxx b/appli/ImageMPR/ImageMPR.cxx
index a83a5d6..69970b7 100644
--- a/appli/ImageMPR/ImageMPR.cxx
+++ b/appli/ImageMPR/ImageMPR.cxx
@@ -15,8 +15,7 @@ ImageMPR::
 ImageMPR( QWidget* parent )
   : QMainWindow( parent ),
     m_UI( new Ui::ImageMPR ),
-    m_Plugins( new TPlugins ),
-    m_MainImage( NULL )
+    m_Plugins( new TPlugins )
 {
   this->m_UI->setupUi( this );
   this->m_Plugins->SetWidget( this );
@@ -53,11 +52,19 @@ ImageMPR::
 void ImageMPR::
 _aOpenImage( )
 {
-  if( this->m_MainImage.IsNotNull( ) )
+  // Clear all, since we are loading the main image
+  if( this->m_Objects.size( ) > 0 )
+  {
     this->m_UI->MPR->ClearAll( );
-  if( this->m_Plugins->ReadImage( this->m_MainImage, true ) )
+    this->m_Objects.clear( );
+
+  } // fi
+
+  // Read and show image, if possible
+  TPlugins::TImage::Pointer image;
+  if( this->m_Plugins->ReadImage( image, true ) )
   {
-    vtkImageData* vimage = this->m_MainImage->GetVTK< vtkImageData >( );
+    vtkImageData* vimage = image->GetVTK< vtkImageData >( );
     if( vimage == NULL )
       QMessageBox::critical(
         this,
@@ -67,7 +74,11 @@ _aOpenImage( )
           )
         );
     else
-      this->m_UI->MPR->ShowImage( vimage );
+      this->m_UI->MPR->ShowImage( vimage, image->GetName( ) );
+
+    // Keep a track on a local data tree
+    this->m_Objects[ image->GetName( ) ] =
+      TTreeNode( "", image.GetPointer( ) );
 
   } // fi
 }
@@ -76,11 +87,19 @@ _aOpenImage( )
 void ImageMPR::
 _aOpenDICOMSeries( )
 {
-  if( this->m_MainImage.IsNotNull( ) )
+  // Clear all, since we are loading the main image
+  if( this->m_Objects.size( ) > 0 )
+  {
     this->m_UI->MPR->ClearAll( );
-  if( this->m_Plugins->ReadDicomSeries( this->m_MainImage ) )
+    this->m_Objects.clear( );
+
+  } // fi
+
+  // Read and show image, if possible
+  TPlugins::TImage::Pointer image;
+  if( this->m_Plugins->ReadDicomSeries( image ) )
   {
-    vtkImageData* vimage = this->m_MainImage->GetVTK< vtkImageData >( );
+    vtkImageData* vimage = image->GetVTK< vtkImageData >( );
     if( vimage == NULL )
       QMessageBox::critical(
         this,
@@ -90,7 +109,11 @@ _aOpenDICOMSeries( )
           )
         );
     else
-      this->m_UI->MPR->ShowImage( vimage );
+      this->m_UI->MPR->ShowImage( vimage, image->GetName( ) );
+
+    // Keep a track on a local data tree
+    this->m_Objects[ image->GetName( ) ] =
+      TTreeNode( "", image.GetPointer( ) );
 
   } // fi
 }
@@ -115,8 +138,13 @@ _aOpenPolyData( )
 void ImageMPR::
 _aSaveImage( )
 {
-  if( this->m_MainImage.IsNotNull( ) )
-    this->m_Plugins->WriteImage( this->m_MainImage, true );
+  std::string data_name = this->m_UI->MPR->GetSelectedData( );
+  TPlugins::TImage* image = dynamic_cast< TPlugins::TImage* >(
+    this->m_Objects[ data_name ].second.GetPointer( )
+    );
+  if( image == NULL )
+    return;
+  this->m_Plugins->WriteImage( image, true );
 }
 
 // -------------------------------------------------------------------------
@@ -129,6 +157,13 @@ _aSaveSegmentation( )
 void ImageMPR::
 _aSavePolyData( )
 {
+  std::string data_name = this->m_UI->MPR->GetSelectedData( );
+  TPlugins::TMesh* mesh = dynamic_cast< TPlugins::TMesh* >(
+    this->m_Objects[ data_name ].second.GetPointer( )
+    );
+  if( mesh == NULL )
+    return;
+  this->m_Plugins->WriteMesh( mesh, true );
 }
 
 // -------------------------------------------------------------------------
@@ -170,95 +205,124 @@ _execPlugin( )
   QMenu* menu = dynamic_cast< QMenu* >( action->parentWidget( ) );
   if( menu == NULL )
     return;
-  std::string cate = menu->title( ).toStdString( );
-  std::string name = action->text( ).toStdString( );
-  std::string err = "";
+  std::string filter_cate = menu->title( ).toStdString( );
+  std::string filter_name = action->text( ).toStdString( );
 
+  // Create filter
   TPlugins::TProcessObject::Pointer filter;
-  if( this->m_Plugins->CreateFilter( filter, name ) )
+  if( !( this->m_Plugins->CreateFilter( filter, filter_name ) ) )
   {
-    if( cate == "ImageToMeshFilter" )
-    {
-      if( filter->ExecConfigurationDialog( this ) )
-      {
-        filter->SetInput( "Input", this->m_MainImage );
-        this->_Block( );
-        err = filter->Update( );
-        this->_Unblock( );
-        TPlugins::TMesh::Pointer mesh =
-          filter->GetOutput< TPlugins::TMesh >( "Output" );
-        this->m_Meshes.push_back( mesh );
-        mesh->CreateVTKActor( );
-        vtkActor* actor = mesh->GetVTKActor( );
-        if( actor != NULL )
-          this->m_UI->MPR->Add3DActor( actor );
+    QMessageBox::critical(
+      this,
+      tr( "Error creating filter" ),
+      tr( (
+            std::string( "No valid filter \"" ) +
+            filter_name +
+            std::string( "\"defined." )
+            ).c_str( ) )
+      );
+    return;
 
-      } // fi
+  } // fi
 
-    } // fi
+  // Configure filter
+  if( !( filter->ExecConfigurationDialog( this ) ) )
+    return;
+
+  // Assign inputs
+  std::string data_name = this->m_UI->MPR->GetSelectedData( );
+  std::vector< std::string > inputs_names = filter->GetInputsNames( );
+  if( inputs_names.size( ) == 1 )
+  {
+    TTree::iterator iIt = this->m_Objects.find( data_name );
+    if( iIt == this->m_Objects.end( ) )
+    {
+      QMessageBox::critical(
+        this,
+        tr( "Error configuring filter" ),
+        tr( "No valid input found. Please select a valid input." )
+        );
+      return;
+
+    } //fi
+    filter->SetInput( inputs_names[ 0 ], iIt->second.second );
   }
-  else
+  else if( inputs_names.size( ) > 1 )
+  {
     QMessageBox::critical(
       this,
-      tr( "Error creating filter" ),
-      tr( "No valid filter defined." )
+      tr( "Error executing" ),
+      tr( "Filter has multiple inputs: NOT YET IMPLEMENTED!!!" )
       );
+    return;
 
-  /*
-    if( name == "cpPlugins::BasicFilters::FloodFillImageFilter" )
-    {
-    this->m_Flooding = true;
-    }
-    else
-    {
-    this->m_Flooding = false;
-    this->m_UI->MPR->ExecuteFilter(
-    name, this->m_ImageLoaded, "SegmentedImage"
-    );
-    
-    } // fi
-  */
-  
-  // Configure filter
-  /*
-    TPluginFilter::Pointer filter =
-    this->m_Plugins.CreateProcessObject( name );
-    bool dlg_ok = filter->ExecConfigurationDialog( NULL );
-    if( !dlg_ok )
+  } // fi
+
+  // Execute filter
+  this->_Block( );
+  std::string filter_err = filter->Update( );
+  this->_Unblock( );
+  if( filter_err != "" )
+  {
+    QMessageBox::critical(
+      this,
+      tr( "Error executing" ),
+      tr( filter_err.c_str( ) )
+      );
     return;
 
-    // Execute filter
-    QApplication::setOverrideCursor( Qt::WaitCursor );
-    this->setEnabled( false );
-    filter->SetInput( 0, this->m_Image );
-    std::string err = filter->Update( );
-    QApplication::restoreOverrideCursor( );
-    this->setEnabled( true );
+  } // fi
 
-    // Update image
-    if( err == "" )
+  // Get outputs
+  std::vector< std::string > outputs_names = filter->GetOutputsNames( );
+  for(
+    auto oIt = outputs_names.begin( );
+    oIt != outputs_names.end( );
+    ++oIt
+    )
+  {
+    std::string out_name = filter_name + "_" + *oIt;
+
+    TPlugins::TImage* image = filter->GetOutput< TPlugins::TImage >( *oIt );
+    if( image != NULL )
     {
-    TPluginImage* result = filter->GetOutput< TPluginImage >( 0 );
-    result->DisconnectPipeline( );
-    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)
+      if( filter_cate == "ImageToBinaryImageFilter" )
+      {
+        this->m_UI->MPR->ShowImage(
+          image->GetVTK< vtkImageData >( ),
+          out_name,
+          data_name, 1, 0, 0
+          );
+      }
+      else if( filter_cate == "ImageToImageFilter" )
+      {
+      } // fi
+
+      // Keep a track on a local data tree and go to next output
+      this->m_Objects[ out_name ] = TTreeNode( data_name, image );
+      continue;
+
+    } // fi
+
+    TPlugins::TMesh* mesh = filter->GetOutput< TPlugins::TMesh >( *oIt );
+    if( mesh != NULL )
     {
-    this->m_max_state = this->m_state;
-    }
-    }
-    else
-    QMessageBox::critical(
-    this,
-    tr( "Error executing filter" ),
-    tr( err.c_str( ) )
-    );
-  */
+      // Show mesh
+      this->_Block( );
+      this->m_UI->MPR->ShowMesh(
+        mesh->GetVTK< vtkPolyData >( ),
+        out_name,
+        data_name
+        );
+      this->_Unblock( );
+
+      // Keep a track on a local data tree and go to next output
+      this->m_Objects[ out_name ] = TTreeNode( data_name, mesh );
+      continue;
+
+    } // fi
+
+  } // rof
 }
 
 // -------------------------------------------------------------------------