Detect VTK version during configure

gandreadis · Oct 26, 2016 · f726437 · f726437
1 parent 9bc1866
commit f726437
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/ANTS.cmake b/ANTS.cmake
@@ -65,10 +65,14 @@ if(USE_VTK)
    vtkImagingStencil
    vtkImagingGeneral
    vtkRenderingAnnotation
-   vtkRenderingVolumeOpenGL
- #   vtkRenderingVolumeOpenGL2 # VTK7
    )
 
+  if(VTK_VERSION_MAJOR GREATER 6)
+    find_package(VTK COMPONENTS vtkRenderingVolumeOpenGL2)
+  else(VTK_VERSION_MAJOR GREATER 6)
+     find_package(VTK COMPONENTS vtkRenderingVolumeOpenGL)
+  endif(VTK_VERSION_MAJOR GREATER 6)
+
   if(VTK_FOUND)
     include(${VTK_USE_FILE})
     include_directories(${VTK_INCLUDE_DIRS})