1 #include <cpPluginsIO/DicomSeriesReader.h>
9 #include <QApplication>
10 #include <QDialogButtonBox>
12 #include <QFileDialog>
13 #include <QGridLayout>
15 #include <QTreeWidget>
16 #include <QVBoxLayout>
18 #include <cpPlugins/ParametersQtDialog.h>
19 #include <cpPlugins_Instances/ImageIO.h>
23 class cpPluginsIO_DicomSeriesReader_ParametersQtDialog
24 : public cpPlugins::ParametersQtDialog
27 cpPluginsIO_DicomSeriesReader_ParametersQtDialog(
28 QWidget* parent = 0, Qt::WindowFlags f = 0
30 : cpPlugins::ParametersQtDialog( parent, f )
33 virtual ~cpPluginsIO_DicomSeriesReader_ParametersQtDialog( )
38 virtual void _dlg_OpenMultipleFiles( )
40 // DICOM series analyzer
41 itk::GDCMSeriesFileNames::GlobalWarningDisplayOff( );
42 itk::GDCMSeriesFileNames::Pointer series =
43 itk::GDCMSeriesFileNames::New( );
45 // Show dialog and check if it was accepted
46 QWidget* parent = dynamic_cast< QWidget* >( this->parent( ) );
47 QFileDialog dialog( parent );
48 dialog.setFileMode( QFileDialog::DirectoryOnly );
49 dialog.setDirectory( QFileDialog::tr( "." ) );
54 QApplication::setOverrideCursor( Qt::WaitCursor );
56 parent->setEnabled( false );
58 QDialog* tree_dialog = new QDialog( parent );
59 QTreeWidget* tree_widget = new QTreeWidget( tree_dialog );
60 QList< QTreeWidgetItem* > tree_items;
63 std::map< std::string, std::vector< std::string > >
66 std::string main_dir_name =
67 dialog.selectedFiles( ).begin( )->toStdString( );
68 std::queue< std::string > q;
69 q.push( main_dir_name );
70 while( !( q.empty( ) ) )
72 std::string dir_name = q.front( );
75 // Get DICOM information
76 series->SetUseSeriesDetails( true );
77 series->AddSeriesRestriction( "0008|0021" );
78 series->SetDirectory( dir_name );
79 const std::vector< std::string >& seriesUID = series->GetSeriesUIDs( );
80 if( seriesUID.size( ) > 0 )
82 QTreeWidgetItem* new_item = new QTreeWidgetItem(
83 ( QTreeWidgetItem* )( NULL ),
84 QStringList( dir_name.c_str( ) )
86 QTreeWidgetItem* new_leaf = NULL;
87 std::vector< std::string >::const_iterator sIt = seriesUID.begin( );
88 for( ; sIt != seriesUID.end( ); ++sIt )
90 std::vector< std::string > filenames = series->GetFileNames( *sIt );
91 if( filenames.size( ) > 0 )
94 ss << "(" << filenames.size( ) << "): " << *sIt;
97 new_item, QStringList( ss.str( ).c_str( ) )
99 new_item->addChild( new_leaf );
100 found_files[ dir_name ][ *sIt ] = filenames;
106 if( new_leaf != NULL )
107 tree_items.append( new_item );
114 QDir dir( dir_name.c_str( ) );
115 QFileInfoList contents = dir.entryInfoList( );
116 QFileInfoList::const_iterator i = contents.begin( );
117 for( ; i != contents.end( ); ++i )
121 std::string new_dir_name = i->absoluteFilePath( ).toStdString( );
122 if( new_dir_name.size( ) > dir_name.size( ) )
123 q.push( new_dir_name );
131 // Show second dialog
132 if( tree_items.size( ) == 0 )
140 QLabel* title = new QLabel( tree_dialog );
141 title->setText( "Choose a DICOM series" );
142 QGridLayout* mainLayout = new QGridLayout( tree_dialog );
143 QVBoxLayout* toolsLayout = new QVBoxLayout( );
144 toolsLayout->addWidget( title );
145 mainLayout->addLayout( toolsLayout, 0, 0, 1, 1 );
147 tree_widget->insertTopLevelItems( 0, tree_items );
148 toolsLayout->addWidget( tree_widget );
150 QDialogButtonBox* bb = new QDialogButtonBox(
151 QDialogButtonBox::Ok | QDialogButtonBox::Cancel
153 tree_dialog->connect(
154 bb, SIGNAL( accepted( ) ), tree_dialog, SLOT( accept( ) )
156 tree_dialog->connect(
157 bb, SIGNAL( rejected( ) ), tree_dialog, SLOT( reject( ) )
159 toolsLayout->addWidget( bb );
161 QApplication::restoreOverrideCursor( );
163 parent->setEnabled( true );
165 if( tree_dialog->exec( ) == 0 )
168 QTreeWidgetItem* item = tree_widget->currentItem( );
171 QTreeWidgetItem* item_parent = item->parent( );
172 if( item_parent != NULL )
174 QApplication::setOverrideCursor( Qt::WaitCursor );
176 parent->setEnabled( false );
178 auto parameters = this->m_ProcessObject->GetParameters( );
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 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 parameters->AddToOpenFileNameList( "FileNames", names[ f ] );
189 QApplication::restoreOverrideCursor( );
191 parent->setEnabled( true );
196 itk::GDCMSeriesFileNames::GlobalWarningDisplayOn( );
200 #endif // cpPlugins_QT4
202 // -------------------------------------------------------------------------
203 cpPlugins::ParametersQtDialog* cpPluginsIO::DicomSeriesReader::
207 cpPluginsIO_DicomSeriesReader_ParametersQtDialog* dlg = NULL;
208 if( QApplication::instance( ) != NULL )
210 dlg = new cpPluginsIO_DicomSeriesReader_ParametersQtDialog( );
211 dlg->setProcessObject( this );
215 #else // cpPlugins_QT4
217 #endif // cpPlugins_QT4
220 // -------------------------------------------------------------------------
221 cpPluginsIO::DicomSeriesReader::
226 // -------------------------------------------------------------------------
227 cpPluginsIO::DicomSeriesReader::
228 ~DicomSeriesReader( )