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