ctkDICOMDatabase.cpp 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632
  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. #include <stdexcept>
  15. // Qt includes
  16. #include <QSqlQuery>
  17. #include <QSqlRecord>
  18. #include <QSqlError>
  19. #include <QVariant>
  20. #include <QDate>
  21. #include <QStringList>
  22. #include <QSet>
  23. #include <QFile>
  24. #include <QDirIterator>
  25. #include <QFileInfo>
  26. #include <QDebug>
  27. #include <QFileSystemWatcher>
  28. // ctkDICOM includes
  29. #include "ctkDICOMDatabase.h"
  30. #include "ctkDICOMAbstractThumbnailGenerator.h"
  31. #include "ctkLogger.h"
  32. // DCMTK includes
  33. #include <dcmtk/dcmdata/dcfilefo.h>
  34. #include <dcmtk/dcmdata/dcfilefo.h>
  35. #include <dcmtk/dcmdata/dcdeftag.h>
  36. #include <dcmtk/dcmdata/dcdatset.h>
  37. #include <dcmtk/ofstd/ofcond.h>
  38. #include <dcmtk/ofstd/ofstring.h>
  39. #include <dcmtk/ofstd/ofstd.h> /* for class OFStandard */
  40. #include <dcmtk/dcmdata/dcddirif.h> /* for class DicomDirInterface */
  41. #include <dcmimage.h>
  42. #include <dcmtk/dcmjpeg/djdecode.h> /* for dcmjpeg decoders */
  43. #include <dcmtk/dcmjpeg/djencode.h> /* for dcmjpeg encoders */
  44. #include <dcmtk/dcmdata/dcrledrg.h> /* for DcmRLEDecoderRegistration */
  45. #include <dcmtk/dcmdata/dcrleerg.h> /* for DcmRLEEncoderRegistration */
  46. //------------------------------------------------------------------------------
  47. static ctkLogger logger("org.commontk.dicom.DICOMDatabase" );
  48. //------------------------------------------------------------------------------
  49. //------------------------------------------------------------------------------
  50. class ctkDICOMDatabasePrivate
  51. {
  52. Q_DECLARE_PUBLIC(ctkDICOMDatabase);
  53. protected:
  54. ctkDICOMDatabase* const q_ptr;
  55. public:
  56. ctkDICOMDatabasePrivate(ctkDICOMDatabase&);
  57. ~ctkDICOMDatabasePrivate();
  58. void init(QString databaseFile);
  59. void registerCompressionLibraries();
  60. bool executeScript(const QString script);
  61. /// Name of the database file (i.e. for SQLITE the sqlite file)
  62. QString DatabaseFileName;
  63. QString LastError;
  64. QSqlDatabase Database;
  65. QMap<QString, QString> LoadedHeader;
  66. ctkDICOMAbstractThumbnailGenerator* thumbnailGenerator;
  67. };
  68. //------------------------------------------------------------------------------
  69. // ctkDICOMDatabasePrivate methods
  70. //------------------------------------------------------------------------------
  71. ctkDICOMDatabasePrivate::ctkDICOMDatabasePrivate(ctkDICOMDatabase& o): q_ptr(&o)
  72. {
  73. this->thumbnailGenerator = NULL;
  74. }
  75. //------------------------------------------------------------------------------
  76. void ctkDICOMDatabasePrivate::init(QString databaseFilename)
  77. {
  78. Q_Q(ctkDICOMDatabase);
  79. q->openDatabase(databaseFilename);
  80. }
  81. //------------------------------------------------------------------------------
  82. void ctkDICOMDatabasePrivate::registerCompressionLibraries(){
  83. logger.debug("Register compression libraries");
  84. // Register the JPEG libraries in case we need them
  85. // (registration only happens once, so it's okay to call repeatedly)
  86. // register global JPEG decompression codecs
  87. DJDecoderRegistration::registerCodecs();
  88. // register global JPEG compression codecs
  89. DJEncoderRegistration::registerCodecs();
  90. // register RLE compression codec
  91. DcmRLEEncoderRegistration::registerCodecs();
  92. // register RLE decompression codec
  93. DcmRLEDecoderRegistration::registerCodecs();
  94. }
  95. //------------------------------------------------------------------------------
  96. ctkDICOMDatabasePrivate::~ctkDICOMDatabasePrivate()
  97. {
  98. }
  99. //------------------------------------------------------------------------------
  100. void ctkDICOMDatabase::openDatabase(const QString databaseFile, const QString& connectionName )
  101. {
  102. Q_D(ctkDICOMDatabase);
  103. d->DatabaseFileName = databaseFile;
  104. d->Database = QSqlDatabase::addDatabase("QSQLITE", connectionName);
  105. d->Database.setDatabaseName(databaseFile);
  106. if ( ! (d->Database.open()) )
  107. {
  108. d->LastError = d->Database.lastError().text();
  109. return;
  110. }
  111. if ( d->Database.tables().empty() )
  112. {
  113. if (!initializeDatabase())
  114. {
  115. d->LastError = QString("Unable to initialize DICOM database!");
  116. return;
  117. }
  118. }
  119. if (!isInMemory())
  120. {
  121. QFileSystemWatcher* watcher = new QFileSystemWatcher(QStringList(databaseFile),this);
  122. connect(watcher, SIGNAL( fileChanged(const QString&)),this, SIGNAL ( databaseChanged() ) );
  123. }
  124. }
  125. //------------------------------------------------------------------------------
  126. // ctkDICOMDatabase methods
  127. //------------------------------------------------------------------------------
  128. ctkDICOMDatabase::ctkDICOMDatabase(QString databaseFile)
  129. : d_ptr(new ctkDICOMDatabasePrivate(*this))
  130. {
  131. Q_D(ctkDICOMDatabase);
  132. d->registerCompressionLibraries();
  133. d->init(databaseFile);
  134. }
  135. ctkDICOMDatabase::ctkDICOMDatabase(QObject* parent)
  136. : d_ptr(new ctkDICOMDatabasePrivate(*this))
  137. {
  138. Q_UNUSED(parent);
  139. Q_D(ctkDICOMDatabase);
  140. d->registerCompressionLibraries();
  141. }
  142. //------------------------------------------------------------------------------
  143. ctkDICOMDatabase::~ctkDICOMDatabase()
  144. {
  145. }
  146. //----------------------------------------------------------------------------
  147. //------------------------------------------------------------------------------
  148. const QString ctkDICOMDatabase::lastError() const {
  149. Q_D(const ctkDICOMDatabase);
  150. return d->LastError;
  151. }
  152. //------------------------------------------------------------------------------
  153. const QString ctkDICOMDatabase::databaseFilename() const {
  154. Q_D(const ctkDICOMDatabase);
  155. return d->DatabaseFileName;
  156. }
  157. //------------------------------------------------------------------------------
  158. const QString ctkDICOMDatabase::databaseDirectory() const {
  159. QString databaseFile = databaseFilename();
  160. if (!QFileInfo(databaseFile).isAbsolute())
  161. {
  162. databaseFile.prepend(QDir::currentPath() + "/");
  163. }
  164. return QFileInfo ( databaseFile ).absoluteDir().path();
  165. }
  166. //------------------------------------------------------------------------------
  167. const QSqlDatabase& ctkDICOMDatabase::database() const {
  168. Q_D(const ctkDICOMDatabase);
  169. return d->Database;
  170. }
  171. void ctkDICOMDatabase::setThumbnailGenerator(ctkDICOMAbstractThumbnailGenerator *generator){
  172. Q_D(ctkDICOMDatabase);
  173. d->thumbnailGenerator = generator;
  174. }
  175. ctkDICOMAbstractThumbnailGenerator* ctkDICOMDatabase::thumbnailGenerator(){
  176. Q_D(const ctkDICOMDatabase);
  177. return d->thumbnailGenerator;
  178. }
  179. //------------------------------------------------------------------------------
  180. bool ctkDICOMDatabasePrivate::executeScript(const QString script) {
  181. QFile scriptFile(script);
  182. scriptFile.open(QIODevice::ReadOnly);
  183. if ( !scriptFile.isOpen() )
  184. {
  185. qDebug() << "Script file " << script << " could not be opened!\n";
  186. return false;
  187. }
  188. QString sqlCommands( QTextStream(&scriptFile).readAll() );
  189. sqlCommands.replace( '\n', ' ' );
  190. sqlCommands.replace("; ", ";\n");
  191. QStringList sqlCommandsLines = sqlCommands.split('\n');
  192. QSqlQuery query(Database);
  193. for (QStringList::iterator it = sqlCommandsLines.begin(); it != sqlCommandsLines.end()-1; ++it)
  194. {
  195. if (! (*it).startsWith("--") )
  196. {
  197. qDebug() << *it << "\n";
  198. query.exec(*it);
  199. if (query.lastError().type())
  200. {
  201. qDebug() << "There was an error during execution of the statement: " << (*it);
  202. qDebug() << "Error message: " << query.lastError().text();
  203. return false;
  204. }
  205. }
  206. }
  207. return true;
  208. }
  209. //------------------------------------------------------------------------------
  210. bool ctkDICOMDatabase::initializeDatabase(const char* sqlFileName)
  211. {
  212. Q_D(ctkDICOMDatabase);
  213. return d->executeScript(sqlFileName);
  214. }
  215. //------------------------------------------------------------------------------
  216. void ctkDICOMDatabase::closeDatabase()
  217. {
  218. Q_D(ctkDICOMDatabase);
  219. d->Database.close();
  220. }
  221. //------------------------------------------------------------------------------
  222. QStringList ctkDICOMDatabase::studiesForPatient(QString patientUID)
  223. {
  224. Q_D(ctkDICOMDatabase);
  225. QSqlQuery query(d->Database);
  226. query.prepare ( "SELECT StudyInstanceUID FROM Studies WHERE PatientsUID = ?" );
  227. query.bindValue ( 0, patientUID );
  228. query.exec();
  229. QStringList result;
  230. while (query.next())
  231. {
  232. result << query.value(0).toString();
  233. }
  234. return( result );
  235. }
  236. //------------------------------------------------------------------------------
  237. QStringList ctkDICOMDatabase::seriesForStudy(QString studyUID)
  238. {
  239. Q_D(ctkDICOMDatabase);
  240. QSqlQuery query(d->Database);
  241. query.prepare ( "SELECT SeriesInstanceUID FROM Series WHERE StudyInstanceUID=?");
  242. query.bindValue ( 0, studyUID );
  243. query.exec();
  244. QStringList result;
  245. while (query.next())
  246. {
  247. result << query.value(0).toString();
  248. }
  249. return( result );
  250. }
  251. //------------------------------------------------------------------------------
  252. QStringList ctkDICOMDatabase::filesForSeries(QString seriesUID)
  253. {
  254. Q_D(ctkDICOMDatabase);
  255. QSqlQuery query(d->Database);
  256. query.prepare ( "SELECT Filename FROM Images WHERE SeriesInstanceUID=?");
  257. query.bindValue ( 0, seriesUID );
  258. query.exec();
  259. QStringList result;
  260. while (query.next())
  261. {
  262. result << query.value(0).toString();
  263. }
  264. return( result );
  265. }
  266. //------------------------------------------------------------------------------
  267. void ctkDICOMDatabase::loadInstanceHeader (QString sopInstanceUID)
  268. {
  269. Q_D(ctkDICOMDatabase);
  270. QSqlQuery query(d->Database);
  271. query.prepare ( "SELECT Filename FROM Images WHERE SOPInstanceUID=?");
  272. query.bindValue ( 0, sopInstanceUID );
  273. query.exec();
  274. d->LoadedHeader.clear();
  275. if (query.next())
  276. {
  277. QString fileName = query.value(0).toString();
  278. this->loadFileHeader(fileName);
  279. }
  280. return;
  281. }
  282. //------------------------------------------------------------------------------
  283. void ctkDICOMDatabase::loadFileHeader (QString fileName)
  284. {
  285. Q_D(ctkDICOMDatabase);
  286. DcmFileFormat fileFormat;
  287. OFCondition status = fileFormat.loadFile(fileName.toLatin1().data());
  288. if (status.good())
  289. {
  290. DcmDataset *dataset = fileFormat.getDataset();
  291. DcmStack stack;
  292. while (dataset->nextObject(stack, true) == EC_Normal)
  293. {
  294. DcmObject *dO = stack.top();
  295. if (dO->isaString())
  296. {
  297. QString tag = QString("%1,%2").arg(
  298. dO->getGTag(),4,16,QLatin1Char('0')).arg(
  299. dO->getETag(),4,16,QLatin1Char('0'));
  300. std::ostringstream s;
  301. dO->print(s);
  302. d->LoadedHeader[tag] = QString(s.str().c_str());
  303. }
  304. }
  305. }
  306. return;
  307. }
  308. //------------------------------------------------------------------------------
  309. QStringList ctkDICOMDatabase::headerKeys ()
  310. {
  311. Q_D(ctkDICOMDatabase);
  312. return (d->LoadedHeader.keys());
  313. }
  314. //------------------------------------------------------------------------------
  315. QString ctkDICOMDatabase::headerValue (QString key)
  316. {
  317. Q_D(ctkDICOMDatabase);
  318. return (d->LoadedHeader[key]);
  319. }
  320. //------------------------------------------------------------------------------
  321. /*
  322. void ctkDICOMDatabase::insert ( DcmDataset *dataset ) {
  323. this->insert ( dataset, QString() );
  324. }
  325. */
  326. //------------------------------------------------------------------------------
  327. void ctkDICOMDatabase::insert ( DcmDataset *dataset, bool storeFile, bool generateThumbnail)
  328. {
  329. Q_D(ctkDICOMDatabase);
  330. if (!dataset)
  331. {
  332. return;
  333. }
  334. // Check to see if the file has already been loaded
  335. OFString sopInstanceUID ;
  336. dataset->findAndGetOFString(DCM_SOPInstanceUID, sopInstanceUID);
  337. QSqlQuery fileExists ( d->Database );
  338. fileExists.prepare("SELECT InsertTimestamp,Filename FROM Images WHERE SOPInstanceUID == ?");
  339. fileExists.bindValue(0,QString(sopInstanceUID.c_str()));
  340. fileExists.exec();
  341. if ( fileExists.next() && QFileInfo(fileExists.value(1).toString()).lastModified() < QDateTime::fromString(fileExists.value(0).toString(),Qt::ISODate) )
  342. {
  343. logger.debug ( "File " + fileExists.value(1).toString() + " already added" );
  344. return;
  345. }
  346. OFString patientsName, patientID, patientsBirthDate, patientsBirthTime, patientsSex,
  347. patientComments, patientsAge;
  348. OFString studyInstanceUID, studyID, studyDate, studyTime,
  349. accessionNumber, modalitiesInStudy, institutionName, performingPhysiciansName, referringPhysician, studyDescription;
  350. OFString seriesInstanceUID, seriesDate, seriesTime,
  351. seriesDescription, bodyPartExamined, frameOfReferenceUID,
  352. contrastAgent, scanningSequence;
  353. OFString instanceNumber;
  354. Sint32 seriesNumber = 0, acquisitionNumber = 0, echoNumber = 0, temporalPosition = 0;
  355. //If the following fields can not be evaluated, cancel evaluation of the DICOM file
  356. dataset->findAndGetOFString(DCM_PatientName, patientsName);
  357. dataset->findAndGetOFString(DCM_StudyInstanceUID, studyInstanceUID);
  358. dataset->findAndGetOFString(DCM_SeriesInstanceUID, seriesInstanceUID);
  359. dataset->findAndGetOFString(DCM_PatientID, patientID);
  360. dataset->findAndGetOFString(DCM_PatientBirthDate, patientsBirthDate);
  361. dataset->findAndGetOFString(DCM_PatientBirthTime, patientsBirthTime);
  362. dataset->findAndGetOFString(DCM_PatientSex, patientsSex);
  363. dataset->findAndGetOFString(DCM_PatientAge, patientsAge);
  364. dataset->findAndGetOFString(DCM_PatientComments, patientComments);
  365. dataset->findAndGetOFString(DCM_StudyID, studyID);
  366. dataset->findAndGetOFString(DCM_StudyDate, studyDate);
  367. dataset->findAndGetOFString(DCM_StudyTime, studyTime);
  368. dataset->findAndGetOFString(DCM_AccessionNumber, accessionNumber);
  369. dataset->findAndGetOFString(DCM_ModalitiesInStudy, modalitiesInStudy);
  370. dataset->findAndGetOFString(DCM_InstitutionName, institutionName);
  371. dataset->findAndGetOFString(DCM_PerformingPhysicianName, performingPhysiciansName);
  372. dataset->findAndGetOFString(DCM_ReferringPhysicianName, referringPhysician);
  373. dataset->findAndGetOFString(DCM_StudyDescription, studyDescription);
  374. dataset->findAndGetOFString(DCM_SeriesDate, seriesDate);
  375. dataset->findAndGetOFString(DCM_SeriesTime, seriesTime);
  376. dataset->findAndGetOFString(DCM_SeriesDescription, seriesDescription);
  377. dataset->findAndGetOFString(DCM_BodyPartExamined, bodyPartExamined);
  378. dataset->findAndGetOFString(DCM_FrameOfReferenceUID, frameOfReferenceUID);
  379. dataset->findAndGetOFString(DCM_ContrastBolusAgent, contrastAgent);
  380. dataset->findAndGetOFString(DCM_ScanningSequence, scanningSequence);
  381. dataset->findAndGetSint32(DCM_SeriesNumber, seriesNumber);
  382. dataset->findAndGetSint32(DCM_AcquisitionNumber, acquisitionNumber);
  383. dataset->findAndGetSint32(DCM_EchoNumbers, echoNumber);
  384. dataset->findAndGetSint32(DCM_TemporalPositionIdentifier, temporalPosition);
  385. // store the file if the database is not in memomry
  386. QString filename;
  387. if ( storeFile && !this->isInMemory() )
  388. {
  389. DcmFileFormat* fileformat = new DcmFileFormat ( dataset );
  390. QString destinationDirectoryName = databaseDirectory() + "/dicom/";
  391. QDir destinationDir(destinationDirectoryName);
  392. QString studySeriesDirectory = QString(studyInstanceUID.c_str()) + "/" + seriesInstanceUID.c_str();
  393. destinationDir.mkpath(studySeriesDirectory);
  394. filename = databaseDirectory() + "/dicom/" + pathForDataset(dataset);
  395. logger.debug ( "Saving file: " + filename );
  396. OFCondition status = fileformat->saveFile ( filename.toAscii() );
  397. if ( !status.good() )
  398. {
  399. logger.error ( "Error saving file: " + filename + "\nError is " + status.text() );
  400. delete fileformat;
  401. return;
  402. }
  403. delete fileformat;
  404. }
  405. QSqlQuery check_exists_query(d->Database);
  406. //The patient UID is a unique number within the database, generated by the sqlite autoincrement
  407. int patientUID = -1;
  408. if ( patientID != "" && patientsName != "" )
  409. {
  410. //Check if patient is already present in the db
  411. check_exists_query.prepare ( "SELECT * FROM Patients WHERE PatientID = ? AND PatientsName = ?" );
  412. check_exists_query.bindValue ( 0, QString ( patientID.c_str() ) );
  413. check_exists_query.bindValue ( 1, QString ( patientsName.c_str() ) );
  414. check_exists_query.exec();
  415. if (check_exists_query.next())
  416. {
  417. patientUID = check_exists_query.value(check_exists_query.record().indexOf("UID")).toInt();
  418. }
  419. else
  420. {
  421. // Insert it
  422. QSqlQuery statement ( d->Database );
  423. statement.prepare ( "INSERT INTO Patients ('UID', 'PatientsName', 'PatientID', 'PatientsBirthDate', 'PatientsBirthTime', 'PatientsSex', 'PatientsAge', 'PatientsComments' ) values ( NULL, ?, ?, ?, ?, ?, ?, ? )" );
  424. statement.bindValue ( 0, QString ( patientsName.c_str() ) );
  425. statement.bindValue ( 1, QString ( patientID.c_str() ) );
  426. statement.bindValue ( 2, QString ( patientsBirthDate.c_str() ) );
  427. statement.bindValue ( 3, QString ( patientsBirthTime.c_str() ) );
  428. statement.bindValue ( 4, QString ( patientsSex.c_str() ) );
  429. // TODO: shift patient's age to study, since this is not a patient level attribute in images
  430. // statement.bindValue ( 5, QString ( patientsAge.c_str() ) );
  431. statement.bindValue ( 6, QString ( patientComments.c_str() ) );
  432. statement.exec ();
  433. patientUID = statement.lastInsertId().toInt();
  434. logger.debug ( "New patient inserted: " + QString().setNum ( patientUID ) );
  435. }
  436. }
  437. if ( studyInstanceUID != "" )
  438. {
  439. check_exists_query.prepare ( "SELECT * FROM Studies WHERE StudyInstanceUID = ?" );
  440. check_exists_query.bindValue ( 0, QString ( studyInstanceUID.c_str() ) );
  441. check_exists_query.exec();
  442. if(!check_exists_query.next())
  443. {
  444. QSqlQuery statement ( d->Database );
  445. statement.prepare ( "INSERT INTO Studies ( 'StudyInstanceUID', 'PatientsUID', 'StudyID', 'StudyDate', 'StudyTime', 'AccessionNumber', 'ModalitiesInStudy', 'InstitutionName', 'ReferringPhysician', 'PerformingPhysiciansName', 'StudyDescription' ) VALUES ( ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ? )" );
  446. statement.bindValue ( 0, QString ( studyInstanceUID.c_str() ) );
  447. statement.bindValue ( 1, patientUID );
  448. statement.bindValue ( 2, QString ( studyID.c_str() ) );
  449. statement.bindValue ( 3, QDate::fromString ( studyDate.c_str(), "yyyyMMdd" ) );
  450. statement.bindValue ( 4, QString ( studyTime.c_str() ) );
  451. statement.bindValue ( 5, QString ( accessionNumber.c_str() ) );
  452. statement.bindValue ( 6, QString ( modalitiesInStudy.c_str() ) );
  453. statement.bindValue ( 7, QString ( institutionName.c_str() ) );
  454. statement.bindValue ( 8, QString ( referringPhysician.c_str() ) );
  455. statement.bindValue ( 9, QString ( performingPhysiciansName.c_str() ) );
  456. statement.bindValue ( 10, QString ( studyDescription.c_str() ) );
  457. if ( !statement.exec() )
  458. {
  459. logger.error ( "Error executing statament: " + statement.lastQuery() + " Error: " + statement.lastError().text() );
  460. }
  461. }
  462. }
  463. if ( seriesInstanceUID != "" )
  464. {
  465. check_exists_query.prepare ( "SELECT * FROM Series WHERE SeriesInstanceUID = ?" );
  466. check_exists_query.bindValue ( 0, QString ( seriesInstanceUID.c_str() ) );
  467. logger.warn ( "Statement: " + check_exists_query.lastQuery() );
  468. check_exists_query.exec();
  469. if(!check_exists_query.next())
  470. {
  471. QSqlQuery statement ( d->Database );
  472. statement.prepare ( "INSERT INTO Series ( 'SeriesInstanceUID', 'StudyInstanceUID', 'SeriesNumber', 'SeriesDate', 'SeriesTime', 'SeriesDescription', 'BodyPartExamined', 'FrameOfReferenceUID', 'AcquisitionNumber', 'ContrastAgent', 'ScanningSequence', 'EchoNumber', 'TemporalPosition' ) VALUES ( ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ? )" );
  473. statement.bindValue ( 0, QString ( seriesInstanceUID.c_str() ) );
  474. statement.bindValue ( 1, QString ( studyInstanceUID.c_str() ) );
  475. statement.bindValue ( 2, static_cast<int>(seriesNumber) );
  476. statement.bindValue ( 3, QString ( seriesDate.c_str() ) );
  477. statement.bindValue ( 4, QDate::fromString ( seriesTime.c_str(), "yyyyMMdd" ) );
  478. statement.bindValue ( 5, QString ( seriesDescription.c_str() ) );
  479. statement.bindValue ( 6, QString ( bodyPartExamined.c_str() ) );
  480. statement.bindValue ( 7, QString ( frameOfReferenceUID.c_str() ) );
  481. statement.bindValue ( 8, static_cast<int>(acquisitionNumber) );
  482. statement.bindValue ( 9, QString ( contrastAgent.c_str() ) );
  483. statement.bindValue ( 10, QString ( scanningSequence.c_str() ) );
  484. statement.bindValue ( 11, static_cast<int>(echoNumber) );
  485. statement.bindValue ( 12, static_cast<int>(temporalPosition) );
  486. if ( !statement.exec() )
  487. {
  488. logger.error ( "Error executing statament: " + statement.lastQuery() + " Error: " + statement.lastError().text() );
  489. }
  490. }
  491. }
  492. if ( !filename.isEmpty() )
  493. {
  494. check_exists_query.prepare ( "SELECT * FROM Images WHERE Filename = ?" );
  495. check_exists_query.bindValue ( 0, filename );
  496. check_exists_query.exec();
  497. if(!check_exists_query.next())
  498. {
  499. QSqlQuery statement ( d->Database );
  500. statement.prepare ( "INSERT INTO Images ( 'SOPInstanceUID', 'Filename', 'SeriesInstanceUID', 'InsertTimestamp' ) VALUES ( ?, ?, ?, ? )" );
  501. statement.bindValue ( 0, QString ( sopInstanceUID.c_str() ) );
  502. statement.bindValue ( 1, filename );
  503. statement.bindValue ( 2, QString ( seriesInstanceUID.c_str() ) );
  504. statement.bindValue ( 3, QDateTime::currentDateTime() );
  505. statement.exec();
  506. }
  507. }
  508. if(generateThumbnail){
  509. if(d->thumbnailGenerator){
  510. QString studySeriesDirectory = QString(studyInstanceUID.c_str()) + "/" + QString(seriesInstanceUID.c_str());
  511. //Create thumbnail here
  512. QString thumbnailPath = databaseDirectory() +
  513. "/thumbs/" + this->pathForDataset(dataset) + ".png";
  514. //QString(studyInstanceUID.c_str()) + "/" +
  515. //QString(seriesInstanceUID.c_str()) + "/" +
  516. //QString(sopInstanceUID.c_str()) + ".png";
  517. QFileInfo thumbnailInfo(thumbnailPath);
  518. if(!(thumbnailInfo.exists() && (thumbnailInfo.lastModified() > QFileInfo(filename).lastModified()))){
  519. QDir(databaseDirectory() + "/thumbs/").mkpath(studySeriesDirectory);
  520. DicomImage dcmImage(QDir::toNativeSeparators(filename).toAscii());
  521. d->thumbnailGenerator->generateThumbnail(&dcmImage, thumbnailPath);
  522. }
  523. }
  524. }
  525. if (isInMemory())
  526. {
  527. emit databaseChanged();
  528. }
  529. }
  530. bool ctkDICOMDatabase::isOpen() const
  531. {
  532. Q_D(const ctkDICOMDatabase);
  533. return d->Database.isOpen();
  534. }
  535. bool ctkDICOMDatabase::isInMemory() const
  536. {
  537. Q_D(const ctkDICOMDatabase);
  538. return d->DatabaseFileName == ":memory:";
  539. }
  540. QString ctkDICOMDatabase::pathForDataset( DcmDataset *dataset)
  541. {
  542. if (!dataset)
  543. {
  544. return QString();
  545. }
  546. OFString studyInstanceUID, seriesInstanceUID, sopInstanceUID;
  547. dataset->findAndGetOFString(DCM_StudyInstanceUID, studyInstanceUID);
  548. dataset->findAndGetOFString(DCM_SeriesInstanceUID, seriesInstanceUID);
  549. dataset->findAndGetOFString(DCM_SOPInstanceUID, sopInstanceUID);
  550. return QString(studyInstanceUID.c_str()) + "/" + seriesInstanceUID.c_str() + "/" + sopInstanceUID.c_str();
  551. }