]> Creatis software - clitk.git/blobdiff - common/vvImage.cxx
Set Directions to CropImage Like
[clitk.git] / common / vvImage.cxx
index 7d68a1438e70e4656a430d153ec0145b247225d2..f75d3212267441b0ee75e2a0ffe221d00ad9066a 100644 (file)
-#ifndef VVIMAGE_CXX
-#define VVIMAGE_CXX
-
 /*=========================================================================
+  Program:   vv                     http://www.creatis.insa-lyon.fr/rio/vv
 
-Program:   vv
-Language:  C++
-Author :   Pierre Seroul (pierre.seroul@gmail.com)
+  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
 
-Copyright (C) 2008
-Léon Bérard cancer center http://oncora1.lyon.fnclcc.fr
-CREATIS-LRMN http://www.creatis.insa-lyon.fr
+  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.
 
-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.
+  It is distributed under dual licence
 
-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.
-
-You should have received a copy of the GNU General Public License
-along with this program.  If not, see <http://www.gnu.org/licenses/>.
-
-=========================================================================*/
+  - 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
 
+// vv
 #include "vvImage.h"
-#include "vtkImageData.h"
+
+// clitk
 #include "clitkCommon.h"
+
+// vtk
+#include <vtkImageData.h>
+#include <vtkTransform.h>
+#include <vtkDataObject.h>
+#include <vtkStreamingDemandDrivenPipeline.h>
+#include <vtkInformation.h>
+#include <vtkVersion.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::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<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<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(mTimeOrigin);
+    else
+      origin.push_back(mVtkImages[0]->GetOrigin()[i]);
+  }
+  return origin;
+}
+//--------------------------------------------------------------------
 
-//====================================================================
-unsigned long vvImage::GetActualMemorySize() {
-    unsigned long size = 0;
-    for (unsigned int i = 0; i < mVtkImages.size(); i++) {
-        size += mVtkImages[i]->GetActualMemorySize();
+//--------------------------------------------------------------------
+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()*1024;
+  }
+  return size;
+}
+//--------------------------------------------------------------------
+
+
+//--------------------------------------------------------------------
+bool vvImage::IsTimeSequence() const
+{
+  return mVtkImages.size()>1;
+}
+//--------------------------------------------------------------------
+
+
+//--------------------------------------------------------------------
+const std::vector<vtkImageData*>& vvImage::GetVTKImages()
+{
+  return mVtkImages;
+}
+//--------------------------------------------------------------------
+
+//--------------------------------------------------------------------
+vtkImageData* vvImage::GetFirstVTKImageData()
+{
+  return mVtkImages[0];
+}
+//--------------------------------------------------------------------
+
+//--------------------------------------------------------------------
+bool vvImage::IsScalarTypeInteger()
+{
+  assert(mVtkImages.size()> 0);
+  int t = mVtkImages[0]->GetScalarType();
+  return IsScalarTypeInteger(t);
+}
+//--------------------------------------------------------------------
+
+//--------------------------------------------------------------------
+bool vvImage::IsScalarTypeInteger(int t)
+{
+  if ((t == VTK_BIT) ||
+      (t == VTK_CHAR) ||
+      (t == VTK_UNSIGNED_CHAR) ||
+      (t == VTK_SHORT) ||
+      (t == VTK_UNSIGNED_SHORT) ||
+      (t == VTK_INT) ||
+      (t == VTK_UNSIGNED_INT) ||
+      (t == VTK_LONG) ||
+      (t == VTK_UNSIGNED_LONG))    {
+    return true;
+  } 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 size;
+  }
+  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