X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=kernel%2Fsrc%2FbbtkVtkBlackBoxMacros.h;h=615af269c35f018852c26bf1ec82a1021b6eebd6;hb=d19c5512748aa8e2dd5c3d01d9ec680ded1f17ce;hp=0bf9acc1a4756efcfcdc6c5f65349bda5e0101ef;hpb=088ed2b7bdbe14eb85a7245afaa693ddb7a53a97;p=bbtk.git

diff --git a/kernel/src/bbtkVtkBlackBoxMacros.h b/kernel/src/bbtkVtkBlackBoxMacros.h
index 0bf9acc..615af26 100644
--- a/kernel/src/bbtkVtkBlackBoxMacros.h
+++ b/kernel/src/bbtkVtkBlackBoxMacros.h
@@ -93,14 +93,29 @@
 */
 
 
+//EED 2017-01-01 Migration VTK7
+#if VTK_MAJOR_VERSION <= 5
+
 //===========================================================================
 /// Declares a vtkImageAlgorithm-inherited AtomicBlackBox input 
-#define BBTK_DECLARE_VTK_IMAGE_ALGORITHM_INPUT(NAME,TYPE)		\
-  public:								\
-  TYPE bbGetInput##NAME ()						\
-  { if (mVtkObject) return mVtkObject->GetImageDataInput(0); return 0; } \
-  void bbSetInput##NAME (TYPE d)					\
-  { if (mVtkObject) mVtkObject->SetInput( (vtkDataObject*) d); }
+  #define BBTK_DECLARE_VTK_IMAGE_ALGORITHM_INPUT(NAME,TYPE)		\
+    public:								\
+    TYPE bbGetInput##NAME ()						\
+    { if (mVtkObject) return mVtkObject->GetImageDataInput(0); return 0; } \
+    void bbSetInput##NAME (TYPE d)					\
+    { if (mVtkObject) mVtkObject->SetInput( (vtkDataObject*) d); }
+#else
+//===========================================================================
+/// Declares a vtkImageAlgorithm-inherited AtomicBlackBox input 
+  #define BBTK_DECLARE_VTK_IMAGE_ALGORITHM_INPUT(NAME,TYPE)		\
+    public:								\
+    TYPE bbGetInput##NAME ()						\
+    { if (mVtkObject) return mVtkObject->GetImageDataInput(0); return 0; } \
+    void bbSetInput##NAME (TYPE d)					\
+    { if (mVtkObject) mVtkObject->SetInputData( (vtkDataObject*) d); }
+
+#endif
+
 
 //===========================================================================
 
@@ -126,6 +141,11 @@
 //===========================================================================
 
 //===========================================================================
+
+//EED 2017-01-01 Migration VTK7 					
+
+#if (VTK_MAJOR_VERSION <= 5) 						
+
 /// Declares a vtkAlgorithm-inherited AtomicBlackBox input 
 #define BBTK_DECLARE_VTK_INPUT(NAME,TYPE)				\
   public:								\
@@ -135,6 +155,24 @@
   void bbSetInput##NAME (TYPE d)					\
   { if (mVtkObject)  mVtkObject->SetInput( (vtkDataObject*) d); }
 
+#endif 									
+
+
+
+#if (VTK_MAJOR_VERSION >= 6) 						
+
+/// Declares a vtkAlgorithm-inherited AtomicBlackBox input 
+#define BBTK_DECLARE_VTK_INPUT(NAME,TYPE)				\
+  public:								\
+  TYPE bbGetInput##NAME ()						\
+  { if (mVtkObject) return dynamic_cast<TYPE>(mVtkObject->GetInput());	\
+    return 0;}								\
+  void bbSetInput##NAME (TYPE d)					\
+  { if (mVtkObject)  mVtkObject->SetInputData( (vtkDataObject*) d); }
+
+#endif 									
+
+
 //===========================================================================
 
 //===========================================================================