ctkDICOMTableView.cpp 9.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306
  1. /*=========================================================================
  2. Library: CTK
  3. Copyright (c) Kitware Inc.
  4. Licensed under the Apache License, Version 2.0 (the "License");
  5. you may not use this file except in compliance with the License.
  6. You may obtain a copy of the License at
  7. http://www.apache.org/licenses/LICENSE-2.0.txt
  8. Unless required by applicable law or agreed to in writing, software
  9. distributed under the License is distributed on an "AS IS" BASIS,
  10. WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  11. See the License for the specific language governing permissions and
  12. limitations under the License.
  13. =========================================================================*/
  14. // ctkDICOMWidget includes
  15. #include "ctkDICOMTableView.h"
  16. #include "ui_ctkDICOMTableView.h"
  17. // Qt includes
  18. #include <QSortFilterProxyModel>
  19. #include <QSqlQueryModel>
  20. //------------------------------------------------------------------------------
  21. class ctkDICOMTableViewPrivate : public Ui_ctkDICOMTableView
  22. {
  23. Q_DECLARE_PUBLIC (ctkDICOMTableView)
  24. protected:
  25. ctkDICOMTableView* const q_ptr;
  26. public:
  27. ctkDICOMTableViewPrivate(ctkDICOMTableView& obj);
  28. ctkDICOMTableViewPrivate(ctkDICOMTableView& obj, ctkDICOMDatabase* db);
  29. ~ctkDICOMTableViewPrivate();
  30. // Initialize UI
  31. void init();
  32. // Setup tableview with tablemodel if database is available
  33. void setUpTableView();
  34. //Temporay solution to hide UID columns
  35. void hideUIDColumns();
  36. QString queryTableName() const;
  37. QStringList uidsForAllRows() const;
  38. ctkDICOMDatabase* dicomDatabase;
  39. QSqlQueryModel dicomSQLModel;
  40. QSortFilterProxyModel* dicomSQLFilterModel;
  41. QString queryForeignKey;
  42. QStringList currentSelection;
  43. };
  44. //------------------------------------------------------------------------------
  45. ctkDICOMTableViewPrivate::ctkDICOMTableViewPrivate(ctkDICOMTableView &obj)
  46. : q_ptr(&obj)
  47. {
  48. this->dicomSQLFilterModel = new QSortFilterProxyModel(&obj);
  49. }
  50. //------------------------------------------------------------------------------
  51. ctkDICOMTableViewPrivate::ctkDICOMTableViewPrivate(ctkDICOMTableView &obj, ctkDICOMDatabase* db)
  52. : q_ptr(&obj)
  53. , dicomDatabase(db)
  54. {
  55. this->dicomSQLFilterModel = new QSortFilterProxyModel(&obj);
  56. }
  57. //------------------------------------------------------------------------------
  58. ctkDICOMTableViewPrivate::~ctkDICOMTableViewPrivate()
  59. {
  60. }
  61. void ctkDICOMTableViewPrivate::init()
  62. {
  63. Q_Q(ctkDICOMTableView);
  64. this->setupUi(q);
  65. this->tblDicomDatabaseView->horizontalHeader()->setResizeMode(QHeaderView::ResizeToContents);
  66. this->leSearchBox->setAlwaysShowClearIcon(true);
  67. this->leSearchBox->setShowSearchIcon(true);
  68. if (this->dicomDatabase != 0)
  69. {
  70. this->setUpTableView();
  71. }
  72. }
  73. //------------------------------------------------------------------------------
  74. void ctkDICOMTableViewPrivate::setUpTableView()
  75. {
  76. Q_Q(ctkDICOMTableView);
  77. if (this->dicomDatabase != 0)
  78. {
  79. q->setQuery();
  80. this->dicomSQLFilterModel->setSourceModel(&this->dicomSQLModel);
  81. this->dicomSQLFilterModel->setFilterKeyColumn(-1);
  82. this->dicomSQLFilterModel->setFilterCaseSensitivity(Qt::CaseInsensitive);
  83. this->tblDicomDatabaseView->setModel(this->dicomSQLFilterModel);
  84. this->tblDicomDatabaseView->setColumnHidden(0, true);
  85. this->tblDicomDatabaseView->setSortingEnabled(true);
  86. this->tblDicomDatabaseView->horizontalHeader()->setResizeMode(QHeaderView::Interactive);
  87. this->tblDicomDatabaseView->verticalHeader()->setResizeMode(QHeaderView::ResizeToContents);
  88. this->hideUIDColumns();
  89. QObject::connect(this->tblDicomDatabaseView->selectionModel(),
  90. SIGNAL(selectionChanged(const QItemSelection&,const QItemSelection&)),
  91. q, SLOT(onSelectionChanged()));
  92. QObject::connect(this->tblDicomDatabaseView->selectionModel(),
  93. SIGNAL(selectionChanged(const QItemSelection&,const QItemSelection&)),
  94. q, SIGNAL(selectionChanged(const QItemSelection&,const QItemSelection&)));
  95. QObject::connect(this->tblDicomDatabaseView, SIGNAL(doubleClicked(const QModelIndex&)),
  96. q, SIGNAL(doubleClicked(const QModelIndex&)));
  97. QObject::connect(this->leSearchBox, SIGNAL(textChanged(QString)),
  98. this->dicomSQLFilterModel, SLOT(setFilterWildcard(QString)));
  99. QObject::connect(this->leSearchBox, SIGNAL(textChanged(QString)), q, SLOT(onFilterChanged()));
  100. QObject::connect(this->dicomDatabase, SIGNAL(schemaUpdated()), q, SLOT(onDatabaseChanged()));
  101. QObject::connect(this->dicomDatabase, SIGNAL(databaseChanged()), q, SLOT(onDatabaseChanged()));
  102. }
  103. }
  104. //------------------------------------------------------------------------------
  105. //Temporay solution to hide UID columns
  106. void ctkDICOMTableViewPrivate::hideUIDColumns()
  107. {
  108. const int numberOfColumns = this->tblDicomDatabaseView->model()->columnCount();
  109. for (int i = 0; i < numberOfColumns; ++i)
  110. {
  111. QString columnName = this->tblDicomDatabaseView->model()->headerData(i, Qt::Horizontal).toString();
  112. if (columnName.contains("UID"))
  113. {
  114. this->tblDicomDatabaseView->hideColumn(i);
  115. }
  116. }
  117. }
  118. //------------------------------------------------------------------------------
  119. QStringList ctkDICOMTableViewPrivate::uidsForAllRows() const
  120. {
  121. QAbstractItemModel* tableModel = this->tblDicomDatabaseView->model();
  122. int numberOfRows = tableModel->rowCount();
  123. QStringList uids;
  124. if (numberOfRows == 0)
  125. {
  126. //Return invalid UID if there are no rows
  127. uids << QString("'#'");
  128. }
  129. else
  130. {
  131. for(int i = 0; i < numberOfRows; ++i)
  132. {
  133. uids << QString("'%1'").arg(tableModel->index(i,0).data().toString());
  134. }
  135. }
  136. return uids;
  137. }
  138. //----------------------------------------------------------------------------
  139. QString ctkDICOMTableViewPrivate::queryTableName() const
  140. {
  141. return this->lblTableName->text();
  142. }
  143. //----------------------------------------------------------------------------
  144. // ctkDICOMTableView methods
  145. //----------------------------------------------------------------------------
  146. ctkDICOMTableView::ctkDICOMTableView(QWidget *parent, QString queryTableName)
  147. : Superclass(parent)
  148. , d_ptr(new ctkDICOMTableViewPrivate(*this))
  149. {
  150. Q_D(ctkDICOMTableView);
  151. d->dicomDatabase = 0;
  152. d->init();
  153. this->setQueryTableName(queryTableName);
  154. d->lblTableName->setText(queryTableName);
  155. }
  156. //------------------------------------------------------------------------------
  157. ctkDICOMTableView::ctkDICOMTableView (ctkDICOMDatabase* dicomDataBase, QWidget* parent, QString queryTableName)
  158. : Superclass(parent)
  159. , d_ptr(new ctkDICOMTableViewPrivate(*this))
  160. {
  161. this->setDicomDataBase(dicomDataBase);
  162. Q_D(ctkDICOMTableView);
  163. this->setQueryTableName(queryTableName);
  164. d->init();
  165. }
  166. //------------------------------------------------------------------------------
  167. ctkDICOMTableView::~ctkDICOMTableView()
  168. {
  169. }
  170. //------------------------------------------------------------------------------
  171. void ctkDICOMTableView::setDicomDataBase(ctkDICOMDatabase *dicomDatabase)
  172. {
  173. Q_D(ctkDICOMTableView);
  174. d->dicomDatabase = dicomDatabase;
  175. d->setUpTableView();
  176. }
  177. //------------------------------------------------------------------------------
  178. void ctkDICOMTableView::setQueryTableName(const QString &tableName)
  179. {
  180. Q_D(ctkDICOMTableView);
  181. d->lblTableName->setText(tableName);
  182. }
  183. //------------------------------------------------------------------------------
  184. void ctkDICOMTableView::setQueryForeignKey(const QString &foreignKey)
  185. {
  186. Q_D(ctkDICOMTableView);
  187. d->queryForeignKey = foreignKey;
  188. }
  189. //------------------------------------------------------------------------------
  190. void ctkDICOMTableView::onSelectionChanged()
  191. {
  192. Q_D(ctkDICOMTableView);
  193. const QModelIndexList selectedRows = d->tblDicomDatabaseView->selectionModel()->selectedRows(0);
  194. QStringList uids;
  195. foreach(QModelIndex i, selectedRows)
  196. {
  197. uids<< (QString("'") + i.data().toString() +"'");
  198. }
  199. emit selectionChanged(uids);
  200. }
  201. //------------------------------------------------------------------------------
  202. void ctkDICOMTableView::onDatabaseChanged()
  203. {
  204. Q_D(ctkDICOMTableView);
  205. setQuery();
  206. }
  207. //------------------------------------------------------------------------------
  208. void ctkDICOMTableView::onUpdateQuery(const QStringList& uids)
  209. {
  210. Q_D(ctkDICOMTableView);
  211. setQuery(uids);
  212. const QStringList& newUIDS = d->uidsForAllRows();
  213. emit queryChanged(newUIDS);
  214. }
  215. //------------------------------------------------------------------------------
  216. void ctkDICOMTableView::onFilterChanged()
  217. {
  218. Q_D(ctkDICOMTableView);
  219. const QStringList uids = d->uidsForAllRows();
  220. emit queryChanged(uids);
  221. }
  222. //------------------------------------------------------------------------------
  223. void ctkDICOMTableView::setQuery(const QStringList &uids)
  224. {
  225. Q_D(ctkDICOMTableView);
  226. QString query= "select * from " + d->queryTableName();
  227. if (!uids.empty() && d->queryForeignKey.length() != 0)
  228. {
  229. query += " where "+d->queryForeignKey+" in ( ";
  230. query.append(uids.join(",")).append(");");
  231. }
  232. if (d->dicomDatabase != 0)
  233. d->dicomSQLModel.setQuery(query, d->dicomDatabase->database());
  234. }
  235. //------------------------------------------------------------------------------
  236. QStringList ctkDICOMTableView::currentSelection() const
  237. {
  238. Q_D(const ctkDICOMTableView);
  239. QModelIndexList currentSelection = d->tblDicomDatabaseView->selectionModel()->selectedRows(0);
  240. QStringList uids;
  241. foreach(QModelIndex i, currentSelection)
  242. {
  243. uids<< /*(QString("'") + */i.data().toString() /*+"'")*/;
  244. }
  245. return uids;
  246. }