]>
Commit | Line | Data |
---|---|---|
257bcf28 | 1 | ///////////////////////////////////////////////////////////////////////////// |
38d4b1e4 | 2 | // Name: src/common/imagtiff.cpp |
8f493002 VS |
3 | // Purpose: wxImage TIFF handler |
4 | // Author: Robert Roebling | |
257bcf28 | 5 | // RCS-ID: $Id$ |
8f493002 | 6 | // Copyright: (c) Robert Roebling |
65571936 | 7 | // Licence: wxWindows licence |
257bcf28 RR |
8 | ///////////////////////////////////////////////////////////////////////////// |
9 | ||
426272a3 VZ |
10 | // ============================================================================ |
11 | // declarations | |
12 | // ============================================================================ | |
13 | ||
14 | // ---------------------------------------------------------------------------- | |
15 | // headers | |
16 | // ---------------------------------------------------------------------------- | |
17 | ||
257bcf28 RR |
18 | // For compilers that support precompilation, includes "wx.h". |
19 | #include "wx/wxprec.h" | |
20 | ||
21 | #ifdef __BORLANDC__ | |
8898456d | 22 | #pragma hdrstop |
257bcf28 RR |
23 | #endif |
24 | ||
c96ea657 | 25 | #if wxUSE_IMAGE && wxUSE_LIBTIFF |
257bcf28 | 26 | |
88a7a4e1 | 27 | #include "wx/imagtiff.h" |
ccec9093 | 28 | #include "wx/versioninfo.h" |
88a7a4e1 | 29 | |
8898456d WS |
30 | #ifndef WX_PRECOMP |
31 | #include "wx/log.h" | |
32 | #include "wx/app.h" | |
88a7a4e1 | 33 | #include "wx/intl.h" |
0bca0373 | 34 | #include "wx/bitmap.h" |
02761f6c | 35 | #include "wx/module.h" |
34deb5cd | 36 | #include "wx/wxcrtvararg.h" |
8898456d WS |
37 | #endif |
38 | ||
257bcf28 RR |
39 | extern "C" |
40 | { | |
8df068ca CE |
41 | #ifdef __DMC__ |
42 | #include "tif_config.h" | |
426272a3 | 43 | #endif |
257bcf28 RR |
44 | #include "tiff.h" |
45 | #include "tiffio.h" | |
257bcf28 RR |
46 | } |
47 | #include "wx/filefn.h" | |
48 | #include "wx/wfstream.h" | |
257bcf28 | 49 | |
0729bd19 | 50 | #ifndef TIFFLINKAGEMODE |
0e1f8ea4 | 51 | #define TIFFLINKAGEMODE LINKAGEMODE |
19193a2c | 52 | #endif |
0729bd19 | 53 | |
426272a3 VZ |
54 | // ============================================================================ |
55 | // implementation | |
56 | // ============================================================================ | |
57 | ||
58 | // ---------------------------------------------------------------------------- | |
59 | // TIFF library error/warning handlers | |
60 | // ---------------------------------------------------------------------------- | |
61 | ||
34deb5cd VS |
62 | static wxString |
63 | FormatTiffMessage(const char *module, const char *fmt, va_list ap) | |
64 | { | |
65 | char buf[512]; | |
66 | if ( wxCRT_VsnprintfA(buf, WXSIZEOF(buf), fmt, ap) <= 0 ) | |
67 | { | |
68 | // this isn't supposed to happen, but if it does, it's better | |
69 | // than nothing | |
70 | strcpy(buf, "Incorrectly formatted TIFF message"); | |
71 | } | |
72 | buf[WXSIZEOF(buf)-1] = 0; // make sure it is always NULL-terminated | |
73 | ||
74 | wxString msg(buf); | |
75 | if ( module ) | |
76 | msg += wxString::Format(_(" (in module \"%s\")"), module); | |
77 | ||
78 | return msg; | |
79 | } | |
80 | ||
426272a3 VZ |
81 | extern "C" |
82 | { | |
83 | ||
84 | static void | |
34deb5cd | 85 | TIFFwxWarningHandler(const char* module, const char *fmt, va_list ap) |
426272a3 | 86 | { |
34deb5cd | 87 | wxLogWarning("%s", FormatTiffMessage(module, fmt, ap)); |
426272a3 VZ |
88 | } |
89 | ||
90 | static void | |
34deb5cd | 91 | TIFFwxErrorHandler(const char* module, const char *fmt, va_list ap) |
426272a3 | 92 | { |
34deb5cd | 93 | wxLogError("%s", FormatTiffMessage(module, fmt, ap)); |
426272a3 VZ |
94 | } |
95 | ||
96 | } // extern "C" | |
97 | ||
257bcf28 RR |
98 | //----------------------------------------------------------------------------- |
99 | // wxTIFFHandler | |
100 | //----------------------------------------------------------------------------- | |
101 | ||
257bcf28 | 102 | IMPLEMENT_DYNAMIC_CLASS(wxTIFFHandler,wxImageHandler) |
257bcf28 | 103 | |
426272a3 VZ |
104 | wxTIFFHandler::wxTIFFHandler() |
105 | { | |
106 | m_name = wxT("TIFF file"); | |
107 | m_extension = wxT("tif"); | |
ba4800d3 | 108 | m_altExtensions.Add(wxT("tiff")); |
4ca8531f | 109 | m_type = wxBITMAP_TYPE_TIFF; |
426272a3 VZ |
110 | m_mime = wxT("image/tiff"); |
111 | TIFFSetWarningHandler((TIFFErrorHandler) TIFFwxWarningHandler); | |
112 | TIFFSetErrorHandler((TIFFErrorHandler) TIFFwxErrorHandler); | |
113 | } | |
114 | ||
e30285ab VZ |
115 | #if wxUSE_STREAMS |
116 | ||
ec524671 VZ |
117 | // helper to translate our, possibly 64 bit, wxFileOffset to TIFF, always 32 |
118 | // bit, toff_t | |
119 | static toff_t wxFileOffsetToTIFF(wxFileOffset ofs) | |
120 | { | |
121 | if ( ofs == wxInvalidOffset ) | |
122 | return (toff_t)-1; | |
123 | ||
124 | toff_t tofs = wx_truncate_cast(toff_t, ofs); | |
38d4b1e4 | 125 | wxCHECK_MSG( (wxFileOffset)tofs == ofs, (toff_t)-1, |
9a83f860 | 126 | wxT("TIFF library doesn't support large files") ); |
ec524671 VZ |
127 | |
128 | return tofs; | |
129 | } | |
130 | ||
131 | // another helper to convert standard seek mode to our | |
132 | static wxSeekMode wxSeekModeFromTIFF(int whence) | |
133 | { | |
134 | switch ( whence ) | |
135 | { | |
136 | case SEEK_SET: | |
137 | return wxFromStart; | |
138 | ||
139 | case SEEK_CUR: | |
140 | return wxFromCurrent; | |
141 | ||
142 | case SEEK_END: | |
143 | return wxFromEnd; | |
144 | ||
145 | default: | |
146 | return wxFromCurrent; | |
147 | } | |
148 | } | |
149 | ||
90350682 VZ |
150 | extern "C" |
151 | { | |
152 | ||
153 | tsize_t TIFFLINKAGEMODE | |
46f36538 | 154 | wxTIFFNullProc(thandle_t WXUNUSED(handle), |
19193a2c KB |
155 | tdata_t WXUNUSED(buf), |
156 | tsize_t WXUNUSED(size)) | |
f6bcfd97 BP |
157 | { |
158 | return (tsize_t) -1; | |
159 | } | |
160 | ||
90350682 | 161 | tsize_t TIFFLINKAGEMODE |
46f36538 | 162 | wxTIFFReadProc(thandle_t handle, tdata_t buf, tsize_t size) |
257bcf28 RR |
163 | { |
164 | wxInputStream *stream = (wxInputStream*) handle; | |
165 | stream->Read( (void*) buf, (size_t) size ); | |
4a10ea8b | 166 | return wx_truncate_cast(tsize_t, stream->LastRead()); |
257bcf28 RR |
167 | } |
168 | ||
90350682 | 169 | tsize_t TIFFLINKAGEMODE |
46f36538 | 170 | wxTIFFWriteProc(thandle_t handle, tdata_t buf, tsize_t size) |
257bcf28 RR |
171 | { |
172 | wxOutputStream *stream = (wxOutputStream*) handle; | |
173 | stream->Write( (void*) buf, (size_t) size ); | |
4a10ea8b | 174 | return wx_truncate_cast(tsize_t, stream->LastWrite()); |
257bcf28 RR |
175 | } |
176 | ||
90350682 | 177 | toff_t TIFFLINKAGEMODE |
46f36538 | 178 | wxTIFFSeekIProc(thandle_t handle, toff_t off, int whence) |
257bcf28 RR |
179 | { |
180 | wxInputStream *stream = (wxInputStream*) handle; | |
13111b2a | 181 | |
ec524671 VZ |
182 | return wxFileOffsetToTIFF(stream->SeekI((wxFileOffset)off, |
183 | wxSeekModeFromTIFF(whence))); | |
257bcf28 RR |
184 | } |
185 | ||
90350682 | 186 | toff_t TIFFLINKAGEMODE |
46f36538 | 187 | wxTIFFSeekOProc(thandle_t handle, toff_t off, int whence) |
f6bcfd97 BP |
188 | { |
189 | wxOutputStream *stream = (wxOutputStream*) handle; | |
f6bcfd97 | 190 | |
7ffc9aff DS |
191 | toff_t offset = wxFileOffsetToTIFF( |
192 | stream->SeekO((wxFileOffset)off, wxSeekModeFromTIFF(whence)) ); | |
193 | ||
194 | if (offset != (toff_t) -1 || whence != SEEK_SET) | |
195 | { | |
196 | return offset; | |
197 | } | |
198 | ||
199 | ||
200 | /* | |
201 | Try to workaround problems with libtiff seeking past the end of streams. | |
202 | ||
203 | This occurs when libtiff is writing tag entries past the end of a | |
204 | stream but hasn't written the directory yet (which will be placed | |
205 | before the tags and contain offsets to the just written tags). | |
206 | The behaviour for seeking past the end of a stream is not consistent | |
207 | and doesn't work with for example wxMemoryOutputStream. When this type | |
208 | of seeking fails (with SEEK_SET), fill in the gap with zeroes and try | |
209 | again. | |
210 | */ | |
211 | ||
212 | wxFileOffset streamLength = stream->GetLength(); | |
213 | if (streamLength != wxInvalidOffset && (wxFileOffset) off > streamLength) | |
214 | { | |
215 | if (stream->SeekO(streamLength, wxFromStart) == wxInvalidOffset) | |
216 | { | |
217 | return (toff_t) -1; | |
218 | } | |
219 | ||
220 | for (wxFileOffset i = 0; i < (wxFileOffset) off - streamLength; ++i) | |
221 | { | |
222 | stream->PutC(0); | |
223 | } | |
224 | } | |
225 | ||
226 | return wxFileOffsetToTIFF( stream->TellO() ); | |
f6bcfd97 BP |
227 | } |
228 | ||
90350682 | 229 | int TIFFLINKAGEMODE |
46f36538 | 230 | wxTIFFCloseIProc(thandle_t WXUNUSED(handle)) |
257bcf28 | 231 | { |
be0d315e VZ |
232 | // there is no need to close the input stream |
233 | return 0; | |
234 | } | |
235 | ||
236 | int TIFFLINKAGEMODE | |
46f36538 | 237 | wxTIFFCloseOProc(thandle_t handle) |
be0d315e VZ |
238 | { |
239 | wxOutputStream *stream = (wxOutputStream*) handle; | |
240 | ||
241 | return stream->Close() ? 0 : -1; | |
257bcf28 RR |
242 | } |
243 | ||
90350682 | 244 | toff_t TIFFLINKAGEMODE |
46f36538 | 245 | wxTIFFSizeProc(thandle_t handle) |
257bcf28 | 246 | { |
f6bcfd97 | 247 | wxStreamBase *stream = (wxStreamBase*) handle; |
0b72db08 | 248 | return (toff_t) stream->GetSize(); |
257bcf28 RR |
249 | } |
250 | ||
90350682 | 251 | int TIFFLINKAGEMODE |
46f36538 | 252 | wxTIFFMapProc(thandle_t WXUNUSED(handle), |
13111b2a VZ |
253 | tdata_t* WXUNUSED(pbase), |
254 | toff_t* WXUNUSED(psize)) | |
257bcf28 RR |
255 | { |
256 | return 0; | |
257 | } | |
258 | ||
90350682 | 259 | void TIFFLINKAGEMODE |
46f36538 | 260 | wxTIFFUnmapProc(thandle_t WXUNUSED(handle), |
13111b2a VZ |
261 | tdata_t WXUNUSED(base), |
262 | toff_t WXUNUSED(size)) | |
257bcf28 RR |
263 | { |
264 | } | |
265 | ||
90350682 VZ |
266 | } // extern "C" |
267 | ||
257bcf28 RR |
268 | TIFF* |
269 | TIFFwxOpen(wxInputStream &stream, const char* name, const char* mode) | |
270 | { | |
271 | TIFF* tif = TIFFClientOpen(name, mode, | |
272 | (thandle_t) &stream, | |
46f36538 VZ |
273 | wxTIFFReadProc, wxTIFFNullProc, |
274 | wxTIFFSeekIProc, wxTIFFCloseIProc, wxTIFFSizeProc, | |
275 | wxTIFFMapProc, wxTIFFUnmapProc); | |
257bcf28 | 276 | |
257bcf28 RR |
277 | return tif; |
278 | } | |
279 | ||
f6bcfd97 BP |
280 | TIFF* |
281 | TIFFwxOpen(wxOutputStream &stream, const char* name, const char* mode) | |
282 | { | |
283 | TIFF* tif = TIFFClientOpen(name, mode, | |
284 | (thandle_t) &stream, | |
46f36538 VZ |
285 | wxTIFFNullProc, wxTIFFWriteProc, |
286 | wxTIFFSeekOProc, wxTIFFCloseOProc, wxTIFFSizeProc, | |
287 | wxTIFFMapProc, wxTIFFUnmapProc); | |
f6bcfd97 BP |
288 | |
289 | return tif; | |
290 | } | |
257bcf28 | 291 | |
700ec454 | 292 | bool wxTIFFHandler::LoadFile( wxImage *image, wxInputStream& stream, bool verbose, int index ) |
257bcf28 | 293 | { |
60d43ad8 VS |
294 | if (index == -1) |
295 | index = 0; | |
296 | ||
257bcf28 | 297 | image->Destroy(); |
13111b2a | 298 | |
0b72db08 | 299 | TIFF *tif = TIFFwxOpen( stream, "image", "r" ); |
13111b2a | 300 | |
257bcf28 RR |
301 | if (!tif) |
302 | { | |
303 | if (verbose) | |
af588446 | 304 | { |
58c837a4 | 305 | wxLogError( _("TIFF: Error loading image.") ); |
af588446 | 306 | } |
13111b2a | 307 | |
7beb59f3 | 308 | return false; |
257bcf28 | 309 | } |
13111b2a | 310 | |
700ec454 RR |
311 | if (!TIFFSetDirectory( tif, (tdir_t)index )) |
312 | { | |
313 | if (verbose) | |
af588446 | 314 | { |
700ec454 | 315 | wxLogError( _("Invalid TIFF image index.") ); |
af588446 | 316 | } |
13111b2a | 317 | |
700ec454 | 318 | TIFFClose( tif ); |
13111b2a | 319 | |
7beb59f3 | 320 | return false; |
700ec454 | 321 | } |
257bcf28 RR |
322 | |
323 | uint32 w, h; | |
257bcf28 | 324 | uint32 *raster; |
13111b2a | 325 | |
257bcf28 RR |
326 | TIFFGetField( tif, TIFFTAG_IMAGEWIDTH, &w ); |
327 | TIFFGetField( tif, TIFFTAG_IMAGELENGTH, &h ); | |
13111b2a | 328 | |
ebbaec82 DS |
329 | uint16 samplesPerPixel = 0; |
330 | (void) TIFFGetFieldDefaulted(tif, TIFFTAG_SAMPLESPERPIXEL, &samplesPerPixel); | |
331 | ||
332 | uint16 bitsPerSample = 0; | |
333 | (void) TIFFGetFieldDefaulted(tif, TIFFTAG_BITSPERSAMPLE, &bitsPerSample); | |
334 | ||
b6ac40dc VS |
335 | uint16 extraSamples; |
336 | uint16* samplesInfo; | |
337 | TIFFGetFieldDefaulted(tif, TIFFTAG_EXTRASAMPLES, | |
338 | &extraSamples, &samplesInfo); | |
ebbaec82 DS |
339 | |
340 | uint16 photometric; | |
3d59540f DS |
341 | if (!TIFFGetField(tif, TIFFTAG_PHOTOMETRIC, &photometric)) |
342 | { | |
343 | photometric = PHOTOMETRIC_MINISWHITE; | |
344 | } | |
345 | const bool hasAlpha = (extraSamples >= 1 | |
ebbaec82 | 346 | && ((samplesInfo[0] == EXTRASAMPLE_UNSPECIFIED) |
3d59540f DS |
347 | || samplesInfo[0] == EXTRASAMPLE_ASSOCALPHA |
348 | || samplesInfo[0] == EXTRASAMPLE_UNASSALPHA)) | |
349 | || (extraSamples == 0 && samplesPerPixel == 4 | |
350 | && photometric == PHOTOMETRIC_RGB); | |
b6ac40dc | 351 | |
5beedebb VZ |
352 | // guard against integer overflow during multiplication which could result |
353 | // in allocating a too small buffer and then overflowing it | |
ca588225 | 354 | const double bytesNeeded = (double)w * (double)h * sizeof(uint32); |
5beedebb VZ |
355 | if ( bytesNeeded >= wxUINT32_MAX ) |
356 | { | |
357 | if ( verbose ) | |
af588446 | 358 | { |
5beedebb | 359 | wxLogError( _("TIFF: Image size is abnormally big.") ); |
af588446 | 360 | } |
5beedebb VZ |
361 | |
362 | TIFFClose(tif); | |
363 | ||
364 | return false; | |
365 | } | |
13111b2a | 366 | |
8f2a8de6 | 367 | raster = (uint32*) _TIFFmalloc( (uint32)bytesNeeded ); |
13111b2a | 368 | |
257bcf28 RR |
369 | if (!raster) |
370 | { | |
371 | if (verbose) | |
af588446 | 372 | { |
58c837a4 | 373 | wxLogError( _("TIFF: Couldn't allocate memory.") ); |
af588446 | 374 | } |
38755449 | 375 | |
f6bcfd97 | 376 | TIFFClose( tif ); |
13111b2a | 377 | |
7beb59f3 | 378 | return false; |
257bcf28 RR |
379 | } |
380 | ||
479cd5de | 381 | image->Create( (int)w, (int)h ); |
a1b806b9 | 382 | if (!image->IsOk()) |
257bcf28 RR |
383 | { |
384 | if (verbose) | |
af588446 | 385 | { |
58c837a4 | 386 | wxLogError( _("TIFF: Couldn't allocate memory.") ); |
af588446 | 387 | } |
13111b2a VZ |
388 | |
389 | _TIFFfree( raster ); | |
f6bcfd97 | 390 | TIFFClose( tif ); |
13111b2a | 391 | |
7beb59f3 | 392 | return false; |
257bcf28 | 393 | } |
13111b2a | 394 | |
b6ac40dc VS |
395 | if ( hasAlpha ) |
396 | image->SetAlpha(); | |
397 | ||
f60e3134 DS |
398 | uint16 planarConfig = PLANARCONFIG_CONTIG; |
399 | (void) TIFFGetField(tif, TIFFTAG_PLANARCONFIG, &planarConfig); | |
400 | ||
401 | bool ok = true; | |
402 | char msg[1024] = ""; | |
ebbaec82 DS |
403 | if |
404 | ( | |
405 | (planarConfig == PLANARCONFIG_CONTIG && samplesPerPixel == 2 | |
406 | && extraSamples == 1) | |
407 | && | |
408 | ( | |
409 | ( !TIFFRGBAImageOK(tif, msg) ) | |
410 | || (bitsPerSample == 8) | |
411 | ) | |
412 | ) | |
f60e3134 | 413 | { |
ebbaec82 | 414 | const bool isGreyScale = (bitsPerSample == 8); |
f60e3134 DS |
415 | unsigned char *buf = (unsigned char *)_TIFFmalloc(TIFFScanlineSize(tif)); |
416 | uint32 pos = 0; | |
ebbaec82 DS |
417 | const bool minIsWhite = (photometric == PHOTOMETRIC_MINISWHITE); |
418 | const int minValue = minIsWhite ? 255 : 0; | |
f60e3134 DS |
419 | const int maxValue = 255 - minValue; |
420 | ||
421 | /* | |
422 | Decode to ABGR format as that is what the code, that converts to | |
423 | wxImage, later on expects (normally TIFFReadRGBAImageOriented is | |
424 | used to decode which uses an ABGR layout). | |
425 | */ | |
426 | for (uint32 y = 0; y < h; ++y) | |
427 | { | |
428 | if (TIFFReadScanline(tif, buf, y, 0) != 1) | |
429 | { | |
430 | ok = false; | |
431 | break; | |
432 | } | |
433 | ||
ebbaec82 | 434 | if (isGreyScale) |
f60e3134 | 435 | { |
ebbaec82 DS |
436 | for (uint32 x = 0; x < w; ++x) |
437 | { | |
438 | uint8 val = minIsWhite ? 255 - buf[x*2] : buf[x*2]; | |
439 | uint8 alpha = minIsWhite ? 255 - buf[x*2+1] : buf[x*2+1]; | |
440 | raster[pos] = val + (val << 8) + (val << 16) | |
441 | + (alpha << 24); | |
442 | pos++; | |
443 | } | |
444 | } | |
445 | else | |
446 | { | |
447 | for (uint32 x = 0; x < w; ++x) | |
448 | { | |
449 | int mask = buf[x*2/8] << ((x*2)%8); | |
450 | ||
451 | uint8 val = mask & 128 ? maxValue : minValue; | |
452 | raster[pos] = val + (val << 8) + (val << 16) | |
453 | + ((mask & 64 ? maxValue : minValue) << 24); | |
454 | pos++; | |
455 | } | |
f60e3134 DS |
456 | } |
457 | } | |
458 | ||
459 | _TIFFfree(buf); | |
460 | } | |
461 | else | |
462 | { | |
463 | ok = TIFFReadRGBAImageOriented( tif, w, h, raster, | |
464 | ORIENTATION_TOPLEFT, 0 ) != 0; | |
465 | } | |
466 | ||
467 | ||
468 | if (!ok) | |
257bcf28 RR |
469 | { |
470 | if (verbose) | |
af588446 | 471 | { |
58c837a4 | 472 | wxLogError( _("TIFF: Error reading image.") ); |
af588446 | 473 | } |
13111b2a VZ |
474 | |
475 | _TIFFfree( raster ); | |
476 | image->Destroy(); | |
f6bcfd97 | 477 | TIFFClose( tif ); |
13111b2a | 478 | |
7beb59f3 | 479 | return false; |
257bcf28 | 480 | } |
13111b2a | 481 | |
257bcf28 | 482 | unsigned char *ptr = image->GetData(); |
b6ac40dc | 483 | |
c6973bab | 484 | unsigned char *alpha = image->GetAlpha(); |
b6ac40dc | 485 | |
257bcf28 | 486 | uint32 pos = 0; |
13111b2a | 487 | |
257bcf28 RR |
488 | for (uint32 i = 0; i < h; i++) |
489 | { | |
ff7c6c9c | 490 | for (uint32 j = 0; j < w; j++) |
13111b2a | 491 | { |
b6ac40dc VS |
492 | *(ptr++) = (unsigned char)TIFFGetR(raster[pos]); |
493 | *(ptr++) = (unsigned char)TIFFGetG(raster[pos]); | |
494 | *(ptr++) = (unsigned char)TIFFGetB(raster[pos]); | |
495 | if ( hasAlpha ) | |
496 | *(alpha++) = (unsigned char)TIFFGetA(raster[pos]); | |
497 | ||
13111b2a VZ |
498 | pos++; |
499 | } | |
257bcf28 | 500 | } |
13111b2a | 501 | |
4d37f425 | 502 | |
fc695138 DS |
503 | image->SetOption(wxIMAGE_OPTION_TIFF_PHOTOMETRIC, photometric); |
504 | ||
ebbaec82 | 505 | uint16 compression; |
4d37f425 | 506 | /* |
ebbaec82 | 507 | Copy some baseline TIFF tags which helps when re-saving a TIFF |
4d37f425 DS |
508 | to be similar to the original image. |
509 | */ | |
ebbaec82 | 510 | if (samplesPerPixel) |
4d37f425 | 511 | { |
ebbaec82 | 512 | image->SetOption(wxIMAGE_OPTION_TIFF_SAMPLESPERPIXEL, samplesPerPixel); |
4d37f425 DS |
513 | } |
514 | ||
ebbaec82 | 515 | if (bitsPerSample) |
4d37f425 | 516 | { |
ebbaec82 | 517 | image->SetOption(wxIMAGE_OPTION_TIFF_BITSPERSAMPLE, bitsPerSample); |
4d37f425 DS |
518 | } |
519 | ||
520 | if ( TIFFGetFieldDefaulted(tif, TIFFTAG_COMPRESSION, &compression) ) | |
521 | { | |
d65172db | 522 | image->SetOption(wxIMAGE_OPTION_TIFF_COMPRESSION, compression); |
4d37f425 DS |
523 | } |
524 | ||
525 | // Set the resolution unit. | |
526 | wxImageResolution resUnit = wxIMAGE_RESOLUTION_NONE; | |
37ba70a5 | 527 | uint16 tiffRes; |
4d37f425 | 528 | if ( TIFFGetFieldDefaulted(tif, TIFFTAG_RESOLUTIONUNIT, &tiffRes) ) |
37ba70a5 | 529 | { |
4d37f425 | 530 | switch (tiffRes) |
37ba70a5 VZ |
531 | { |
532 | default: | |
533 | wxLogWarning(_("Unknown TIFF resolution unit %d ignored"), | |
4d37f425 | 534 | tiffRes); |
37ba70a5 VZ |
535 | // fall through |
536 | ||
537 | case RESUNIT_NONE: | |
4d37f425 | 538 | resUnit = wxIMAGE_RESOLUTION_NONE; |
37ba70a5 VZ |
539 | break; |
540 | ||
541 | case RESUNIT_INCH: | |
4d37f425 | 542 | resUnit = wxIMAGE_RESOLUTION_INCHES; |
37ba70a5 VZ |
543 | break; |
544 | ||
545 | case RESUNIT_CENTIMETER: | |
4d37f425 | 546 | resUnit = wxIMAGE_RESOLUTION_CM; |
37ba70a5 VZ |
547 | break; |
548 | } | |
4d37f425 | 549 | } |
37ba70a5 | 550 | |
4d37f425 | 551 | image->SetOption(wxIMAGE_OPTION_RESOLUTIONUNIT, resUnit); |
37ba70a5 | 552 | |
4d37f425 DS |
553 | /* |
554 | Set the image resolution if it's available. Resolution tag is not | |
555 | dependant on RESOLUTIONUNIT != RESUNIT_NONE (according to TIFF spec). | |
556 | */ | |
557 | float resX, resY; | |
558 | ||
559 | if ( TIFFGetField(tif, TIFFTAG_XRESOLUTION, &resX) ) | |
560 | { | |
561 | /* | |
562 | Use a string value to not lose precision. | |
563 | rounding to int as cm and then converting to inch may | |
564 | result in whole integer rounding error, eg. 201 instead of 200 dpi. | |
565 | If an app wants an int, GetOptionInt will convert and round down. | |
566 | */ | |
567 | image->SetOption(wxIMAGE_OPTION_RESOLUTIONX, | |
568 | wxString::FromCDouble((double) resX)); | |
37ba70a5 VZ |
569 | } |
570 | ||
4d37f425 DS |
571 | if ( TIFFGetField(tif, TIFFTAG_YRESOLUTION, &resY) ) |
572 | { | |
573 | image->SetOption(wxIMAGE_OPTION_RESOLUTIONY, | |
574 | wxString::FromCDouble((double) resY)); | |
575 | } | |
37ba70a5 | 576 | |
257bcf28 | 577 | _TIFFfree( raster ); |
13111b2a | 578 | |
257bcf28 | 579 | TIFFClose( tif ); |
13111b2a | 580 | |
7beb59f3 | 581 | return true; |
257bcf28 RR |
582 | } |
583 | ||
8faef7cc | 584 | int wxTIFFHandler::DoGetImageCount( wxInputStream& stream ) |
700ec454 RR |
585 | { |
586 | TIFF *tif = TIFFwxOpen( stream, "image", "r" ); | |
13111b2a | 587 | |
700ec454 | 588 | if (!tif) |
13111b2a | 589 | return 0; |
257bcf28 | 590 | |
700ec454 RR |
591 | int dircount = 0; // according to the libtiff docs, dircount should be set to 1 here??? |
592 | do { | |
593 | dircount++; | |
594 | } while (TIFFReadDirectory(tif)); | |
13111b2a | 595 | |
700ec454 | 596 | TIFFClose( tif ); |
03647350 | 597 | |
8faef7cc FM |
598 | // NOTE: this function modifies the current stream position but it's ok |
599 | // (see wxImageHandler::GetImageCount) | |
13111b2a | 600 | |
700ec454 RR |
601 | return dircount; |
602 | } | |
257bcf28 | 603 | |
f6bcfd97 | 604 | bool wxTIFFHandler::SaveFile( wxImage *image, wxOutputStream& stream, bool verbose ) |
257bcf28 | 605 | { |
f6bcfd97 BP |
606 | TIFF *tif = TIFFwxOpen( stream, "image", "w" ); |
607 | ||
608 | if (!tif) | |
609 | { | |
610 | if (verbose) | |
af588446 | 611 | { |
f6bcfd97 | 612 | wxLogError( _("TIFF: Error saving image.") ); |
af588446 | 613 | } |
f6bcfd97 | 614 | |
7beb59f3 | 615 | return false; |
f6bcfd97 BP |
616 | } |
617 | ||
ab176b4b DS |
618 | const int imageWidth = image->GetWidth(); |
619 | TIFFSetField(tif, TIFFTAG_IMAGEWIDTH, (uint32) imageWidth); | |
f6bcfd97 BP |
620 | TIFFSetField(tif, TIFFTAG_IMAGELENGTH, (uint32)image->GetHeight()); |
621 | TIFFSetField(tif, TIFFTAG_ORIENTATION, ORIENTATION_TOPLEFT); | |
f6bcfd97 | 622 | TIFFSetField(tif, TIFFTAG_PLANARCONFIG, PLANARCONFIG_CONTIG); |
38755449 | 623 | |
37ba70a5 VZ |
624 | // save the image resolution if we have it |
625 | int xres, yres; | |
626 | const wxImageResolution res = GetResolutionFromOptions(*image, &xres, &yres); | |
627 | uint16 tiffRes; | |
628 | switch ( res ) | |
fe9308c6 | 629 | { |
37ba70a5 | 630 | default: |
9a83f860 | 631 | wxFAIL_MSG( wxT("unknown image resolution units") ); |
37ba70a5 VZ |
632 | // fall through |
633 | ||
634 | case wxIMAGE_RESOLUTION_NONE: | |
635 | tiffRes = RESUNIT_NONE; | |
636 | break; | |
637 | ||
638 | case wxIMAGE_RESOLUTION_INCHES: | |
639 | tiffRes = RESUNIT_INCH; | |
640 | break; | |
641 | ||
642 | case wxIMAGE_RESOLUTION_CM: | |
643 | tiffRes = RESUNIT_CENTIMETER; | |
644 | break; | |
fe9308c6 VZ |
645 | } |
646 | ||
37ba70a5 VZ |
647 | if ( tiffRes != RESUNIT_NONE ) |
648 | { | |
649 | TIFFSetField(tif, TIFFTAG_RESOLUTIONUNIT, tiffRes); | |
650 | TIFFSetField(tif, TIFFTAG_XRESOLUTION, (float)xres); | |
651 | TIFFSetField(tif, TIFFTAG_YRESOLUTION, (float)yres); | |
652 | } | |
653 | ||
654 | ||
d65172db | 655 | int spp = image->GetOptionInt(wxIMAGE_OPTION_TIFF_SAMPLESPERPIXEL); |
fe9308c6 VZ |
656 | if ( !spp ) |
657 | spp = 3; | |
658 | ||
096c930c DS |
659 | int bps = image->GetOptionInt(wxIMAGE_OPTION_TIFF_BITSPERSAMPLE); |
660 | if ( !bps ) | |
0e11090e | 661 | { |
096c930c | 662 | bps = 8; |
0e11090e DS |
663 | } |
664 | else if (bps == 1) | |
665 | { | |
666 | // One bit per sample combined with 3 samples per pixel is | |
667 | // not allowed and crashes libtiff. | |
668 | spp = 1; | |
669 | } | |
fe9308c6 | 670 | |
fc695138 DS |
671 | int photometric = PHOTOMETRIC_RGB; |
672 | ||
673 | if ( image->HasOption(wxIMAGE_OPTION_TIFF_PHOTOMETRIC) ) | |
674 | { | |
675 | photometric = image->GetOptionInt(wxIMAGE_OPTION_TIFF_PHOTOMETRIC); | |
676 | if (photometric == PHOTOMETRIC_MINISWHITE | |
677 | || photometric == PHOTOMETRIC_MINISBLACK) | |
678 | { | |
679 | // either b/w or greyscale | |
680 | spp = 1; | |
681 | } | |
682 | } | |
ab176b4b | 683 | else if (spp <= 2) |
fc695138 | 684 | { |
2453ceee | 685 | photometric = PHOTOMETRIC_MINISWHITE; |
fc695138 DS |
686 | } |
687 | ||
ab176b4b | 688 | const bool hasAlpha = image->HasAlpha(); |
fc695138 | 689 | |
d65172db | 690 | int compression = image->GetOptionInt(wxIMAGE_OPTION_TIFF_COMPRESSION); |
ab176b4b | 691 | if ( !compression || (compression == COMPRESSION_JPEG && hasAlpha) ) |
f5e20985 | 692 | { |
ab176b4b | 693 | // We can't use COMPRESSION_LZW because current version of libtiff |
f5e20985 VZ |
694 | // doesn't implement it ("no longer implemented due to Unisys patent |
695 | // enforcement") and other compression methods are lossy so we | |
ab176b4b DS |
696 | // shouldn't use them by default -- and the only remaining one is none. |
697 | // Also JPEG compression for alpha images is not a good idea (viewers | |
698 | // not opening the image properly). | |
f5e20985 VZ |
699 | compression = COMPRESSION_NONE; |
700 | } | |
fe9308c6 | 701 | |
ab176b4b DS |
702 | if |
703 | ( | |
704 | (photometric == PHOTOMETRIC_RGB && spp == 4) | |
705 | || (photometric <= PHOTOMETRIC_MINISBLACK && spp == 2) | |
706 | ) | |
707 | { | |
708 | // Compensate for user passing a SamplesPerPixel that includes | |
709 | // the alpha channel. | |
710 | spp--; | |
711 | } | |
712 | ||
713 | ||
714 | int extraSamples = hasAlpha ? 1 : 0; | |
715 | ||
716 | TIFFSetField(tif, TIFFTAG_SAMPLESPERPIXEL, spp + extraSamples); | |
096c930c | 717 | TIFFSetField(tif, TIFFTAG_BITSPERSAMPLE, bps); |
fc695138 | 718 | TIFFSetField(tif, TIFFTAG_PHOTOMETRIC, photometric); |
fe9308c6 VZ |
719 | TIFFSetField(tif, TIFFTAG_COMPRESSION, compression); |
720 | ||
ab176b4b DS |
721 | if (extraSamples) |
722 | { | |
723 | uint16 extra[] = { EXTRASAMPLE_UNSPECIFIED }; | |
724 | TIFFSetField(tif, TIFFTAG_EXTRASAMPLES, (long) 1, &extra); | |
725 | } | |
726 | ||
727 | // scanlinesize is determined by spp+extraSamples and bps | |
649df4bf | 728 | const tsize_t linebytes = |
ab176b4b | 729 | (tsize_t)((imageWidth * (spp + extraSamples) * bps + 7) / 8); |
fe9308c6 | 730 | |
f6bcfd97 | 731 | unsigned char *buf; |
38755449 | 732 | |
ab176b4b DS |
733 | const bool isColouredImage = (spp > 1) |
734 | && (photometric != PHOTOMETRIC_MINISWHITE) | |
735 | && (photometric != PHOTOMETRIC_MINISBLACK); | |
736 | ||
737 | ||
738 | if (TIFFScanlineSize(tif) > linebytes || !isColouredImage || hasAlpha) | |
f6bcfd97 BP |
739 | { |
740 | buf = (unsigned char *)_TIFFmalloc(TIFFScanlineSize(tif)); | |
741 | if (!buf) | |
742 | { | |
743 | if (verbose) | |
af588446 | 744 | { |
f6bcfd97 | 745 | wxLogError( _("TIFF: Couldn't allocate memory.") ); |
af588446 | 746 | } |
f6bcfd97 BP |
747 | |
748 | TIFFClose( tif ); | |
749 | ||
7beb59f3 | 750 | return false; |
f6bcfd97 | 751 | } |
38755449 DW |
752 | } |
753 | else | |
f6bcfd97 BP |
754 | { |
755 | buf = NULL; | |
756 | } | |
757 | ||
fe9308c6 VZ |
758 | TIFFSetField(tif, TIFFTAG_ROWSPERSTRIP,TIFFDefaultStripSize(tif, (uint32) -1)); |
759 | ||
ab176b4b DS |
760 | const int bitsPerPixel = (spp + extraSamples) * bps; |
761 | const int bytesPerPixel = (bitsPerPixel + 7) / 8; | |
f2de33ed DS |
762 | const int pixelsPerByte = 8 / bitsPerPixel; |
763 | int remainingPixelCount = 0; | |
764 | ||
765 | if (pixelsPerByte) | |
766 | { | |
767 | // How many pixels to write in the last byte column? | |
ab176b4b DS |
768 | remainingPixelCount = imageWidth % pixelsPerByte; |
769 | if (!remainingPixelCount) remainingPixelCount = pixelsPerByte; | |
f2de33ed DS |
770 | } |
771 | ||
fc695138 | 772 | const bool minIsWhite = (photometric == PHOTOMETRIC_MINISWHITE); |
f6bcfd97 | 773 | unsigned char *ptr = image->GetData(); |
fe9308c6 | 774 | for ( int row = 0; row < image->GetHeight(); row++ ) |
f6bcfd97 | 775 | { |
fe9308c6 VZ |
776 | if ( buf ) |
777 | { | |
fc695138 | 778 | if (isColouredImage) |
fe9308c6 | 779 | { |
ab176b4b DS |
780 | // colour image |
781 | if (hasAlpha) | |
782 | { | |
783 | for ( int column = 0; column < imageWidth; column++ ) | |
784 | { | |
785 | buf[column*4 ] = ptr[column*3 ]; | |
786 | buf[column*4 + 1] = ptr[column*3 + 1]; | |
787 | buf[column*4 + 2] = ptr[column*3 + 2]; | |
788 | buf[column*4 + 3] = image->GetAlpha(column, row); | |
789 | } | |
790 | } | |
791 | else | |
792 | { | |
793 | memcpy(buf, ptr, imageWidth * 3); | |
794 | } | |
fe9308c6 | 795 | } |
8e28cc47 DS |
796 | else if (spp * bps == 8) // greyscale image |
797 | { | |
ab176b4b | 798 | for ( int column = 0; column < imageWidth; column++ ) |
8e28cc47 | 799 | { |
fc695138 DS |
800 | uint8 value = ptr[column*3 + 1]; |
801 | if (minIsWhite) | |
802 | { | |
803 | value = 255 - value; | |
804 | } | |
805 | ||
ab176b4b DS |
806 | buf[column * bytesPerPixel] = value; |
807 | ||
808 | if (hasAlpha) | |
809 | { | |
810 | value = image->GetAlpha(column, row); | |
811 | buf[column*bytesPerPixel+1] | |
812 | = minIsWhite ? 255 - value : value; | |
813 | } | |
8e28cc47 DS |
814 | } |
815 | } | |
fe9308c6 VZ |
816 | else // black and white image |
817 | { | |
818 | for ( int column = 0; column < linebytes; column++ ) | |
819 | { | |
820 | uint8 reverse = 0; | |
f2de33ed DS |
821 | int pixelsPerByteCount = (column + 1 != linebytes) |
822 | ? pixelsPerByte | |
823 | : remainingPixelCount; | |
824 | for ( int bp = 0; bp < pixelsPerByteCount; bp++ ) | |
fe9308c6 | 825 | { |
ab176b4b DS |
826 | if ( (ptr[column * 3 * pixelsPerByte + bp*3 + 1] <=127) |
827 | == minIsWhite ) | |
fe9308c6 | 828 | { |
7f77b25e | 829 | // check only green as this is sufficient |
ab176b4b DS |
830 | reverse |= (uint8) (128 >> (bp * bitsPerPixel)); |
831 | } | |
832 | ||
833 | if (hasAlpha | |
834 | && (image->GetAlpha(column * pixelsPerByte + bp, | |
835 | row) <= 127) == minIsWhite) | |
836 | { | |
837 | reverse |= (uint8) (64 >> (bp * bitsPerPixel)); | |
fe9308c6 | 838 | } |
fe9308c6 VZ |
839 | } |
840 | ||
3ba9ea72 | 841 | buf[column] = reverse; |
fe9308c6 VZ |
842 | } |
843 | } | |
844 | } | |
38755449 | 845 | |
fe9308c6 | 846 | if ( TIFFWriteScanline(tif, buf ? buf : ptr, (uint32)row, 0) < 0 ) |
f6bcfd97 | 847 | { |
19193a2c | 848 | if (verbose) |
af588446 | 849 | { |
19193a2c | 850 | wxLogError( _("TIFF: Error writing image.") ); |
af588446 | 851 | } |
38755449 | 852 | |
f6bcfd97 BP |
853 | TIFFClose( tif ); |
854 | if (buf) | |
855 | _TIFFfree(buf); | |
38755449 | 856 | |
7beb59f3 | 857 | return false; |
f6bcfd97 | 858 | } |
fe9308c6 | 859 | |
ab176b4b | 860 | ptr += imageWidth * 3; |
f6bcfd97 BP |
861 | } |
862 | ||
863 | (void) TIFFClose(tif); | |
864 | ||
865 | if (buf) | |
fe9308c6 | 866 | _TIFFfree(buf); |
f6bcfd97 | 867 | |
7beb59f3 | 868 | return true; |
257bcf28 RR |
869 | } |
870 | ||
871 | bool wxTIFFHandler::DoCanRead( wxInputStream& stream ) | |
872 | { | |
0b72db08 | 873 | unsigned char hdr[2]; |
257bcf28 | 874 | |
8faef7cc | 875 | if ( !stream.Read(&hdr[0], WXSIZEOF(hdr)) ) // it's ok to modify the stream position here |
7beb59f3 | 876 | return false; |
79fa2374 | 877 | |
79fa2374 VZ |
878 | return (hdr[0] == 'I' && hdr[1] == 'I') || |
879 | (hdr[0] == 'M' && hdr[1] == 'M'); | |
257bcf28 RR |
880 | } |
881 | ||
e30285ab | 882 | #endif // wxUSE_STREAMS |
257bcf28 | 883 | |
ccec9093 VZ |
884 | /*static*/ wxVersionInfo wxTIFFHandler::GetLibraryVersionInfo() |
885 | { | |
886 | int major, | |
887 | minor, | |
888 | micro; | |
889 | ||
890 | const wxString ver(::TIFFGetVersion()); | |
9c1f960f | 891 | if ( wxSscanf(ver, "LIBTIFF, Version %d.%d.%d", &major, &minor, µ) != 3 ) |
ccec9093 VZ |
892 | { |
893 | wxLogDebug("Unrecognized libtiff version string \"%s\"", ver); | |
894 | ||
895 | major = | |
896 | minor = | |
897 | micro = 0; | |
898 | } | |
899 | ||
900 | wxString copyright; | |
901 | const wxString desc = ver.BeforeFirst('\n', ©right); | |
902 | copyright.Replace("\n", ""); | |
903 | ||
904 | return wxVersionInfo("libtiff", major, minor, micro, desc, copyright); | |
905 | } | |
906 | ||
e30285ab | 907 | #endif // wxUSE_LIBTIFF |