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