]> Creatis software - clitk.git/blobdiff - common/vvImage.cxx
Enable resliced images, put in mTransformedVtkImages
[clitk.git] / common / vvImage.cxx
index 9b9ca8289956b832b19e9dca0b6b47300208c6e1..1f5116849cb711a73cc80cfa19954c1bcfd7c6ab 100644 (file)
@@ -1,7 +1,7 @@
 /*=========================================================================
   Program:   vv                     http://www.creatis.insa-lyon.fr/rio/vv
 
-  Authors belong to: 
+  Authors belong to:
   - University of LYON              http://www.universite-lyon.fr/
   - Léon Bérard cancer center       http://oncora1.lyon.fnclcc.fr
   - CREATIS CNRS laboratory         http://www.creatis.insa-lyon.fr
 ======================================================================-====*/
 #ifndef VVIMAGE_CXX
 #define VVIMAGE_CXX
+
+// vv
 #include "vvImage.h"
-#include "vtkImageData.h"
+
+// clitk
 #include "clitkCommon.h"
-#include <cassert>
 
-//--------------------------------------------------------------------
-vvImage::vvImage() {
-  mVtkImages.resize(0);
-}
-//--------------------------------------------------------------------
+// vtk
+#include <vtkImageData.h>
+#include <vtkImageReslice.h>
+#include <vtkTransform.h>
 
+// std
+#include <cassert>
 
 //--------------------------------------------------------------------
