Replace vvImage::Pointer with raw pointer for function

[clitk.git] / common / vvImage.txx

diff --git a/common/vvImage.txx b/common/vvImage.txx
index 06f44af400ef2d75dc0bb12936ed80289672d84f..a84fa76669d7a317655d725e9e2e8371fd8f133c 100755 (executable)
--- a/common/vvImage.txx
+++ b/common/vvImage.txx
@@ -13,11 +13,14 @@ void vvImage::AddItkImage(TItkImageType *input)
   converter->Update();
   mVtkImages.push_back( converter->GetOutput() );
   
-  // Account for direction in transform. The offset is already accounted for
+ // Account for direction in transform. The offset is already accounted for
   // in the VTK image coordinates, no need to put it in the transform.
+  vtkSmartPointer<vtkMatrix4x4> matrix = vtkSmartPointer<vtkMatrix4x4>::New();
+  matrix->Identity();
   for(unsigned int j=0; j<input->GetImageDimension(); j++)
     for(unsigned int i=0; i<input->GetImageDimension(); i++)
-      mTransform->GetMatrix()->SetElement(i,j, input->GetDirection()[i][j]);
+      (*matrix)[i][j] = input->GetDirection()[i][j];
+  mTransform->SetMatrix(matrix);
 
   // Create the corresponding transformed image
   mVtkImageReslice.push_back(vtkSmartPointer<vtkImageReslice>::New());