plugins/IO/ImageWriter.cxx

   1 #include <IO/ImageWriter.h>
   2 #include <cpPlugins/DataObjects/Image.h>
   3 #include <cpPlugins/DataObjects/Image_Demanglers.h>
   4 #include <cpPlugins/QT/SaveFileDialog.h>
   5 #include <cpInstances_SimpleImages.h>
   6
   7 #include <itkImageFileWriter.h>
   8
   9 #ifdef cpPlugins_QT4
  10 #  include <QApplication>
  11 #endif // cpPlugins_QT4
  12
  13 // -------------------------------------------------------------------------
  14 QDialog* cpPluginsIO::ImageWriter::
  15 CreateQDialog( )
  16 {
  17 #ifdef cpPlugins_QT4
  18   cpPlugins::QT::SaveFileDialog* dlg = NULL;
  19   if( QApplication::instance( ) != NULL )
  20   {
  21     dlg = new cpPlugins::QT::SaveFileDialog( );
  22     dlg->SetParameters( &( this->m_Parameters ), "FileName" );
  23
  24   } // fi
  25   return( dlg );
  26 #else // cpPlugins_QT4
  27   return( NULL );
  28 #endif // cpPlugins_QT4
  29 }
  30
  31 // -------------------------------------------------------------------------
  32 cpPluginsIO::ImageWriter::
  33 ImageWriter( )
  34   : Superclass( )
  35 {
  36   this->_ConfigureInput< cpPlugins::DataObjects::Image >( "Input", true, false );
  37   this->m_Parameters.Clear( );
  38   this->m_Parameters.ConfigureAsSaveFileName( "FileName", "" );
  39   this->m_Parameters.SetAcceptedFileExtensions(
  40     "FileName",
  41     "Image files (*.bmp *.png *.jpg *.jpeg *.dcm *.mhd *.nhdr *.nrrd *.tiff)"
  42     );
  43 }
  44
  45 // -------------------------------------------------------------------------
  46 cpPluginsIO::ImageWriter::
  47 ~ImageWriter( )
  48 {
  49 }
  50
  51 // -------------------------------------------------------------------------
  52 void cpPluginsIO::ImageWriter::
  53 _GenerateData( )
  54 {
  55   auto o = this->GetInputData( "Input" );
  56   cpPlugins_Demangle_Image_ProcessDims_1( o, _GD0 )
  57     cpPlugins_Demangle_Image_DiffTensors3D_1( o, _GD1 )
  58     this->_Error( "Invalid input image dimension." );
  59 }
  60
  61 // -------------------------------------------------------------------------
  62 template< class _TImage >
  63 void cpPluginsIO::ImageWriter::
  64 _GD0( _TImage* image )
  65 {
  66   cpPlugins_Demangle_Image_ScalarPixels_1( image, _GD1, _TImage::ImageDimension )
  67     cpPlugins_Demangle_Image_ComplexPixels_1( image, _GD1, _TImage::ImageDimension )
  68     cpPlugins_Demangle_Image_ColorPixels_1( image, _GD1, _TImage::ImageDimension )
  69     cpPlugins_Demangle_Image_VectorPixels_1( image, _GD1, _TImage::ImageDimension )
  70     this->_Error( "Invalid input image pixel type." );
  71 }
  72
  73 // -------------------------------------------------------------------------
  74 template< class _TImage >
  75 void cpPluginsIO::ImageWriter::
  76 _GD1( _TImage* image )
  77 {
  78   typedef itk::ImageFileWriter< _TImage > _TFilter;
  79
  80   auto f = this->_CreateITK< _TFilter >( );
  81   f->SetFileName( this->m_Parameters.GetSaveFileName( "FileName" ) );
  82   f->SetInput( image );
  83   try
  84   {
  85     f->Update( );
  86   }
  87   catch( itk::ExceptionObject& err )
  88   {
  89     this->_Error( err.GetDescription( ) );
  90
  91   } // yrt
  92 }
  93
  94 // eof - $RCSfile$