]> Creatis software - cpPlugins.git/blob - lib/cpPlugins/Plugins/BasicFilters/ImageInterpolatorSource.cxx
...
[cpPlugins.git] / lib / cpPlugins / Plugins / BasicFilters / ImageInterpolatorSource.cxx
1 #include "ImageInterpolatorSource.h"
2
3 #include <cpPlugins/Interface/Image.h>
4 #include <itkLinearInterpolateImageFunction.h>
5 #include <itkNearestNeighborInterpolateImageFunction.h>
6
7 // -------------------------------------------------------------------------
8 cpPlugins::BasicFilters::ImageInterpolatorSource::
9 ImageInterpolatorSource( )
10   : Superclass( )
11 {
12   this->_AddInput( "ReferenceImage" );
13   this->_AddOutput< cpPlugins::Interface::DataObject >( "Output" );
14
15   std::vector< std::string > type_choices;
16   type_choices.push_back( "Linear" );
17   type_choices.push_back( "NearestNeighbor" );
18   this->m_Parameters->ConfigureAsChoices( "InterpolationType", type_choices );
19   this->m_Parameters->SetSelectedChoice( "InterpolationType", "Linear" );
20
21   std::vector< std::string > scalar_choices;
22   scalar_choices.push_back( "float" );
23   scalar_choices.push_back( "double" );
24   this->m_Parameters->ConfigureAsChoices( "ScalarType", scalar_choices );
25   this->m_Parameters->SetSelectedChoice( "ScalarType", "float" );
26 }
27
28 // -------------------------------------------------------------------------
29 cpPlugins::BasicFilters::ImageInterpolatorSource::
30 ~ImageInterpolatorSource( )
31 {
32 }
33
34 // -------------------------------------------------------------------------
35 std::string cpPlugins::BasicFilters::ImageInterpolatorSource::
36 _GenerateData( )
37 {
38   auto image =
39     this->GetInputData< cpPlugins::Interface::Image >( "ReferenceImage" );
40   itk::DataObject* itk_image = NULL;
41   std::string r = "";
42   cpPlugins_Image_Demangle_AllScalarTypes( 2, image, itk_image, r, _GD0 );
43   else cpPlugins_Image_Demangle_AllScalarTypes( 3, image, itk_image, r, _GD0 );
44   else cpPlugins_Image_Demangle_AllScalarTypes( 4, image, itk_image, r, _GD0 );
45   else r = "ImageInterpolatorSource: Input image type not supported.";
46   return( r );
47 }
48
49 // -------------------------------------------------------------------------
50 template< class I >
51 std::string cpPlugins::BasicFilters::ImageInterpolatorSource::
52 _GD0( itk::DataObject* dobj )
53 {
54   std::string int_choice =
55     this->m_Parameters->GetSelectedChoice( "InterpolationType" );
56   std::string scalar_choice =
57     this->m_Parameters->GetSelectedChoice( "ScalarType" );
58
59   if( int_choice == "Linear" )
60   {
61     if( scalar_choice == "float" )
62       return(
63         this->_GD1< itk::LinearInterpolateImageFunction< I, float > >( )
64         );
65     else if( scalar_choice == "double" )
66       return(
67         this->_GD1< itk::LinearInterpolateImageFunction< I, double > >( )
68         );
69     else
70       return( "ImageInterpolatorSource: Invalid scalar type" );
71   }
72   else if( int_choice == "NearestNeighbor" )
73   {
74     if( scalar_choice == "float" )
75       return(
76         this->_GD1< itk::NearestNeighborInterpolateImageFunction< I, float > >( )
77         );
78     else if( scalar_choice == "double" )
79       return(
80         this->_GD1< itk::NearestNeighborInterpolateImageFunction< I, double > >( )
81         );
82     else
83       return( "ImageInterpolatorSource: Invalid scalar type" );
84   }
85   else
86     return( "ImageInterpolatorSource: Invalid interpolator" );
87 }
88
89 // -------------------------------------------------------------------------
90 template< class T >
91 std::string cpPlugins::BasicFilters::ImageInterpolatorSource::
92 _GD1( )
93 {
94   auto out =
95     this->GetOutputData< cpPlugins::Interface::DataObject >( "Output" );
96   if( out->GetITK< T >( ) == NULL )
97   {
98     typename T::Pointer res = T::New( );
99     out->SetITK( res.GetPointer( ) );
100
101   } // fi
102   return( "" );
103 }
104
105 // eof - $RCSfile$