X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=common%2FvvFromITK.h;h=a51cfe2d7c150caaf78f862b2ee0cda325d3e739;hb=49169efa175f39cb8052cea08186c032835bb2f2;hp=9e1f7c1c61e8aee0d3eab373c1bdf012359efa9f;hpb=cc1685b591796baaed0e3c9327cb83a218347b27;p=clitk.git

diff --git a/common/vvFromITK.h b/common/vvFromITK.h
index 9e1f7c1..a51cfe2 100644
--- a/common/vvFromITK.h
+++ b/common/vvFromITK.h
@@ -1,113 +1,101 @@
 /*=========================================================================
+  Program:   vv                     http://www.creatis.insa-lyon.fr/rio/vv
 
- Program:   vv
- Language:  C++
- Author :   Joel Schaerer (joel.schaerer@insa-lyon.fr)
+  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.
-
-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/>.
-
-=========================================================================*/
+  It is distributed under dual licence
 
+  - BSD        See included LICENSE.txt file
+  - CeCILL-B   http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.html
+===========================================================================**/
 #ifndef VVIMAGEFROMITK_H
 #define VVIMAGEFROMITK_H
 
+// vv
 #include "vvImage.h"
-//#include <itkImage.h>
+
+// itk
 #include <itkExtractImageFilter.h>
-#include <itkImageToVTKImageFilter.h>
 
+//------------------------------------------------------------------------------
 /**Converts the itk image to vv, handling the 4D problem
  * The time_sequence boolean specifies that the image is to be interpreted as a time sequence,
  * even if its dim is < 4. */
