]> git.saurik.com Git - wxWidgets.git/blame_incremental - src/common/imagtiff.cpp
set string length properly in wxString(p, conv, len) ctor (patch 755593)
[wxWidgets.git] / src / common / imagtiff.cpp
... / ...
CommitLineData
1/////////////////////////////////////////////////////////////////////////////
2// Name: imagtiff.cpp
3// Purpose: wxImage TIFF handler
4// Author: Robert Roebling
5// RCS-ID: $Id$
6// Copyright: (c) Robert Roebling
7// Licence: wxWindows licence
8/////////////////////////////////////////////////////////////////////////////
9
10#ifdef __GNUG__
11#pragma implementation "imagtiff.h"
12#endif
13
14// For compilers that support precompilation, includes "wx.h".
15#include "wx/wxprec.h"
16
17#ifdef __BORLANDC__
18#pragma hdrstop
19#endif
20
21#include "wx/defs.h"
22
23#if wxUSE_IMAGE && wxUSE_LIBTIFF
24
25#include "wx/imagtiff.h"
26#include "wx/bitmap.h"
27#include "wx/debug.h"
28#include "wx/log.h"
29#include "wx/app.h"
30extern "C"
31{
32 #include "tiff.h"
33 #include "tiffio.h"
34}
35#include "wx/filefn.h"
36#include "wx/wfstream.h"
37#include "wx/intl.h"
38#include "wx/module.h"
39
40#ifndef TIFFLINKAGEMODE
41 #define TIFFLINKAGEMODE LINKAGEMODE
42#endif
43
44//-----------------------------------------------------------------------------
45// wxTIFFHandler
46//-----------------------------------------------------------------------------
47
48IMPLEMENT_DYNAMIC_CLASS(wxTIFFHandler,wxImageHandler)
49
50#if wxUSE_STREAMS
51
52extern "C"
53{
54
55tsize_t TIFFLINKAGEMODE
56_tiffNullProc(thandle_t WXUNUSED(handle),
57 tdata_t WXUNUSED(buf),
58 tsize_t WXUNUSED(size))
59{
60 return (tsize_t) -1;
61}
62
63tsize_t TIFFLINKAGEMODE
64_tiffReadProc(thandle_t handle, tdata_t buf, tsize_t size)
65{
66 wxInputStream *stream = (wxInputStream*) handle;
67 stream->Read( (void*) buf, (size_t) size );
68 return stream->LastRead();
69}
70
71tsize_t TIFFLINKAGEMODE
72_tiffWriteProc(thandle_t handle, tdata_t buf, tsize_t size)
73{
74 wxOutputStream *stream = (wxOutputStream*) handle;
75 stream->Write( (void*) buf, (size_t) size );
76 return stream->LastWrite();
77}
78
79toff_t TIFFLINKAGEMODE
80_tiffSeekIProc(thandle_t handle, toff_t off, int whence)
81{
82 wxInputStream *stream = (wxInputStream*) handle;
83 wxSeekMode mode;
84 switch (whence)
85 {
86 case SEEK_SET: mode = wxFromStart; break;
87 case SEEK_CUR: mode = wxFromCurrent; break;
88 case SEEK_END: mode = wxFromEnd; break;
89 default: mode = wxFromCurrent; break;
90 }
91
92 return (toff_t)stream->SeekI( (off_t)off, mode );
93}
94
95toff_t TIFFLINKAGEMODE
96_tiffSeekOProc(thandle_t handle, toff_t off, int whence)
97{
98 wxOutputStream *stream = (wxOutputStream*) handle;
99 wxSeekMode mode;
100 switch (whence)
101 {
102 case SEEK_SET: mode = wxFromStart; break;
103 case SEEK_CUR: mode = wxFromCurrent; break;
104 case SEEK_END: mode = wxFromEnd; break;
105 default: mode = wxFromCurrent; break;
106 }
107
108 return (toff_t)stream->SeekO( (off_t)off, mode );
109}
110
111int TIFFLINKAGEMODE
112_tiffCloseProc(thandle_t WXUNUSED(handle))
113{
114 return 0; // ?
115}
116
117toff_t TIFFLINKAGEMODE
118_tiffSizeProc(thandle_t handle)
119{
120 wxStreamBase *stream = (wxStreamBase*) handle;
121 return (toff_t) stream->GetSize();
122}
123
124int TIFFLINKAGEMODE
125_tiffMapProc(thandle_t WXUNUSED(handle),
126 tdata_t* WXUNUSED(pbase),
127 toff_t* WXUNUSED(psize))
128{
129 return 0;
130}
131
132void TIFFLINKAGEMODE
133_tiffUnmapProc(thandle_t WXUNUSED(handle),
134 tdata_t WXUNUSED(base),
135 toff_t WXUNUSED(size))
136{
137}
138
139static void
140TIFFwxWarningHandler(const char* module, const char* fmt, va_list ap)
141{
142 if (module != NULL)
143 wxLogWarning(_("tiff module: %s"), module);
144 wxLogWarning((wxChar *) fmt, ap);
145}
146
147static void
148TIFFwxErrorHandler(const char* module, const char* fmt, va_list ap)
149{
150 if (module != NULL)
151 wxLogError(_("tiff module: %s"), module);
152 wxVLogError((wxChar *) fmt, ap);
153}
154
155} // extern "C"
156
157TIFF*
158TIFFwxOpen(wxInputStream &stream, const char* name, const char* mode)
159{
160 TIFF* tif = TIFFClientOpen(name, mode,
161 (thandle_t) &stream,
162 _tiffReadProc, _tiffNullProc,
163 _tiffSeekIProc, _tiffCloseProc, _tiffSizeProc,
164 _tiffMapProc, _tiffUnmapProc);
165
166 return tif;
167}
168
169TIFF*
170TIFFwxOpen(wxOutputStream &stream, const char* name, const char* mode)
171{
172 TIFF* tif = TIFFClientOpen(name, mode,
173 (thandle_t) &stream,
174 _tiffNullProc, _tiffWriteProc,
175 _tiffSeekOProc, _tiffCloseProc, _tiffSizeProc,
176 _tiffMapProc, _tiffUnmapProc);
177
178 return tif;
179}
180
181wxTIFFHandler::wxTIFFHandler()
182{
183 m_name = wxT("TIFF file");
184 m_extension = wxT("tif");
185 m_type = wxBITMAP_TYPE_TIF;
186 m_mime = wxT("image/tiff");
187 TIFFSetWarningHandler((TIFFErrorHandler) TIFFwxWarningHandler);
188 TIFFSetErrorHandler((TIFFErrorHandler) TIFFwxErrorHandler);
189}
190
191bool wxTIFFHandler::LoadFile( wxImage *image, wxInputStream& stream, bool verbose, int index )
192{
193 if (index == -1)
194 index = 0;
195
196 image->Destroy();
197
198 TIFF *tif = TIFFwxOpen( stream, "image", "r" );
199
200 if (!tif)
201 {
202 if (verbose)
203 wxLogError( _("TIFF: Error loading image.") );
204
205 return FALSE;
206 }
207
208 if (!TIFFSetDirectory( tif, (tdir_t)index ))
209 {
210 if (verbose)
211 wxLogError( _("Invalid TIFF image index.") );
212
213 TIFFClose( tif );
214
215 return FALSE;
216 }
217
218 uint32 w, h;
219 uint32 npixels;
220 uint32 *raster;
221
222 TIFFGetField( tif, TIFFTAG_IMAGEWIDTH, &w );
223 TIFFGetField( tif, TIFFTAG_IMAGELENGTH, &h );
224
225 npixels = w * h;
226
227 raster = (uint32*) _TIFFmalloc( npixels * sizeof(uint32) );
228
229 if (!raster)
230 {
231 if (verbose)
232 wxLogError( _("TIFF: Couldn't allocate memory.") );
233
234 TIFFClose( tif );
235
236 return FALSE;
237 }
238
239 image->Create( (int)w, (int)h );
240 if (!image->Ok())
241 {
242 if (verbose)
243 wxLogError( _("TIFF: Couldn't allocate memory.") );
244
245 _TIFFfree( raster );
246 TIFFClose( tif );
247
248 return FALSE;
249 }
250
251 if (!TIFFReadRGBAImage( tif, w, h, raster, 0 ))
252 {
253 if (verbose)
254 wxLogError( _("TIFF: Error reading image.") );
255
256 _TIFFfree( raster );
257 image->Destroy();
258 TIFFClose( tif );
259
260 return FALSE;
261 }
262
263 bool hasmask = FALSE;
264
265 unsigned char *ptr = image->GetData();
266 ptr += w*3*(h-1);
267 uint32 pos = 0;
268
269 for (uint32 i = 0; i < h; i++)
270 {
271 for (uint32 j = 0; j < w; j++)
272 {
273 unsigned char alpha = (unsigned char)TIFFGetA(raster[pos]);
274 if (alpha < 127)
275 {
276 hasmask = TRUE;
277 ptr[0] = image->GetMaskRed();
278 ptr++;
279 ptr[0] = image->GetMaskGreen();
280 ptr++;
281 ptr[0] = image->GetMaskBlue();
282 ptr++;
283 }
284 else
285 {
286 ptr[0] = (unsigned char)TIFFGetR(raster[pos]);
287 ptr++;
288 ptr[0] = (unsigned char)TIFFGetG(raster[pos]);
289 ptr++;
290 ptr[0] = (unsigned char)TIFFGetB(raster[pos]);
291 ptr++;
292 }
293 pos++;
294 }
295 ptr -= 2*w*3; // subtract line we just added plus one line
296 }
297
298 _TIFFfree( raster );
299
300 TIFFClose( tif );
301
302 image->SetMask( hasmask );
303
304 return TRUE;
305}
306
307int wxTIFFHandler::GetImageCount( wxInputStream& stream )
308{
309 TIFF *tif = TIFFwxOpen( stream, "image", "r" );
310
311 if (!tif)
312 return 0;
313
314 int dircount = 0; // according to the libtiff docs, dircount should be set to 1 here???
315 do {
316 dircount++;
317 } while (TIFFReadDirectory(tif));
318
319 TIFFClose( tif );
320
321 return dircount;
322}
323
324bool wxTIFFHandler::SaveFile( wxImage *image, wxOutputStream& stream, bool verbose )
325{
326 TIFF *tif = TIFFwxOpen( stream, "image", "w" );
327
328 if (!tif)
329 {
330 if (verbose)
331 wxLogError( _("TIFF: Error saving image.") );
332
333 return FALSE;
334 }
335
336 TIFFSetField(tif, TIFFTAG_IMAGEWIDTH, (uint32)image->GetWidth());
337 TIFFSetField(tif, TIFFTAG_IMAGELENGTH, (uint32)image->GetHeight());
338 TIFFSetField(tif, TIFFTAG_ORIENTATION, ORIENTATION_TOPLEFT);
339 TIFFSetField(tif, TIFFTAG_SAMPLESPERPIXEL, 3);
340 TIFFSetField(tif, TIFFTAG_BITSPERSAMPLE, 8);
341 TIFFSetField(tif, TIFFTAG_PLANARCONFIG, PLANARCONFIG_CONTIG);
342 TIFFSetField(tif, TIFFTAG_PHOTOMETRIC, PHOTOMETRIC_RGB);
343 TIFFSetField(tif, TIFFTAG_COMPRESSION, COMPRESSION_LZW);
344
345 tsize_t linebytes = (tsize_t)image->GetWidth() * 3;
346 unsigned char *buf;
347
348 if (TIFFScanlineSize(tif) > linebytes)
349 {
350 buf = (unsigned char *)_TIFFmalloc(TIFFScanlineSize(tif));
351 if (!buf)
352 {
353 if (verbose)
354 wxLogError( _("TIFF: Couldn't allocate memory.") );
355
356 TIFFClose( tif );
357
358 return FALSE;
359 }
360 }
361 else
362 {
363 buf = NULL;
364 }
365
366 TIFFSetField(tif, TIFFTAG_ROWSPERSTRIP,
367 TIFFDefaultStripSize(tif, (uint32) -1));
368
369 unsigned char *ptr = image->GetData();
370 for (int row = 0; row < image->GetHeight(); row++)
371 {
372 if (buf)
373 memcpy(buf, ptr, image->GetWidth());
374
375 if (TIFFWriteScanline(tif, buf ? buf : ptr, (uint32)row, 0) < 0)
376 {
377 if (verbose)
378 wxLogError( _("TIFF: Error writing image.") );
379
380 TIFFClose( tif );
381 if (buf)
382 _TIFFfree(buf);
383
384 return FALSE;
385 }
386 ptr += image->GetWidth()*3;
387 }
388
389 (void) TIFFClose(tif);
390
391 if (buf)
392 _TIFFfree(buf);
393
394 return TRUE;
395}
396
397bool wxTIFFHandler::DoCanRead( wxInputStream& stream )
398{
399 unsigned char hdr[2];
400
401 if ( !stream.Read(&hdr, WXSIZEOF(hdr)) )
402 return FALSE;
403
404 return (hdr[0] == 'I' && hdr[1] == 'I') ||
405 (hdr[0] == 'M' && hdr[1] == 'M');
406}
407
408#endif // wxUSE_STREAMS
409
410#endif // wxUSE_LIBTIFF
411