X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=common%2FvvImage.cxx;h=f75d3212267441b0ee75e2a0ffe221d00ad9066a;hb=d55f025b18f68066a52b8f33c2dc6481e82c2580;hp=ae5c5a645ca5c842524be2c8b03c617485254571;hpb=9bcf64b7f7c5a7f0a39843df634c05b5d9e4e237;p=clitk.git

diff --git a/common/vvImage.cxx b/common/vvImage.cxx
index ae5c5a6..f75d321 100644
--- a/common/vvImage.cxx
+++ b/common/vvImage.cxx
@@ -1,215 +1,233 @@
-#ifndef VVIMAGE_CXX
-#define VVIMAGE_CXX
+/*=========================================================================
+  Program:   vv                     http://www.creatis.insa-lyon.fr/rio/vv
 
-/*-------------------------------------------------------------------------
+  Authors belong to:
+  - University of LYON              http://www.universite-lyon.fr/
+  - Léon Bérard cancer center       http://www.centreleonberard.fr
+  - CREATIS CNRS laboratory         http://www.creatis.insa-lyon.fr
 
-Program:   vv
-Language:  C++
-Author :   Pierre Seroul (pierre.seroul@gmail.com)
+  This software is distributed WITHOUT ANY WARRANTY; without even
+  the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
+  PURPOSE.  See the copyright notices for more information.
 
-Copyright (C) 2008
-Léon Bérard cancer center http://oncora1.lyon.fnclcc.fr
-CREATIS-LRMN http://www.creatis.insa-lyon.fr
+  It is distributed under dual licence
 
-This program is free software: you can redistribute it and/or modify
-it under the terms of the GNU General Public License as published by
-the Free Software Foundation, version 3 of the License.
+  - BSD        See included LICENSE.txt file
+  - CeCILL-B   http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.html
+===========================================================================**/
+#ifndef VVIMAGE_CXX
+#define VVIMAGE_CXX
 
-This program is distributed in the hope that it will be useful,
-but WITHOUT ANY WARRANTY; without even the implied warranty of
-MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
-GNU General Public License for more details.
+// vv
+#include "vvImage.h"
 
-You should have received a copy of the GNU General Public License
-along with this program.  If not, see <http://www.gnu.org/licenses/>.
+// clitk
+#include "clitkCommon.h"
 
--------------------------------------------------------------------------*/
+// vtk
+#include <vtkImageData.h>
+#include <vtkTransform.h>
+#include <vtkDataObject.h>
+#include <vtkStreamingDemandDrivenPipeline.h>
+#include <vtkInformation.h>
+#include <vtkVersion.h>
 
-#include "vvImage.h"
-#include "vtkImageData.h"
-#include "clitkCommon.h"
+// std
 #include <cassert>
 
 //--------------------------------------------------------------------
-vvImage::vvImage() {
-    mVtkImages.resize(0);
+vvImage::vvImage()
+{
+  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;
+  mImageDimension = 0;
+  mrange[0] = std::numeric_limits<int>::max();;//min
+  mrange[1] = std::numeric_limits<int>::min();;//max
 }
 //--------------------------------------------------------------------
 
 //--------------------------------------------------------------------
-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]);
-    }
+vvImage::~vvImage()
+{
+  Reset();
 }
 //--------------------------------------------------------------------
 
 //--------------------------------------------------------------------
-void vvImage::AddImage(vtkImageData* image) {
-    mVtkImages.push_back(image);
+void vvImage::Reset()
+{
+  mVtkImages.clear();
+  Init();
 }
 //--------------------------------------------------------------------
 
 //--------------------------------------------------------------------
-void vvImage::Init() {
-    for (unsigned int i = 0; i < mVtkImages.size(); i++) {
-        if (mVtkImages[i] != NULL)
-            mVtkImages[i]->Delete();
-    }
-    mVtkImages.resize(0);
+void vvImage::AddVtkImage(vtkImageData* input)
+{
+  // RP: 20/12/2011
+  // Note that we're simply adding a new image to the vector.
+  // mItkToVtkConverters is therefore not being updated, but 
+  // up to here it's not being used anyway...
+  mImageDimension = 0;
+#if VTK_MAJOR_VERSION <= 5
+  int* extent = input->GetWholeExtent();
+#else
+  int* extent = input->GetInformation()->Get(vtkDataObject::DATA_EXTENT());
+#endif
+  if (extent[4] != extent[5])
+    mImageDimension = 3;
+  else if (extent[3] != extent[4])
+    mImageDimension = 2;
+  else if (extent[0] != extent[1])
+    mImageDimension = 1;
+  
+  mVtkImages.push_back(input);
 }
+
 //--------------------------------------------------------------------
 
 //--------------------------------------------------------------------
-int vvImage::GetNumberOfSpatialDimensions() {
-    int dim=GetNumberOfDimensions();
+int vvImage::GetNumberOfSpatialDimensions()
+{
+  return mImageDimension;
+}
+//--------------------------------------------------------------------
+
+//--------------------------------------------------------------------
+int vvImage::GetNumberOfDimensions() const
+{
+  if (mVtkImages.size()) {
     if (IsTimeSequence())
-        return dim-1;
+      return mImageDimension+1;
     else
-        return dim;
-}
-//--------------------------------------------------------------------
-//
-//--------------------------------------------------------------------
-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;
+      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]=mrange[0];
+  range[1]=mrange[1];
 }
+//--------------------------------------------------------------------
 
 //--------------------------------------------------------------------
-std::string vvImage::GetScalarTypeAsString() {
-    return mVtkImages[0]->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";
+  if (vtktype == "unsigned short") return "unsigned_short";
+  if (vtktype == "unsigned int") return "unsigned_int";
+  return vtktype;
 }
 //--------------------------------------------------------------------
 
 //--------------------------------------------------------------------
-int vvImage::GetNumberOfScalarComponents() {
-    return mVtkImages[0]->GetNumberOfScalarComponents();
+int vvImage::GetNumberOfScalarComponents()
+{
+  return mVtkImages[0]->GetNumberOfScalarComponents();
 }
 //--------------------------------------------------------------------
 
 //--------------------------------------------------------------------
-int vvImage::GetScalarSize() {
-    return mVtkImages[0]->GetScalarSize();
+int vvImage::GetScalarSize()
+{
+  return mVtkImages[0]->GetScalarSize();
 }
 //--------------------------------------------------------------------
 
 //--------------------------------------------------------------------
-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]);
-    }
-    return spacing;
+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]);
+  }
+  return spacing;
 }
 //--------------------------------------------------------------------
