1 #include "ImageReader.h"
2 #include <cpPlugins/Interface/Image.h>
4 #include <itkImageFileReader.h>
5 #include <itkImageSeriesReader.h>
7 #ifdef cpPlugins_Interface_QT4
9 #endif // cpPlugins_Interface_QT4
11 // -------------------------------------------------------------------------
12 bool cpPlugins::IO::ImageReader::
13 ExecConfigurationDialog( QWidget* parent )
17 #ifdef cpPlugins_Interface_QT4
21 << "Image files (*.bmp *.png *.jpg *.jpeg *.dcm *.mhd *.nhdr *.nrrd *.tiff)"
24 std::vector< std::string > names;
25 this->m_Parameters->GetStringList( names, "FileNames" );
26 std::string name = ( names.size( ) > 0 )? names[ 0 ]: ".";
28 // Show dialog and check if it was accepted
29 QFileDialog dialog( parent );
30 dialog.setFileMode( QFileDialog::ExistingFiles );
31 dialog.setDirectory( QFileDialog::tr( name.c_str( ) ) );
32 dialog.setNameFilters( filters );
35 QStringList names = dialog.selectedFiles( );
36 QStringList::const_iterator qIt = names.begin( );
37 for( ; qIt != names.end( ); ++qIt )
38 this->m_Parameters->AddToStringList(
39 "FileNames", qIt->toStdString( )
41 this->m_Parameters->SetBool( "VectorType", false );
46 #endif // cpPlugins_Interface_QT4
51 // -------------------------------------------------------------------------
52 cpPlugins::IO::ImageReader::
56 this->_MakeOutput< cpPlugins::Interface::Image >( "Output" );
58 this->m_Parameters->ConfigureAsStringList( "FileNames" );
59 this->m_Parameters->ConfigureAsBool( "VectorType", false );
62 // -------------------------------------------------------------------------
63 cpPlugins::IO::ImageReader::
68 // -------------------------------------------------------------------------
69 std::string cpPlugins::IO::ImageReader::
74 this->m_Parameters->GetStringList( names, "FileNames" );
77 if( names.size( ) >= 1 )
79 // Guess image properties
80 itk::ImageIOBase::Pointer io =
81 itk::ImageIOFactory::CreateImageIO(
83 itk::ImageIOFactory::ReadMode
87 io->SetFileName( names[ 0 ] );
88 io->ReadImageInformation( );
89 if( names.size( ) >= 1 )
91 switch( io->GetNumberOfDimensions( ) )
93 case 2: r = this->_GD0< 2 >( io, names ); break;
94 case 3: r = this->_GD0< 3 >( io, names ); break;
95 case 4: r = this->_GD0< 4 >( io, names ); break;
97 r = "ImageReader: Image dimension not supported.";
104 r = "ImageReader: Could not CreateImageIO for \"" + names[ 0 ] + "\"";
107 r = "No image files given";
111 // -------------------------------------------------------------------------
112 template< unsigned int D >
113 std::string cpPlugins::IO::ImageReader::
114 _GD0( itk::ImageIOBase* io, const TStringList& names )
116 itk::ImageIOBase::IOComponentType ct = io->GetComponentType( );
117 itk::ImageIOBase::IOPixelType pt = io->GetPixelType( );
120 if( pt == itk::ImageIOBase::SCALAR )
124 case itk::ImageIOBase::UCHAR:
125 r = this->_GD1< unsigned char, D >( names );
127 case itk::ImageIOBase::CHAR:
128 r = this->_GD1< char, D >( names );
130 case itk::ImageIOBase::USHORT:
131 r = this->_GD1< unsigned short, D >( names );
133 case itk::ImageIOBase::SHORT:
134 r = this->_GD1< short, D >( names );
136 case itk::ImageIOBase::UINT:
137 r = this->_GD1< unsigned int, D >( names );
139 case itk::ImageIOBase::INT:
140 r = this->_GD1< int, D >( names );
142 case itk::ImageIOBase::ULONG:
143 r = this->_GD1< unsigned long, D >( names );
145 case itk::ImageIOBase::LONG:
146 r = this->_GD1< long, D >( names );
148 case itk::ImageIOBase::FLOAT:
149 r = this->_GD1< float, D >( names );
151 case itk::ImageIOBase::DOUBLE:
152 r = this->_GD1< double, D >( names );
155 r = "ImageReader: Scalar pixel type not supported.";
159 else if( pt == itk::ImageIOBase::RGB )
163 case itk::ImageIOBase::UCHAR:
164 r = this->_GD1< itk::RGBPixel< unsigned char >, D >( names );
166 case itk::ImageIOBase::CHAR:
167 r = this->_GD1< itk::RGBPixel< char >, D >( names );
169 case itk::ImageIOBase::USHORT:
170 r = this->_GD1< itk::RGBPixel< unsigned short >, D >( names );
172 case itk::ImageIOBase::SHORT:
173 r = this->_GD1< itk::RGBPixel< short >, D >( names );
175 case itk::ImageIOBase::UINT:
176 r = this->_GD1< itk::RGBPixel< unsigned int >, D >( names );
178 case itk::ImageIOBase::INT:
179 r = this->_GD1< itk::RGBPixel< int >, D >( names );
181 case itk::ImageIOBase::ULONG:
182 r = this->_GD1< itk::RGBPixel< unsigned long >, D >( names );
184 case itk::ImageIOBase::LONG:
185 r = this->_GD1< itk::RGBPixel< long >, D >( names );
187 case itk::ImageIOBase::FLOAT:
188 r = this->_GD1< itk::RGBPixel< float >, D >( names );
190 case itk::ImageIOBase::DOUBLE:
191 r = this->_GD1< itk::RGBPixel< double >, D >( names );
194 r = "ImageReader: RGB pixel type not supported.";
198 else if( pt == itk::ImageIOBase::RGBA )
202 case itk::ImageIOBase::UCHAR:
203 r = this->_GD1< itk::RGBAPixel< unsigned char >, D >( names );
205 case itk::ImageIOBase::CHAR:
206 r = this->_GD1< itk::RGBAPixel< char >, D >( names );
208 case itk::ImageIOBase::USHORT:
209 r = this->_GD1< itk::RGBAPixel< unsigned short >, D >( names );
211 case itk::ImageIOBase::SHORT:
212 r = this->_GD1< itk::RGBAPixel< short >, D >( names );
214 case itk::ImageIOBase::UINT:
215 r = this->_GD1< itk::RGBAPixel< unsigned int >, D >( names );
217 case itk::ImageIOBase::INT:
218 r = this->_GD1< itk::RGBAPixel< int >, D >( names );
220 case itk::ImageIOBase::ULONG:
221 r = this->_GD1< itk::RGBAPixel< unsigned long >, D >( names );
223 case itk::ImageIOBase::LONG:
224 r = this->_GD1< itk::RGBAPixel< long >, D >( names );
226 case itk::ImageIOBase::FLOAT:
227 r = this->_GD1< itk::RGBAPixel< float >, D >( names );
229 case itk::ImageIOBase::DOUBLE:
230 r = this->_GD1< itk::RGBAPixel< double >, D >( names );
233 r = "ImageReader: RGBA pixel type not supported.";
237 else if( pt == itk::ImageIOBase::VECTOR )
241 case itk::ImageIOBase::FLOAT:
242 r = this->_GD1< itk::Vector< float, D >, D >( names );
244 case itk::ImageIOBase::DOUBLE:
245 r = this->_GD1< itk::Vector< double, D >, D >( names );
248 r = "ImageReader: Vector type not supported.";
252 else if( pt == itk::ImageIOBase::POINT )
256 case itk::ImageIOBase::FLOAT:
257 r = this->_GD1< itk::Point< float, D >, D >( names );
259 case itk::ImageIOBase::DOUBLE:
260 r = this->_GD1< itk::Point< double, D >, D >( names );
263 r = "ImageReader: Point type not supported.";
267 else if( pt == itk::ImageIOBase::COVARIANTVECTOR )
271 case itk::ImageIOBase::FLOAT:
272 r = this->_GD1< itk::CovariantVector< float, D >, D >( names );
274 case itk::ImageIOBase::DOUBLE:
275 r = this->_GD1< itk::CovariantVector< double, D >, D >( names );
278 r = "ImageReader: CovariantVector type not supported.";
282 else if( pt == itk::ImageIOBase::SYMMETRICSECONDRANKTENSOR )
286 case itk::ImageIOBase::FLOAT:
288 this->_GD1< itk::SymmetricSecondRankTensor< float, D >, D >(
292 case itk::ImageIOBase::DOUBLE:
294 this->_GD1< itk::SymmetricSecondRankTensor< double, D >, D >(
299 r = "ImageReader: Vector type not supported.";
303 else if( pt == itk::ImageIOBase::DIFFUSIONTENSOR3D )
307 case itk::ImageIOBase::FLOAT:
308 r = this->_GD1< itk::DiffusionTensor3D< float >, D >( names );
310 case itk::ImageIOBase::DOUBLE:
311 r = this->_GD1< itk::DiffusionTensor3D< double >, D >( names );
314 r = "ImageReader: Diffusion tensor type not supported.";
318 else if( pt == itk::ImageIOBase::COMPLEX )
322 case itk::ImageIOBase::FLOAT:
323 r = this->_GD1< std::complex< float >, D >( names );
325 case itk::ImageIOBase::DOUBLE:
326 r = this->_GD1< std::complex< double >, D >( names );
329 r = "ImageReader: Complex type not supported.";
333 else if( pt == itk::ImageIOBase::OFFSET )
334 r = this->_GD1< itk::Offset< D >, D >( names );
336 r = "ImageReader: Image pixel type not yet supported.";
338 itk::ImageIOBase::FIXEDARRAY
339 itk::ImageIOBase::MATRIX
344 // -------------------------------------------------------------------------
345 template< class P, unsigned int D >
346 std::string cpPlugins::IO::ImageReader::
347 _GD1( const TStringList& names )
349 if( this->m_Parameters->GetBool( "VectorType" ) )
350 return( this->_RealGD< itk::VectorImage< P, D > >( names ) );
352 return( this->_RealGD< itk::Image< P, D > >( names ) );
355 // -------------------------------------------------------------------------
357 std::string cpPlugins::IO::ImageReader::
358 _RealGD( const TStringList& names )
360 cpPlugins::Interface::Image* out =
361 this->GetOutput< cpPlugins::Interface::Image >( "Output" );
363 return( "ImageReader: No output object properly created." );
366 if( names.size( ) == 1 )
369 typedef itk::ImageFileReader< I > _SR;
370 _SR* reader = this->_CreateITK< _SR >( );
371 reader->SetFileName( names[ 0 ] );
375 out->SetITK< I >( reader->GetOutput( ) );
376 out->SetName( names[ 0 ] );
378 catch( itk::ExceptionObject& err )
380 r = "ImageReader: " + std::string( err.GetDescription( ) );
381 out->SetITK< I >( NULL );
386 else if( names.size( ) > 1 )
389 typedef itk::ImageSeriesReader< I > _MR;
390 _MR* reader = this->_CreateITK< _MR >( );
391 for( unsigned int i = 0; i < names.size( ); ++i )
392 reader->AddFileName( names[ i ] );
396 out->SetITK< I >( reader->GetOutput( ) );
397 out->SetName( names[ 0 ] );
399 catch( itk::ExceptionObject& err )
401 r = "ImageReader: " + std::string( err.GetDescription( ) );
402 out->SetITK< I >( NULL );
408 r = "ImageReader: No image files given";