...

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

diff --git a/plugins/IO/ImageWriter.cxx b/plugins/IO/ImageWriter.cxx
index 2058339b615061034cbc0c86f53d2eb6d7ebca55..8fcf08e7dce05f1a705936975403e724365f7b03 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,27 @@ void cpPluginsIO::ImageWriter::
 _GenerateData( )
 {
   auto o = this->GetInputData( "Input" );
-  cpPlugins_Demangle_ImageAll_Dims( o, _GD0 );
-  else this->_Error( "Invalid input image." );
+  cpPlugins_Demangle_Image_ProcessDims_1( o, _GD0 )
+    cpPlugins_Demangle_Image_DiffTensors3D_1( o, _GD1 )
+    this->_Error( "Invalid input image dimension." );
 }
 
 // -------------------------------------------------------------------------
 template< class _TImage >
 void cpPluginsIO::ImageWriter::
 _GD0( _TImage* image )
+{
+  cpPlugins_Demangle_Image_ScalarPixels_1( image, _GD1, _TImage::ImageDimension )
+    cpPlugins_Demangle_Image_ComplexPixels_1( image, _GD1, _TImage::ImageDimension )
+    cpPlugins_Demangle_Image_ColorPixels_1( image, _GD1, _TImage::ImageDimension )
+    cpPlugins_Demangle_Image_VectorPixels_1( 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;