78
|
1 /**
|
|
2 * Orthanc - A Lightweight, RESTful DICOM Store
|
122
|
3 * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics
|
78
|
4 * Department, University Hospital of Liege, 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 * 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.
|
|
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 "../PrecompiledHeaders.h"
|
|
34 #include "ImageProcessing.h"
|
|
35
|
|
36 #include "../OrthancException.h"
|
|
37
|
|
38 #include <boost/math/special_functions/round.hpp>
|
|
39
|
|
40 #include <cassert>
|
|
41 #include <string.h>
|
|
42 #include <limits>
|
|
43 #include <stdint.h>
|
|
44
|
|
45 namespace Orthanc
|
|
46 {
|
|
47 template <typename TargetType, typename SourceType>
|
|
48 static void ConvertInternal(ImageAccessor& target,
|
|
49 const ImageAccessor& source)
|
|
50 {
|
|
51 const TargetType minValue = std::numeric_limits<TargetType>::min();
|
|
52 const TargetType maxValue = std::numeric_limits<TargetType>::max();
|
|
53
|
|
54 for (unsigned int y = 0; y < source.GetHeight(); y++)
|
|
55 {
|
|
56 TargetType* t = reinterpret_cast<TargetType*>(target.GetRow(y));
|
|
57 const SourceType* s = reinterpret_cast<const SourceType*>(source.GetConstRow(y));
|
|
58
|
|
59 for (unsigned int x = 0; x < source.GetWidth(); x++, t++, s++)
|
|
60 {
|
|
61 if (static_cast<int32_t>(*s) < static_cast<int32_t>(minValue))
|
|
62 {
|
|
63 *t = minValue;
|
|
64 }
|
|
65 else if (static_cast<int32_t>(*s) > static_cast<int32_t>(maxValue))
|
|
66 {
|
|
67 *t = maxValue;
|
|
68 }
|
|
69 else
|
|
70 {
|
|
71 *t = static_cast<TargetType>(*s);
|
|
72 }
|
|
73 }
|
|
74 }
|
|
75 }
|
|
76
|
|
77
|
|
78 template <typename TargetType>
|
|
79 static void ConvertColorToGrayscale(ImageAccessor& target,
|
|
80 const ImageAccessor& source)
|
|
81 {
|
|
82 assert(source.GetFormat() == PixelFormat_RGB24);
|
|
83
|
|
84 const TargetType minValue = std::numeric_limits<TargetType>::min();
|
|
85 const TargetType maxValue = std::numeric_limits<TargetType>::max();
|
|
86
|
|
87 for (unsigned int y = 0; y < source.GetHeight(); y++)
|
|
88 {
|
|
89 TargetType* t = reinterpret_cast<TargetType*>(target.GetRow(y));
|
|
90 const uint8_t* s = reinterpret_cast<const uint8_t*>(source.GetConstRow(y));
|
|
91
|
|
92 for (unsigned int x = 0; x < source.GetWidth(); x++, t++, s += 3)
|
|
93 {
|
|
94 // Y = 0.2126 R + 0.7152 G + 0.0722 B
|
|
95 int32_t v = (2126 * static_cast<int32_t>(s[0]) +
|
|
96 7152 * static_cast<int32_t>(s[1]) +
|
|
97 0722 * static_cast<int32_t>(s[2])) / 1000;
|
|
98
|
|
99 if (static_cast<int32_t>(v) < static_cast<int32_t>(minValue))
|
|
100 {
|
|
101 *t = minValue;
|
|
102 }
|
|
103 else if (static_cast<int32_t>(v) > static_cast<int32_t>(maxValue))
|
|
104 {
|
|
105 *t = maxValue;
|
|
106 }
|
|
107 else
|
|
108 {
|
|
109 *t = static_cast<TargetType>(v);
|
|
110 }
|
|
111 }
|
|
112 }
|
|
113 }
|
|
114
|
|
115
|
|
116 template <typename PixelType>
|
|
117 static void SetInternal(ImageAccessor& image,
|
|
118 int64_t constant)
|
|
119 {
|
|
120 for (unsigned int y = 0; y < image.GetHeight(); y++)
|
|
121 {
|
|
122 PixelType* p = reinterpret_cast<PixelType*>(image.GetRow(y));
|
|
123
|
|
124 for (unsigned int x = 0; x < image.GetWidth(); x++, p++)
|
|
125 {
|
|
126 *p = static_cast<PixelType>(constant);
|
|
127 }
|
|
128 }
|
|
129 }
|
|
130
|
|
131
|
|
132 template <typename PixelType>
|
|
133 static void GetMinMaxValueInternal(PixelType& minValue,
|
|
134 PixelType& maxValue,
|
|
135 const ImageAccessor& source)
|
|
136 {
|
|
137 // Deal with the special case of empty image
|
|
138 if (source.GetWidth() == 0 ||
|
|
139 source.GetHeight() == 0)
|
|
140 {
|
|
141 minValue = 0;
|
|
142 maxValue = 0;
|
|
143 return;
|
|
144 }
|
|
145
|
|
146 minValue = std::numeric_limits<PixelType>::max();
|
|
147 maxValue = std::numeric_limits<PixelType>::min();
|
|
148
|
|
149 for (unsigned int y = 0; y < source.GetHeight(); y++)
|
|
150 {
|
|
151 const PixelType* p = reinterpret_cast<const PixelType*>(source.GetConstRow(y));
|
|
152
|
|
153 for (unsigned int x = 0; x < source.GetWidth(); x++, p++)
|
|
154 {
|
|
155 if (*p < minValue)
|
|
156 {
|
|
157 minValue = *p;
|
|
158 }
|
|
159
|
|
160 if (*p > maxValue)
|
|
161 {
|
|
162 maxValue = *p;
|
|
163 }
|
|
164 }
|
|
165 }
|
|
166 }
|
|
167
|
|
168
|
|
169
|
|
170 template <typename PixelType>
|
|
171 static void AddConstantInternal(ImageAccessor& image,
|
|
172 int64_t constant)
|
|
173 {
|
|
174 if (constant == 0)
|
|
175 {
|
|
176 return;
|
|
177 }
|
|
178
|
|
179 const int64_t minValue = std::numeric_limits<PixelType>::min();
|
|
180 const int64_t maxValue = std::numeric_limits<PixelType>::max();
|
|
181
|
|
182 for (unsigned int y = 0; y < image.GetHeight(); y++)
|
|
183 {
|
|
184 PixelType* p = reinterpret_cast<PixelType*>(image.GetRow(y));
|
|
185
|
|
186 for (unsigned int x = 0; x < image.GetWidth(); x++, p++)
|
|
187 {
|
|
188 int64_t v = static_cast<int64_t>(*p) + constant;
|
|
189
|
|
190 if (v > maxValue)
|
|
191 {
|
|
192 *p = std::numeric_limits<PixelType>::max();
|
|
193 }
|
|
194 else if (v < minValue)
|
|
195 {
|
|
196 *p = std::numeric_limits<PixelType>::min();
|
|
197 }
|
|
198 else
|
|
199 {
|
|
200 *p = static_cast<PixelType>(v);
|
|
201 }
|
|
202 }
|
|
203 }
|
|
204 }
|
|
205
|
|
206
|
|
207
|
|
208 template <typename PixelType>
|
|
209 void MultiplyConstantInternal(ImageAccessor& image,
|
|
210 float factor)
|
|
211 {
|
|
212 if (std::abs(factor - 1.0f) <= std::numeric_limits<float>::epsilon())
|
|
213 {
|
|
214 return;
|
|
215 }
|
|
216
|
|
217 const int64_t minValue = std::numeric_limits<PixelType>::min();
|
|
218 const int64_t maxValue = std::numeric_limits<PixelType>::max();
|
|
219
|
|
220 for (unsigned int y = 0; y < image.GetHeight(); y++)
|
|
221 {
|
|
222 PixelType* p = reinterpret_cast<PixelType*>(image.GetRow(y));
|
|
223
|
|
224 for (unsigned int x = 0; x < image.GetWidth(); x++, p++)
|
|
225 {
|
|
226 int64_t v = boost::math::llround(static_cast<float>(*p) * factor);
|
|
227
|
|
228 if (v > maxValue)
|
|
229 {
|
|
230 *p = std::numeric_limits<PixelType>::max();
|
|
231 }
|
|
232 else if (v < minValue)
|
|
233 {
|
|
234 *p = std::numeric_limits<PixelType>::min();
|
|
235 }
|
|
236 else
|
|
237 {
|
|
238 *p = static_cast<PixelType>(v);
|
|
239 }
|
|
240 }
|
|
241 }
|
|
242 }
|
|
243
|
|
244
|
|
245 template <typename PixelType>
|
|
246 void ShiftScaleInternal(ImageAccessor& image,
|
|
247 float offset,
|
|
248 float scaling)
|
|
249 {
|
|
250 const float minValue = static_cast<float>(std::numeric_limits<PixelType>::min());
|
|
251 const float maxValue = static_cast<float>(std::numeric_limits<PixelType>::max());
|
|
252
|
|
253 for (unsigned int y = 0; y < image.GetHeight(); y++)
|
|
254 {
|
|
255 PixelType* p = reinterpret_cast<PixelType*>(image.GetRow(y));
|
|
256
|
|
257 for (unsigned int x = 0; x < image.GetWidth(); x++, p++)
|
|
258 {
|
|
259 float v = (static_cast<float>(*p) + offset) * scaling;
|
|
260
|
|
261 if (v > maxValue)
|
|
262 {
|
|
263 *p = std::numeric_limits<PixelType>::max();
|
|
264 }
|
|
265 else if (v < minValue)
|
|
266 {
|
|
267 *p = std::numeric_limits<PixelType>::min();
|
|
268 }
|
|
269 else
|
|
270 {
|
|
271 *p = static_cast<PixelType>(boost::math::iround(v));
|
|
272 }
|
|
273 }
|
|
274 }
|
|
275 }
|
|
276
|
|
277
|
|
278 void ImageProcessing::Copy(ImageAccessor& target,
|
|
279 const ImageAccessor& source)
|
|
280 {
|
|
281 if (target.GetWidth() != source.GetWidth() ||
|
|
282 target.GetHeight() != source.GetHeight())
|
|
283 {
|
|
284 throw OrthancException(ErrorCode_IncompatibleImageSize);
|
|
285 }
|
|
286
|
|
287 if (target.GetFormat() != source.GetFormat())
|
|
288 {
|
|
289 throw OrthancException(ErrorCode_IncompatibleImageFormat);
|
|
290 }
|
|
291
|
|
292 unsigned int lineSize = GetBytesPerPixel(source.GetFormat()) * source.GetWidth();
|
|
293
|
|
294 assert(source.GetPitch() >= lineSize && target.GetPitch() >= lineSize);
|
|
295
|
|
296 for (unsigned int y = 0; y < source.GetHeight(); y++)
|
|
297 {
|
|
298 memcpy(target.GetRow(y), source.GetConstRow(y), lineSize);
|
|
299 }
|
|
300 }
|
|
301
|
|
302
|
|
303 void ImageProcessing::Convert(ImageAccessor& target,
|
|
304 const ImageAccessor& source)
|
|
305 {
|
|
306 if (target.GetWidth() != source.GetWidth() ||
|
|
307 target.GetHeight() != source.GetHeight())
|
|
308 {
|
|
309 throw OrthancException(ErrorCode_IncompatibleImageSize);
|
|
310 }
|
|
311
|
|
312 if (source.GetFormat() == target.GetFormat())
|
|
313 {
|
|
314 Copy(target, source);
|
|
315 return;
|
|
316 }
|
|
317
|
|
318 if (target.GetFormat() == PixelFormat_Grayscale16 &&
|
|
319 source.GetFormat() == PixelFormat_Grayscale8)
|
|
320 {
|
|
321 ConvertInternal<uint16_t, uint8_t>(target, source);
|
|
322 return;
|
|
323 }
|
|
324
|
|
325 if (target.GetFormat() == PixelFormat_SignedGrayscale16 &&
|
|
326 source.GetFormat() == PixelFormat_Grayscale8)
|
|
327 {
|
|
328 ConvertInternal<int16_t, uint8_t>(target, source);
|
|
329 return;
|
|
330 }
|
|
331
|
|
332 if (target.GetFormat() == PixelFormat_Grayscale8 &&
|
|
333 source.GetFormat() == PixelFormat_Grayscale16)
|
|
334 {
|
|
335 ConvertInternal<uint8_t, uint16_t>(target, source);
|
|
336 return;
|
|
337 }
|
|
338
|
|
339 if (target.GetFormat() == PixelFormat_SignedGrayscale16 &&
|
|
340 source.GetFormat() == PixelFormat_Grayscale16)
|
|
341 {
|
|
342 ConvertInternal<int16_t, uint16_t>(target, source);
|
|
343 return;
|
|
344 }
|
|
345
|
|
346 if (target.GetFormat() == PixelFormat_Grayscale8 &&
|
|
347 source.GetFormat() == PixelFormat_SignedGrayscale16)
|
|
348 {
|
|
349 ConvertInternal<uint8_t, int16_t>(target, source);
|
|
350 return;
|
|
351 }
|
|
352
|
|
353 if (target.GetFormat() == PixelFormat_Grayscale16 &&
|
|
354 source.GetFormat() == PixelFormat_SignedGrayscale16)
|
|
355 {
|
|
356 ConvertInternal<uint16_t, int16_t>(target, source);
|
|
357 return;
|
|
358 }
|
|
359
|
|
360 if (target.GetFormat() == PixelFormat_Grayscale8 &&
|
|
361 source.GetFormat() == PixelFormat_RGB24)
|
|
362 {
|
|
363 ConvertColorToGrayscale<uint8_t>(target, source);
|
|
364 return;
|
|
365 }
|
|
366
|
|
367 if (target.GetFormat() == PixelFormat_Grayscale16 &&
|
|
368 source.GetFormat() == PixelFormat_RGB24)
|
|
369 {
|
|
370 ConvertColorToGrayscale<uint16_t>(target, source);
|
|
371 return;
|
|
372 }
|
|
373
|
|
374 if (target.GetFormat() == PixelFormat_SignedGrayscale16 &&
|
|
375 source.GetFormat() == PixelFormat_RGB24)
|
|
376 {
|
|
377 ConvertColorToGrayscale<int16_t>(target, source);
|
|
378 return;
|
|
379 }
|
|
380
|
|
381 if (target.GetFormat() == PixelFormat_Grayscale8 &&
|
|
382 source.GetFormat() == PixelFormat_RGBA32)
|
|
383 {
|
|
384 for (unsigned int y = 0; y < source.GetHeight(); y++)
|
|
385 {
|
|
386 const uint8_t* p = reinterpret_cast<const uint8_t*>(source.GetConstRow(y));
|
|
387 uint8_t* q = reinterpret_cast<uint8_t*>(target.GetRow(y));
|
|
388 for (unsigned int x = 0; x < source.GetWidth(); x++, q++)
|
|
389 {
|
|
390 *q = static_cast<uint8_t>((2126 * static_cast<uint32_t>(p[0]) +
|
|
391 7152 * static_cast<uint32_t>(p[1]) +
|
|
392 0722 * static_cast<uint32_t>(p[2])) / 10000);
|
|
393 p += 4;
|
|
394 }
|
|
395 }
|
|
396
|
|
397 return;
|
|
398 }
|
|
399
|
|
400 if (target.GetFormat() == PixelFormat_RGB24 &&
|
|
401 source.GetFormat() == PixelFormat_RGBA32)
|
|
402 {
|
|
403 for (unsigned int y = 0; y < source.GetHeight(); y++)
|
|
404 {
|
|
405 const uint8_t* p = reinterpret_cast<const uint8_t*>(source.GetConstRow(y));
|
|
406 uint8_t* q = reinterpret_cast<uint8_t*>(target.GetRow(y));
|
|
407 for (unsigned int x = 0; x < source.GetWidth(); x++)
|
|
408 {
|
|
409 q[0] = p[0];
|
|
410 q[1] = p[1];
|
|
411 q[2] = p[2];
|
|
412 p += 4;
|
|
413 q += 3;
|
|
414 }
|
|
415 }
|
|
416
|
|
417 return;
|
|
418 }
|
|
419
|
|
420 if (target.GetFormat() == PixelFormat_RGBA32 &&
|
|
421 source.GetFormat() == PixelFormat_RGB24)
|
|
422 {
|
|
423 for (unsigned int y = 0; y < source.GetHeight(); y++)
|
|
424 {
|
|
425 const uint8_t* p = reinterpret_cast<const uint8_t*>(source.GetConstRow(y));
|
|
426 uint8_t* q = reinterpret_cast<uint8_t*>(target.GetRow(y));
|
|
427 for (unsigned int x = 0; x < source.GetWidth(); x++)
|
|
428 {
|
|
429 q[0] = p[0];
|
|
430 q[1] = p[1];
|
|
431 q[2] = p[2];
|
|
432 q[3] = 255; // Set the alpha channel to full opacity
|
|
433 p += 3;
|
|
434 q += 4;
|
|
435 }
|
|
436 }
|
|
437
|
|
438 return;
|
|
439 }
|
|
440
|
|
441 throw OrthancException(ErrorCode_NotImplemented);
|
|
442 }
|
|
443
|
|
444
|
|
445
|
|
446 void ImageProcessing::Set(ImageAccessor& image,
|
|
447 int64_t value)
|
|
448 {
|
|
449 switch (image.GetFormat())
|
|
450 {
|
|
451 case PixelFormat_Grayscale8:
|
|
452 SetInternal<uint8_t>(image, value);
|
|
453 return;
|
|
454
|
|
455 case PixelFormat_Grayscale16:
|
|
456 SetInternal<uint16_t>(image, value);
|
|
457 return;
|
|
458
|
|
459 case PixelFormat_SignedGrayscale16:
|
|
460 SetInternal<int16_t>(image, value);
|
|
461 return;
|
|
462
|
|
463 default:
|
|
464 throw OrthancException(ErrorCode_NotImplemented);
|
|
465 }
|
|
466 }
|
|
467
|
|
468
|
|
469 void ImageProcessing::ShiftRight(ImageAccessor& image,
|
|
470 unsigned int shift)
|
|
471 {
|
|
472 if (image.GetWidth() == 0 ||
|
|
473 image.GetHeight() == 0 ||
|
|
474 shift == 0)
|
|
475 {
|
|
476 // Nothing to do
|
|
477 return;
|
|
478 }
|
|
479
|
|
480 throw OrthancException(ErrorCode_NotImplemented);
|
|
481 }
|
|
482
|
|
483
|
|
484 void ImageProcessing::GetMinMaxValue(int64_t& minValue,
|
|
485 int64_t& maxValue,
|
|
486 const ImageAccessor& image)
|
|
487 {
|
|
488 switch (image.GetFormat())
|
|
489 {
|
|
490 case PixelFormat_Grayscale8:
|
|
491 {
|
|
492 uint8_t a, b;
|
|
493 GetMinMaxValueInternal<uint8_t>(a, b, image);
|
|
494 minValue = a;
|
|
495 maxValue = b;
|
|
496 break;
|
|
497 }
|
|
498
|
|
499 case PixelFormat_Grayscale16:
|
|
500 {
|
|
501 uint16_t a, b;
|
|
502 GetMinMaxValueInternal<uint16_t>(a, b, image);
|
|
503 minValue = a;
|
|
504 maxValue = b;
|
|
505 break;
|
|
506 }
|
|
507
|
|
508 case PixelFormat_SignedGrayscale16:
|
|
509 {
|
|
510 int16_t a, b;
|
|
511 GetMinMaxValueInternal<int16_t>(a, b, image);
|
|
512 minValue = a;
|
|
513 maxValue = b;
|
|
514 break;
|
|
515 }
|
|
516
|
|
517 default:
|
|
518 throw OrthancException(ErrorCode_NotImplemented);
|
|
519 }
|
|
520 }
|
|
521
|
|
522
|
|
523
|
|
524 void ImageProcessing::AddConstant(ImageAccessor& image,
|
|
525 int64_t value)
|
|
526 {
|
|
527 switch (image.GetFormat())
|
|
528 {
|
|
529 case PixelFormat_Grayscale8:
|
|
530 AddConstantInternal<uint8_t>(image, value);
|
|
531 return;
|
|
532
|
|
533 case PixelFormat_Grayscale16:
|
|
534 AddConstantInternal<uint16_t>(image, value);
|
|
535 return;
|
|
536
|
|
537 case PixelFormat_SignedGrayscale16:
|
|
538 AddConstantInternal<int16_t>(image, value);
|
|
539 return;
|
|
540
|
|
541 default:
|
|
542 throw OrthancException(ErrorCode_NotImplemented);
|
|
543 }
|
|
544 }
|
|
545
|
|
546
|
|
547 void ImageProcessing::MultiplyConstant(ImageAccessor& image,
|
|
548 float factor)
|
|
549 {
|
|
550 switch (image.GetFormat())
|
|
551 {
|
|
552 case PixelFormat_Grayscale8:
|
|
553 MultiplyConstantInternal<uint8_t>(image, factor);
|
|
554 return;
|
|
555
|
|
556 case PixelFormat_Grayscale16:
|
|
557 MultiplyConstantInternal<uint16_t>(image, factor);
|
|
558 return;
|
|
559
|
|
560 case PixelFormat_SignedGrayscale16:
|
|
561 MultiplyConstantInternal<int16_t>(image, factor);
|
|
562 return;
|
|
563
|
|
564 default:
|
|
565 throw OrthancException(ErrorCode_NotImplemented);
|
|
566 }
|
|
567 }
|
|
568
|
|
569
|
|
570 void ImageProcessing::ShiftScale(ImageAccessor& image,
|
|
571 float offset,
|
|
572 float scaling)
|
|
573 {
|
|
574 switch (image.GetFormat())
|
|
575 {
|
|
576 case PixelFormat_Grayscale8:
|
|
577 ShiftScaleInternal<uint8_t>(image, offset, scaling);
|
|
578 return;
|
|
579
|
|
580 case PixelFormat_Grayscale16:
|
|
581 ShiftScaleInternal<uint16_t>(image, offset, scaling);
|
|
582 return;
|
|
583
|
|
584 case PixelFormat_SignedGrayscale16:
|
|
585 ShiftScaleInternal<int16_t>(image, offset, scaling);
|
|
586 return;
|
|
587
|
|
588 default:
|
|
589 throw OrthancException(ErrorCode_NotImplemented);
|
|
590 }
|
|
591 }
|
|
592 }
|