Mercurial > hg > orthanc-wsi
annotate Framework/Jpeg2000Writer.cpp @ 112:999a17ecf9b6
fix
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Sat, 26 Aug 2017 12:23:09 +0200 |
parents | ff0ef01c332c |
children | a51dee6a1515 |
rev | line source |
---|---|
0 | 1 /** |
2 * Orthanc - A Lightweight, RESTful DICOM Store | |
3 * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics | |
4 * Department, University Hospital of Liege, Belgium | |
98
ff0ef01c332c
shared copyright with osimis
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
59
diff
changeset
|
5 * Copyright (C) 2017 Osimis, Belgium |
0 | 6 * |
7 * This program is free software: you can redistribute it and/or | |
8 * modify it under the terms of the GNU Affero General Public License | |
9 * as published by the Free Software Foundation, either version 3 of | |
10 * the License, or (at your option) any later version. | |
11 * | |
12 * This program is distributed in the hope that it will be useful, but | |
13 * WITHOUT ANY WARRANTY; without even the implied warranty of | |
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
15 * Affero General Public License for more details. | |
16 * | |
17 * You should have received a copy of the GNU Affero General Public License | |
18 * along with this program. If not, see <http://www.gnu.org/licenses/>. | |
19 **/ | |
20 | |
21 | |
16
7a88c614be04
preparing for precompiled headers
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
22 #include "PrecompiledHeadersWSI.h" |
0 | 23 #include "Jpeg2000Writer.h" |
24 | |
59
7a3853d51c45
Move "Framework/Orthanc/" as "Resources/Orthanc/"
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
16
diff
changeset
|
25 #include "../Resources/Orthanc/Core/ChunkedBuffer.h" |
7a3853d51c45
Move "Framework/Orthanc/" as "Resources/Orthanc/"
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
16
diff
changeset
|
26 #include "../Resources/Orthanc/Core/OrthancException.h" |
0 | 27 |
28 #include <openjpeg.h> | |
29 #include <string.h> | |
30 #include <vector> | |
31 | |
32 #if ORTHANC_OPENJPEG_MAJOR_VERSION == 1 | |
33 # define OPJ_CLRSPC_GRAY CLRSPC_GRAY | |
34 # define OPJ_CLRSPC_SRGB CLRSPC_SRGB | |
35 # define OPJ_CODEC_J2K CODEC_J2K | |
36 typedef opj_cinfo_t opj_codec_t; | |
37 typedef opj_cio_t opj_stream_t; | |
38 #elif ORTHANC_OPENJPEG_MAJOR_VERSION == 2 | |
39 #else | |
40 #error Unsupported version of OpenJpeg | |
41 #endif | |
42 | |
43 namespace OrthancWSI | |
44 { | |
45 namespace | |
46 { | |
47 class OpenJpegImage : public boost::noncopyable | |
48 { | |
49 private: | |
50 std::vector<opj_image_cmptparm_t> components_; | |
51 COLOR_SPACE colorspace_; | |
52 opj_image_t* image_; | |
53 | |
54 void SetupComponents(unsigned int width, | |
55 unsigned int height, | |
56 Orthanc::PixelFormat format) | |
57 { | |
58 switch (format) | |
59 { | |
60 case Orthanc::PixelFormat_Grayscale8: | |
61 colorspace_ = OPJ_CLRSPC_GRAY; | |
62 components_.resize(1); | |
63 break; | |
64 | |
65 case Orthanc::PixelFormat_RGB24: | |
66 colorspace_ = OPJ_CLRSPC_SRGB; | |
67 components_.resize(3); | |
68 break; | |
69 | |
70 default: | |
71 throw Orthanc::OrthancException(Orthanc::ErrorCode_ParameterOutOfRange); | |
72 } | |
73 | |
74 for (size_t i = 0; i < components_.size(); i++) | |
75 { | |
76 memset(&components_[i], 0, sizeof(opj_image_cmptparm_t)); | |
77 components_[i].dx = 1; | |
78 components_[i].dy = 1; | |
79 components_[i].x0 = 0; | |
80 components_[i].y0 = 0; | |
81 components_[i].w = width; | |
82 components_[i].h = height; | |
83 components_[i].prec = 8; | |
84 components_[i].bpp = 8; | |
85 components_[i].sgnd = 0; | |
86 } | |
87 } | |
88 | |
89 void CopyRGB24(unsigned int width, | |
90 unsigned int height, | |
91 unsigned int pitch, | |
92 const void* buffer) | |
93 { | |
94 int32_t* r = image_->comps[0].data; | |
95 int32_t* g = image_->comps[1].data; | |
96 int32_t* b = image_->comps[2].data; | |
97 | |
98 for (unsigned int y = 0; y < height; y++) | |
99 { | |
100 const uint8_t *p = reinterpret_cast<const uint8_t*>(buffer) + y * pitch; | |
101 | |
102 for (unsigned int x = 0; x < width; x++) | |
103 { | |
104 *r = p[0]; | |
105 *g = p[1]; | |
106 *b = p[2]; | |
107 p += 3; | |
108 r++; | |
109 g++; | |
110 b++; | |
111 } | |
112 } | |
113 } | |
114 | |
115 void CopyGrayscale8(unsigned int width, | |
116 unsigned int height, | |
117 unsigned int pitch, | |
118 const void* buffer) | |
119 { | |
120 int32_t* q = image_->comps[0].data; | |
121 | |
122 for (unsigned int y = 0; y < height; y++) | |
123 { | |
124 const uint8_t *p = reinterpret_cast<const uint8_t*>(buffer) + y * pitch; | |
125 | |
126 for (unsigned int x = 0; x < width; x++) | |
127 { | |
128 *q = *p; | |
129 p++; | |
130 q++; | |
131 } | |
132 } | |
133 } | |
134 | |
135 public: | |
136 OpenJpegImage(unsigned int width, | |
137 unsigned int height, | |
138 unsigned int pitch, | |
139 Orthanc::PixelFormat format, | |
140 const void* buffer) : image_(NULL) | |
141 { | |
142 SetupComponents(width, height, format); | |
143 | |
144 image_ = opj_image_create(components_.size(), &components_[0], colorspace_); | |
145 if (!image_) | |
146 { | |
147 throw Orthanc::OrthancException(Orthanc::ErrorCode_InternalError); | |
148 } | |
149 | |
150 image_->x0 = 0; | |
151 image_->y0 = 0; | |
152 image_->x1 = width; | |
153 image_->y1 = height; | |
154 | |
155 switch (format) | |
156 { | |
157 case Orthanc::PixelFormat_Grayscale8: | |
158 CopyGrayscale8(width, height, pitch, buffer); | |
159 break; | |
160 | |
161 case Orthanc::PixelFormat_RGB24: | |
162 CopyRGB24(width, height, pitch, buffer); | |
163 break; | |
164 | |
165 default: | |
166 throw Orthanc::OrthancException(Orthanc::ErrorCode_ParameterOutOfRange); | |
167 } | |
168 } | |
169 | |
170 | |
171 ~OpenJpegImage() | |
172 { | |
173 if (image_) | |
174 { | |
175 opj_image_destroy(image_); | |
176 image_ = NULL; | |
177 } | |
178 } | |
179 | |
180 | |
181 opj_image_t* GetObject() | |
182 { | |
183 return image_; | |
184 } | |
185 }; | |
186 | |
187 | |
188 class OpenJpegEncoder : public boost::noncopyable | |
189 { | |
190 private: | |
191 opj_codec_t* cinfo_; | |
192 | |
193 public: | |
194 OpenJpegEncoder(opj_cparameters_t& parameters, | |
195 OpenJpegImage& image) : cinfo_(NULL) | |
196 { | |
197 cinfo_ = opj_create_compress(OPJ_CODEC_J2K); | |
198 if (!cinfo_) | |
199 { | |
200 throw Orthanc::OrthancException(Orthanc::ErrorCode_InternalError); | |
201 } | |
202 | |
203 opj_setup_encoder(cinfo_, ¶meters, image.GetObject()); | |
204 } | |
205 | |
206 ~OpenJpegEncoder() | |
207 { | |
208 if (cinfo_ != NULL) | |
209 { | |
210 #if ORTHANC_OPENJPEG_MAJOR_VERSION == 1 | |
211 opj_destroy_compress(cinfo_); | |
212 #else | |
213 opj_destroy_codec(cinfo_); | |
214 #endif | |
215 cinfo_ = NULL; | |
216 } | |
217 } | |
218 | |
219 opj_codec_t* GetObject() | |
220 { | |
221 return cinfo_; | |
222 } | |
223 }; | |
224 | |
225 | |
226 class OpenJpegOutput : public boost::noncopyable | |
227 { | |
228 private: | |
229 opj_stream_t* cio_; | |
230 | |
231 #if ORTHANC_OPENJPEG_MAJOR_VERSION == 2 | |
232 Orthanc::ChunkedBuffer buffer_; | |
233 | |
234 static void Free(void *userData) | |
235 { | |
236 } | |
237 | |
238 static OPJ_SIZE_T Write(void *buffer, | |
239 OPJ_SIZE_T size, | |
240 void *userData) | |
241 { | |
242 OpenJpegOutput* that = reinterpret_cast<OpenJpegOutput*>(userData); | |
243 that->buffer_.AddChunk(reinterpret_cast<const char*>(buffer), size); | |
244 return size; | |
245 } | |
246 #endif | |
247 | |
248 public: | |
249 OpenJpegOutput(OpenJpegEncoder& encoder) : cio_(NULL) | |
250 { | |
251 #if ORTHANC_OPENJPEG_MAJOR_VERSION == 1 | |
252 cio_ = opj_cio_open(reinterpret_cast<opj_common_ptr>(encoder.GetObject()), NULL, 0); | |
253 #else | |
254 cio_ = opj_stream_default_create(0 /* output stream */); | |
255 if (!cio_) | |
256 { | |
257 throw Orthanc::OrthancException(Orthanc::ErrorCode_InternalError); | |
258 } | |
259 | |
260 opj_stream_set_user_data(cio_, this, Free); | |
261 opj_stream_set_write_function(cio_, Write); | |
262 #endif | |
263 } | |
264 | |
265 ~OpenJpegOutput() | |
266 { | |
267 if (cio_) | |
268 { | |
269 #if ORTHANC_OPENJPEG_MAJOR_VERSION == 1 | |
270 opj_cio_close(cio_); | |
271 #else | |
272 opj_stream_destroy(cio_); | |
273 #endif | |
274 cio_ = NULL; | |
275 } | |
276 } | |
277 | |
278 opj_stream_t* GetObject() | |
279 { | |
280 return cio_; | |
281 } | |
282 | |
283 void Flatten(std::string& target) | |
284 { | |
285 #if ORTHANC_OPENJPEG_MAJOR_VERSION == 1 | |
286 target.assign(reinterpret_cast<const char*>(cio_->buffer), cio_tell(cio_)); | |
287 #else | |
288 buffer_.Flatten(target); | |
289 #endif | |
290 } | |
291 }; | |
292 } | |
293 | |
294 | |
295 static void SetupParameters(opj_cparameters_t& parameters, | |
296 Orthanc::PixelFormat format, | |
297 bool isLossless) | |
298 { | |
299 opj_set_default_encoder_parameters(¶meters); | |
300 parameters.cp_disto_alloc = 1; | |
301 | |
302 if (isLossless) | |
303 { | |
304 parameters.tcp_numlayers = 1; | |
305 parameters.tcp_rates[0] = 0; | |
306 } | |
307 else | |
308 { | |
309 parameters.tcp_numlayers = 5; | |
310 parameters.tcp_rates[0] = 1920; | |
311 parameters.tcp_rates[1] = 480; | |
312 parameters.tcp_rates[2] = 120; | |
313 parameters.tcp_rates[3] = 30; | |
314 parameters.tcp_rates[4] = 10; | |
315 parameters.irreversible = 1; | |
316 | |
317 if (format == Orthanc::PixelFormat_RGB24 || | |
318 format == Orthanc::PixelFormat_RGBA32) | |
319 { | |
320 // This must be set to 1 if the number of color channels is >= 3 | |
321 parameters.tcp_mct = 1; | |
322 } | |
323 } | |
324 | |
325 parameters.cp_comment = const_cast<char*>(""); | |
326 } | |
327 | |
328 | |
329 void Jpeg2000Writer::WriteToMemoryInternal(std::string& compressed, | |
330 unsigned int width, | |
331 unsigned int height, | |
332 unsigned int pitch, | |
333 Orthanc::PixelFormat format, | |
334 const void* buffer) | |
335 { | |
336 if (format != Orthanc::PixelFormat_Grayscale8 && | |
337 format != Orthanc::PixelFormat_RGB24 && | |
338 format != Orthanc::PixelFormat_RGBA32) | |
339 { | |
340 throw Orthanc::OrthancException(Orthanc::ErrorCode_ParameterOutOfRange); | |
341 } | |
342 | |
343 opj_cparameters_t parameters; | |
344 SetupParameters(parameters, format, isLossless_); | |
345 | |
346 OpenJpegImage image(width, height, pitch, format, buffer); | |
347 OpenJpegEncoder encoder(parameters, image); | |
348 OpenJpegOutput output(encoder); | |
349 | |
350 #if ORTHANC_OPENJPEG_MAJOR_VERSION == 1 | |
351 if (!opj_encode(encoder.GetObject(), output.GetObject(), image.GetObject(), parameters.index)) | |
352 { | |
353 throw Orthanc::OrthancException(Orthanc::ErrorCode_InternalError); | |
354 } | |
355 #else | |
356 if (!opj_start_compress(encoder.GetObject(), image.GetObject(), output.GetObject()) || | |
357 !opj_encode(encoder.GetObject(), output.GetObject()) || | |
358 !opj_end_compress(encoder.GetObject(), output.GetObject())) | |
359 { | |
360 throw Orthanc::OrthancException(Orthanc::ErrorCode_InternalError); | |
361 } | |
362 #endif | |
363 | |
364 output.Flatten(compressed); | |
365 } | |
366 } |