Mercurial > hg > orthanc-wsi
annotate Framework/Jpeg2000Writer.cpp @ 161:dd736430e3a4 OrthancWSI-0.6
OrthancWSI-0.6
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Sat, 26 Jan 2019 10:48:48 +0100 |
parents | 6b8ccfc02051 |
children | e3cbf890b588 |
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 | |
157 | 5 * Copyright (C) 2017-2019 Osimis S.A., 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 | |
140
a0f9a3df1110
resort to Orthanc framework
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
115
diff
changeset
|
25 #include <Core/ChunkedBuffer.h> |
a0f9a3df1110
resort to Orthanc framework
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
115
diff
changeset
|
26 #include <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: | |
145 | 249 explicit OpenJpegOutput(OpenJpegEncoder& encoder) : |
250 cio_(NULL) | |
0 | 251 { |
252 #if ORTHANC_OPENJPEG_MAJOR_VERSION == 1 | |
253 cio_ = opj_cio_open(reinterpret_cast<opj_common_ptr>(encoder.GetObject()), NULL, 0); | |
254 #else | |
255 cio_ = opj_stream_default_create(0 /* output stream */); | |
256 if (!cio_) | |
257 { | |
258 throw Orthanc::OrthancException(Orthanc::ErrorCode_InternalError); | |
259 } | |
260 | |
261 opj_stream_set_user_data(cio_, this, Free); | |
262 opj_stream_set_write_function(cio_, Write); | |
263 #endif | |
264 } | |
265 | |
266 ~OpenJpegOutput() | |
267 { | |
268 if (cio_) | |
269 { | |
270 #if ORTHANC_OPENJPEG_MAJOR_VERSION == 1 | |
271 opj_cio_close(cio_); | |
272 #else | |
273 opj_stream_destroy(cio_); | |
274 #endif | |
275 cio_ = NULL; | |
276 } | |
277 } | |
278 | |
279 opj_stream_t* GetObject() | |
280 { | |
281 return cio_; | |
282 } | |
283 | |
284 void Flatten(std::string& target) | |
285 { | |
286 #if ORTHANC_OPENJPEG_MAJOR_VERSION == 1 | |
287 target.assign(reinterpret_cast<const char*>(cio_->buffer), cio_tell(cio_)); | |
288 #else | |
289 buffer_.Flatten(target); | |
290 #endif | |
291 } | |
292 }; | |
293 } | |
294 | |
295 | |
296 static void SetupParameters(opj_cparameters_t& parameters, | |
297 Orthanc::PixelFormat format, | |
298 bool isLossless) | |
299 { | |
300 opj_set_default_encoder_parameters(¶meters); | |
301 parameters.cp_disto_alloc = 1; | |
302 | |
303 if (isLossless) | |
304 { | |
305 parameters.tcp_numlayers = 1; | |
306 parameters.tcp_rates[0] = 0; | |
307 } | |
308 else | |
309 { | |
310 parameters.tcp_numlayers = 5; | |
311 parameters.tcp_rates[0] = 1920; | |
312 parameters.tcp_rates[1] = 480; | |
313 parameters.tcp_rates[2] = 120; | |
314 parameters.tcp_rates[3] = 30; | |
315 parameters.tcp_rates[4] = 10; | |
316 parameters.irreversible = 1; | |
317 | |
318 if (format == Orthanc::PixelFormat_RGB24 || | |
319 format == Orthanc::PixelFormat_RGBA32) | |
320 { | |
321 // This must be set to 1 if the number of color channels is >= 3 | |
322 parameters.tcp_mct = 1; | |
323 } | |
324 } | |
325 | |
326 parameters.cp_comment = const_cast<char*>(""); | |
327 } | |
328 | |
329 | |
330 void Jpeg2000Writer::WriteToMemoryInternal(std::string& compressed, | |
331 unsigned int width, | |
332 unsigned int height, | |
333 unsigned int pitch, | |
334 Orthanc::PixelFormat format, | |
335 const void* buffer) | |
336 { | |
337 if (format != Orthanc::PixelFormat_Grayscale8 && | |
338 format != Orthanc::PixelFormat_RGB24 && | |
339 format != Orthanc::PixelFormat_RGBA32) | |
340 { | |
341 throw Orthanc::OrthancException(Orthanc::ErrorCode_ParameterOutOfRange); | |
342 } | |
343 | |
344 opj_cparameters_t parameters; | |
345 SetupParameters(parameters, format, isLossless_); | |
346 | |
347 OpenJpegImage image(width, height, pitch, format, buffer); | |
348 OpenJpegEncoder encoder(parameters, image); | |
349 OpenJpegOutput output(encoder); | |
350 | |
351 #if ORTHANC_OPENJPEG_MAJOR_VERSION == 1 | |
352 if (!opj_encode(encoder.GetObject(), output.GetObject(), image.GetObject(), parameters.index)) | |
353 { | |
354 throw Orthanc::OrthancException(Orthanc::ErrorCode_InternalError); | |
355 } | |
356 #else | |
357 if (!opj_start_compress(encoder.GetObject(), image.GetObject(), output.GetObject()) || | |
358 !opj_encode(encoder.GetObject(), output.GetObject()) || | |
359 !opj_end_compress(encoder.GetObject(), output.GetObject())) | |
360 { | |
361 throw Orthanc::OrthancException(Orthanc::ErrorCode_InternalError); | |
362 } | |
363 #endif | |
364 | |
365 output.Flatten(compressed); | |
366 } | |
367 } |