1 #include "DicomSeriesReader.h"
3 #ifdef cpPlugins_Interface_QT4
9 #include <QApplication>
10 #include <QDialogButtonBox>
12 #include <QFileDialog>
13 #include <QGridLayout>
15 #include <QTreeWidget>
16 #include <QVBoxLayout>
18 #include <cpPlugins/Interface/ParametersQtDialog.h>
20 #include <itkGDCMSeriesFileNames.h>
24 class cpPlugins_IO_DicomSeriesReader_ParametersQtDialog
25 : public cpPlugins::Interface::ParametersQtDialog
30 cpPlugins_IO_DicomSeriesReader_ParametersQtDialog(
31 QWidget* parent = 0, Qt::WindowFlags f = 0
33 : cpPlugins::Interface::ParametersQtDialog( parent, f )
36 virtual ~cpPlugins_IO_DicomSeriesReader_ParametersQtDialog( )
41 virtual void _dlg_OpenMultipleFiles( )
43 // DICOM series analyzer
44 itk::GDCMSeriesFileNames::GlobalWarningDisplayOff( );
45 itk::GDCMSeriesFileNames::Pointer series =
46 itk::GDCMSeriesFileNames::New( );
48 // Show dialog and check if it was accepted
49 QWidget* parent = dynamic_cast< QWidget* >( this->parent( ) );
50 QFileDialog dialog( parent );
51 dialog.setFileMode( QFileDialog::DirectoryOnly );
52 dialog.setDirectory( QFileDialog::tr( "." ) );
57 QApplication::setOverrideCursor( Qt::WaitCursor );
59 parent->setEnabled( false );
61 QDialog* tree_dialog = new QDialog( parent );
62 QTreeWidget* tree_widget = new QTreeWidget( tree_dialog );
63 QList< QTreeWidgetItem* > tree_items;
64 std::map< std::string, std::map< std::string, std::vector< std::string > > >
67 std::string main_dir_name =
68 dialog.selectedFiles( ).begin( )->toStdString( );
69 std::queue< std::string > q;
70 q.push( main_dir_name );
71 while( !( q.empty( ) ) )
73 std::string dir_name = q.front( );
76 // Get DICOM information
77 series->SetUseSeriesDetails( true );
78 series->AddSeriesRestriction( "0008|0021" );
79 series->SetDirectory( dir_name );
80 const std::vector< std::string >& seriesUID = series->GetSeriesUIDs( );
81 if( seriesUID.size( ) > 0 )
83 QTreeWidgetItem* new_item = new QTreeWidgetItem(
84 ( QTreeWidgetItem* )( NULL ),
85 QStringList( dir_name.c_str( ) )
87 QTreeWidgetItem* new_leaf = NULL;
88 std::vector< std::string >::const_iterator sIt = seriesUID.begin( );
89 for( ; sIt != seriesUID.end( ); ++sIt )
91 std::vector< std::string > filenames = series->GetFileNames( *sIt );
92 if( filenames.size( ) > 0 )
95 ss << "(" << filenames.size( ) << "): " << *sIt;
98 new_item, QStringList( ss.str( ).c_str( ) )
100 new_item->addChild( new_leaf );
101 found_files[ dir_name ][ *sIt ] = filenames;
107 if( new_leaf != NULL )
108 tree_items.append( new_item );
115 QDir dir( dir_name.c_str( ) );
116 QFileInfoList contents = dir.entryInfoList( );
117 QFileInfoList::const_iterator i = contents.begin( );
118 for( ; i != contents.end( ); ++i )
122 std::string new_dir_name = i->absoluteFilePath( ).toStdString( );
123 if( new_dir_name.size( ) > dir_name.size( ) )
124 q.push( new_dir_name );
132 // Show second dialog
133 if( tree_items.size( ) == 0 )
141 QLabel* title = new QLabel( tree_dialog );
142 title->setText( "Choose a DICOM series" );
143 QGridLayout* mainLayout = new QGridLayout( tree_dialog );
144 QVBoxLayout* toolsLayout = new QVBoxLayout( );
145 toolsLayout->addWidget( title );
146 mainLayout->addLayout( toolsLayout, 0, 0, 1, 1 );
148 tree_widget->insertTopLevelItems( 0, tree_items );
149 toolsLayout->addWidget( tree_widget );
151 QDialogButtonBox* bb = new QDialogButtonBox(
152 QDialogButtonBox::Ok | QDialogButtonBox::Cancel
154 tree_dialog->connect(
155 bb, SIGNAL( accepted( ) ), tree_dialog, SLOT( accept( ) )
157 tree_dialog->connect(
158 bb, SIGNAL( rejected( ) ), tree_dialog, SLOT( reject( ) )
160 toolsLayout->addWidget( bb );
162 QApplication::restoreOverrideCursor( );
164 parent->setEnabled( true );
166 if( tree_dialog->exec( ) == 0 )
169 QTreeWidgetItem* item = tree_widget->currentItem( );
172 QTreeWidgetItem* item_parent = item->parent( );
173 if( item_parent != NULL )
175 QApplication::setOverrideCursor( Qt::WaitCursor );
177 parent->setEnabled( false );
179 std::string serie_dir = item_parent->text( 0 ).toStdString( );
180 std::string serie_id = item->text( 0 ).toStdString( );
181 serie_id = serie_id.substr( serie_id.find_first_of( " " ) + 1 );
182 this->m_Parameters->ClearOpenFileNameList( "FileNames" );
183 const std::vector< std::string >& names =
184 found_files[ serie_dir ][ serie_id ];
185 for( unsigned int f = 0; f < names.size( ); ++f )
186 this->m_Parameters->AddToOpenFileNameList( "FileNames", names[ f ] );
189 QApplication::restoreOverrideCursor( );
191 parent->setEnabled( true );
196 itk::GDCMSeriesFileNames::GlobalWarningDisplayOn( );
200 #endif // cpPlugins_Interface_QT4
202 // -------------------------------------------------------------------------
203 cpPlugins::IO::DicomSeriesReader::
207 #ifdef cpPlugins_Interface_QT4
208 if( QApplication::instance( ) != NULL )
210 this->m_ParametersDialog =
211 new cpPlugins_IO_DicomSeriesReader_ParametersQtDialog( );
212 this->m_ParametersDialog->setParameters( this->m_Parameters );
215 #endif // cpPlugins_Interface_QT4
218 // -------------------------------------------------------------------------
219 cpPlugins::IO::DicomSeriesReader::
220 ~DicomSeriesReader( )