Debug Rigid Registration tool cancel

[clitk.git] / registration / clitkMatrixTransformToVFGenericFilter.h

diff --git a/registration/clitkMatrixTransformToVFGenericFilter.h b/registration/clitkMatrixTransformToVFGenericFilter.h
index 34648cb928e7035329a764b79c9f4b5ed8f67adc..9aa54895d3fb409b4144843b4e607e56c8ef2298 100644 (file)
--- a/registration/clitkMatrixTransformToVFGenericFilter.h
+++ b/registration/clitkMatrixTransformToVFGenericFilter.h
@@ -36,7 +36,13 @@
 
 //itk include
 #include "itkLightObject.h"
-#include "itkTransformToDeformationFieldSource.h"
+#if ITK_VERSION_MAJOR >= 4
+#  if ITK_VERSION_MINOR < 6
+#    include "itkTransformToDisplacementFieldSource.h"
+#  else
+#    include "itkTransformToDisplacementFieldFilter.h"
+#  endif
+#endif
 #include "itkAffineTransform.h"
 
 namespace clitk