lib/cpPlugins/Plugins/IO/DicomSeriesReader.cxx

   1 #include "DicomSeriesReader.h"
   2
   3 #ifdef cpPlugins_Interface_QT4
   4 #include <queue>
   5 #include <map>
   6 #include <sstream>
   7 #include <vector>
   8
   9 #include <QApplication>
  10 #include <QDialogButtonBox>
  11 #include <QDir>
  12 #include <QFileDialog>
  13 #include <QGridLayout>
  14 #include <QLabel>
  15 #include <QTreeWidget>
  16 #include <QVBoxLayout>
  17
  18 #include <cpPlugins/Interface/ParametersQtDialog.h>
  19
  20 #include <itkGDCMSeriesFileNames.h>
  21
  22 /**
  23  */
  24 class cpPlugins_IO_DicomSeriesReader_ParametersQtDialog
  25   : public cpPlugins::Interface::ParametersQtDialog
  26 {
  27 public:
  28   cpPlugins_IO_DicomSeriesReader_ParametersQtDialog(
  29     QWidget* parent = 0, Qt::WindowFlags f = 0
  30     )
  31     : cpPlugins::Interface::ParametersQtDialog( parent, f )
  32     {
  33     }
  34   virtual ~cpPlugins_IO_DicomSeriesReader_ParametersQtDialog( )
  35     {
  36     }
  37
  38 protected:
  39   virtual void _dlg_OpenMultipleFiles( )
  40     {
  41       // DICOM series analyzer
  42       itk::GDCMSeriesFileNames::GlobalWarningDisplayOff( );
  43       itk::GDCMSeriesFileNames::Pointer series =
  44         itk::GDCMSeriesFileNames::New( );
  45
  46       // Show dialog and check if it was accepted
  47       QWidget* parent = dynamic_cast< QWidget* >( this->parent( ) );
  48       QFileDialog dialog( parent );
  49       dialog.setFileMode( QFileDialog::DirectoryOnly );
  50       dialog.setDirectory( QFileDialog::tr( "." ) );
  51       if( !dialog.exec( ) )
  52         return;
  53
  54       // Prepare dialog
  55       QApplication::setOverrideCursor( Qt::WaitCursor );
  56       if( parent != NULL )
  57         parent->setEnabled( false );
  58
  59       QDialog* tree_dialog = new QDialog( parent );
  60       QTreeWidget* tree_widget = new QTreeWidget( tree_dialog );
  61       QList< QTreeWidgetItem* > tree_items;
  62       std::map<
  63         std::string,
  64         std::map< std::string, std::vector< std::string > >
  65         > found_files;
  66
  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( ) ) )
  72       {
  73         std::string dir_name = q.front( );
  74         q.pop( );
  75
  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 )
  82         {
  83           QTreeWidgetItem* new_item = new QTreeWidgetItem(
  84             ( QTreeWidgetItem* )( NULL ),
  85             QStringList( dir_name.c_str( ) )
  86             );
  87           QTreeWidgetItem* new_leaf = NULL;
  88           std::vector< std::string >::const_iterator sIt = seriesUID.begin( );
  89           for( ; sIt != seriesUID.end( ); ++sIt )
  90           {
  91             std::vector< std::string > filenames = series->GetFileNames( *sIt );
  92             if( filenames.size( ) > 0 )
  93             {
  94               std::stringstream ss;
  95               ss << "(" << filenames.size( ) << "): " << *sIt;
  96               new_leaf =
  97                 new QTreeWidgetItem(
  98                   new_item, QStringList( ss.str( ).c_str( ) )
  99                   );
 100               new_item->addChild( new_leaf );
 101               found_files[ dir_name ][ *sIt ] = filenames;
 102
 103             } // fi
 104
 105           } // rof
 106
 107           if( new_leaf != NULL )
 108             tree_items.append( new_item );
 109           else
 110             delete new_item;
 111
 112         } // fi
 113
 114         // Update queue
 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 )
 119         {
 120           if( i->isDir( ) )
 121           {
 122             std::string new_dir_name = i->absoluteFilePath( ).toStdString( );
 123             if( new_dir_name.size( ) > dir_name.size( ) )
 124               q.push( new_dir_name );
 125
 126           } // fi
 127
 128         } // rof
 129
 130       } // elihw
 131
 132       // Show second dialog
 133       if( tree_items.size( ) == 0 )
 134       {
 135         delete tree_widget;
 136         delete tree_dialog;
 137         return;
 138
 139       } // fi
 140
 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 );
 147
 148       tree_widget->insertTopLevelItems( 0, tree_items );
 149       toolsLayout->addWidget( tree_widget );
 150
 151       QDialogButtonBox* bb = new QDialogButtonBox(
 152         QDialogButtonBox::Ok | QDialogButtonBox::Cancel
 153         );
 154       tree_dialog->connect(
 155         bb, SIGNAL( accepted( ) ), tree_dialog, SLOT( accept( ) )
 156         );
 157       tree_dialog->connect(
 158         bb, SIGNAL( rejected( ) ), tree_dialog, SLOT( reject( ) )
 159         );
 160       toolsLayout->addWidget( bb );
 161
 162       QApplication::restoreOverrideCursor( );
 163       if( parent != NULL )
 164         parent->setEnabled( true );
 165
 166       if( tree_dialog->exec( ) == 0 )
 167         return;
 168
 169       QTreeWidgetItem* item = tree_widget->currentItem( );
 170       if( item != NULL )
 171       {
 172         QTreeWidgetItem* item_parent = item->parent( );
 173         if( item_parent != NULL )
 174         {
 175           QApplication::setOverrideCursor( Qt::WaitCursor );
 176           if( parent != NULL )
 177             parent->setEnabled( false );
 178
 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 ] );
 187           this->updateView( );
 188
 189           QApplication::restoreOverrideCursor( );
 190           if( parent != NULL )
 191             parent->setEnabled( true );
 192
 193         } // fi
 194
 195       } // fi
 196       itk::GDCMSeriesFileNames::GlobalWarningDisplayOn( );
 197     }
 198 };
 199
 200 #endif // cpPlugins_Interface_QT4
 201
 202 // -------------------------------------------------------------------------
 203 cpPlugins::IO::DicomSeriesReader::
 204 DicomSeriesReader( )
 205   : Superclass( )
 206 {
 207 #ifdef cpPlugins_Interface_QT4
 208   if( QApplication::instance( ) != NULL )
 209   {
 210     this->m_ParametersDialog =
 211       new cpPlugins_IO_DicomSeriesReader_ParametersQtDialog( );
 212     this->m_ParametersDialog->setParameters( this->m_Parameters );
 213
 214   } // fi
 215 #endif // cpPlugins_Interface_QT4
 216 }
 217
 218 // -------------------------------------------------------------------------
 219 cpPlugins::IO::DicomSeriesReader::
 220 ~DicomSeriesReader( )
 221 {
 222 }
 223
 224 // eof - $RCSfile$