make difference between 2D and 3D with a single slice image (Dim obtained from ITK...

[clitk.git] / common / vvImage.txx

diff --git a/common/vvImage.txx b/common/vvImage.txx
index 4532a68f13819fb53024905b6ed2c29e1db5d00d..1561a13d447a52256d0f02c02b323a63a4aeffe3 100755 (executable)
--- a/common/vvImage.txx
+++ b/common/vvImage.txx
@@ -4,6 +4,7 @@
 template<class TItkImageType>
 void vvImage::AddItkImage(TItkImageType *input)
 {
+  mImageDimension = TItkImageType::ImageDimension; 
   typedef itk::ImageToVTKImageFilter <TItkImageType> ConverterType;
   typename ConverterType::Pointer converter = ConverterType::New();
   converter->SetInput(input);