-vvImage::~vvImage() {
-  for (unsigned int i = 0; i < mVtkImages.size(); i++) {
-    if (mVtkImages[i] != NULL)
-      mVtkImages[i]->Delete();
-  }
+vvImage::vvImage():mTransform(vtkSmartPointer<vtkTransform>::New())
+{
+  Init();
 }
 //--------------------------------------------------------------------
 
 
 //--------------------------------------------------------------------
-void vvImage::SetImage(std::vector<vtkImageData*> images) {
-  for (unsigned int i = 0; i < mVtkImages.size(); i++) {
-    if (mVtkImages[i] != NULL)
-      mVtkImages[i]->Delete();
-  }
-  mVtkImages.resize(0);
-  for (unsigned int i = 0; i < images.size(); i++) {
-    mVtkImages.push_back(images[i]);
-  }
+void vvImage::Init()
+{
+  mTimeSpacing = 1;
+  mTimeOrigin = 0;
+  mImageDimension = 0;
 }
 //--------------------------------------------------------------------
 
 //--------------------------------------------------------------------
-void vvImage::AddImage(vtkImageData* image) {
-  mVtkImages.push_back(image);
+vvImage::~vvImage()
+{
+  //  DD("vvImage got deleted!");
+  Reset();
 }
 //--------------------------------------------------------------------
 
 //--------------------------------------------------------------------
-void vvImage::Init() {
-  for (unsigned int i = 0; i < mVtkImages.size(); i++) {
-    if (mVtkImages[i] != NULL)
-      mVtkImages[i]->Delete();
-  }
+void vvImage::Reset()
+{
   mVtkImages.resize(0);
+  mTransformedVtkImages.resize(0);
+  mVtkImageReslice.resize(0);
+
+  Init();
 }
 //--------------------------------------------------------------------
 
-
 //--------------------------------------------------------------------
-int vvImage::GetNumberOfSpatialDimensions() {
-  int dim=GetNumberOfDimensions();
-  if (IsTimeSequence())
-    return dim-1;
-  else
-    return dim;
+int vvImage::GetNumberOfSpatialDimensions()
+{
+  return mImageDimension;
 }
 //--------------------------------------------------------------------
 
-
 //--------------------------------------------------------------------
-int vvImage::GetNumberOfDimensions() const {
-  if (mVtkImages.size())
-    {
-      int dimension = 2;
-      int extent[6];
-      mVtkImages[0]->GetWholeExtent(extent);
-      if (extent[5] - extent[4] >= 1)
-        dimension++;
-      if (mVtkImages.size() > 1)
-        dimension++;
-      return dimension;
-    }
+int vvImage::GetNumberOfDimensions() const
+{
+  if (mVtkImages.size()) {
+    if (IsTimeSequence())
+      return mImageDimension+1;
+    else
+      return mImageDimension;
+  }
   return 0;
 }
 //--------------------------------------------------------------------
 
-
 //--------------------------------------------------------------------
 void vvImage::GetScalarRange(double* range)
 {
   assert(mVtkImages.size());
   double * temp = mVtkImages[0]->GetScalarRange();
-  range[0]=temp[0];range[1]=temp[1];
-  for (unsigned int i=1;i<mVtkImages.size();i++)
-    {
-      temp = mVtkImages[i]->GetScalarRange();
-      if (temp[0] < range[0]) range[0]=temp[0];
-      if (temp[1] > range[1]) range[1]=temp[1];
-    }
+  range[0]=temp[0];
+  range[1]=temp[1];
+  for (unsigned int i=1; i<mVtkImages.size(); i++) {
+    temp = mVtkImages[i]->GetScalarRange();
+    if (temp[0] < range[0]) range[0]=temp[0];
+    if (temp[1] > range[1]) range[1]=temp[1];
+  }
 }
 //--------------------------------------------------------------------
 
-
 //--------------------------------------------------------------------
-std::string vvImage::GetScalarTypeAsString() {
+std::string vvImage::GetScalarTypeAsITKString()
+{
   // WARNING VTK pixel type different from ITK Pixel type
   std::string vtktype = mVtkImages[0]->GetScalarTypeAsString();
   if (vtktype == "unsigned char") return "unsigned_char";
@@ -125,71 +115,66 @@ std::string vvImage::GetScalarTypeAsString() {
 }
 //--------------------------------------------------------------------
 
-
 //--------------------------------------------------------------------
-int vvImage::GetNumberOfScalarComponents() {
+int vvImage::GetNumberOfScalarComponents()
+{
   return mVtkImages[0]->GetNumberOfScalarComponents();
 }
 //--------------------------------------------------------------------
 
-
 //--------------------------------------------------------------------
-int vvImage::GetScalarSize() {
+int vvImage::GetScalarSize()
+{
   return mVtkImages[0]->GetScalarSize();
 }
 //--------------------------------------------------------------------
 
-
 //--------------------------------------------------------------------
-std::vector<double> vvImage::GetSpacing() {
+std::vector<double> vvImage::GetSpacing()
+{
   std::vector<double> spacing;
   int dim = this->GetNumberOfDimensions();
-  for (int i = 0; i < dim; i++)
-    {
-      if (i == 3)
-        spacing.push_back(1);
-      else
-        spacing.push_back(mVtkImages[0]->GetSpacing()[i]);
-    }
+  for (int i = 0; i < dim; i++) {
+    if (i == 3)
+      spacing.push_back(mTimeSpacing);
+    else
+      spacing.push_back(mVtkImages[0]->GetSpacing()[i]);
+  }
   return spacing;
 }
 //--------------------------------------------------------------------
-
-
-//--------------------------------------------------------------------
-std::vector<double> vvImage::GetOrigin() const {
+std::vector<double> vvImage::GetOrigin() const
+{
   std::vector<double> origin;
   int dim = this->GetNumberOfDimensions();
-  for (int i = 0; i < dim; i++)
-    {
-      if (i == 3)
-        origin.push_back(0);
-      else
-        origin.push_back(mVtkImages[0]->GetOrigin()[i]);
-    }
+  for (int i = 0; i < dim; i++) {
+    if (i == 3)
+      origin.push_back(mTimeOrigin);
+    else
+      origin.push_back(mVtkImages[0]->GetOrigin()[i]);
+  }
   return origin;
 }
 //--------------------------------------------------------------------
 
-
 //--------------------------------------------------------------------
-std::vector<int> vvImage::GetSize() {
+std::vector<int> vvImage::GetSize()
+{
   std::vector<int> size0;
   int dim = this->GetNumberOfDimensions();
-  for (int i = 0; i < dim; i++)
-    {
-      if (i == 3)
-        size0.push_back(mVtkImages.size());
-      else
-        size0.push_back(mVtkImages[0]->GetDimensions()[i]);
-    }
+  for (int i = 0; i < dim; i++) {
+    if (i == 3)
+      size0.push_back(mVtkImages.size());
+    else
+      size0.push_back(mVtkImages[0]->GetDimensions()[i]);
+  }
   return size0;
 }
 //--------------------------------------------------------------------
 
-
 //--------------------------------------------------------------------
-unsigned long vvImage::GetActualMemorySize() {
+unsigned long vvImage::GetActualMemorySize()
+{
   unsigned long size = 0;
   for (unsigned int i = 0; i < mVtkImages.size(); i++) {
     size += mVtkImages[i]->GetActualMemorySize();
@@ -200,37 +185,46 @@ unsigned long vvImage::GetActualMemorySize() {
 
 
 //--------------------------------------------------------------------
-bool vvImage::IsTimeSequence() {
+bool vvImage::IsTimeSequence() const
+{
   return mVtkImages.size()>1;
 }
 //--------------------------------------------------------------------
 
 
 //--------------------------------------------------------------------
-const std::vector<vtkImageData*>& vvImage::GetVTKImages() {
+const std::vector<vtkImageData*>& vvImage::GetVTKImages()
+{
   return mVtkImages;
 }
 //--------------------------------------------------------------------
 
-
 //--------------------------------------------------------------------
-vtkImageData* vvImage::GetFirstVTKImageData() { 
-  return mVtkImages[0]; 
+const std::vector<vtkImageData*>& vvImage::GetTransformedVTKImages()
+{
+  return mTransformedVtkImages;
 }
 //--------------------------------------------------------------------
 
+//--------------------------------------------------------------------
+vtkImageData* vvImage::GetFirstVTKImageData()
+{
+  return mVtkImages[0];
+}
+//--------------------------------------------------------------------
 
 //--------------------------------------------------------------------
-bool vvImage::IsScalarTypeInteger() {
+bool vvImage::IsScalarTypeInteger()
+{
   assert(mVtkImages.size()> 0);
   int t = mVtkImages[0]->GetScalarType();
   return IsScalarTypeInteger(t);
 }
 //--------------------------------------------------------------------
 
-
 //--------------------------------------------------------------------
-bool vvImage::IsScalarTypeInteger(int t) {
+bool vvImage::IsScalarTypeInteger(int t)
+{
   if ((t == VTK_BIT) ||
       (t == VTK_CHAR) ||
       (t == VTK_UNSIGNED_CHAR) ||
@@ -241,12 +235,27 @@ bool vvImage::IsScalarTypeInteger(int t) {
       (t == VTK_LONG) ||
       (t == VTK_UNSIGNED_LONG))    {
     return true;
-  }
-  else {
+  } else {
     return false;
   }
 }
+
 //--------------------------------------------------------------------
 
+//--------------------------------------------------------------------
+vtkSmartPointer<vtkTransform> vvImage::GetTransform()
+{
+  return this->mTransform;
+}
+//--------------------------------------------------------------------
+
+//--------------------------------------------------------------------
+void vvImage::UpdateReslice()
+{
+  for (unsigned int i=0; i<mVtkImageReslice.size(); i++) {
+    mVtkImageReslice[i]->Update();
+  }
+}
+//--------------------------------------------------------------------
 
 #endif // VVIMAGE_CXX