X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=common%2FvvFromITK.h;h=d4232b1be5f4d0604895edbb473bc081c3ccf514;hb=d5a2dc4d106f7c63c7f3a42b47e7466b7eb4fe0f;hp=4d43c61d22d386e7b79792981ac233c5fab98aac;hpb=758de1c14c7f16d924e1a2a575d52cd3e48ec503;p=clitk.git diff --git a/common/vvFromITK.h b/common/vvFromITK.h index 4d43c61..d4232b1 100644 --- a/common/vvFromITK.h +++ b/common/vvFromITK.h @@ -1,113 +1,125 @@ /*========================================================================= + 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. + 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 . - -=========================================================================*/ - -#ifndef vvImageFromITK_h -#define vvImageFromITK_h + - 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 + +// itk #include -#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 vvImage::Pointer vvImageFromITK(typename itk::Image::Pointer input, bool time_sequence=false) + +template +struct vvImageFromITK_Impl { - assert(Dim < 5 && Dim > 0); // We don't handle anything higher than 4-dimensional (for the moment :-p) + static vvImage::Pointer Do (typename itk::Image::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 + if (time_sequence) //The time sequence case: create a series of VTK images { - typedef itk::Image< PixelType, Dim - 1 > ConnectorImageType; - typedef itk::ImageToVTKImageFilter ConnectorType; - typedef itk::ExtractImageFilter FilterType; + typedef itk::Image< PixelType, Dim - 1 > ItkImageType; + typedef itk::ExtractImageFilter FilterType; + + //extract the 3D slices and put them in a std::vector + 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(); - typename ConnectorType::Pointer connector = ConnectorType::New(); - - //extract the 3D slices and put them in a std::vector - 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); - +#if ITK_VERSION_MAJOR == 4 + filter->SetDirectionCollapseToSubmatrix(); +#endif + filter->SetExtractionRegion(extractedRegion); 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); - } + filter->ReleaseDataFlagOn(); + vv_image->AddItkImage(filter->GetOutput()); + } + vv_image->SetTimeSpacing(input->GetSpacing()[Dim-1]); + vv_image->SetTimeOrigin(input->GetOrigin()[Dim-1]); } else //Dim == 1,2,3 and not time_sequence { - typedef itk::Image< PixelType, Dim > ConnectorImageType; - typedef itk::ImageToVTKImageFilter 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); + vv_image->AddItkImage(input); + } + return vv_image; + } +}; + +template +struct vvImageFromITK_Impl<4u, PixelType> +{ + static vvImage::Pointer Do (typename itk::Image::Pointer input, bool time_sequence=false) + { + vvImage::Pointer vv_image=vvImage::New(); + typedef itk::Image< PixelType, 4 > InputImageType; + + typedef itk::Image< PixelType, 3 > ItkImageType; + typedef itk::ExtractImageFilter FilterType; + + //extract the 3D slices and put them in a std::vector + 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[3] = 0; + extractedRegion.SetSize(extractedRegionSize); + + for (unsigned int i = 0; i < inputSize[3]; i++) { + start[3] = i; + extractedRegion.SetIndex(start); + + typename FilterType::Pointer filter = FilterType::New(); +#if ITK_VERSION_MAJOR == 4 + filter->SetDirectionCollapseToSubmatrix(); +#endif + filter->SetExtractionRegion(extractedRegion); + filter->SetInput(input); + filter->ReleaseDataFlagOn(); + vv_image->AddItkImage(filter->GetOutput()); } + vv_image->SetTimeSpacing(input->GetSpacing()[3]); + vv_image->SetTimeOrigin(input->GetOrigin()[3]); return vv_image; + } +}; + +template vvImage::Pointer vvImageFromITK(typename itk::Image::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::Do(input, time_sequence); } +//------------------------------------------------------------------------------ #endif //vvImageFromITK