...

[cpPlugins.git] / plugins / IO / ImageWriter.cxx

diff --git a/plugins/IO/ImageWriter.cxx b/plugins/IO/ImageWriter.cxx
index ac61cb94936cd6964808ed65ea59e09b4bbbedc7..d6d8f87c62cb8f34e58a73327d1bf99e9d347dce 100644 (file)
--- a/plugins/IO/ImageWriter.cxx
+++ b/plugins/IO/ImageWriter.cxx
@@ -1,11 +1,11 @@
-#include <plugins/IO/ImageWriter.h>
-#include <plugins/IO/ImageWriterQDialog.h>
+#include <IO/ImageWriter.h>
+#include <IO/ImageWriterQDialog.h>
 #include <cpPlugins/DataObjects/Image.h>
+#include <cpPlugins/DataObjects/Image_Demanglers.h>
+#include <cpInstances_SimpleImages.h>
 
 #include <itkImageFileWriter.h>
 
-#include <itkImageFileWriter.hxx>
-
 #ifdef cpPlugins_QT4
 
 #include <QApplication>
@@ -118,14 +118,28 @@ void cpPluginsIO::ImageWriter::
 _GenerateData( )
 {
   auto o = this->GetInputData( "Input" );
-  cpPlugins_Demangle_ImageScalars_Dims( o, _GD0 );
-  else this->_Error( "Invalid input image." );
+  cpPlugins_Demangle_Image_ProcessDims( o, _GD0, );
+  cpPlugins_Demangle_Image_DiffTensors3D( o, _GD1, );
+  this->_Error( "Invalid input image dimension." );
 }
 
 // -------------------------------------------------------------------------
 template< class _TImage >
 void cpPluginsIO::ImageWriter::
 _GD0( _TImage* image )
+{
+  cpPlugins_Demangle_Image_IntPixels( image, _GD1, _TImage::ImageDimension, );
+  cpPlugins_Demangle_Image_RealPixels( image, _GD1, _TImage::ImageDimension, );
+  cpPlugins_Demangle_Image_ScalarPixels( image, _GD1, _TImage::ImageDimension, );
+  cpPlugins_Demangle_Image_ColorPixels( image, _GD1, _TImage::ImageDimension, );
+  cpPlugins_Demangle_Image_VectorPixels( image, _GD1, _TImage::ImageDimension, );
+  this->_Error( "Invalid input image pixel type." );
+}
+
+// -------------------------------------------------------------------------
+template< class _TImage >
+void cpPluginsIO::ImageWriter::
+_GD1( _TImage* image )
 {
   typedef itk::ImageFileWriter< _TImage > _TFilter;