ctkDICOMAppWidget.cpp 15 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450
  1. // std includes
  2. #include <iostream>
  3. #include <dcmimage.h>
  4. // Qt includes
  5. #include <QDebug>
  6. #include <QTreeView>
  7. #include <QTabBar>
  8. #include <QSettings>
  9. #include <QAction>
  10. #include <QModelIndex>
  11. #include <QCheckBox>
  12. // ctkDICOMCore includes
  13. #include "ctkDICOMDatabase.h"
  14. #include "ctkDICOMIndexer.h"
  15. // ctkDICOMWidgets includes
  16. #include "ctkDICOMImage.h"
  17. #include "ctkDICOMModel.h"
  18. #include "ctkDICOMAppWidget.h"
  19. #include "ctkDICOMQueryResultsTabWidget.h"
  20. #include "ui_ctkDICOMAppWidget.h"
  21. #include "ctkDirectoryButton.h"
  22. #include "ctkFileDialog.h"
  23. #include "ctkDICOMQueryRetrieveWidget.h"
  24. #include "ctkDICOMImportWidget.h"
  25. #include "ctkDICOMThumbnailWidget.h"
  26. //logger
  27. #include <ctkLogger.h>
  28. static ctkLogger logger("org.commontk.DICOM.Widgets.ctkDICOMAppWidget");
  29. //----------------------------------------------------------------------------
  30. class ctkDICOMAppWidgetPrivate: public Ui_ctkDICOMAppWidget
  31. {
  32. public:
  33. ctkDICOMAppWidgetPrivate();
  34. ctkFileDialog* ImportDialog;
  35. ctkDICOMQueryRetrieveWidget* QueryRetrieveWidget;
  36. QSharedPointer<ctkDICOMDatabase> DICOMDatabase;
  37. ctkDICOMModel DICOMModel;
  38. QSharedPointer<ctkDICOMIndexer> DICOMIndexer;
  39. };
  40. //----------------------------------------------------------------------------
  41. // ctkDICOMAppWidgetPrivate methods
  42. ctkDICOMAppWidgetPrivate::ctkDICOMAppWidgetPrivate(){
  43. DICOMDatabase = QSharedPointer<ctkDICOMDatabase> (new ctkDICOMDatabase);
  44. DICOMIndexer = QSharedPointer<ctkDICOMIndexer> (new ctkDICOMIndexer);
  45. }
  46. //----------------------------------------------------------------------------
  47. // ctkDICOMAppWidget methods
  48. //----------------------------------------------------------------------------
  49. ctkDICOMAppWidget::ctkDICOMAppWidget(QWidget* _parent):Superclass(_parent),
  50. d_ptr(new ctkDICOMAppWidgetPrivate)
  51. {
  52. Q_D(ctkDICOMAppWidget);
  53. d->setupUi(this);
  54. //Hide image previewer buttons
  55. d->nextImageButton->hide();
  56. d->prevImageButton->hide();
  57. d->nextSeriesButton->hide();
  58. d->prevSeriesButton->hide();
  59. d->nextStudyButton->hide();
  60. d->prevStudyButton->hide();
  61. //Enable sorting in tree view
  62. d->treeView->setSortingEnabled(true);
  63. d->treeView->setSelectionBehavior(QAbstractItemView::SelectRows);
  64. connect(d->treeView, SIGNAL(collapsed(QModelIndex)), this, SLOT(onTreeCollapsed(QModelIndex)));
  65. connect(d->treeView, SIGNAL(expanded(QModelIndex)), this, SLOT(onTreeExpanded(QModelIndex)));
  66. //Set toolbar button style
  67. d->toolBar->setToolButtonStyle(Qt::ToolButtonTextUnderIcon);
  68. //Initialize Q/R widget
  69. d->QueryRetrieveWidget = new ctkDICOMQueryRetrieveWidget();
  70. d->QueryRetrieveWidget->setWindowModality ( Qt::ApplicationModal );
  71. //initialize directory from settings, then listen for changes
  72. QSettings settings;
  73. if ( settings.value("DatabaseDirectory", "") == "" )
  74. {
  75. QString directory = QString("./ctkDICOM-Database");
  76. settings.setValue("DatabaseDirectory", directory);
  77. settings.sync();
  78. }
  79. QString databaseDirectory = settings.value("DatabaseDirectory").toString();
  80. this->setDatabaseDirectory(databaseDirectory);
  81. d->directoryButton->setDirectory(databaseDirectory);
  82. connect(d->directoryButton, SIGNAL(directoryChanged(const QString&)), this, SLOT(setDatabaseDirectory(const QString&)));
  83. //Initialize import widget
  84. d->ImportDialog = new ctkFileDialog();
  85. QCheckBox* importCheckbox = new QCheckBox("Copy on import", d->ImportDialog);
  86. d->ImportDialog->setBottomWidget(importCheckbox);
  87. d->ImportDialog->setFileMode(QFileDialog::Directory);
  88. d->ImportDialog->setLabelText(QFileDialog::Accept,"Import");
  89. d->ImportDialog->setWindowTitle("Import DICOM files from directory ...");
  90. d->ImportDialog->setWindowModality(Qt::ApplicationModal);
  91. //connect signal and slots
  92. connect(d->treeView, SIGNAL(clicked(const QModelIndex&)), d->thumbnailsWidget, SLOT(onModelSelected(const QModelIndex &)));
  93. connect(d->treeView, SIGNAL(clicked(const QModelIndex&)), d->imagePreview, SLOT(onModelSelected(const QModelIndex &)));
  94. connect(d->treeView, SIGNAL(clicked(const QModelIndex&)), this, SLOT(onModelSelected(const QModelIndex &)));
  95. connect(d->thumbnailsWidget, SIGNAL(selected(const ctkDICOMThumbnailWidget&)), this, SLOT(onThumbnailSelected(const ctkDICOMThumbnailWidget&)));
  96. connect(d->ImportDialog, SIGNAL(fileSelected(QString)),this,SLOT(onImportDirectory(QString)));
  97. connect(d->DICOMDatabase.data(), SIGNAL( databaseChanged() ), &(d->DICOMModel), SLOT( reset() ) );
  98. connect(d->QueryRetrieveWidget, SIGNAL( canceled() ), d->QueryRetrieveWidget, SLOT( hide() ) );
  99. }
  100. //----------------------------------------------------------------------------
  101. ctkDICOMAppWidget::~ctkDICOMAppWidget()
  102. {
  103. Q_D(ctkDICOMAppWidget);
  104. d->QueryRetrieveWidget->deleteLater();
  105. d->ImportDialog->deleteLater();
  106. }
  107. //----------------------------------------------------------------------------
  108. void ctkDICOMAppWidget::setDatabaseDirectory(const QString& directory)
  109. {
  110. Q_D(ctkDICOMAppWidget);
  111. QSettings settings;
  112. settings.setValue("DatabaseDirectory", directory);
  113. settings.sync();
  114. //close the active DICOM database
  115. d->DICOMDatabase->closeDatabase();
  116. //open DICOM database on the directory
  117. QString databaseFileName = directory + QString("/ctkDICOM.sql");
  118. try { d->DICOMDatabase->openDatabase( databaseFileName ); }
  119. catch (std::exception e)
  120. {
  121. std::cerr << "Database error: " << qPrintable(d->DICOMDatabase->lastError()) << "\n";
  122. d->DICOMDatabase->closeDatabase();
  123. return;
  124. }
  125. d->DICOMModel.setDatabase(d->DICOMDatabase->database());
  126. d->treeView->setModel(&d->DICOMModel);
  127. d->treeView->resizeColumnToContents(0);
  128. //pass DICOM database instance to Import widget
  129. // d->ImportDialog->setDICOMDatabase(d->DICOMDatabase);
  130. d->QueryRetrieveWidget->setRetrieveDatabase(d->DICOMDatabase);
  131. // update the button and let any connected slots know about the change
  132. d->directoryButton->setDirectory(directory);
  133. d->thumbnailsWidget->setDatabaseDirectory(directory);
  134. d->imagePreview->setDatabaseDirectory(directory);
  135. emit databaseDirectoryChanged(directory);
  136. }
  137. //----------------------------------------------------------------------------
  138. QString ctkDICOMAppWidget::databaseDirectory() const
  139. {
  140. QSettings settings;
  141. return settings.value("DatabaseDirectory").toString();
  142. }
  143. void ctkDICOMAppWidget::onAddToDatabase()
  144. {
  145. //Q_D(ctkDICOMAppWidget);
  146. //d->
  147. }
  148. //----------------------------------------------------------------------------
  149. void ctkDICOMAppWidget::openImportDialog()
  150. {
  151. Q_D(ctkDICOMAppWidget);
  152. d->ImportDialog->show();
  153. d->ImportDialog->raise();
  154. }
  155. //----------------------------------------------------------------------------
  156. void ctkDICOMAppWidget::openExportDialog()
  157. {
  158. }
  159. //----------------------------------------------------------------------------
  160. void ctkDICOMAppWidget::openQueryDialog()
  161. {
  162. Q_D(ctkDICOMAppWidget);
  163. d->QueryRetrieveWidget->show();
  164. d->QueryRetrieveWidget->raise();
  165. }
  166. //----------------------------------------------------------------------------
  167. void ctkDICOMAppWidget::onThumbnailSelected(const ctkDICOMThumbnailWidget& widget)
  168. {
  169. Q_D(ctkDICOMAppWidget);
  170. d->imagePreview->onModelSelected(widget.sourceIndex());
  171. }
  172. //----------------------------------------------------------------------------
  173. void ctkDICOMAppWidget::onImportDirectory(QString directory)
  174. {
  175. Q_D(ctkDICOMAppWidget);
  176. if (QDir(directory).exists())
  177. {
  178. QCheckBox* copyOnImport = qobject_cast<QCheckBox*>(d->ImportDialog->bottomWidget());
  179. QString targetDirectory;
  180. if (copyOnImport->isEnabled())
  181. {
  182. targetDirectory = d->DICOMDatabase->databaseDirectory();
  183. }
  184. d->DICOMIndexer->addDirectory(*d->DICOMDatabase,directory,targetDirectory);
  185. d->DICOMModel.reset();
  186. }
  187. }
  188. //----------------------------------------------------------------------------
  189. void ctkDICOMAppWidget::onModelSelected(const QModelIndex &index){
  190. Q_D(ctkDICOMAppWidget);
  191. ctkDICOMModel* model = const_cast<ctkDICOMModel*>(qobject_cast<const ctkDICOMModel*>(index.model()));
  192. if(model){
  193. QModelIndex index0 = index.sibling(index.row(), 0);
  194. if ( model->data(index0,ctkDICOMModel::TypeRole) == ctkDICOMModel::PatientType ){
  195. d->nextImageButton->show();
  196. d->prevImageButton->show();
  197. d->nextSeriesButton->show();
  198. d->prevSeriesButton->show();
  199. d->nextStudyButton->show();
  200. d->prevStudyButton->show();
  201. }else if ( model->data(index0,ctkDICOMModel::TypeRole) == ctkDICOMModel::StudyType ){
  202. d->nextImageButton->show();
  203. d->prevImageButton->show();
  204. d->nextSeriesButton->show();
  205. d->prevSeriesButton->show();
  206. d->nextStudyButton->hide();
  207. d->prevStudyButton->hide();
  208. }else if ( model->data(index0,ctkDICOMModel::TypeRole) == ctkDICOMModel::SeriesType ){
  209. d->nextImageButton->show();
  210. d->prevImageButton->show();
  211. d->nextSeriesButton->hide();
  212. d->prevSeriesButton->hide();
  213. d->nextStudyButton->hide();
  214. d->prevStudyButton->hide();
  215. }else{
  216. d->nextImageButton->hide();
  217. d->prevImageButton->hide();
  218. d->nextSeriesButton->hide();
  219. d->prevSeriesButton->hide();
  220. d->nextStudyButton->hide();
  221. d->prevStudyButton->hide();
  222. }
  223. }else{
  224. d->nextImageButton->hide();
  225. d->prevImageButton->hide();
  226. d->nextSeriesButton->hide();
  227. d->prevSeriesButton->hide();
  228. d->nextStudyButton->hide();
  229. d->prevStudyButton->hide();
  230. }
  231. }
  232. //----------------------------------------------------------------------------
  233. void ctkDICOMAppWidget::onNextImage(){
  234. Q_D(ctkDICOMAppWidget);
  235. QModelIndex currentIndex = d->imagePreview->currentImageIndex();
  236. if(currentIndex.isValid()){
  237. ctkDICOMModel* model = const_cast<ctkDICOMModel*>(qobject_cast<const ctkDICOMModel*>(currentIndex.model()));
  238. if(model){
  239. QModelIndex seriesIndex = currentIndex.parent();
  240. int imageCount = model->rowCount(seriesIndex);
  241. int imageID = currentIndex.row();
  242. imageID = (imageID+1)%imageCount;
  243. QModelIndex nextIndex = currentIndex.sibling(imageID, 0);
  244. d->imagePreview->onModelSelected(nextIndex);
  245. }
  246. }
  247. }
  248. //----------------------------------------------------------------------------
  249. void ctkDICOMAppWidget::onPreviousImage(){
  250. Q_D(ctkDICOMAppWidget);
  251. QModelIndex currentIndex = d->imagePreview->currentImageIndex();
  252. if(currentIndex.isValid()){
  253. ctkDICOMModel* model = const_cast<ctkDICOMModel*>(qobject_cast<const ctkDICOMModel*>(currentIndex.model()));
  254. if(model){
  255. QModelIndex seriesIndex = currentIndex.parent();
  256. int imageCount = model->rowCount(seriesIndex);
  257. int imageID = currentIndex.row();
  258. imageID--;
  259. if(imageID < 0) imageID += imageCount;
  260. QModelIndex prevIndex = currentIndex.sibling(imageID, 0);
  261. d->imagePreview->onModelSelected(prevIndex);
  262. }
  263. }
  264. }
  265. //----------------------------------------------------------------------------
  266. void ctkDICOMAppWidget::onNextSeries(){
  267. Q_D(ctkDICOMAppWidget);
  268. QModelIndex currentIndex = d->imagePreview->currentImageIndex();
  269. if(currentIndex.isValid()){
  270. ctkDICOMModel* model = const_cast<ctkDICOMModel*>(qobject_cast<const ctkDICOMModel*>(currentIndex.model()));
  271. if(model){
  272. QModelIndex seriesIndex = currentIndex.parent();
  273. QModelIndex studyIndex = seriesIndex.parent();
  274. int seriesCount = model->rowCount(studyIndex);
  275. int seriesID = seriesIndex.row();
  276. seriesID = (seriesID + 1)%seriesCount;
  277. QModelIndex nextIndex = seriesIndex.sibling(seriesID, 0);
  278. d->imagePreview->onModelSelected(nextIndex);
  279. }
  280. }
  281. }
  282. //----------------------------------------------------------------------------
  283. void ctkDICOMAppWidget::onPreviousSeries(){
  284. Q_D(ctkDICOMAppWidget);
  285. QModelIndex currentIndex = d->imagePreview->currentImageIndex();
  286. if(currentIndex.isValid()){
  287. ctkDICOMModel* model = const_cast<ctkDICOMModel*>(qobject_cast<const ctkDICOMModel*>(currentIndex.model()));
  288. if(model){
  289. QModelIndex seriesIndex = currentIndex.parent();
  290. QModelIndex studyIndex = seriesIndex.parent();
  291. int seriesCount = model->rowCount(studyIndex);
  292. int seriesID = seriesIndex.row();
  293. seriesID--;
  294. if(seriesID < 0) seriesID += seriesCount;
  295. QModelIndex nextIndex = seriesIndex.sibling(seriesID, 0);
  296. d->imagePreview->onModelSelected(nextIndex);
  297. }
  298. }
  299. }
  300. //----------------------------------------------------------------------------
  301. void ctkDICOMAppWidget::onNextStudy(){
  302. Q_D(ctkDICOMAppWidget);
  303. QModelIndex currentIndex = d->imagePreview->currentImageIndex();
  304. if(currentIndex.isValid()){
  305. ctkDICOMModel* model = const_cast<ctkDICOMModel*>(qobject_cast<const ctkDICOMModel*>(currentIndex.model()));
  306. if(model){
  307. QModelIndex seriesIndex = currentIndex.parent();
  308. QModelIndex studyIndex = seriesIndex.parent();
  309. QModelIndex patientIndex = studyIndex.parent();
  310. int studyCount = model->rowCount(patientIndex);
  311. int studyID = studyIndex.row();
  312. studyID = (studyID + 1)%studyCount;
  313. QModelIndex nextIndex = studyIndex.sibling(studyID, 0);
  314. d->imagePreview->onModelSelected(nextIndex);
  315. }
  316. }
  317. }
  318. //----------------------------------------------------------------------------
  319. void ctkDICOMAppWidget::onPreviousStudy(){
  320. Q_D(ctkDICOMAppWidget);
  321. QModelIndex currentIndex = d->imagePreview->currentImageIndex();
  322. if(currentIndex.isValid()){
  323. ctkDICOMModel* model = const_cast<ctkDICOMModel*>(qobject_cast<const ctkDICOMModel*>(currentIndex.model()));
  324. if(model){
  325. QModelIndex seriesIndex = currentIndex.parent();
  326. QModelIndex studyIndex = seriesIndex.parent();
  327. QModelIndex patientIndex = studyIndex.parent();
  328. int studyCount = model->rowCount(patientIndex);
  329. int studyID = studyIndex.row();
  330. studyID--;
  331. if(studyID < 0) studyID += studyCount;
  332. QModelIndex nextIndex = studyIndex.sibling(studyID, 0);
  333. d->imagePreview->onModelSelected(nextIndex);
  334. }
  335. }
  336. }
  337. //----------------------------------------------------------------------------
  338. void ctkDICOMAppWidget::onTreeCollapsed(const QModelIndex &index){
  339. Q_D(ctkDICOMAppWidget);
  340. d->treeView->resizeColumnToContents(0);
  341. }
  342. //----------------------------------------------------------------------------
  343. void ctkDICOMAppWidget::onTreeExpanded(const QModelIndex &index){
  344. Q_D(ctkDICOMAppWidget);
  345. d->treeView->resizeColumnToContents(0);
  346. }