...

[cpPlugins.git] / lib / cpPlugins / Plugins / IO / ImageWriter.cxx

diff --git a/lib/cpPlugins/Plugins/IO/ImageWriter.cxx b/lib/cpPlugins/Plugins/IO/ImageWriter.cxx
index 6a0de2ed09dc730cdeab376e3c7ce3c9684e7b43..234c916f99f5955407294607dc9edd948ebc1eeb 100644 (file)
--- a/lib/cpPlugins/Plugins/IO/ImageWriter.cxx
+++ b/lib/cpPlugins/Plugins/IO/ImageWriter.cxx
@@ -8,13 +8,12 @@ cpPlugins::IO::ImageWriter::
 ImageWriter( )
   : Superclass( )
 {
-  this->m_ClassName = "cpPlugins::IO::ImageWriter";
-  this->m_ClassCategory = "ImageWriter";
-  this->SetNumberOfInputs( 1 );
-
-  using namespace cpPlugins::Interface;
-  this->m_DefaultParameters.Configure( Parameters::String, "FileName" );
-  this->m_Parameters = this->m_DefaultParameters;
+  this->_AddInput( "Input" );
+  this->m_Parameters->ConfigureAsSaveFileName( "FileName" );
+  this->m_Parameters->SetAcceptedFileExtensions(
+    "FileName",
+    "Image files (*.bmp *.png *.jpg *.jpeg *.dcm *.mhd *.nhdr *.nrrd *.tiff)"
+    );
 }
 
 // -------------------------------------------------------------------------
@@ -27,32 +26,39 @@ cpPlugins::IO::ImageWriter::
 std::string cpPlugins::IO::ImageWriter::
 _GenerateData( )
 {
-  cpPlugins::Interface::Image* image =
-    this->GetInput< cpPlugins::Interface::Image >( 0 );
-  if( image == NULL )
-    return( "ImageWriter: No input image." );
+  // Thank you very much Microsoft !!! :-@
+  std::string   r = this->_GD0_Image< 2 >( );
+  if( r != "" ) r = this->_GD0_Image< 3 >( );
+  if( r != "" ) r = this->_GD0_Image< 4 >( );
+  if( r != "" ) r = this->_GD0_VectorImage< 2 >( );
+  if( r != "" ) r = this->_GD0_VectorImage< 3 >( );
+  if( r != "" ) r = this->_GD0_VectorImage< 4 >( );
+  return( r );
+}
 
+// -------------------------------------------------------------------------
+template< unsigned int D >
+std::string cpPlugins::IO::ImageWriter::
+_GD0_Image( )
+{
+  auto image = this->GetInputData< cpPlugins::Interface::Image >( "Input" );
   itk::DataObject* itk_image = NULL;
   std::string r = "";
-  cpPlugins_Image_Input_Demangle_Dimension_AllTypes(
-    2, image, itk_image, r, _RealGD
-    );
-  else cpPlugins_Image_Input_Demangle_Dimension_AllTypes(
-    3, image, itk_image, r, _RealGD
-    );
-  else cpPlugins_Image_Input_Demangle_Dimension_AllTypes(
-    4, image, itk_image, r, _RealGD
-    );
-#ifndef _WIN32 // win32: Nested loops error ????
-  else cpPlugins_Image_Input_Demangle(
-    itk::DiffusionTensor3D< float >, 3, image, itk_image, r, _RealGD
-    );
-  else cpPlugins_Image_Input_Demangle(
-    itk::DiffusionTensor3D< double >, 3, image, itk_image, r, _RealGD
-    );
-#endif // _WIN32
+  cpPlugins_Image_Demangle_AllTypes( D, image, itk_image, r, _RealGD );
   else r = "ImageWriter: Input image type not supported.";
+  return( r );
+}
 
+// -------------------------------------------------------------------------
+template< unsigned int D >
+std::string cpPlugins::IO::ImageWriter::
+_GD0_VectorImage( )
+{
+  auto image = this->GetInputData< cpPlugins::Interface::Image >( "Input" );
+  itk::DataObject* itk_image = NULL;
+  std::string r = "";
+  cpPlugins_VectorImage_Demangle_AllTypes( D, image, itk_image, r, _RealGD );
+  else r = "ImageWriter: Input image type not supported.";
   return( r );
 }
 
@@ -62,17 +68,8 @@ std::string cpPlugins::IO::ImageWriter::
 _RealGD( itk::DataObject* image )
 {
   typedef itk::ImageFileWriter< I > _W;
-  
-  // Get filename
-  std::string fname = this->m_Parameters.GetValueAsString( "FileName" );
-  _W* writer = dynamic_cast< _W* >( this->m_RealProcessObject.GetPointer( ) );
-  if( writer == NULL )
-  {
-    this->m_RealProcessObject = _W::New( );
-    writer = dynamic_cast< _W* >( this->m_RealProcessObject.GetPointer( ) );
-
-  } // fi
-  writer->SetFileName( fname );
+  _W* writer = this->_CreateITK< _W >( );
+  writer->SetFileName( this->m_Parameters->GetSaveFileName( "FileName" ) );
   writer->SetInput( dynamic_cast< I* >( image ) );
   try
   {