]> Creatis software - clitk.git/blobdiff - common/vvImage.cxx
Added Varian OBI file format
[clitk.git] / common / vvImage.cxx
index 1d718bce26d6fb5d24498d02bbe01e5f3bddf5dc..55a56ef2f3e75cd3351b28b3bd475f6afafbace0 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
 #include "vvImage.h"
-#include "vtkImageData.h"
 #include "clitkCommon.h"
+#include <vtkImageData.h>
+#include <vtkImageReslice.h>
 #include <cassert>
 
+#include <vtkGeneralTransform.h>
+#include <vtkTransform.h>
+#include <vtkGridTransform.h>
+
 //--------------------------------------------------------------------
-vvImage::vvImage() {
-  mVtkImages.resize(0);
-  mTimeSpacing = 1;
-  mTimeOrigin = 0;
+vvImage::vvImage():mGeneralTransform(vtkSmartPointer<vtkGeneralTransform>::New()),
+                  mTransform       (vtkSmartPointer<vtkTransform>::New()),
+                  mGridTransform   (vtkSmartPointer<vtkGridTransform>::New())
+{
+  mGeneralTransform->Concatenate(mTransform);
+  mGeneralTransform->Concatenate(mGridTransform);
+  Init();
 }
 //--------------------------------------------------------------------
 
 
 //--------------------------------------------------------------------
-vvImage::~vvImage() {
-  for (unsigned int i = 0; i < mVtkImages.size(); i++) {
-    if (mVtkImages[i] != NULL)
-      mVtkImages[i]->Delete();
-  }
+void vvImage::Init()
+{
+  mTimeSpacing = 1;
+  mTimeOrigin = 0;
 }
 //--------------------------------------------------------------------
 
+//--------------------------------------------------------------------
+vvImage::~vvImage()
+{
+  Reset();
+}
+//--------------------------------------------------------------------
 
 //--------------------------------------------------------------------
-void vvImage::SetImage(std::vector<vtkImageData*> images) {
-  for (unsigned int i = 0; i < mVtkImages.size(); i++) {
-    if (mVtkImages[i] != NULL)
-      mVtkImages[i]->Delete();
-  }
+void vvImage::Reset()
+{
+  for (unsigned int i = 0; i < mVtkImages.size(); i++)
+    mVtkImageReslice[i]->GetInput()->Delete();
+
   mVtkImages.resize(0);
-  for (unsigned int i = 0; i < images.size(); i++) {
-    mVtkImages.push_back(images[i]);
-  }
+  mVtkImageReslice.resize(0);
+
+  Init();
 }
 //--------------------------------------------------------------------
 
 //--------------------------------------------------------------------
-void vvImage::AddImage(vtkImageData* image) {
-  mVtkImages.push_back(image);
+void vvImage::SetImage(std::vector< vtkImageData* > images)
+{
+  Reset();
+  for (unsigned int i = 0; i < images.size(); i++)
+    AddImage(images[i]);
 }
 //--------------------------------------------------------------------
 
+
 //--------------------------------------------------------------------
-void vvImage::Init() {
-  for (unsigned int i = 0; i < mVtkImages.size(); i++) {
-    if (mVtkImages[i] != NULL)
-      mVtkImages[i]->Delete();
-  }
-  mVtkImages.resize(0);
+void vvImage::AddImage(vtkImageData* image)
+{
+  mVtkImageReslice.push_back(vtkSmartPointer<vtkImageReslice>::New());
+
+  mVtkImageReslice.back()->SetInterpolationModeToLinear();
+  mVtkImageReslice.back()->AutoCropOutputOn();
+  mVtkImageReslice.back()->SetBackgroundColor(-1000,-1000,-1000,1);
+  mVtkImageReslice.back()->SetResliceTransform(mGeneralTransform);
+  mVtkImageReslice.back()->SetInput(0, image);
+  mVtkImageReslice.back()->Update();
+
+  mVtkImages.push_back( mVtkImageReslice.back()->GetOutput(0) );
 }
 //--------------------------------------------------------------------
 
-
 //--------------------------------------------------------------------
-int vvImage::GetNumberOfSpatialDimensions() {
+int vvImage::GetNumberOfSpatialDimensions()
+{
   int dim=GetNumberOfDimensions();
   if (IsTimeSequence())
     return dim-1;
@@ -81,43 +104,41 @@ int vvImage::GetNumberOfSpatialDimensions() {
 }
 //--------------------------------------------------------------------
 
-
 //--------------------------------------------------------------------
-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()) {
+    int dimension = 2;
+    int extent[6];
+    mVtkImages[0]->GetWholeExtent(extent);
+    if (extent[5] - extent[4] >= 1)
+      dimension++;
+    if (mVtkImages.size() > 1)
+      dimension++;
+    return dimension;
+  }
   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";
@@ -127,71 +148,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(mTimeSpacing);
-      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(mTimeOrigin);
-      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();
@@ -202,37 +218,39 @@ unsigned long vvImage::GetActualMemorySize() {
 
 
 //--------------------------------------------------------------------
-bool vvImage::IsTimeSequence() {
+bool vvImage::IsTimeSequence()
+{
   return mVtkImages.size()>1;
 }
 //--------------------------------------------------------------------
 
 
 //--------------------------------------------------------------------
-const std::vector<vtkImageData*>& vvImage::GetVTKImages() {
+const std::vector<vtkImageData*>& vvImage::GetVTKImages()
+{
   return mVtkImages;
 }
 //--------------------------------------------------------------------
 
-
 //--------------------------------------------------------------------
-vtkImageData* vvImage::GetFirstVTKImageData() { 
-  return mVtkImages[0]; 
+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) ||
@@ -243,11 +261,47 @@ bool vvImage::IsScalarTypeInteger(int t) {
       (t == VTK_LONG) ||
       (t == VTK_UNSIGNED_LONG))    {
     return true;
-  }
-  else {
+  } else {
     return false;
   }
 }
+
+//--------------------------------------------------------------------
+
+//--------------------------------------------------------------------
+vtkSmartPointer<vtkTransform> vvImage::GetTransform()
+{
+  return mTransform;
+}
+//--------------------------------------------------------------------
+
+//--------------------------------------------------------------------
+vtkSmartPointer<vtkGridTransform> vvImage::GetGridTransform()
+{
+  return mGridTransform;
+}
+//--------------------------------------------------------------------
+
+//--------------------------------------------------------------------
+void vvImage::UpdateReslice()
+{
+  for (unsigned int i=0; i<mVtkImageReslice.size(); i++) {
+    mVtkImageReslice[i]->Update();
+  }
+}
+//--------------------------------------------------------------------
+
+
+//--------------------------------------------------------------------
+vtkImageData * CopyAndCastToFloatFrom(vtkImageData * input)
+{
+  vtkImageData * p = vtkImageData::New();
+  p->SetExtent(input->GetExtent ()); // Only first ! could not be 4D
+  p->SetScalarTypeToFloat();
+  p->AllocateScalars();
+  p->CopyAndCastFrom(input, input->GetExtent());
+  return p;
+}
 //--------------------------------------------------------------------