Mercurial > hg > orthanc
annotate Core/FileFormats/PngReader.cpp @ 797:37adac56017a
ImageAccessor abstraction
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Tue, 06 May 2014 12:47:26 +0200 |
parents | 2d0a347e8cfc |
children | e1d27ee2114a |
rev | line source |
---|---|
455 | 1 /** |
2 * Orthanc - A Lightweight, RESTful DICOM Store | |
689 | 3 * Copyright (C) 2012-2014 Medical Physics Department, CHU of Liege, |
455 | 4 * 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 "PngReader.h" | |
34 | |
35 #include "../OrthancException.h" | |
36 #include "../Toolbox.h" | |
37 | |
38 #include <png.h> | |
492
f3d4193c571a
switch to jsoncpp-0.6.0-rc2
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
455
diff
changeset
|
39 #include <string.h> // For memcpy() |
455 | 40 |
41 namespace Orthanc | |
42 { | |
43 namespace | |
44 { | |
45 struct FileRabi | |
46 { | |
47 FILE* fp_; | |
48 | |
49 FileRabi(const char* filename) | |
50 { | |
51 fp_ = fopen(filename, "rb"); | |
52 if (!fp_) | |
53 { | |
54 throw OrthancException(ErrorCode_InexistentFile); | |
55 } | |
56 } | |
57 | |
58 ~FileRabi() | |
59 { | |
60 if (fp_) | |
61 fclose(fp_); | |
62 } | |
63 }; | |
64 } | |
65 | |
66 | |
67 struct PngReader::PngRabi | |
68 { | |
69 png_structp png_; | |
70 png_infop info_; | |
71 png_infop endInfo_; | |
72 | |
73 void Destruct() | |
74 { | |
75 if (png_) | |
76 { | |
77 png_destroy_read_struct(&png_, &info_, &endInfo_); | |
78 | |
79 png_ = NULL; | |
80 info_ = NULL; | |
81 endInfo_ = NULL; | |
82 } | |
83 } | |
84 | |
85 PngRabi() | |
86 { | |
87 png_ = NULL; | |
88 info_ = NULL; | |
89 endInfo_ = NULL; | |
90 | |
91 png_ = png_create_read_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL); | |
92 if (!png_) | |
93 { | |
94 throw OrthancException(ErrorCode_NotEnoughMemory); | |
95 } | |
96 | |
97 info_ = png_create_info_struct(png_); | |
98 if (!info_) | |
99 { | |
100 png_destroy_read_struct(&png_, NULL, NULL); | |
101 throw OrthancException(ErrorCode_NotEnoughMemory); | |
102 } | |
103 | |
104 endInfo_ = png_create_info_struct(png_); | |
105 if (!info_) | |
106 { | |
107 png_destroy_read_struct(&png_, &info_, NULL); | |
108 throw OrthancException(ErrorCode_NotEnoughMemory); | |
109 } | |
110 } | |
111 | |
112 ~PngRabi() | |
113 { | |
114 Destruct(); | |
115 } | |
116 | |
117 static void MemoryCallback(png_structp png_ptr, | |
118 png_bytep data, | |
119 png_size_t size); | |
120 }; | |
121 | |
122 | |
123 void PngReader::CheckHeader(const void* header) | |
124 { | |
125 int is_png = !png_sig_cmp((png_bytep) header, 0, 8); | |
126 if (!is_png) | |
127 { | |
128 throw OrthancException(ErrorCode_BadFileFormat); | |
129 } | |
130 } | |
131 | |
132 PngReader::PngReader() | |
133 { | |
134 } | |
135 | |
136 void PngReader::Read(PngRabi& rabi) | |
137 { | |
138 png_set_sig_bytes(rabi.png_, 8); | |
139 | |
140 png_read_info(rabi.png_, rabi.info_); | |
141 | |
142 png_uint_32 width, height; | |
143 int bit_depth, color_type, interlace_type; | |
144 int compression_type, filter_method; | |
145 // get size and bit-depth of the PNG-image | |
146 png_get_IHDR(rabi.png_, rabi.info_, | |
147 &width, &height, | |
148 &bit_depth, &color_type, &interlace_type, | |
149 &compression_type, &filter_method); | |
150 | |
797
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
151 PixelFormat format; |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
152 unsigned int pitch; |
455 | 153 |
154 if (color_type == PNG_COLOR_TYPE_GRAY && bit_depth == 8) | |
155 { | |
797
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
156 format = PixelFormat_Grayscale8; |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
157 pitch = width; |
455 | 158 } |
159 else if (color_type == PNG_COLOR_TYPE_GRAY && bit_depth == 16) | |
160 { | |
797
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
161 format = PixelFormat_Grayscale16; |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
162 pitch = 2 * width; |
455 | 163 |
164 if (Toolbox::DetectEndianness() == Endianness_Little) | |
165 { | |
166 png_set_swap(rabi.png_); | |
167 } | |
168 } | |
169 else if (color_type == PNG_COLOR_TYPE_RGB && bit_depth == 8) | |
170 { | |
797
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
171 format = PixelFormat_RGB24; |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
172 pitch = 3 * width; |
455 | 173 } |
174 else | |
175 { | |
176 throw OrthancException(ErrorCode_NotImplemented); | |
177 } | |
178 | |
797
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
179 data_.resize(height * pitch); |
455 | 180 |
797
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
181 if (height == 0 || width == 0) |
455 | 182 { |
183 // Empty image, we are done | |
797
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
184 AssignEmpty(format); |
455 | 185 return; |
186 } | |
797
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
187 |
455 | 188 png_read_update_info(rabi.png_, rabi.info_); |
189 | |
797
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
190 std::vector<png_bytep> rows(height); |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
191 for (size_t i = 0; i < height; i++) |
455 | 192 { |
797
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
193 rows[i] = &data_[0] + i * pitch; |
455 | 194 } |
195 | |
196 png_read_image(rabi.png_, &rows[0]); | |
797
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
197 |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
198 AssignReadOnly(format, width, height, pitch, &data_[0]); |
455 | 199 } |
200 | |
201 void PngReader::ReadFromFile(const char* filename) | |
202 { | |
203 FileRabi f(filename); | |
204 | |
205 char header[8]; | |
206 if (fread(header, 1, 8, f.fp_) != 8) | |
207 { | |
208 throw OrthancException(ErrorCode_BadFileFormat); | |
209 } | |
210 | |
211 CheckHeader(header); | |
212 | |
213 PngRabi rabi; | |
214 | |
215 if (setjmp(png_jmpbuf(rabi.png_))) | |
216 { | |
217 rabi.Destruct(); | |
218 throw OrthancException(ErrorCode_BadFileFormat); | |
219 } | |
220 | |
221 png_init_io(rabi.png_, f.fp_); | |
222 | |
223 Read(rabi); | |
224 } | |
225 | |
226 | |
797
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
227 void* ImageAccessor::GetBuffer() |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
228 { |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
229 if (readOnly_) |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
230 { |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
231 throw OrthancException(ErrorCode_ReadOnly); |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
232 } |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
233 |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
234 return buffer_; |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
235 } |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
236 |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
237 |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
238 const void* ImageAccessor::GetConstRow(unsigned int y) const |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
239 { |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
240 if (buffer_ != NULL) |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
241 { |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
242 return reinterpret_cast<const uint8_t*>(buffer_) + y * pitch_; |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
243 } |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
244 else |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
245 { |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
246 return NULL; |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
247 } |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
248 } |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
249 |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
250 |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
251 void* ImageAccessor::GetRow(unsigned int y) |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
252 { |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
253 if (readOnly_) |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
254 { |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
255 throw OrthancException(ErrorCode_ReadOnly); |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
256 } |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
257 |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
258 if (buffer_ != NULL) |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
259 { |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
260 return reinterpret_cast<uint8_t*>(buffer_) + y * pitch_; |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
261 } |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
262 else |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
263 { |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
264 return NULL; |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
265 } |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
266 } |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
267 |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
268 |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
269 void ImageAccessor::AssignEmpty(PixelFormat format) |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
270 { |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
271 readOnly_ = false; |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
272 format_ = format; |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
273 width_ = 0; |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
274 height_ = 0; |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
275 pitch_ = 0; |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
276 buffer_ = NULL; |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
277 } |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
278 |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
279 |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
280 void ImageAccessor::AssignReadOnly(PixelFormat format, |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
281 unsigned int width, |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
282 unsigned int height, |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
283 unsigned int pitch, |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
284 const void *buffer) |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
285 { |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
286 readOnly_ = true; |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
287 format_ = format; |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
288 width_ = width; |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
289 height_ = height; |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
290 pitch_ = pitch; |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
291 buffer_ = const_cast<void*>(buffer); |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
292 } |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
293 |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
294 |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
295 void ImageAccessor::AssignWritable(PixelFormat format, |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
296 unsigned int width, |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
297 unsigned int height, |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
298 unsigned int pitch, |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
299 void *buffer) |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
300 { |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
301 readOnly_ = false; |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
302 format_ = format; |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
303 width_ = width; |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
304 height_ = height; |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
305 pitch_ = pitch; |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
306 buffer_ = buffer; |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
307 } |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
308 |
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
309 |
455 | 310 |
311 namespace | |
312 { | |
313 struct MemoryBuffer | |
314 { | |
315 const uint8_t* buffer_; | |
316 size_t size_; | |
317 size_t pos_; | |
318 bool ok_; | |
319 }; | |
320 } | |
321 | |
322 | |
323 void PngReader::PngRabi::MemoryCallback(png_structp png_ptr, | |
324 png_bytep outBytes, | |
325 png_size_t byteCountToRead) | |
326 { | |
656 | 327 MemoryBuffer* from = reinterpret_cast<MemoryBuffer*>(png_get_io_ptr(png_ptr)); |
455 | 328 |
329 if (!from->ok_) | |
330 { | |
331 return; | |
332 } | |
333 | |
334 if (from->pos_ + byteCountToRead > from->size_) | |
335 { | |
336 from->ok_ = false; | |
337 return; | |
338 } | |
339 | |
340 memcpy(outBytes, from->buffer_ + from->pos_, byteCountToRead); | |
341 | |
342 from->pos_ += byteCountToRead; | |
343 } | |
344 | |
345 | |
346 void PngReader::ReadFromMemory(const void* buffer, | |
347 size_t size) | |
348 { | |
349 if (size < 8) | |
350 { | |
351 throw OrthancException(ErrorCode_BadFileFormat); | |
352 } | |
353 | |
354 CheckHeader(buffer); | |
355 | |
356 PngRabi rabi; | |
357 | |
358 if (setjmp(png_jmpbuf(rabi.png_))) | |
359 { | |
360 rabi.Destruct(); | |
361 throw OrthancException(ErrorCode_BadFileFormat); | |
362 } | |
363 | |
364 MemoryBuffer tmp; | |
365 tmp.buffer_ = reinterpret_cast<const uint8_t*>(buffer) + 8; // We skip the header | |
366 tmp.size_ = size - 8; | |
367 tmp.pos_ = 0; | |
368 tmp.ok_ = true; | |
369 | |
370 png_set_read_fn(rabi.png_, &tmp, PngRabi::MemoryCallback); | |
371 | |
372 Read(rabi); | |
373 | |
374 if (!tmp.ok_) | |
375 { | |
376 throw OrthancException(ErrorCode_BadFileFormat); | |
377 } | |
378 } | |
379 | |
380 void PngReader::ReadFromMemory(const std::string& buffer) | |
381 { | |
382 if (buffer.size() != 0) | |
797
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
383 { |
455 | 384 ReadFromMemory(&buffer[0], buffer.size()); |
797
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
385 } |
455 | 386 else |
797
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
387 { |
455 | 388 ReadFromMemory(NULL, 0); |
797
37adac56017a
ImageAccessor abstraction
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
389 } |
455 | 390 } |
391 } |