lib/cpExtensions/DataStructures/ITKAndVTKImage.h

   1 #ifndef __CPEXTENSIONS__DATASTRUCTURES__ITKANDVTKIMAGE__H__
   2 #define __CPEXTENSIONS__DATASTRUCTURES__ITKANDVTKIMAGE__H__
   3
   4 #include <itkImage.h>
   5 #include <itkImageFileReader.h>
   6 #include <itkImageFileWriter.h>
   7 #include <itkImageToVTKImageFilter.h>
   8
   9 namespace cpExtensions
  10 {
  11   namespace DataStructures
  12   {
  13     /**
  14      */
  15     template< typename P, unsigned int D >
  16     struct ITKAndVTKImage
  17     {
  18       typedef itk::Image< P, D >                   TImage;
  19       typedef itk::ImageToVTKImageFilter< TImage > TItk2Vtk;
  20
  21       typename TImage::Pointer   ITK;
  22       typename TItk2Vtk::Pointer VTK;
  23
  24       void CopyInformation( const itk::ImageBase< D >* i )
  25         {
  26           this->ITK = TImage::New( );
  27           this->ITK->SetLargestPossibleRegion( i->GetLargestPossibleRegion( ) );
  28           this->ITK->SetRequestedRegion( i->GetRequestedRegion( ) );
  29           this->ITK->SetBufferedRegion( i->GetBufferedRegion( ) );
  30           this->ITK->SetSpacing( i->GetSpacing( ) );
  31           this->ITK->SetOrigin( i->GetOrigin( ) );
  32           this->ITK->SetDirection( i->GetDirection( ) );
  33           this->ITK->Allocate( );
  34         }
  35
  36       void SetItkImage( TImage* image )
  37         {
  38           this->ITK = image;
  39           this->_Synch( );
  40         }
  41
  42       std::string ReadFromFile( const std::string& filename )
  43         {
  44           typename itk::ImageFileReader< TImage >::Pointer r =
  45             itk::ImageFileReader< TImage >::New( );
  46           r->SetFileName( filename );
  47           try
  48           {
  49             r->Update( );
  50           }
  51           catch( itk::ExceptionObject& err )
  52           {
  53             this->ITK = NULL;
  54             return( err.GetDescription( ) );
  55
  56           } // fi
  57           this->ITK = r->GetOutput( );
  58           this->ITK->DisconnectPipeline( );
  59           this->_Synch( );
  60           return( "" );
  61         }
  62
  63       std::string WriteToFile( const std::string& filename ) const
  64         {
  65           typename itk::ImageFileWriter< TImage >::Pointer w =
  66             itk::ImageFileWriter< TImage >::New( );
  67           w->SetFileName( filename );
  68           w->SetInput( this->ITK );
  69           try
  70           {
  71             w->Update( );
  72           }
  73           catch( itk::ExceptionObject& err )
  74           {
  75             return( err.GetDescription( ) );
  76
  77           } // fi
  78           return( "" );
  79         }
  80
  81       TImage* GetItkImage( )
  82         { return( this->ITK ); }
  83       vtkImageData* GetVtkImage( )
  84         { return( this->VTK->GetOutput( ) ); }
  85       const TImage* GetItkImage( ) const
  86         { return( this->ITK ); }
  87       const vtkImageData* GetVtkImage( ) const
  88         { return( this->VTK->GetOutput( ) ); }
  89       void DisconnectPipeline( )
  90         { this->ITK->DisconnectPipeline( ); }
  91
  92       void _Synch( )
  93         {
  94           this->VTK = TItk2Vtk::New( );
  95           this->VTK->SetInput( this->ITK );
  96           this->VTK->Update( );
  97         }
  98
  99     };
 100
 101   } // ecapseman
 102
 103 } // ecapseman
 104
 105 #endif // __CPEXTENSIONS__DATASTRUCTURES__ITKANDVTKIMAGE__H__
 106
 107 // eof - $RCSfile$