Mercurial > hg > orthanc
annotate OrthancServer/ServerIndex.cpp @ 184:d4e967d401d3
debugging for msvc
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Mon, 12 Nov 2012 15:01:39 +0100 |
parents | 626777d01dc4 |
children | f68c039b0571 |
rev | line source |
---|---|
0 | 1 /** |
62 | 2 * Orthanc - A Lightweight, RESTful DICOM Store |
0 | 3 * Copyright (C) 2012 Medical Physics Department, CHU of Liege, |
4 * Belgium | |
5 * | |
6 * This program is free software: you can redistribute it and/or | |
7 * modify it under the terms of the GNU General Public License as | |
8 * published by the Free Software Foundation, either version 3 of the | |
9 * License, or (at your option) any later version. | |
136 | 10 * |
11 * In addition, as a special exception, the copyright holders of this | |
12 * program give permission to link the code of its release with the | |
13 * OpenSSL project's "OpenSSL" library (or with modified versions of it | |
14 * that use the same license as the "OpenSSL" library), and distribute | |
15 * the linked executables. You must obey the GNU General Public License | |
16 * in all respects for all of the code used other than "OpenSSL". If you | |
17 * modify file(s) with this exception, you may extend this exception to | |
18 * your version of the file(s), but you are not obligated to do so. If | |
19 * you do not wish to do so, delete this exception statement from your | |
20 * version. If you delete this exception statement from all source files | |
21 * in the program, then also delete it here. | |
0 | 22 * |
23 * This program is distributed in the hope that it will be useful, but | |
24 * WITHOUT ANY WARRANTY; without even the implied warranty of | |
25 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
26 * General Public License for more details. | |
27 * | |
28 * You should have received a copy of the GNU General Public License | |
29 * along with this program. If not, see <http://www.gnu.org/licenses/>. | |
30 **/ | |
31 | |
32 | |
33 #include "ServerIndex.h" | |
34 | |
62 | 35 using namespace Orthanc; |
0 | 36 |
6 | 37 #ifndef NOMINMAX |
2 | 38 #define NOMINMAX |
6 | 39 #endif |
40 | |
8 | 41 #include "EmbeddedResources.h" |
0 | 42 #include "../Core/Toolbox.h" |
43 #include "../Core/Uuid.h" | |
44 #include "../Core/DicomFormat/DicomArray.h" | |
45 #include "../Core/SQLite/Transaction.h" | |
46 #include "FromDcmtkBridge.h" | |
47 | |
48 #include <boost/lexical_cast.hpp> | |
49 #include <stdio.h> | |
108 | 50 #include <glog/logging.h> |
0 | 51 |
62 | 52 namespace Orthanc |
0 | 53 { |
54 namespace Internals | |
55 { | |
56 class DeleteFromFileStorageFunction : public SQLite::IScalarFunction | |
57 { | |
58 private: | |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
59 std::auto_ptr<FileStorage> fileStorage_; |
0 | 60 |
61 public: | |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
62 DeleteFromFileStorageFunction(const std::string& path) |
0 | 63 { |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
64 if (path != ":memory:") |
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
65 { |
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
66 fileStorage_.reset(new FileStorage(path)); |
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
67 } |
0 | 68 } |
69 | |
70 virtual const char* GetName() const | |
71 { | |
72 return "DeleteFromFileStorage"; | |
73 } | |
74 | |
75 virtual unsigned int GetCardinality() const | |
76 { | |
77 return 1; | |
78 } | |
79 | |
80 virtual void Compute(SQLite::FunctionContext& context) | |
81 { | |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
82 if (fileStorage_.get() == NULL) |
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
83 { |
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
84 // In-memory index, for unit tests |
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
85 return; |
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
86 } |
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
87 |
0 | 88 std::string fileUuid = context.GetStringValue(0); |
108 | 89 LOG(INFO) << "Removing file [" << fileUuid << "]"; |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
90 |
0 | 91 if (Toolbox::IsUuid(fileUuid)) |
92 { | |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
93 fileStorage_->Remove(fileUuid); |
0 | 94 } |
95 } | |
96 }; | |
97 | |
98 | |
99 class SignalDeletedLevelFunction : public SQLite::IScalarFunction | |
100 { | |
101 private: | |
102 int remainingLevel_; | |
103 std::string remainingLevelUuid_; | |
104 | |
105 public: | |
106 void Clear() | |
107 { | |
108 remainingLevel_ = std::numeric_limits<int>::max(); | |
109 } | |
110 | |
111 bool HasRemainingLevel() const | |
112 { | |
113 return (remainingLevel_ != 0 && | |
114 remainingLevel_ != std::numeric_limits<int>::max()); | |
115 } | |
116 | |
117 const std::string& GetRemainingLevelUuid() const | |
118 { | |
119 assert(HasRemainingLevel()); | |
120 return remainingLevelUuid_; | |
121 } | |
122 | |
123 const char* GetRemainingLevelType() const | |
124 { | |
125 assert(HasRemainingLevel()); | |
126 switch (remainingLevel_) | |
127 { | |
128 case 1: | |
129 return "patient"; | |
130 case 2: | |
131 return "study"; | |
132 case 3: | |
133 return "series"; | |
134 default: | |
62 | 135 throw OrthancException(ErrorCode_InternalError); |
0 | 136 } |
137 } | |
138 | |
139 virtual const char* GetName() const | |
140 { | |
141 return "SignalDeletedLevel"; | |
142 } | |
143 | |
144 virtual unsigned int GetCardinality() const | |
145 { | |
146 return 2; | |
147 } | |
148 | |
149 virtual void Compute(SQLite::FunctionContext& context) | |
150 { | |
151 int level = context.GetIntValue(0); | |
152 if (level < remainingLevel_) | |
153 { | |
154 remainingLevel_ = level; | |
155 remainingLevelUuid_ = context.GetStringValue(1); | |
156 } | |
157 | |
158 //printf("deleted level [%d] [%s]\n", level, context.GetStringValue(1).c_str()); | |
159 } | |
160 }; | |
161 } | |
162 | |
163 | |
164 void ServerIndex::StoreMainDicomTags(const std::string& uuid, | |
165 const DicomMap& map) | |
166 { | |
167 DicomArray flattened(map); | |
168 for (size_t i = 0; i < flattened.GetSize(); i++) | |
169 { | |
138 | 170 SQLite::Statement s(db_, SQLITE_FROM_HERE, "INSERT INTO MainDicomTags VALUES(?, ?, ?, ?)"); |
0 | 171 s.BindString(0, uuid); |
172 s.BindInt(1, flattened.GetElement(i).GetTag().GetGroup()); | |
173 s.BindInt(2, flattened.GetElement(i).GetTag().GetElement()); | |
174 s.BindString(3, flattened.GetElement(i).GetValue().AsString()); | |
175 s.Run(); | |
176 } | |
177 } | |
178 | |
179 bool ServerIndex::GetMainDicomStringTag(std::string& result, | |
180 const std::string& uuid, | |
181 const DicomTag& tag) | |
182 { | |
183 SQLite::Statement s(db_, SQLITE_FROM_HERE, | |
184 "SELECT * FROM MainDicomTags WHERE uuid=? AND tagGroup=? AND tagElement=?"); | |
185 s.BindString(0, uuid); | |
186 s.BindInt(1, tag.GetGroup()); | |
187 s.BindInt(2, tag.GetElement()); | |
188 if (!s.Step()) | |
189 { | |
190 return false; | |
191 } | |
192 | |
193 result = s.ColumnString(0); | |
194 return true; | |
195 } | |
196 | |
197 bool ServerIndex::GetMainDicomIntTag(int& result, | |
198 const std::string& uuid, | |
199 const DicomTag& tag) | |
200 { | |
201 std::string s; | |
202 if (!GetMainDicomStringTag(s, uuid, tag)) | |
203 { | |
204 return false; | |
205 } | |
206 | |
207 try | |
208 { | |
209 result = boost::lexical_cast<int>(s); | |
210 return true; | |
211 } | |
212 catch (boost::bad_lexical_cast) | |
213 { | |
214 return false; | |
215 } | |
216 } | |
217 | |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
218 |
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
219 bool ServerIndex::HasInstance(DicomInstanceHasher& hasher) |
0 | 220 { |
221 SQLite::Statement s(db_, SQLITE_FROM_HERE, "SELECT uuid FROM Instances WHERE dicomInstance=?"); | |
179 | 222 s.BindString(0, hasher.GetInstanceUid()); |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
223 return s.Step(); |
0 | 224 } |
225 | |
226 | |
227 void ServerIndex::RecordChange(const std::string& resourceType, | |
228 const std::string& uuid) | |
229 { | |
230 SQLite::Statement s(db_, SQLITE_FROM_HERE, "INSERT INTO Changes VALUES(NULL, ?, ?)"); | |
231 s.BindString(0, resourceType); | |
232 s.BindString(1, uuid); | |
233 s.Run(); | |
234 } | |
235 | |
236 | |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
237 void ServerIndex::CreateInstance(DicomInstanceHasher& hasher, |
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
238 const DicomMap& dicomSummary, |
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
239 const std::string& fileUuid, |
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
240 uint64_t fileSize, |
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
241 const std::string& jsonUuid, |
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
242 const std::string& distantAet) |
0 | 243 { |
82 | 244 SQLite::Statement s2(db_, SQLITE_FROM_HERE, "INSERT INTO Resources VALUES(?, ?)"); |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
245 s2.BindString(0, hasher.HashInstance()); |
82 | 246 s2.BindInt(1, ResourceType_Instance); |
247 s2.Run(); | |
248 | |
80 | 249 SQLite::Statement s(db_, SQLITE_FROM_HERE, "INSERT INTO Instances VALUES(?, ?, ?, ?, ?, ?, ?, ?)"); |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
250 s.BindString(0, hasher.HashInstance()); |
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
251 s.BindString(1, hasher.HashSeries()); |
179 | 252 s.BindString(2, hasher.GetInstanceUid()); |
0 | 253 s.BindString(3, fileUuid); |
254 s.BindInt64(4, fileSize); | |
255 s.BindString(5, jsonUuid); | |
256 s.BindString(6, distantAet); | |
80 | 257 |
258 const DicomValue* indexInSeries; | |
259 if ((indexInSeries = dicomSummary.TestAndGetValue(DICOM_TAG_INSTANCE_NUMBER)) != NULL || | |
260 (indexInSeries = dicomSummary.TestAndGetValue(DICOM_TAG_IMAGE_INDEX)) != NULL) | |
261 { | |
262 s.BindInt(7, boost::lexical_cast<unsigned int>(indexInSeries->AsString())); | |
263 } | |
264 else | |
265 { | |
266 s.BindNull(7); | |
267 } | |
268 | |
0 | 269 s.Run(); |
270 | |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
271 RecordChange("instances", hasher.HashInstance()); |
0 | 272 |
273 DicomMap dicom; | |
274 dicomSummary.ExtractInstanceInformation(dicom); | |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
275 StoreMainDicomTags(hasher.HashInstance(), dicom); |
0 | 276 } |
277 | |
278 void ServerIndex::RemoveInstance(const std::string& uuid) | |
279 { | |
280 SQLite::Statement s(db_, SQLITE_FROM_HERE, "DELETE FROM Instances WHERE uuid=?"); | |
281 s.BindString(0, uuid); | |
282 s.Run(); | |
283 } | |
284 | |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
285 bool ServerIndex::HasSeries(DicomInstanceHasher& hasher) |
0 | 286 { |
287 SQLite::Statement s(db_, SQLITE_FROM_HERE, "SELECT uuid FROM Series WHERE dicomSeries=?"); | |
179 | 288 s.BindString(0, hasher.GetSeriesUid()); |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
289 return s.Step(); |
0 | 290 } |
291 | |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
292 void ServerIndex::CreateSeries(DicomInstanceHasher& hasher, |
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
293 const DicomMap& dicomSummary) |
0 | 294 { |
82 | 295 SQLite::Statement s2(db_, SQLITE_FROM_HERE, "INSERT INTO Resources VALUES(?, ?)"); |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
296 s2.BindString(0, hasher.HashSeries()); |
82 | 297 s2.BindInt(1, ResourceType_Series); |
298 s2.Run(); | |
299 | |
77 | 300 SQLite::Statement s(db_, SQLITE_FROM_HERE, "INSERT INTO Series VALUES(?, ?, ?, ?)"); |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
301 s.BindString(0, hasher.HashSeries()); |
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
302 s.BindString(1, hasher.HashStudy()); |
179 | 303 s.BindString(2, hasher.GetSeriesUid()); |
80 | 304 |
305 const DicomValue* expectedNumberOfInstances; | |
84 | 306 if (//(expectedNumberOfInstances = dicomSummary.TestAndGetValue(DICOM_TAG_NUMBER_OF_FRAMES)) != NULL || |
80 | 307 (expectedNumberOfInstances = dicomSummary.TestAndGetValue(DICOM_TAG_NUMBER_OF_SLICES)) != NULL || |
82 | 308 //(expectedNumberOfInstances = dicomSummary.TestAndGetValue(DICOM_TAG_CARDIAC_NUMBER_OF_IMAGES)) != NULL || |
80 | 309 (expectedNumberOfInstances = dicomSummary.TestAndGetValue(DICOM_TAG_IMAGES_IN_ACQUISITION)) != NULL) |
310 { | |
311 s.BindInt(3, boost::lexical_cast<unsigned int>(expectedNumberOfInstances->AsString())); | |
312 } | |
313 else | |
314 { | |
315 s.BindNull(3); | |
316 } | |
317 | |
0 | 318 s.Run(); |
319 | |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
320 RecordChange("series", hasher.HashSeries()); |
0 | 321 |
322 DicomMap dicom; | |
323 dicomSummary.ExtractSeriesInformation(dicom); | |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
324 StoreMainDicomTags(hasher.HashSeries(), dicom); |
0 | 325 } |
326 | |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
327 bool ServerIndex::HasStudy(DicomInstanceHasher& hasher) |
0 | 328 { |
329 SQLite::Statement s(db_, SQLITE_FROM_HERE, "SELECT uuid FROM Studies WHERE dicomStudy=?"); | |
179 | 330 s.BindString(0, hasher.GetStudyUid()); |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
331 return s.Step(); |
0 | 332 } |
333 | |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
334 void ServerIndex::CreateStudy(DicomInstanceHasher& hasher, |
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
335 const DicomMap& dicomSummary) |
0 | 336 { |
82 | 337 SQLite::Statement s2(db_, SQLITE_FROM_HERE, "INSERT INTO Resources VALUES(?, ?)"); |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
338 s2.BindString(0, hasher.HashStudy()); |
82 | 339 s2.BindInt(1, ResourceType_Study); |
340 s2.Run(); | |
341 | |
0 | 342 SQLite::Statement s(db_, SQLITE_FROM_HERE, "INSERT INTO Studies VALUES(?, ?, ?)"); |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
343 s.BindString(0, hasher.HashStudy()); |
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
344 s.BindString(1, hasher.HashPatient()); |
179 | 345 s.BindString(2, hasher.GetStudyUid()); |
0 | 346 s.Run(); |
347 | |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
348 RecordChange("studies", hasher.HashStudy()); |
0 | 349 |
350 DicomMap dicom; | |
351 dicomSummary.ExtractStudyInformation(dicom); | |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
352 StoreMainDicomTags(hasher.HashStudy(), dicom); |
0 | 353 } |
354 | |
355 | |
356 | |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
357 bool ServerIndex::HasPatient(DicomInstanceHasher& hasher) |
0 | 358 { |
359 SQLite::Statement s(db_, SQLITE_FROM_HERE, "SELECT uuid FROM Patients WHERE dicomPatientId=?"); | |
179 | 360 s.BindString(0,hasher.GetPatientId()); |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
361 return s.Step(); |
0 | 362 } |
363 | |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
364 void ServerIndex::CreatePatient(DicomInstanceHasher& hasher, |
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
365 const DicomMap& dicomSummary) |
0 | 366 { |
80 | 367 std::string dicomPatientId = dicomSummary.GetValue(DICOM_TAG_PATIENT_ID).AsString(); |
0 | 368 |
82 | 369 SQLite::Statement s2(db_, SQLITE_FROM_HERE, "INSERT INTO Resources VALUES(?, ?)"); |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
370 s2.BindString(0, hasher.HashPatient()); |
82 | 371 s2.BindInt(1, ResourceType_Patient); |
372 s2.Run(); | |
373 | |
0 | 374 SQLite::Statement s(db_, SQLITE_FROM_HERE, "INSERT INTO Patients VALUES(?, ?)"); |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
375 s.BindString(0, hasher.HashPatient()); |
0 | 376 s.BindString(1, dicomPatientId); |
377 s.Run(); | |
378 | |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
379 RecordChange("patients", hasher.HashPatient()); |
0 | 380 |
381 DicomMap dicom; | |
382 dicomSummary.ExtractPatientInformation(dicom); | |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
383 StoreMainDicomTags(hasher.HashPatient(), dicom); |
0 | 384 } |
385 | |
386 | |
387 void ServerIndex::GetMainDicomTags(DicomMap& map, | |
388 const std::string& uuid) | |
389 { | |
390 map.Clear(); | |
391 | |
392 SQLite::Statement s(db_, SQLITE_FROM_HERE, "SELECT * FROM MainDicomTags WHERE uuid=?"); | |
393 s.BindString(0, uuid); | |
394 while (s.Step()) | |
395 { | |
396 map.SetValue(s.ColumnInt(1), | |
397 s.ColumnInt(2), | |
398 s.ColumnString(3)); | |
399 } | |
400 } | |
401 | |
402 void ServerIndex::MainDicomTagsToJson(Json::Value& target, | |
403 const std::string& uuid) | |
404 { | |
405 DicomMap map; | |
406 GetMainDicomTags(map, uuid); | |
407 target["MainDicomTags"] = Json::objectValue; | |
408 FromDcmtkBridge::ToJson(target["MainDicomTags"], map); | |
409 } | |
410 | |
411 | |
412 bool ServerIndex::DeleteInternal(Json::Value& target, | |
413 const std::string& uuid, | |
414 const std::string& tableName) | |
415 { | |
416 boost::mutex::scoped_lock scoped_lock(mutex_); | |
417 | |
418 deletedLevels_->Clear(); | |
419 | |
420 SQLite::Statement s(db_, "DELETE FROM " + tableName + " WHERE uuid=?"); | |
421 s.BindString(0, uuid); | |
422 | |
423 if (!s.Run()) | |
424 { | |
425 return false; | |
426 } | |
427 | |
428 if (db_.GetLastChangeCount() == 0) | |
429 { | |
430 // Nothing was deleted, inexistent UUID | |
431 return false; | |
432 } | |
433 | |
434 if (deletedLevels_->HasRemainingLevel()) | |
435 { | |
436 std::string type(deletedLevels_->GetRemainingLevelType()); | |
437 const std::string& uuid = deletedLevels_->GetRemainingLevelUuid(); | |
438 | |
439 target["RemainingAncestor"] = Json::Value(Json::objectValue); | |
440 target["RemainingAncestor"]["Path"] = "/" + type + "/" + uuid; | |
441 target["RemainingAncestor"]["Type"] = type; | |
442 target["RemainingAncestor"]["ID"] = uuid; | |
443 } | |
444 else | |
445 { | |
446 target["RemainingAncestor"] = Json::nullValue; | |
447 } | |
448 | |
449 return true; | |
450 } | |
451 | |
452 | |
453 ServerIndex::ServerIndex(const std::string& storagePath) | |
454 { | |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
455 if (storagePath == ":memory:") |
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
456 { |
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
457 db_.OpenInMemory(); |
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
458 } |
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
459 else |
0 | 460 { |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
461 boost::filesystem::path p = storagePath; |
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
462 |
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
463 try |
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
464 { |
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
465 boost::filesystem::create_directories(storagePath); |
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
466 } |
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
467 catch (boost::filesystem::filesystem_error) |
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
468 { |
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
469 } |
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
470 |
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
471 p /= "index"; |
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
472 db_.Open(p.string()); |
0 | 473 } |
474 | |
475 db_.Register(new Internals::DeleteFromFileStorageFunction(storagePath)); | |
476 deletedLevels_ = (Internals::SignalDeletedLevelFunction*) | |
477 db_.Register(new Internals::SignalDeletedLevelFunction); | |
478 | |
479 if (!db_.DoesTableExist("GlobalProperties")) | |
480 { | |
108 | 481 LOG(INFO) << "Creating the database"; |
0 | 482 std::string query; |
483 EmbeddedResources::GetFileResource(query, EmbeddedResources::PREPARE_DATABASE); | |
484 db_.Execute(query); | |
485 } | |
486 } | |
487 | |
488 | |
489 StoreStatus ServerIndex::Store(std::string& instanceUuid, | |
490 const DicomMap& dicomSummary, | |
491 const std::string& fileUuid, | |
492 uint64_t uncompressedFileSize, | |
493 const std::string& jsonUuid, | |
494 const std::string& distantAet) | |
495 { | |
496 boost::mutex::scoped_lock scoped_lock(mutex_); | |
497 | |
178 | 498 DicomInstanceHasher hasher(dicomSummary); |
0 | 499 |
500 try | |
501 { | |
502 SQLite::Transaction t(db_); | |
503 t.Begin(); | |
504 | |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
505 if (HasInstance(hasher)) |
0 | 506 { |
507 return StoreStatus_AlreadyStored; | |
508 // TODO: Check consistency? | |
509 } | |
510 | |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
511 if (HasPatient(hasher)) |
0 | 512 { |
513 // TODO: Check consistency? | |
514 } | |
515 else | |
516 { | |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
517 CreatePatient(hasher, dicomSummary); |
0 | 518 } |
519 | |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
520 if (HasStudy(hasher)) |
0 | 521 { |
522 // TODO: Check consistency? | |
523 } | |
524 else | |
525 { | |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
526 CreateStudy(hasher, dicomSummary); |
0 | 527 } |
528 | |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
529 if (HasSeries(hasher)) |
0 | 530 { |
531 // TODO: Check consistency? | |
532 } | |
533 else | |
534 { | |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
535 CreateSeries(hasher, dicomSummary); |
0 | 536 } |
537 | |
180
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
538 CreateInstance(hasher, dicomSummary, fileUuid, |
626777d01dc4
use of hashes to index dicom objects
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
179
diff
changeset
|
539 uncompressedFileSize, jsonUuid, distantAet); |
0 | 540 |
541 t.Commit(); | |
542 return StoreStatus_Success; | |
543 //t.Rollback(); | |
544 } | |
62 | 545 catch (OrthancException& e) |
0 | 546 { |
108 | 547 LOG(ERROR) << "EXCEPTION [" << e.What() << "]" << " " << db_.GetErrorMessage(); |
0 | 548 } |
549 | |
550 return StoreStatus_Failure; | |
551 } | |
552 | |
553 | |
554 StoreStatus ServerIndex::Store(std::string& instanceUuid, | |
555 FileStorage& storage, | |
556 const char* dicomFile, | |
557 size_t dicomSize, | |
558 const DicomMap& dicomSummary, | |
559 const Json::Value& dicomJson, | |
560 const std::string& distantAet) | |
561 { | |
562 std::string fileUuid = storage.Create(dicomFile, dicomSize); | |
563 std::string jsonUuid = storage.Create(dicomJson.toStyledString()); | |
564 StoreStatus status = Store(instanceUuid, dicomSummary, fileUuid, | |
565 dicomSize, jsonUuid, distantAet); | |
566 | |
567 if (status != StoreStatus_Success) | |
568 { | |
147 | 569 storage.Remove(fileUuid); |
570 storage.Remove(jsonUuid); | |
0 | 571 } |
572 | |
573 switch (status) | |
574 { | |
575 case StoreStatus_Success: | |
138 | 576 LOG(WARNING) << "New instance stored: " << GetTotalSize() << " bytes"; |
0 | 577 break; |
578 | |
579 case StoreStatus_AlreadyStored: | |
138 | 580 LOG(WARNING) << "Already stored"; |
0 | 581 break; |
582 | |
583 case StoreStatus_Failure: | |
138 | 584 LOG(ERROR) << "Store failure"; |
0 | 585 break; |
586 } | |
587 | |
588 return status; | |
589 } | |
590 | |
591 uint64_t ServerIndex::GetTotalSize() | |
592 { | |
593 boost::mutex::scoped_lock scoped_lock(mutex_); | |
594 SQLite::Statement s(db_, SQLITE_FROM_HERE, "SELECT SUM(fileSize) FROM Instances"); | |
595 s.Run(); | |
596 return s.ColumnInt64(0); | |
597 } | |
598 | |
599 | |
600 SeriesStatus ServerIndex::GetSeriesStatus(const std::string& seriesUuid) | |
601 { | |
80 | 602 SQLite::Statement s1(db_, SQLITE_FROM_HERE, "SELECT expectedNumberOfInstances FROM Series WHERE uuid=?"); |
603 s1.BindString(0, seriesUuid); | |
82 | 604 if (!s1.Step() || s1.ColumnIsNull(0)) |
80 | 605 { |
606 return SeriesStatus_Unknown; | |
607 } | |
608 | |
609 int numberOfInstances = s1.ColumnInt(0); | |
610 if (numberOfInstances < 0) | |
0 | 611 { |
612 return SeriesStatus_Unknown; | |
613 } | |
614 | |
80 | 615 std::set<int> instances; |
616 SQLite::Statement s2(db_, SQLITE_FROM_HERE, "SELECT indexInSeries FROM Instances WHERE parentSeries=?"); | |
617 s2.BindString(0, seriesUuid); | |
618 while (s2.Step()) | |
619 { | |
620 int index = s2.ColumnInt(0); | |
621 if (index <= 0 || index > numberOfInstances) | |
622 { | |
623 // Out-of-range instance index | |
624 return SeriesStatus_Inconsistent; | |
625 } | |
0 | 626 |
80 | 627 if (instances.find(index) != instances.end()) |
628 { | |
629 // Twice the same instance index | |
630 return SeriesStatus_Inconsistent; | |
631 } | |
632 | |
633 instances.insert(index); | |
634 } | |
635 | |
636 for (int i = 1; i <= numberOfInstances; i++) | |
637 { | |
638 if (instances.find(i) == instances.end()) | |
639 { | |
640 return SeriesStatus_Missing; | |
641 } | |
642 } | |
643 | |
644 return SeriesStatus_Complete; | |
0 | 645 } |
646 | |
647 | |
648 | |
649 bool ServerIndex::GetInstance(Json::Value& result, | |
650 const std::string& instanceUuid) | |
651 { | |
652 assert(result.type() == Json::objectValue); | |
653 boost::mutex::scoped_lock scoped_lock(mutex_); | |
654 | |
80 | 655 SQLite::Statement s(db_, SQLITE_FROM_HERE, "SELECT parentSeries, dicomInstance, fileSize, fileUuid, indexInSeries FROM Instances WHERE uuid=?"); |
0 | 656 s.BindString(0, instanceUuid); |
657 if (!s.Step()) | |
658 { | |
659 return false; | |
660 } | |
661 else | |
662 { | |
663 result["ID"] = instanceUuid; | |
664 result["ParentSeries"] = s.ColumnString(0); | |
665 result["FileSize"] = s.ColumnInt(2); // TODO switch to 64bit with JsonCpp 0.6? | |
666 result["FileUuid"] = s.ColumnString(3); | |
667 MainDicomTagsToJson(result, instanceUuid); | |
80 | 668 |
669 if (s.ColumnIsNull(4)) | |
670 { | |
82 | 671 result["IndexInSeries"] = Json::nullValue; |
80 | 672 } |
673 else | |
674 { | |
675 result["IndexInSeries"] = s.ColumnInt(4); | |
676 } | |
677 | |
0 | 678 return true; |
679 } | |
680 } | |
681 | |
682 | |
683 bool ServerIndex::GetSeries(Json::Value& result, | |
684 const std::string& seriesUuid) | |
685 { | |
686 assert(result.type() == Json::objectValue); | |
687 boost::mutex::scoped_lock scoped_lock(mutex_); | |
688 | |
80 | 689 SQLite::Statement s1(db_, SQLITE_FROM_HERE, "SELECT parentStudy, dicomSeries, expectedNumberOfInstances FROM Series WHERE uuid=?"); |
0 | 690 s1.BindString(0, seriesUuid); |
691 if (!s1.Step()) | |
692 { | |
693 return false; | |
694 } | |
695 | |
696 result["ID"] = seriesUuid; | |
697 result["ParentStudy"] = s1.ColumnString(0); | |
698 MainDicomTagsToJson(result, seriesUuid); | |
699 | |
700 Json::Value instances(Json::arrayValue); | |
701 SQLite::Statement s2(db_, SQLITE_FROM_HERE, "SELECT uuid FROM Instances WHERE parentSeries=?"); | |
702 s2.BindString(0, seriesUuid); | |
703 while (s2.Step()) | |
704 { | |
705 instances.append(s2.ColumnString(0)); | |
706 } | |
707 | |
708 result["Instances"] = instances; | |
709 | |
80 | 710 if (s1.ColumnIsNull(2)) |
711 { | |
82 | 712 result["ExpectedNumberOfInstances"] = Json::nullValue; |
80 | 713 } |
714 else | |
715 { | |
716 result["ExpectedNumberOfInstances"] = s1.ColumnInt(2); | |
717 } | |
718 | |
719 SeriesStatus status = GetSeriesStatus(seriesUuid); | |
720 | |
721 switch (status) | |
722 { | |
723 case SeriesStatus_Complete: | |
724 result["Status"] = "Complete"; | |
725 break; | |
726 | |
727 case SeriesStatus_Missing: | |
728 result["Status"] = "Missing"; | |
729 break; | |
730 | |
731 case SeriesStatus_Inconsistent: | |
732 result["Status"] = "Inconsistent"; | |
733 break; | |
734 | |
735 default: | |
736 case SeriesStatus_Unknown: | |
737 result["Status"] = "Unknown"; | |
738 break; | |
739 } | |
740 | |
0 | 741 return true; |
742 } | |
743 | |
744 | |
745 bool ServerIndex::GetStudy(Json::Value& result, | |
746 const std::string& studyUuid) | |
747 { | |
748 assert(result.type() == Json::objectValue); | |
749 boost::mutex::scoped_lock scoped_lock(mutex_); | |
750 | |
751 SQLite::Statement s1(db_, SQLITE_FROM_HERE, "SELECT parentPatient, dicomStudy FROM Studies WHERE uuid=?"); | |
752 s1.BindString(0, studyUuid); | |
753 if (!s1.Step()) | |
754 { | |
755 return false; | |
756 } | |
757 | |
758 result["ID"] = studyUuid; | |
759 result["ParentPatient"] = s1.ColumnString(0); | |
760 MainDicomTagsToJson(result, studyUuid); | |
761 | |
762 Json::Value series(Json::arrayValue); | |
763 SQLite::Statement s2(db_, SQLITE_FROM_HERE, "SELECT uuid FROM Series WHERE parentStudy=?"); | |
764 s2.BindString(0, studyUuid); | |
765 while (s2.Step()) | |
766 { | |
767 series.append(s2.ColumnString(0)); | |
768 } | |
769 | |
770 result["Series"] = series; | |
771 return true; | |
772 } | |
773 | |
774 | |
775 bool ServerIndex::GetPatient(Json::Value& result, | |
776 const std::string& patientUuid) | |
777 { | |
778 assert(result.type() == Json::objectValue); | |
779 boost::mutex::scoped_lock scoped_lock(mutex_); | |
780 | |
781 SQLite::Statement s1(db_, SQLITE_FROM_HERE, "SELECT dicomPatientId FROM Patients WHERE uuid=?"); | |
782 s1.BindString(0, patientUuid); | |
783 if (!s1.Step()) | |
784 { | |
785 return false; | |
786 } | |
787 | |
788 result["ID"] = patientUuid; | |
789 MainDicomTagsToJson(result, patientUuid); | |
790 | |
791 Json::Value studies(Json::arrayValue); | |
792 SQLite::Statement s2(db_, SQLITE_FROM_HERE, "SELECT uuid FROM Studies WHERE parentPatient=?"); | |
793 s2.BindString(0, patientUuid); | |
794 while (s2.Step()) | |
795 { | |
796 studies.append(s2.ColumnString(0)); | |
797 } | |
798 | |
799 result["Studies"] = studies; | |
800 return true; | |
801 } | |
802 | |
803 | |
804 bool ServerIndex::GetJsonFile(std::string& fileUuid, | |
805 const std::string& instanceUuid) | |
806 { | |
807 boost::mutex::scoped_lock scoped_lock(mutex_); | |
808 | |
809 SQLite::Statement s(db_, SQLITE_FROM_HERE, "SELECT jsonUuid FROM Instances WHERE uuid=?"); | |
810 s.BindString(0, instanceUuid); | |
811 if (s.Step()) | |
812 { | |
813 fileUuid = s.ColumnString(0); | |
814 return true; | |
815 } | |
816 else | |
817 { | |
818 return false; | |
819 } | |
820 } | |
821 | |
822 bool ServerIndex::GetDicomFile(std::string& fileUuid, | |
823 const std::string& instanceUuid) | |
824 { | |
825 boost::mutex::scoped_lock scoped_lock(mutex_); | |
826 | |
827 SQLite::Statement s(db_, SQLITE_FROM_HERE, "SELECT fileUuid FROM Instances WHERE uuid=?"); | |
828 s.BindString(0, instanceUuid); | |
829 if (s.Step()) | |
830 { | |
831 fileUuid = s.ColumnString(0); | |
832 return true; | |
833 } | |
834 else | |
835 { | |
836 return false; | |
837 } | |
838 } | |
839 | |
840 | |
841 void ServerIndex::GetAllUuids(Json::Value& target, | |
842 const std::string& tableName) | |
843 { | |
844 assert(target.type() == Json::arrayValue); | |
845 boost::mutex::scoped_lock scoped_lock(mutex_); | |
846 | |
847 std::string query = "SELECT uuid FROM " + tableName; | |
848 SQLite::Statement s(db_, query); | |
849 while (s.Step()) | |
850 { | |
851 target.append(s.ColumnString(0)); | |
852 } | |
853 } | |
854 | |
855 | |
856 bool ServerIndex::GetChanges(Json::Value& target, | |
857 int64_t since, | |
858 const std::string& filter, | |
859 unsigned int maxResults) | |
860 { | |
861 assert(target.type() == Json::objectValue); | |
862 boost::mutex::scoped_lock scoped_lock(mutex_); | |
863 | |
864 if (filter.size() != 0 && | |
865 filter != "instances" && | |
866 filter != "series" && | |
867 filter != "studies" && | |
868 filter != "patients") | |
869 { | |
870 return false; | |
871 } | |
872 | |
873 std::auto_ptr<SQLite::Statement> s; | |
874 if (filter.size() == 0) | |
875 { | |
876 s.reset(new SQLite::Statement(db_, SQLITE_FROM_HERE, "SELECT * FROM Changes WHERE seq>? " | |
877 "ORDER BY seq LIMIT ?")); | |
878 s->BindInt64(0, since); | |
879 s->BindInt(1, maxResults); | |
880 } | |
881 else | |
882 { | |
883 s.reset(new SQLite::Statement(db_, SQLITE_FROM_HERE, "SELECT * FROM Changes WHERE seq>? " | |
884 "AND basePath=? ORDER BY seq LIMIT ?")); | |
885 s->BindInt64(0, since); | |
886 s->BindString(1, filter); | |
887 s->BindInt(2, maxResults); | |
888 } | |
889 | |
890 int64_t lastSeq = 0; | |
891 Json::Value results(Json::arrayValue); | |
892 while (s->Step()) | |
893 { | |
894 int64_t seq = s->ColumnInt64(0); | |
895 std::string basePath = s->ColumnString(1); | |
896 std::string uuid = s->ColumnString(2); | |
897 | |
898 if (filter.size() == 0 || | |
899 filter == basePath) | |
900 { | |
901 Json::Value change(Json::objectValue); | |
902 change["Seq"] = static_cast<int>(seq); // TODO JsonCpp in 64bit | |
903 change["BasePath"] = basePath; | |
904 change["ID"] = uuid; | |
905 results.append(change); | |
906 } | |
907 | |
908 if (seq > lastSeq) | |
909 { | |
910 lastSeq = seq; | |
911 } | |
912 } | |
913 | |
914 target["Results"] = results; | |
915 target["LastSeq"] = static_cast<int>(lastSeq); // TODO JsonCpp in 64bit | |
916 | |
917 return true; | |
918 } | |
82 | 919 |
0 | 920 } |