fix VF problem.

[clitk.git] / filters / clitkGuerreroVentilationGenericFilter.cxx

diff --git a/filters/clitkGuerreroVentilationGenericFilter.cxx b/filters/clitkGuerreroVentilationGenericFilter.cxx
index 426765ba89397249030e918a6260c17fa5bcfea2..1a2dcd66e74155daf09c5723165409d38cff9127 100644 (file)
--- a/filters/clitkGuerreroVentilationGenericFilter.cxx
+++ b/filters/clitkGuerreroVentilationGenericFilter.cxx
@@ -19,69 +19,45 @@
  * @brief  
  -------------------------------------------------------------------*/
 
+#include <sstream>
+#include <vector>
 #include "clitkGuerreroVentilationGenericFilter.h"
 #include <itkBinaryGuerreroFilter.h>
 #include <itkImageDuplicator.h>
-//--------------------------------------------------------------------
-clitk::GuerreroVentilationGenericFilter::GuerreroVentilationGenericFilter() 
-{
-    blood_mass_factor=1.;
-}
-//--------------------------------------------------------------------
+#include <itkExtractImageFilter.h>
 
 //--------------------------------------------------------------------
-void clitk::GuerreroVentilationGenericFilter::Update () {
-  
-  // Determine dim, pixel type, number of components
-  this->GetInputImageDimensionAndPixelType(mDim,mPixelTypeName,mNbOfComponents);
-  
-  // Switch by dimension
-  if (mDim == 3) { Update_WithDim<3>(); return; }
-  if (mDim == 2) { Update_WithDim<2>(); return; }
-  std::cerr << "Error, dimension of input image is " << mDim << ", but I only work with 2 or 3." << std::endl;
-  exit(0);
+clitk::GuerreroVentilationGenericFilter::GuerreroVentilationGenericFilter() 
+  :ImageToImageGenericFilter<Self>("GuerreroVentilationGenericFilter") {
+  blood_mass_factor=1.;
+  InitializeImageType<2>();
+  InitializeImageType<3>();
 }
 //--------------------------------------------------------------------
 
-//This is where you put the actual implementation
-
-#include <sstream>
-#include <itkExtractImageFilter.h>
-
-
 //--------------------------------------------------------------------
 template<unsigned int Dim>
-void clitk::GuerreroVentilationGenericFilter::Update_WithDim() { 
-#define TRY_TYPE(TYPE)                                                 \
-  if (IsSameType<TYPE>(mPixelTypeName)) { Update_WithDimAndPixelType<Dim, TYPE>(); return; } 
-  // TRY_TYPE(signed char);
-  // TRY_TYPE(uchar);
-  TRY_TYPE(short);
-  //TRY_TYPE(ushort);
-  // TRY_TYPE(int);
-//   TRY_TYPE(unsigned int); 
-  //TRY_TYPE(float);
-  // TRY_TYPE(double);
-#undef TRY_TYPE
-
-  std::string list = CreateListOfTypes<uchar, short, ushort, int, uint, float, double>();
-  std::cerr << "Error, I don't know the type '" << mPixelTypeName << "' for the input image '"
-           << mInputFilenames[0] << "'." << std::endl << "Known types are " << list << std::endl;
-  exit(0);
+void clitk::GuerreroVentilationGenericFilter::InitializeImageType() {      
+  ADD_IMAGE_TYPE(Dim, short);
 }
 //--------------------------------------------------------------------
 
+
 //--------------------------------------------------------------------
-template<unsigned int Dim, class PixelType>
-void clitk::GuerreroVentilationGenericFilter::Update_WithDimAndPixelType() {
+template<class ImageType>
+void clitk::GuerreroVentilationGenericFilter::UpdateWithInputImageType() {
 
-    // Read input
+    // Input should be 2
     assert(mInputFilenames.size() == 2);
-    typedef itk::Image<PixelType,Dim> ImageType;
-    typedef itk::Image<float,Dim> OutputImageType;
-    typename ImageType::Pointer input = clitk::readImage<ImageType>(mInputFilenames[0], mIOVerbose);
-    typename ImageType::Pointer ref = clitk::readImage<ImageType>(mInputFilenames[1], mIOVerbose);
 
+    // Reading input
+    typedef ImageType InputImageType;
+    typename InputImageType::Pointer input = this->template GetInput<InputImageType>(0);
+    typename InputImageType::Pointer ref = this->template GetInput<InputImageType>(1);
+
+    typedef itk::Image<float,InputImageType::ImageDimension> OutputImageType;
+    //    typename ImageType::Pointer input = clitk::readImage<ImageType>(mInputFilenames[0], mIOVerbose);
+    //typename ImageType::Pointer ref = clitk::readImage<ImageType>(mInputFilenames[1], mIOVerbose);
 
     typedef itk::BinaryGuerreroFilter<ImageType,ImageType,OutputImageType> GFilterType;
     typename GFilterType::Pointer filter = GFilterType::New();
@@ -90,6 +66,7 @@ void clitk::GuerreroVentilationGenericFilter::Update_WithDimAndPixelType() {
     filter->SetBloodCorrectionFactor(blood_mass_factor);
     filter->SetUseCorrectFormula(use_correct_formula);
     filter->Update();
+
     this->SetNextOutput<OutputImageType>(filter->GetOutput());
     //clitk::writeImage<OutputImageType>(filter->GetOutput(), mOutputFilename, mIOVerbose);
     //std::cout << "Warning: removed " << filter->GetFunctor().aberant_voxels << " aberant voxels from the ventilation image"