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 #endif // cpPlugins_Interface_QT4
20 #include <itkGDCMSeriesFileNames.h>
22 // -------------------------------------------------------------------------
23 bool cpPlugins::IO::DicomSeriesReader::
24 ExecConfigurationDialog( QWidget* parent )
28 #ifdef cpPlugins_Interface_QT4
30 // DICOM series analyzer
31 itk::GDCMSeriesFileNames::GlobalWarningDisplayOff( );
32 itk::GDCMSeriesFileNames::Pointer series =
33 itk::GDCMSeriesFileNames::New( );
35 // Show dialog and check if it was accepted
36 QFileDialog dialog( parent );
37 dialog.setFileMode( QFileDialog::DirectoryOnly );
38 dialog.setDirectory( QFileDialog::tr( "." ) );
43 QApplication::setOverrideCursor( Qt::WaitCursor );
45 parent->setEnabled( false );
47 QDialog* tree_dialog = new QDialog( parent );
48 QTreeWidget* tree_widget = new QTreeWidget( tree_dialog );
49 QList< QTreeWidgetItem* > tree_items;
50 std::map< std::string, std::map< std::string, TStringList > > found_files;
52 std::string main_dir_name =
53 dialog.selectedFiles( ).begin( )->toStdString( );
54 std::queue< std::string > q;
55 q.push( main_dir_name );
56 while( !( q.empty( ) ) )
58 std::string dir_name = q.front( );
61 // Get DICOM information
62 series->SetUseSeriesDetails( true );
63 series->AddSeriesRestriction( "0008|0021" );
64 series->SetDirectory( dir_name );
65 const TStringList& seriesUID = series->GetSeriesUIDs( );
66 if( seriesUID.size( ) > 0 )
68 QTreeWidgetItem* new_item = new QTreeWidgetItem(
69 ( QTreeWidgetItem* )( NULL ),
70 QStringList( dir_name.c_str( ) )
72 QTreeWidgetItem* new_leaf = NULL;
73 TStringList::const_iterator sIt = seriesUID.begin( );
74 for( ; sIt != seriesUID.end( ); ++sIt )
76 TStringList filenames = series->GetFileNames( *sIt );
77 if( filenames.size( ) > 0 )
80 ss << "(" << filenames.size( ) << "): " << *sIt;
83 new_item, QStringList( ss.str( ).c_str( ) )
85 new_item->addChild( new_leaf );
86 found_files[ dir_name ][ *sIt ] = filenames;
92 if( new_leaf != NULL )
93 tree_items.append( new_item );
100 QDir dir( dir_name.c_str( ) );
101 QFileInfoList contents = dir.entryInfoList( );
102 QFileInfoList::const_iterator i = contents.begin( );
103 for( ; i != contents.end( ); ++i )
107 std::string new_dir_name = i->absoluteFilePath( ).toStdString( );
108 if( new_dir_name.size( ) > dir_name.size( ) )
109 q.push( new_dir_name );
117 // Show second dialog
118 if( tree_items.size( ) == 0 )
126 QLabel* title = new QLabel( tree_dialog );
127 title->setText( "Choose a DICOM series" );
128 QGridLayout* mainLayout = new QGridLayout( tree_dialog );
129 QVBoxLayout* toolsLayout = new QVBoxLayout( );
130 toolsLayout->addWidget( title );
131 mainLayout->addLayout( toolsLayout, 0, 0, 1, 1 );
133 tree_widget->insertTopLevelItems( 0, tree_items );
134 toolsLayout->addWidget( tree_widget );
136 QDialogButtonBox* bb = new QDialogButtonBox(
137 QDialogButtonBox::Ok | QDialogButtonBox::Cancel
139 tree_dialog->connect(
140 bb, SIGNAL( accepted( ) ), tree_dialog, SLOT( accept( ) )
142 tree_dialog->connect(
143 bb, SIGNAL( rejected( ) ), tree_dialog, SLOT( reject( ) )
145 toolsLayout->addWidget( bb );
147 QApplication::restoreOverrideCursor( );
149 parent->setEnabled( true );
151 if( tree_dialog->exec( ) == 0 )
154 QTreeWidgetItem* item = tree_widget->currentItem( );
157 QTreeWidgetItem* item_parent = item->parent( );
158 if( item_parent != NULL )
160 QApplication::setOverrideCursor( Qt::WaitCursor );
162 parent->setEnabled( false );
164 std::string serie_dir = item_parent->text( 0 ).toStdString( );
165 std::string serie_id = item->text( 0 ).toStdString( );
166 serie_id = serie_id.substr( serie_id.find_first_of( " " ) + 1 );
167 this->m_Parameters->ClearStringList( "FileNames" );
168 const TStringList& names = found_files[ serie_dir ][ serie_id ];
169 for( unsigned int f = 0; f < names.size( ); ++f )
170 this->m_Parameters->AddToStringList( "FileNames", names[ f ] );
174 QApplication::restoreOverrideCursor( );
176 parent->setEnabled( true );
182 itk::GDCMSeriesFileNames::GlobalWarningDisplayOn( );
184 #endif // cpPlugins_Interface_QT4
189 // -------------------------------------------------------------------------
190 cpPlugins::IO::DicomSeriesReader::
196 // -------------------------------------------------------------------------
197 cpPlugins::IO::DicomSeriesReader::
198 ~DicomSeriesReader( )