-
-//--------------------------------------------------------------------
-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]);
-    }
-    return origin;
+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]);
+  }
+  return origin;
 }
 //--------------------------------------------------------------------
 
 //--------------------------------------------------------------------
-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]);
-    }
-    return size0;
+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]);
+  }
+  return size0;
 }
 //--------------------------------------------------------------------
 
 //--------------------------------------------------------------------
-unsigned long vvImage::GetActualMemorySize() {
-    unsigned long size = 0;
-    for (unsigned int i = 0; i < mVtkImages.size(); i++) {
-        size += mVtkImages[i]->GetActualMemorySize();
-    }
-    return size;
+unsigned long vvImage::GetActualMemorySize()
+{
+  unsigned long size = 0;
+  for (unsigned int i = 0; i < mVtkImages.size(); i++) {
+    size += mVtkImages[i]->GetActualMemorySize()*1024;
+  }
+  return size;
 }
 //--------------------------------------------------------------------
 
 
 //--------------------------------------------------------------------
-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]; 
+vtkImageData* vvImage::GetFirstVTKImageData()
+{
+  return mVtkImages[0];
 }
 //--------------------------------------------------------------------
 
-
 //--------------------------------------------------------------------
-bool vvImage::IsScalarTypeInteger() {
+bool vvImage::IsScalarTypeInteger()
+{
   assert(mVtkImages.size()> 0);
   int t = mVtkImages[0]->GetScalarType();
   return IsScalarTypeInteger(t);
@@ -217,7 +235,8 @@ bool vvImage::IsScalarTypeInteger() {
 //--------------------------------------------------------------------
 
 //--------------------------------------------------------------------
-bool vvImage::IsScalarTypeInteger(int t) {
+bool vvImage::IsScalarTypeInteger(int t)
+{
   if ((t == VTK_BIT) ||
       (t == VTK_CHAR) ||
       (t == VTK_UNSIGNED_CHAR) ||
@@ -228,12 +247,77 @@ bool vvImage::IsScalarTypeInteger(int t) {
       (t == VTK_LONG) ||
       (t == VTK_UNSIGNED_LONG))    {
     return true;
-  }
-  else {
+  } else {
     return false;
   }
 }
 //--------------------------------------------------------------------
 
+//--------------------------------------------------------------------
+const std::vector< vtkSmartPointer<vtkTransform> >& vvImage::GetTransform()
+{
+  return this->mTransform;
+}
+//--------------------------------------------------------------------
+
+
+//--------------------------------------------------------------------
+void vvImage::InitializeTransform()
+{
+  for (int i = 0; i < mTransform.size(); i++)
+    mTransform[i]->Identity();
+}
+//--------------------------------------------------------------------
+
+
+//--------------------------------------------------------------------
+std::vector< std::vector<double> > vvImage::GetDirection()
+{
+  int dim = this->GetNumberOfDimensions();
+  vtkSmartPointer<vtkTransform> transform = vtkSmartPointer<vtkTransform>::New();
+  transform->Identity();
+  for (int i = 0; i < mTransform.size(); i++)
+    transform->Concatenate(this->mTransform[i]);
+
+  vtkSmartPointer<vtkMatrix4x4> matrix = transform->GetMatrix();
+  matrix->Invert();
+  std::vector<std::vector<double> > direction0;
+  for (int i = 0; i < dim; i++) {
+    if (i != 3) {
+      std::vector<double> direction1;
+      for (int j = 0; j < dim; j++) {
+        if (j != 3)
+#if VTK_MAJOR_VERSION <= 6
+          direction1.push_back((*matrix)[i][j]);
+#else
+          direction1.push_back((*matrix).GetElement(i,j));
+#endif
+      }
+      direction0.push_back(direction1);
+    }
+  }
+  return direction0;
+}
+//--------------------------------------------------------------------
+
+
+
+//--------------------------------------------------------------------
+bool vvImage::HaveSameSizeAndSpacingThan(vvImage * other)
+{
+  bool same = true;
+  for(int i=0; i<GetNumberOfDimensions(); i++) {
+    if ( GetSize()[i] != other->GetSize()[i]) same = false;
+    if ( GetSpacing()[i] != other->GetSpacing()[i]) same = false;
+  }
+  return same;
+}
+//--------------------------------------------------------------------
+itk::MetaDataDictionary* vvImage::GetFirstMetaDataDictionary()
+{
+    return mDictionary[0];
+}
+//--------------------------------------------------------------------
+
 
 #endif // VVIMAGE_CXX