Mercurial > hg > orthanc
comparison OrthancServer/FromDcmtkBridge.cpp @ 57:4bc019d2f969 orthanc-renaming
renaming
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Sun, 16 Sep 2012 09:22:48 +0200 |
parents | PalanthirServer/FromDcmtkBridge.cpp@293038baf8f1 |
children | a70bb32802ae |
comparison
equal
deleted
inserted
replaced
56:088c4f23e2c8 | 57:4bc019d2f969 |
---|---|
1 /** | |
2 * Palanthir - A Lightweight, RESTful DICOM Store | |
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. | |
10 * | |
11 * This program is distributed in the hope that it will be useful, but | |
12 * WITHOUT ANY WARRANTY; without even the implied warranty of | |
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
14 * General Public License for more details. | |
15 * | |
16 * You should have received a copy of the GNU General Public License | |
17 * along with this program. If not, see <http://www.gnu.org/licenses/>. | |
18 **/ | |
19 | |
20 | |
21 #include "FromDcmtkBridge.h" | |
22 | |
23 #include "ToDcmtkBridge.h" | |
24 #include "DicomIntegerPixelAccessor.h" | |
25 #include "../Core/PalanthirException.h" | |
26 #include "../Core/PngWriter.h" | |
27 #include "../Core/DicomFormat/DicomString.h" | |
28 #include "../Core/DicomFormat/DicomNullValue.h" | |
29 | |
30 #include <boost/locale.hpp> | |
31 #include <boost/lexical_cast.hpp> | |
32 | |
33 #include <dcmtk/dcmdata/dcdicent.h> | |
34 #include <dcmtk/dcmdata/dcdict.h> | |
35 #include <dcmtk/dcmdata/dcelem.h> | |
36 #include <dcmtk/dcmdata/dcfilefo.h> | |
37 #include <dcmtk/dcmdata/dcistrmb.h> | |
38 #include <dcmtk/dcmdata/dcsequen.h> | |
39 #include <dcmtk/dcmdata/dcvrfd.h> | |
40 #include <dcmtk/dcmdata/dcvrfl.h> | |
41 #include <dcmtk/dcmdata/dcvrsl.h> | |
42 #include <dcmtk/dcmdata/dcvrss.h> | |
43 #include <dcmtk/dcmdata/dcvrul.h> | |
44 #include <dcmtk/dcmdata/dcvrus.h> | |
45 | |
46 #include <boost/math/special_functions/round.hpp> | |
47 | |
48 namespace Palanthir | |
49 { | |
50 void FromDcmtkBridge::Convert(DicomMap& target, DcmDataset& dataset) | |
51 { | |
52 target.Clear(); | |
53 for (unsigned long i = 0; i < dataset.card(); i++) | |
54 { | |
55 DcmElement* element = dataset.getElement(i); | |
56 if (element && element->isLeaf()) | |
57 { | |
58 target.SetValue(element->getTag().getGTag(), | |
59 element->getTag().getETag(), | |
60 ConvertLeafElement(*element)); | |
61 } | |
62 } | |
63 } | |
64 | |
65 | |
66 DicomTag FromDcmtkBridge::GetTag(const DcmElement& element) | |
67 { | |
68 return DicomTag(element.getGTag(), element.getETag()); | |
69 } | |
70 | |
71 | |
72 DicomValue* FromDcmtkBridge::ConvertLeafElement(DcmElement& element) | |
73 { | |
74 if (!element.isLeaf()) | |
75 { | |
76 throw PalanthirException("Only applicable to leaf elements"); | |
77 } | |
78 | |
79 if (element.isaString()) | |
80 { | |
81 char *c; | |
82 if (element.getString(c).good() && | |
83 c != NULL) | |
84 { | |
85 std::string s(c); | |
86 std::string utf8; | |
87 try | |
88 { | |
89 utf8 = boost::locale::conv::to_utf<char>(s, "ISO-8859-1"); // TODO Parameter? | |
90 } | |
91 catch (std::runtime_error&) | |
92 { | |
93 // Bad input string or bad encoding | |
94 utf8 = s; | |
95 } | |
96 | |
97 return new DicomString(utf8); | |
98 } | |
99 else | |
100 { | |
101 return new DicomNullValue; | |
102 } | |
103 } | |
104 | |
105 try | |
106 { | |
107 // http://support.dcmtk.org/docs/dcvr_8h-source.html | |
108 switch (element.getVR()) | |
109 { | |
110 | |
111 /** | |
112 * TODO. | |
113 **/ | |
114 | |
115 case EVR_DS: // decimal string | |
116 case EVR_IS: // integer string | |
117 case EVR_OB: // other byte | |
118 case EVR_OF: // other float | |
119 case EVR_OW: // other word | |
120 case EVR_AS: // age string | |
121 case EVR_AT: // attribute tag | |
122 case EVR_DA: // date string | |
123 case EVR_DT: // date time string | |
124 case EVR_TM: // time string | |
125 case EVR_UN: // unknown value representation | |
126 return new DicomNullValue(); | |
127 | |
128 | |
129 /** | |
130 * String types, should never happen at this point because of | |
131 * "element.isaString()". | |
132 **/ | |
133 | |
134 case EVR_AE: // application entity title | |
135 case EVR_CS: // code string | |
136 case EVR_SH: // short string | |
137 case EVR_LO: // long string | |
138 case EVR_ST: // short text | |
139 case EVR_LT: // long text | |
140 case EVR_UT: // unlimited text | |
141 case EVR_PN: // person name | |
142 case EVR_UI: // unique identifier | |
143 return new DicomNullValue(); | |
144 | |
145 | |
146 /** | |
147 * Numerical types | |
148 **/ | |
149 | |
150 case EVR_SL: // signed long | |
151 { | |
152 Sint32 f; | |
153 if (dynamic_cast<DcmSignedLong&>(element).getSint32(f).good()) | |
154 { | |
155 return new DicomString(boost::lexical_cast<std::string>(f)); | |
156 } | |
157 else | |
158 { | |
159 return new DicomNullValue(); | |
160 } | |
161 } | |
162 | |
163 case EVR_SS: // signed short | |
164 { | |
165 Sint16 f; | |
166 if (dynamic_cast<DcmSignedShort&>(element).getSint16(f).good()) | |
167 { | |
168 return new DicomString(boost::lexical_cast<std::string>(f)); | |
169 } | |
170 else | |
171 { | |
172 return new DicomNullValue(); | |
173 } | |
174 } | |
175 | |
176 case EVR_UL: // unsigned long | |
177 { | |
178 Uint32 f; | |
179 if (dynamic_cast<DcmUnsignedLong&>(element).getUint32(f).good()) | |
180 { | |
181 return new DicomString(boost::lexical_cast<std::string>(f)); | |
182 } | |
183 else | |
184 { | |
185 return new DicomNullValue(); | |
186 } | |
187 } | |
188 | |
189 case EVR_US: // unsigned short | |
190 { | |
191 Uint16 f; | |
192 if (dynamic_cast<DcmUnsignedShort&>(element).getUint16(f).good()) | |
193 { | |
194 return new DicomString(boost::lexical_cast<std::string>(f)); | |
195 } | |
196 else | |
197 { | |
198 return new DicomNullValue(); | |
199 } | |
200 } | |
201 | |
202 case EVR_FL: // float single-precision | |
203 { | |
204 Float32 f; | |
205 if (dynamic_cast<DcmFloatingPointSingle&>(element).getFloat32(f).good()) | |
206 { | |
207 return new DicomString(boost::lexical_cast<std::string>(f)); | |
208 } | |
209 else | |
210 { | |
211 return new DicomNullValue(); | |
212 } | |
213 } | |
214 | |
215 case EVR_FD: // float double-precision | |
216 { | |
217 Float64 f; | |
218 if (dynamic_cast<DcmFloatingPointDouble&>(element).getFloat64(f).good()) | |
219 { | |
220 return new DicomString(boost::lexical_cast<std::string>(f)); | |
221 } | |
222 else | |
223 { | |
224 return new DicomNullValue(); | |
225 } | |
226 } | |
227 | |
228 | |
229 /** | |
230 * Sequence types, should never occur at this point because of | |
231 * "element.isLeaf()". | |
232 **/ | |
233 | |
234 case EVR_SQ: // sequence of items | |
235 return new DicomNullValue; | |
236 | |
237 | |
238 /** | |
239 * Internal to DCMTK. | |
240 **/ | |
241 | |
242 case EVR_ox: // OB or OW depending on context | |
243 case EVR_xs: // SS or US depending on context | |
244 case EVR_lt: // US, SS or OW depending on context, used for LUT Data (thus the name) | |
245 case EVR_na: // na="not applicable", for data which has no VR | |
246 case EVR_up: // up="unsigned pointer", used internally for DICOMDIR suppor | |
247 case EVR_item: // used internally for items | |
248 case EVR_metainfo: // used internally for meta info datasets | |
249 case EVR_dataset: // used internally for datasets | |
250 case EVR_fileFormat: // used internally for DICOM files | |
251 case EVR_dicomDir: // used internally for DICOMDIR objects | |
252 case EVR_dirRecord: // used internally for DICOMDIR records | |
253 case EVR_pixelSQ: // used internally for pixel sequences in a compressed image | |
254 case EVR_pixelItem: // used internally for pixel items in a compressed image | |
255 case EVR_UNKNOWN: // used internally for elements with unknown VR (encoded with 4-byte length field in explicit VR) | |
256 case EVR_PixelData: // used internally for uncompressed pixeld data | |
257 case EVR_OverlayData: // used internally for overlay data | |
258 case EVR_UNKNOWN2B: // used internally for elements with unknown VR with 2-byte length field in explicit VR | |
259 return new DicomNullValue; | |
260 | |
261 | |
262 /** | |
263 * Default case. | |
264 **/ | |
265 | |
266 default: | |
267 return new DicomNullValue; | |
268 } | |
269 } | |
270 catch (boost::bad_lexical_cast) | |
271 { | |
272 return new DicomNullValue; | |
273 } | |
274 } | |
275 | |
276 | |
277 static void StoreElement(Json::Value& target, | |
278 DcmElement& element, | |
279 unsigned int maxStringLength); | |
280 | |
281 static void StoreItem(Json::Value& target, | |
282 DcmItem& item, | |
283 unsigned int maxStringLength) | |
284 { | |
285 target = Json::Value(Json::objectValue); | |
286 | |
287 for (unsigned long i = 0; i < item.card(); i++) | |
288 { | |
289 DcmElement* element = item.getElement(i); | |
290 StoreElement(target, *element, maxStringLength); | |
291 } | |
292 } | |
293 | |
294 | |
295 static void StoreElement(Json::Value& target, | |
296 DcmElement& element, | |
297 unsigned int maxStringLength) | |
298 { | |
299 assert(target.type() == Json::objectValue); | |
300 | |
301 DicomTag tag(FromDcmtkBridge::GetTag(element)); | |
302 const std::string tagName = FromDcmtkBridge::GetName(tag); | |
303 const std::string formattedTag = tag.Format(); | |
304 | |
305 if (element.isLeaf()) | |
306 { | |
307 Json::Value value(Json::objectValue); | |
308 value["Name"] = tagName; | |
309 | |
310 std::auto_ptr<DicomValue> v(FromDcmtkBridge::ConvertLeafElement(element)); | |
311 if (v->IsNull()) | |
312 { | |
313 value["Type"] = "Null"; | |
314 value["Value"] = Json::nullValue; | |
315 } | |
316 else | |
317 { | |
318 std::string s = v->AsString(); | |
319 if (maxStringLength == 0 || | |
320 s.size() <= maxStringLength) | |
321 { | |
322 value["Type"] = "String"; | |
323 value["Value"] = s; | |
324 } | |
325 else | |
326 { | |
327 value["Type"] = "TooLong"; | |
328 value["Value"] = Json::nullValue; | |
329 } | |
330 } | |
331 | |
332 target[formattedTag] = value; | |
333 } | |
334 else | |
335 { | |
336 Json::Value children(Json::arrayValue); | |
337 | |
338 // "All subclasses of DcmElement except for DcmSequenceOfItems | |
339 // are leaf nodes, while DcmSequenceOfItems, DcmItem, DcmDataset | |
340 // etc. are not." The following cast is thus OK. | |
341 DcmSequenceOfItems& sequence = dynamic_cast<DcmSequenceOfItems&>(element); | |
342 | |
343 for (unsigned long i = 0; i < sequence.card(); i++) | |
344 { | |
345 DcmItem* child = sequence.getItem(i); | |
346 Json::Value& v = children.append(Json::objectValue); | |
347 StoreItem(v, *child, maxStringLength); | |
348 } | |
349 | |
350 target[formattedTag]["Name"] = tagName; | |
351 target[formattedTag]["Type"] = "Sequence"; | |
352 target[formattedTag]["Value"] = children; | |
353 } | |
354 } | |
355 | |
356 | |
357 void FromDcmtkBridge::ToJson(Json::Value& root, | |
358 DcmDataset& dataset, | |
359 unsigned int maxStringLength) | |
360 { | |
361 StoreItem(root, dataset, maxStringLength); | |
362 } | |
363 | |
364 | |
365 | |
366 void FromDcmtkBridge::ToJson(Json::Value& target, | |
367 const std::string& path, | |
368 unsigned int maxStringLength) | |
369 { | |
370 DcmFileFormat dicom; | |
371 if (!dicom.loadFile(path.c_str()).good()) | |
372 { | |
373 throw PalanthirException(ErrorCode_BadFileFormat); | |
374 } | |
375 else | |
376 { | |
377 FromDcmtkBridge::ToJson(target, *dicom.getDataset(), maxStringLength); | |
378 } | |
379 } | |
380 | |
381 | |
382 static void ExtractPngImagePreview(std::string& result, | |
383 DicomIntegerPixelAccessor& accessor) | |
384 { | |
385 PngWriter w; | |
386 | |
387 int32_t min, max; | |
388 accessor.GetExtremeValues(min, max); | |
389 | |
390 std::vector<uint8_t> image(accessor.GetWidth() * accessor.GetHeight(), 0); | |
391 if (min != max) | |
392 { | |
393 uint8_t* pixel = &image[0]; | |
394 for (unsigned int y = 0; y < accessor.GetHeight(); y++) | |
395 { | |
396 for (unsigned int x = 0; x < accessor.GetWidth(); x++, pixel++) | |
397 { | |
398 int32_t v = accessor.GetValue(x, y); | |
399 *pixel = static_cast<uint8_t>( | |
400 boost::math::lround(static_cast<float>(v - min) / | |
401 static_cast<float>(max - min) * 255.0f)); | |
402 } | |
403 } | |
404 } | |
405 | |
406 w.WriteToMemory(result, accessor.GetWidth(), accessor.GetHeight(), | |
407 accessor.GetWidth(), PixelFormat_Grayscale8, &image[0]); | |
408 } | |
409 | |
410 | |
411 template <typename T> | |
412 static void ExtractPngImageTruncate(std::string& result, | |
413 DicomIntegerPixelAccessor& accessor, | |
414 PixelFormat format) | |
415 { | |
416 PngWriter w; | |
417 | |
418 std::vector<T> image(accessor.GetWidth() * accessor.GetHeight(), 0); | |
419 T* pixel = &image[0]; | |
420 for (unsigned int y = 0; y < accessor.GetHeight(); y++) | |
421 { | |
422 for (unsigned int x = 0; x < accessor.GetWidth(); x++, pixel++) | |
423 { | |
424 int32_t v = accessor.GetValue(x, y); | |
425 if (v < std::numeric_limits<T>::min()) | |
426 *pixel = std::numeric_limits<T>::min(); | |
427 else if (v > std::numeric_limits<T>::max()) | |
428 *pixel = std::numeric_limits<T>::max(); | |
429 else | |
430 *pixel = static_cast<T>(v); | |
431 } | |
432 } | |
433 | |
434 w.WriteToMemory(result, accessor.GetWidth(), accessor.GetHeight(), | |
435 accessor.GetWidth() * sizeof(T), format, &image[0]); | |
436 } | |
437 | |
438 | |
439 void FromDcmtkBridge::ExtractPngImage(std::string& result, | |
440 DcmDataset& dataset, | |
441 unsigned int frame, | |
442 ImageExtractionMode mode) | |
443 { | |
444 // See also: http://support.dcmtk.org/wiki/dcmtk/howto/accessing-compressed-data | |
445 | |
446 std::auto_ptr<DicomIntegerPixelAccessor> accessor; | |
447 | |
448 DicomMap m; | |
449 FromDcmtkBridge::Convert(m, dataset); | |
450 | |
451 DcmElement* e; | |
452 if (dataset.findAndGetElement(ToDcmtkBridge::Convert(DicomTag::PIXEL_DATA), e).good() && | |
453 e != NULL) | |
454 { | |
455 Uint8* pixData = NULL; | |
456 if (e->getUint8Array(pixData) == EC_Normal) | |
457 { | |
458 accessor.reset(new DicomIntegerPixelAccessor(m, pixData, e->getLength())); | |
459 accessor->SetCurrentFrame(frame); | |
460 } | |
461 } | |
462 | |
463 PixelFormat format; | |
464 switch (mode) | |
465 { | |
466 case ImageExtractionMode_Preview: | |
467 case ImageExtractionMode_UInt8: | |
468 format = PixelFormat_Grayscale8; | |
469 break; | |
470 | |
471 case ImageExtractionMode_UInt16: | |
472 format = PixelFormat_Grayscale16; | |
473 break; | |
474 | |
475 default: | |
476 throw PalanthirException(ErrorCode_NotImplemented); | |
477 } | |
478 | |
479 if (accessor.get() == NULL || | |
480 accessor->GetWidth() == 0 || | |
481 accessor->GetHeight() == 0) | |
482 { | |
483 PngWriter w; | |
484 w.WriteToMemory(result, 0, 0, 0, format, NULL); | |
485 } | |
486 else | |
487 { | |
488 switch (mode) | |
489 { | |
490 case ImageExtractionMode_Preview: | |
491 ExtractPngImagePreview(result, *accessor); | |
492 break; | |
493 | |
494 case ImageExtractionMode_UInt8: | |
495 ExtractPngImageTruncate<uint8_t>(result, *accessor, format); | |
496 break; | |
497 | |
498 case ImageExtractionMode_UInt16: | |
499 ExtractPngImageTruncate<uint16_t>(result, *accessor, format); | |
500 break; | |
501 | |
502 default: | |
503 throw PalanthirException(ErrorCode_NotImplemented); | |
504 } | |
505 } | |
506 } | |
507 | |
508 | |
509 void FromDcmtkBridge::ExtractPngImage(std::string& result, | |
510 const std::string& dicomContent, | |
511 unsigned int frame, | |
512 ImageExtractionMode mode) | |
513 { | |
514 DcmInputBufferStream is; | |
515 if (dicomContent.size() > 0) | |
516 { | |
517 is.setBuffer(&dicomContent[0], dicomContent.size()); | |
518 } | |
519 is.setEos(); | |
520 | |
521 DcmFileFormat dicom; | |
522 if (dicom.read(is).good()) | |
523 { | |
524 ExtractPngImage(result, *dicom.getDataset(), frame, mode); | |
525 } | |
526 else | |
527 { | |
528 throw PalanthirException(ErrorCode_BadFileFormat); | |
529 } | |
530 } | |
531 | |
532 | |
533 | |
534 std::string FromDcmtkBridge::GetName(const DicomTag& t) | |
535 { | |
536 DcmTagKey tag(t.GetGroup(), t.GetElement()); | |
537 const DcmDataDictionary& dict = dcmDataDict.rdlock(); | |
538 const DcmDictEntry* entry = dict.findEntry(tag, NULL); | |
539 | |
540 std::string s("Unknown"); | |
541 if (entry != NULL) | |
542 { | |
543 s = std::string(entry->getTagName()); | |
544 } | |
545 | |
546 dcmDataDict.unlock(); | |
547 return s; | |
548 } | |
549 | |
550 | |
551 DicomTag FromDcmtkBridge::FindTag(const char* name) | |
552 { | |
553 const DcmDataDictionary& dict = dcmDataDict.rdlock(); | |
554 const DcmDictEntry* entry = dict.findEntry(name); | |
555 | |
556 if (entry == NULL) | |
557 { | |
558 dcmDataDict.unlock(); | |
559 throw PalanthirException("Unknown DICOM tag"); | |
560 } | |
561 else | |
562 { | |
563 DcmTagKey key = entry->getKey(); | |
564 DicomTag tag(key.getGroup(), key.getElement()); | |
565 dcmDataDict.unlock(); | |
566 return tag; | |
567 } | |
568 } | |
569 | |
570 | |
571 void FromDcmtkBridge::Print(FILE* fp, const DicomMap& m) | |
572 { | |
573 for (DicomMap::Map::const_iterator | |
574 it = m.map_.begin(); it != m.map_.end(); it++) | |
575 { | |
576 DicomTag t = it->first; | |
577 std::string s = it->second->AsString(); | |
578 printf("0x%04x 0x%04x (%s) [%s]\n", t.GetGroup(), t.GetElement(), GetName(t).c_str(), s.c_str()); | |
579 } | |
580 } | |
581 | |
582 | |
583 void FromDcmtkBridge::ToJson(Json::Value& result, | |
584 const DicomMap& values) | |
585 { | |
586 if (result.type() != Json::objectValue) | |
587 { | |
588 throw PalanthirException(ErrorCode_BadParameterType); | |
589 } | |
590 | |
591 result.clear(); | |
592 | |
593 for (DicomMap::Map::const_iterator | |
594 it = values.map_.begin(); it != values.map_.end(); it++) | |
595 { | |
596 result[GetName(it->first)] = it->second->AsString(); | |
597 } | |
598 } | |
599 } |