-template<unsigned int Dim, class PixelType> vvImage::Pointer vvImageFromITK(typename itk::Image<PixelType,Dim>::Pointer input, bool time_sequence=false)
+
+template<unsigned int Dim, class PixelType>
+static inline void ReadTimeSequence (vvImage::Pointer& vv_image, typename itk::Image<PixelType,Dim>::Pointer input, bool time_sequence=false)
 {
-    assert(Dim < 5 && Dim > 0); // We don't handle anything higher than 4-dimensional (for the moment :-p)
+  typedef itk::Image< PixelType, Dim > InputImageType;
+  typedef itk::Image< PixelType,  Dim - 1 >    ItkImageType;
+  typedef itk::ExtractImageFilter<InputImageType, ItkImageType> FilterType;
+
+  //extract the 3D slices and put them in a std::vector<vtkImageData*>
+  input->UpdateOutputInformation();
+  typename InputImageType::RegionType inputRegion = input->GetLargestPossibleRegion();
+  typename InputImageType::SizeType inputSize = inputRegion.GetSize();
+  typename InputImageType::IndexType start = inputRegion.GetIndex();
+  typename InputImageType::SizeType extractedRegionSize = inputSize;
+  typename InputImageType::RegionType extractedRegion;
+  extractedRegionSize[Dim - 1] = 0;
+  extractedRegion.SetSize(extractedRegionSize);
+
+  for (unsigned int i = 0; i < inputSize[Dim - 1]; i++) {
+    start[Dim - 1] = i;
+    extractedRegion.SetIndex(start);
+
+    typename FilterType::Pointer filter = FilterType::New();
+    filter->SetDirectionCollapseToSubmatrix();
+    filter->SetExtractionRegion(extractedRegion);
+    filter->SetInput(input);
+    filter->ReleaseDataFlagOn();
+    vv_image->AddItkImage<ItkImageType>(filter->GetOutput());
+    vv_image->ComputeScalarRangeBase<PixelType, Dim-1>(filter->GetOutput());
+  }
+  vv_image->SetTimeSpacing(input->GetSpacing()[Dim-1]);
+  vv_image->SetTimeOrigin(input->GetOrigin()[Dim-1]);
+}
+
+template<unsigned int Dim, class PixelType>
+struct vvImageFromITK_Impl
+{
+  static vvImage::Pointer Do (typename itk::Image<PixelType,Dim>::Pointer input, bool time_sequence=false)
+  {
     vvImage::Pointer vv_image=vvImage::New();
-    vv_image->Init(); //Delete any existing images
     typedef itk::Image< PixelType, Dim > InputImageType;
 
-    if (Dim == 4 || time_sequence) //The time sequence case: create a series of VTK images
-    {
-        typedef itk::Image< PixelType,  Dim - 1 >    ConnectorImageType;
-        typedef itk::ImageToVTKImageFilter <ConnectorImageType> ConnectorType;
-        typedef itk::ExtractImageFilter<InputImageType,ConnectorImageType> FilterType;
-
-        typename FilterType::Pointer filter = FilterType::New();
-        typename ConnectorType::Pointer connector = ConnectorType::New();
-
-        //extract the 3D slices and put them in a std::vector<vtkImageData*>
-        typename InputImageType::RegionType inputRegion = input->GetLargestPossibleRegion();
-        typename InputImageType::SizeType inputSize = inputRegion.GetSize();
-
-        typename InputImageType::SizeType extractedRegionSize = inputSize;
-        typename InputImageType::RegionType extractedRegion;
-        extractedRegionSize[Dim - 1] = 0;
-        extractedRegion.SetSize(extractedRegionSize);
-
-        filter->SetInput(input);
-        connector->SetInput(filter->GetOutput());
-
-        typename InputImageType::IndexType start = inputRegion.GetIndex();
-
-        for (unsigned int i = 0; i < inputSize[Dim - 1]; i++) {
-            start[Dim - 1] = i;
-            extractedRegion.SetIndex(start);
-            filter->SetExtractionRegion(extractedRegion);
-            try {
-                filter->Update();
-            }
-            catch ( itk::ExceptionObject & err ) {
-                std::cerr << "Error while setting vvImage from ITK (Dim==4) [Extract phase]"
-                          << " " << err << std::endl;
-                return vv_image;
-            }
-            try {
-                connector->Update();
-            }
-            catch ( itk::ExceptionObject & err ) {
-                std::cerr << "Error while setting vvImage from ITK (Dim==4) [Connect phase]"
-                          << " " << err << std::endl;
-                return vv_image;
-            }
-            vtkImageData *image = vtkImageData::New();
-            image->DeepCopy(connector->GetOutput());
-            vv_image->AddImage(image);
-        }
-    }
-    else //Dim == 1,2,3 and not time_sequence
-    {
-        typedef itk::Image< PixelType,  Dim >    ConnectorImageType;
-        typedef itk::ImageToVTKImageFilter <ConnectorImageType> ConnectorType;
-        typename ConnectorType::Pointer connector = ConnectorType::New();
-        connector->SetInput(input);
-
-        try {
-            connector->Update();
-        }
-        catch ( itk::ExceptionObject & err ) {
-            std::cerr << "Error while setting vvImage from ITK (Dim==3)"
-                      << " " << err << std::endl;
-            return vv_image;
-        }
-        vtkImageData *image = vtkImageData::New();
-        image->DeepCopy(connector->GetOutput());
-        vv_image->AddImage(image);
+    if (time_sequence) //The time sequence case: create a series of VTK images
+      ReadTimeSequence<Dim,PixelType>(vv_image, input, time_sequence);
+    else { //Dim == 1,2,3 and not time_sequence
+      vv_image->AddItkImage<InputImageType>(input);
+      vv_image->ComputeScalarRangeBase<PixelType, Dim>(input);
     }
     return vv_image;
+  }
+};
+
+template<class PixelType>
+struct vvImageFromITK_Impl<4u, PixelType>
+{
+  static vvImage::Pointer Do (typename itk::Image<PixelType,4u>::Pointer input, bool time_sequence=false)
+  {
+    vvImage::Pointer vv_image = vvImage::New();
+    ReadTimeSequence<4u,PixelType>(vv_image, input, time_sequence);
+    return vv_image;
+  }
+};
+
+template<unsigned int Dim, class PixelType> vvImage::Pointer vvImageFromITK(typename itk::Image<PixelType,Dim>::Pointer input, bool time_sequence=false)
+{
+    assert(Dim < 5 && Dim > 0); // We don't handle anything higher than 4-dimensional (for the moment :-p)
+    return vvImageFromITK_Impl<Dim, PixelType>::Do(input, time_sequence);
 }
+//------------------------------------------------------------------------------
 
 #endif //vvImageFromITK