1 #include <plugins/IO/ImageReader.h>
2 #include <plugins/IO/ImageReaderQDialog.h>
3 #include <cpPlugins/Image.h>
4 #include <cpPlugins_Instances_ImageReaders.h>
5 #include <itkImageIOFactory.h>
9 #include <QApplication>
12 // -------------------------------------------------------------------------
13 cpPluginsIO::ImageReaderQDialog::
14 ImageReaderQDialog( QWidget* parent )
15 : QFileDialog( parent ),
16 m_ProcessObject( NULL )
19 this, SIGNAL( accepted( ) ), this, SLOT( _dlg_Accepted( ) )
21 this->setWindowTitle( "Open an(some) image(s)" );
24 // -------------------------------------------------------------------------
25 cpPluginsIO::ImageReaderQDialog::
26 ~ImageReaderQDialog( )
30 // -------------------------------------------------------------------------
31 void cpPluginsIO::ImageReaderQDialog::
32 setProcessObject( cpPlugins::ProcessObject* obj )
36 this->m_ProcessObject = obj;
37 auto param = this->m_ProcessObject->GetParameters( );
38 auto extensions = param->GetAcceptedFileExtensions( "FileNames" );
39 auto files = param->GetOpenFileNameList( "FileNames" );
42 if( extensions != "" )
43 filters << extensions.c_str( );
44 filters << "Any file (*)";
45 this->setFileMode( QFileDialog::ExistingFiles );
46 this->setNameFilters( filters );
47 this->setAcceptMode( QFileDialog::AcceptOpen );
48 if( files.size( ) > 0 )
50 QFileInfo info( files[ 0 ].c_str( ) );
51 this->setDirectory( info.canonicalPath( ) );
56 // -------------------------------------------------------------------------
57 void cpPluginsIO::ImageReaderQDialog::
60 if( this->m_ProcessObject != NULL )
62 auto param = this->m_ProcessObject->GetParameters( );
63 auto files = this->selectedFiles( );
64 param->ClearOpenFileNameList( "FileNames" );
65 for( auto fIt = files.begin( ); fIt != files.end( ); ++fIt )
66 param->AddToOpenFileNameList( "FileNames", fIt->toStdString( ) );
71 #endif // cpPlugins_QT4
73 // -------------------------------------------------------------------------
74 QDialog* cpPluginsIO::ImageReader::
78 ImageReaderQDialog* dlg = NULL;
79 if( QApplication::instance( ) != NULL )
81 dlg = new ImageReaderQDialog( );
82 dlg->setProcessObject( this );
86 #else // cpPlugins_QT4
88 #endif // cpPlugins_QT4
91 // -------------------------------------------------------------------------
92 cpPluginsIO::ImageReader::
96 this->_AddOutput< cpPlugins::Image >( "Output" );
97 this->m_Parameters.Clear( );
98 this->m_Parameters.ConfigureAsOpenFileNameList( "FileNames" );
99 this->m_Parameters.SetAcceptedFileExtensions(
101 "Image files (*.bmp *.png *.jpg *.jpeg *.dcm *.mhd *.nhdr *.nrrd *.tiff)"
105 // -------------------------------------------------------------------------
106 cpPluginsIO::ImageReader::
111 // -------------------------------------------------------------------------
112 void cpPluginsIO::ImageReader::
116 auto fnames = this->m_Parameters.GetOpenFileNameList( "FileNames" );
117 if( fnames.size( ) >= 1 )
119 // Guess image properties
120 itk::ImageIOBase::Pointer io =
121 itk::ImageIOFactory::CreateImageIO(
122 fnames[ 0 ].c_str( ),
123 itk::ImageIOFactory::ReadMode
125 if( io.IsNotNull( ) )
127 io->SetFileName( fnames[ 0 ] );
128 io->ReadImageInformation( );
129 if( fnames.size( ) >= 1 )
131 switch( io->GetNumberOfDimensions( ) )
133 case 1: this->_GD0< 1 >( io ); break;
134 case 2: this->_GD0< 2 >( io ); break;
135 case 3: this->_GD0< 3 >( io ); break;
136 case 4: this->_GD0< 4 >( io ); break;
138 this->_Error( "Image dimension not supported." );
146 std::string( "Could not create an ImageIO for \"" ) +
152 this->_Error( "No image(s) given" );
155 // -------------------------------------------------------------------------
156 template< unsigned int _Dim >
157 void cpPluginsIO::ImageReader::
158 _GD0( itk::ImageIOBase* io )
160 typedef unsigned char uchar;
161 typedef unsigned short ushort;
162 typedef unsigned int uint;
163 typedef unsigned long ulong;
165 itk::ImageIOBase::IOComponentType ct = io->GetComponentType( );
166 itk::ImageIOBase::IOPixelType pt = io->GetPixelType( );
168 if( pt == itk::ImageIOBase::SCALAR )
172 case itk::ImageIOBase::CHAR : this->_GD1< char, _Dim >( io ); break;
173 case itk::ImageIOBase::SHORT : this->_GD1< short, _Dim >( io ); break;
174 case itk::ImageIOBase::INT : this->_GD1< int, _Dim >( io ); break;
175 case itk::ImageIOBase::LONG : this->_GD1< long, _Dim >( io ); break;
176 case itk::ImageIOBase::FLOAT : this->_GD1< float, _Dim >( io ); break;
177 case itk::ImageIOBase::DOUBLE : this->_GD1< double, _Dim >( io ); break;
178 case itk::ImageIOBase::UCHAR : this->_GD1< uchar, _Dim >( io ); break;
179 case itk::ImageIOBase::USHORT : this->_GD1< ushort, _Dim >( io ); break;
180 case itk::ImageIOBase::UINT : this->_GD1< uint, _Dim >( io ); break;
181 case itk::ImageIOBase::ULONG : this->_GD1< ulong, _Dim >( io ); break;
182 default: this->_Error( "Scalar pixel type not supported." ); break;
185 else if( pt == itk::ImageIOBase::RGB )
189 case itk::ImageIOBase::CHAR : this->_GD1< itk::RGBPixel< char >, _Dim >( io ); break;
190 case itk::ImageIOBase::SHORT : this->_GD1< itk::RGBPixel< short >, _Dim >( io ); break;
191 case itk::ImageIOBase::INT : this->_GD1< itk::RGBPixel< int >, _Dim >( io ); break;
192 case itk::ImageIOBase::LONG : this->_GD1< itk::RGBPixel< long >, _Dim >( io ); break;
193 case itk::ImageIOBase::FLOAT : this->_GD1< itk::RGBPixel< float >, _Dim >( io ); break;
194 case itk::ImageIOBase::DOUBLE : this->_GD1< itk::RGBPixel< double >, _Dim >( io ); break;
195 case itk::ImageIOBase::UCHAR : this->_GD1< itk::RGBPixel< uchar >, _Dim >( io ); break;
196 case itk::ImageIOBase::USHORT : this->_GD1< itk::RGBPixel< ushort >, _Dim >( io ); break;
197 case itk::ImageIOBase::UINT : this->_GD1< itk::RGBPixel< uint >, _Dim >( io ); break;
198 case itk::ImageIOBase::ULONG : this->_GD1< itk::RGBPixel< ulong >, _Dim >( io ); break;
199 default: this->_Error( "RGB pixel type not supported." ); break;
202 else if( pt == itk::ImageIOBase::RGBA )
206 case itk::ImageIOBase::CHAR : this->_GD1< itk::RGBAPixel< char >, _Dim >( io ); break;
207 case itk::ImageIOBase::SHORT : this->_GD1< itk::RGBAPixel< short >, _Dim >( io ); break;
208 case itk::ImageIOBase::INT : this->_GD1< itk::RGBAPixel< int >, _Dim >( io ); break;
209 case itk::ImageIOBase::LONG : this->_GD1< itk::RGBAPixel< long >, _Dim >( io ); break;
210 case itk::ImageIOBase::FLOAT : this->_GD1< itk::RGBAPixel< float >, _Dim >( io ); break;
211 case itk::ImageIOBase::DOUBLE : this->_GD1< itk::RGBAPixel< double >, _Dim >( io ); break;
212 case itk::ImageIOBase::UCHAR : this->_GD1< itk::RGBAPixel< uchar >, _Dim >( io ); break;
213 case itk::ImageIOBase::USHORT : this->_GD1< itk::RGBAPixel< ushort >, _Dim >( io ); break;
214 case itk::ImageIOBase::UINT : this->_GD1< itk::RGBAPixel< uint >, _Dim >( io ); break;
215 case itk::ImageIOBase::ULONG : this->_GD1< itk::RGBAPixel< ulong >, _Dim >( io ); break;
216 default: this->_Error( "RGBA pixel type not supported." ); break;
219 else if( pt == itk::ImageIOBase::COMPLEX )
223 case itk::ImageIOBase::FLOAT : this->_GD1< std::complex< float >, _Dim >( io ); break;
224 case itk::ImageIOBase::DOUBLE : this->_GD1< std::complex< double >, _Dim >( io ); break;
225 default: this->_Error( "Complex pixel type not supported." ); break;
228 else if( pt == itk::ImageIOBase::COVARIANTVECTOR )
232 case itk::ImageIOBase::FLOAT : this->_GD1< itk::CovariantVector< float, _Dim >, _Dim >( io ); break;
233 case itk::ImageIOBase::DOUBLE : this->_GD1< itk::CovariantVector< double, _Dim >, _Dim >( io ); break;
234 default: this->_Error( "CovariantVector pixel type not supported." ); break;
237 else if( pt == itk::ImageIOBase::POINT )
241 case itk::ImageIOBase::FLOAT : this->_GD1< itk::Point< float, _Dim >, _Dim >( io ); break;
242 case itk::ImageIOBase::DOUBLE : this->_GD1< itk::Point< double, _Dim >, _Dim >( io ); break;
243 default: this->_Error( "Point pixel type not supported." ); break;
246 else if( pt == itk::ImageIOBase::VECTOR )
250 case itk::ImageIOBase::FLOAT : this->_GD1< itk::Vector< float, _Dim >, _Dim >( io ); break;
251 case itk::ImageIOBase::DOUBLE : this->_GD1< itk::Vector< double, _Dim >, _Dim >( io ); break;
252 default: this->_Error( "Vector pixel type not supported." ); break;
255 else if( pt == itk::ImageIOBase::SYMMETRICSECONDRANKTENSOR )
259 case itk::ImageIOBase::FLOAT : this->_GD1< itk::SymmetricSecondRankTensor< float, _Dim >, _Dim >( io ); break;
260 case itk::ImageIOBase::DOUBLE : this->_GD1< itk::SymmetricSecondRankTensor< double, _Dim >, _Dim >( io ); break;
261 default: this->_Error( "SymmetricSecondRankTensor pixel type not supported." ); break;
264 else if( pt == itk::ImageIOBase::DIFFUSIONTENSOR3D )
270 case itk::ImageIOBase::FLOAT : this->_GD1< itk::DiffusionTensor3D< float >, 3 >( io ); break;
271 case itk::ImageIOBase::DOUBLE : this->_GD1< itk::DiffusionTensor3D< double >, 3 >( io ); break;
272 default: this->_Error( "DiffusionTensor3D pixel type not supported." ); break;
276 this->_Error( "DiffusionTensor3D dimension not supported." );
279 else if( pt == itk::ImageIOBase::OFFSET )
282 else if( pt == itk::ImageIOBase::FIXEDARRAY )
285 else if( pt == itk::ImageIOBase::MATRIX )
290 this->_Error( "Image pixel type not yet supported." );
293 // -------------------------------------------------------------------------
294 template< class _TPixel, unsigned int _Dim >
295 void cpPluginsIO::ImageReader::
296 _GD1( itk::ImageIOBase* io )
298 typedef itk::Image< _TPixel, _Dim > _TImage;
301 auto fnames = this->m_Parameters.GetOpenFileNameList( "FileNames" );
302 if( fnames.size( ) == 1 )
304 auto f = this->_CreateITK< itk::ImageFileReader< _TImage > >( );
305 f->SetFileName( fnames[ 0 ] );
310 this->GetOutput( "Output" )->SetITK( f->GetOutput( ) );
312 catch( itk::ExceptionObject& err )
314 this->_Error( err.GetDescription( ) );
317 else // if( fnames.size( ) > 1 )
319 auto f = this->_CreateITK< itk::ImageSeriesReader< _TImage > >( );
320 for( auto i = fnames.begin( ); i != fnames.end( ); ++i )
321 f->AddFileName( *i );
326 this->GetOutput( "Output" )->SetITK( f->GetOutput( ) );
328 catch( itk::ExceptionObject& err )
330 this->_Error( err.GetDescription( ) );