- multiple inputs in vv

[clitk.git] / tools / clitkImageArithm.cxx

diff --git a/tools/clitkImageArithm.cxx b/tools/clitkImageArithm.cxx
index 0b7b36e6bb47b90188b2f7a62b458980f372d14f..24e374844838dd8163a6b9bfab33746d6c84403a 100644 (file)
--- a/tools/clitkImageArithm.cxx
+++ b/tools/clitkImageArithm.cxx
@@ -23,88 +23,23 @@
 // clitk include
 #include "clitkImageArithm_ggo.h"
 #include "clitkImageArithmGenericFilter.h"
-#include "clitkIO.h"
 
 //--------------------------------------------------------------------
 int main(int argc, char * argv[]) {
 
   // Init command line
   GGO(clitkImageArithm, args_info);
-  CLITK_INIT;
 
-  // Check that we have either the 2nd image or the scalar 
-  if ((args_info.input2_given) && (args_info.scalar_given)) {
-    std::cerr << "ERROR : you cannot provide both --scalar and --input2 option" << std::endl;
-    exit(-1);
-  }
-  if ((!args_info.input2_given) && (!args_info.scalar_given)) {
-    if (args_info.operation_arg < 5) {
-      std::cerr << "Such operation need the --scalar option." << std::endl;
-      exit(-1);
-    }
-  }
-  
-  // Read input image header1 to check image dimension
-  itk::ImageIOBase::Pointer header1 = clitk::readImageHeader(args_info.input1_arg);
-  unsigned int dim1 = header1->GetNumberOfDimensions();
-  std::string pixelTypeName = header1->GetComponentTypeAsString(header1->GetComponentType());
-
-  // Options for arithm operation between 2 images
-  if (args_info.input2_given) {
-
-    itk::ImageIOBase::Pointer header2 = clitk::readImageHeader(args_info.input2_arg);
-    unsigned int dim2 = header2->GetNumberOfDimensions();
-    std::string pixelTypeName2 = header2->GetComponentTypeAsString(header1->GetComponentType());
-    
-    // Check dimension
-    if (dim1 != dim2) {
-      std::cerr << "Images should have the same dimension : " <<std::endl;
-      std::cerr  << "Dim input1 = " << dim1 << std::endl;
-      std::cerr  << "Dim input2 = " << dim2 << std::endl;
-      exit(-1);
-    }
-
-    // Check same type of pixel in both images
-    if (pixelTypeName2 != pixelTypeName) {
-      std::cerr << "Input images do not have the same pixel type: " <<std::endl;
-      std::cerr  << "Type input1 = " << pixelTypeName << std::endl;
-      std::cerr  << "Type input2 = " << pixelTypeName2 << std::endl;
-      exit(-1);
-    }
-    
-    // Check same X,Y dimensions in both images
-    for(int i=0; i<2; i++) {
-      if ( header1->GetDimensions(i) != header2->GetDimensions(i) ) {
-       std::cerr << "ERROR: input images should have same X and Y size. " <<  std::endl;
-       std::cerr << "Dimensions X and Y of input1: = " << header1->GetDimensions(0) <<"  "<< header1->GetDimensions(1)<< std::endl;
-       std::cerr << "Dimensions X and Y of input2: = " << header2->GetDimensions(0) <<"  "<< header2->GetDimensions(1)<< std::endl;
-       exit(-1);
-      }      
-    }//end for
-    if (dim1 == 3) {
-      if (header1->GetDimensions(2) < header2->GetDimensions(2)) {
-       std::cerr << "ERROR: Z size in input1 is greater than in input2. " <<  std::endl;
-       std::cerr << "Size input1 : " << header1->GetDimensions(0) << " " << header1->GetDimensions(1)<< " " << header1->GetDimensions(2) << std::endl;
-       std::cerr << "Size input2 : " << header2->GetDimensions(0) << " " << header2->GetDimensions(1)<< " " << header2->GetDimensions(2) << std::endl;
-      }
-    }
-  } //end if operation between 2 images        
-  
   // Creation of a generic filter
-  clitk::ImageArithmGenericFilter::Pointer filter = clitk::ImageArithmGenericFilter::New();
-  filter->AddInputFilename(args_info.input1_arg);
-  if (args_info.input2_given) filter->AddInputFilename(args_info.input2_arg);
-  else filter->SetScalar(args_info.scalar_arg);
-  //if (args_info.binary_given) filter->SetBinaryMaskFilename(args_info.binary_arg);   
-  filter->SetTypeOfOperation(args_info.operation_arg);   
-  filter->SetDefaultPixelValue(args_info.pixelValue_arg);
-  filter->SetOutputFilename(args_info.output_arg);
+  typedef clitk::ImageArithmGenericFilter<args_info_clitkImageArithm> FilterType;
+  FilterType::Pointer filter = FilterType::New();
 
-  // Go ! 
+  // Go !
+  filter->SetArgsInfo(args_info);
   filter->Update();
-
+  
   // this is the end my friend  
-  return 0;
+  return EXIT_SUCCESS;
 } // end main
 
 #endif //define CLITKIMAGEARITHM_CXX