Bugfix: Handle treatment plans that contain number of protons

[clitk.git] / registration / clitkMatrixTransformToVFGenericFilter.h

diff --git a/registration/clitkMatrixTransformToVFGenericFilter.h b/registration/clitkMatrixTransformToVFGenericFilter.h
index 717e938a4b667b86dfaebc3eea570d578789e222..9aa54895d3fb409b4144843b4e607e56c8ef2298 100644 (file)
--- a/registration/clitkMatrixTransformToVFGenericFilter.h
+++ b/registration/clitkMatrixTransformToVFGenericFilter.h
@@ -37,9 +37,11 @@
 //itk include
 #include "itkLightObject.h"
 #if ITK_VERSION_MAJOR >= 4
-  #include "itkTransformToDisplacementFieldSource.h"
-#else
-  #include "itkTransformToDeformationFieldSource.h"
+#  if ITK_VERSION_MINOR < 6
+#    include "itkTransformToDisplacementFieldSource.h"
+#  else
+#    include "itkTransformToDisplacementFieldFilter.h"
+#  endif
 #endif
 #include "itkAffineTransform.h"