VF composition with BSpline coefficients

[clitk.git] / tools / clitkComposeVF.cxx

diff --git a/tools/clitkComposeVF.cxx b/tools/clitkComposeVF.cxx
index 08cff6585d0e52022ad4ac23e83be208b36b7c4d..c2cead24fd92e323f35d0a70ed2839c576b874ce 100644 (file)
--- a/tools/clitkComposeVF.cxx
+++ b/tools/clitkComposeVF.cxx
@@ -28,18 +28,27 @@ int main( int argc, char *argv[] )
   // Init command line
   GGO(clitkComposeVF, args_info);
   CLITK_INIT;
+
+  if (args_info.type_arg == 1 && !args_info.like_given) {
+    std::cerr << "--like must be used when type = 1" << std::endl;
+    return -1;
+  }
+
   
   //Creation of the generic filter
   clitk::ComposeVFGenericFilter::Pointer ComposeVFGenericFilter= clitk::ComposeVFGenericFilter::New();
-
+  
   //Pass the parameters
   ComposeVFGenericFilter->SetInput1(args_info.input1_arg);
   ComposeVFGenericFilter->SetInput2(args_info.input2_arg);
   ComposeVFGenericFilter->SetOutput(args_info.output_arg);
+  ComposeVFGenericFilter->SetInputType(args_info.type_arg);
+  if (args_info.like_given)
+    ComposeVFGenericFilter->SetLikeImage(args_info.like_arg);
   //JV how to pass for different dims?
   //ComposeVFGenericFilter->SetEdgePaddingValue(args_info.pad_arg);
   ComposeVFGenericFilter->SetVerbose(args_info.verbose_flag);
-  ComposeVFGenericFilter->Update();  
+  CLITK_TRY_CATCH_EXIT(ComposeVFGenericFilter->Update());  
 
   return EXIT_SUCCESS;
 }