]> git.saurik.com Git - wxWidgets.git/blame - src/tiff/libtiff/tif_write.c
Support Unicode module names in wxDynamicLibrary::MSWGetModuleHandle().
[wxWidgets.git] / src / tiff / libtiff / tif_write.c
CommitLineData
8414a40c
VZ
1/* $Id$ */
2
3/*
4 * Copyright (c) 1988-1997 Sam Leffler
5 * Copyright (c) 1991-1997 Silicon Graphics, Inc.
6 *
80ed523f 7 * Permission to use, copy, modify, distribute, and sell this software and
8414a40c
VZ
8 * its documentation for any purpose is hereby granted without fee, provided
9 * that (i) the above copyright notices and this permission notice appear in
10 * all copies of the software and related documentation, and (ii) the names of
11 * Sam Leffler and Silicon Graphics may not be used in any advertising or
12 * publicity relating to the software without the specific, prior written
13 * permission of Sam Leffler and Silicon Graphics.
80ed523f
VZ
14 *
15 * THE SOFTWARE IS PROVIDED "AS-IS" AND WITHOUT WARRANTY OF ANY KIND,
16 * EXPRESS, IMPLIED OR OTHERWISE, INCLUDING WITHOUT LIMITATION, ANY
17 * WARRANTY OF MERCHANTABILITY OR FITNESS FOR A PARTICULAR PURPOSE.
18 *
8414a40c
VZ
19 * IN NO EVENT SHALL SAM LEFFLER OR SILICON GRAPHICS BE LIABLE FOR
20 * ANY SPECIAL, INCIDENTAL, INDIRECT OR CONSEQUENTIAL DAMAGES OF ANY KIND,
21 * OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS,
80ed523f
VZ
22 * WHETHER OR NOT ADVISED OF THE POSSIBILITY OF DAMAGE, AND ON ANY THEORY OF
23 * LIABILITY, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE
8414a40c
VZ
24 * OF THIS SOFTWARE.
25 */
26
27/*
28 * TIFF Library.
29 *
30 * Scanline-oriented Write Support
31 */
32#include "tiffiop.h"
33#include <stdio.h>
34
80ed523f 35#define STRIPINCR 20 /* expansion factor on strip array */
8414a40c 36
80ed523f 37#define WRITECHECKSTRIPS(tif, module) \
8414a40c 38 (((tif)->tif_flags&TIFF_BEENWRITING) || TIFFWriteCheck((tif),0,module))
80ed523f 39#define WRITECHECKTILES(tif, module) \
8414a40c 40 (((tif)->tif_flags&TIFF_BEENWRITING) || TIFFWriteCheck((tif),1,module))
80ed523f 41#define BUFFERCHECK(tif) \
8414a40c 42 ((((tif)->tif_flags & TIFF_BUFFERSETUP) && tif->tif_rawdata) || \
80ed523f 43 TIFFWriteBufferSetup((tif), NULL, (tmsize_t) -1))
8414a40c 44
80ed523f
VZ
45static int TIFFGrowStrips(TIFF* tif, uint32 delta, const char* module);
46static int TIFFAppendToStrip(TIFF* tif, uint32 strip, uint8* data, tmsize_t cc);
8414a40c
VZ
47
48int
80ed523f 49TIFFWriteScanline(TIFF* tif, void* buf, uint32 row, uint16 sample)
8414a40c
VZ
50{
51 static const char module[] = "TIFFWriteScanline";
52 register TIFFDirectory *td;
53 int status, imagegrew = 0;
80ed523f 54 uint32 strip;
8414a40c
VZ
55
56 if (!WRITECHECKSTRIPS(tif, module))
57 return (-1);
58 /*
59 * Handle delayed allocation of data buffer. This
60 * permits it to be sized more intelligently (using
61 * directory information).
62 */
63 if (!BUFFERCHECK(tif))
64 return (-1);
80ed523f
VZ
65 tif->tif_flags |= TIFF_BUF4WRITE; /* not strictly sure this is right*/
66
8414a40c
VZ
67 td = &tif->tif_dir;
68 /*
69 * Extend image length if needed
70 * (but only for PlanarConfig=1).
71 */
72 if (row >= td->td_imagelength) { /* extend image */
73 if (td->td_planarconfig == PLANARCONFIG_SEPARATE) {
80ed523f
VZ
74 TIFFErrorExt(tif->tif_clientdata, module,
75 "Can not change \"ImageLength\" when using separate planes");
8414a40c
VZ
76 return (-1);
77 }
78 td->td_imagelength = row+1;
79 imagegrew = 1;
80 }
81 /*
82 * Calculate strip and check for crossings.
83 */
84 if (td->td_planarconfig == PLANARCONFIG_SEPARATE) {
85 if (sample >= td->td_samplesperpixel) {
80ed523f
VZ
86 TIFFErrorExt(tif->tif_clientdata, module,
87 "%lu: Sample out of range, max %lu",
88 (unsigned long) sample, (unsigned long) td->td_samplesperpixel);
8414a40c
VZ
89 return (-1);
90 }
91 strip = sample*td->td_stripsperimage + row/td->td_rowsperstrip;
92 } else
93 strip = row / td->td_rowsperstrip;
94 /*
95 * Check strip array to make sure there's space. We don't support
96 * dynamically growing files that have data organized in separate
97 * bitplanes because it's too painful. In that case we require that
98 * the imagelength be set properly before the first write (so that the
99 * strips array will be fully allocated above).
100 */
101 if (strip >= td->td_nstrips && !TIFFGrowStrips(tif, 1, module))
102 return (-1);
103 if (strip != tif->tif_curstrip) {
104 /*
105 * Changing strips -- flush any data present.
106 */
107 if (!TIFFFlushData(tif))
108 return (-1);
109 tif->tif_curstrip = strip;
110 /*
111 * Watch out for a growing image. The value of strips/image
112 * will initially be 1 (since it can't be deduced until the
113 * imagelength is known).
114 */
115 if (strip >= td->td_stripsperimage && imagegrew)
116 td->td_stripsperimage =
80ed523f 117 TIFFhowmany_32(td->td_imagelength,td->td_rowsperstrip);
8414a40c
VZ
118 tif->tif_row =
119 (strip % td->td_stripsperimage) * td->td_rowsperstrip;
120 if ((tif->tif_flags & TIFF_CODERSETUP) == 0) {
121 if (!(*tif->tif_setupencode)(tif))
122 return (-1);
123 tif->tif_flags |= TIFF_CODERSETUP;
124 }
125
126 tif->tif_rawcc = 0;
127 tif->tif_rawcp = tif->tif_rawdata;
128
129 if( td->td_stripbytecount[strip] > 0 )
130 {
131 /* if we are writing over existing tiles, zero length */
132 td->td_stripbytecount[strip] = 0;
133
134 /* this forces TIFFAppendToStrip() to do a seek */
135 tif->tif_curoff = 0;
136 }
137
138 if (!(*tif->tif_preencode)(tif, sample))
139 return (-1);
140 tif->tif_flags |= TIFF_POSTENCODE;
141 }
142 /*
143 * Ensure the write is either sequential or at the
144 * beginning of a strip (or that we can randomly
145 * access the data -- i.e. no encoding).
146 */
147 if (row != tif->tif_row) {
148 if (row < tif->tif_row) {
149 /*
150 * Moving backwards within the same strip:
151 * backup to the start and then decode
152 * forward (below).
153 */
154 tif->tif_row = (strip % td->td_stripsperimage) *
155 td->td_rowsperstrip;
156 tif->tif_rawcp = tif->tif_rawdata;
157 }
158 /*
159 * Seek forward to the desired row.
160 */
161 if (!(*tif->tif_seek)(tif, row - tif->tif_row))
162 return (-1);
163 tif->tif_row = row;
164 }
165
80ed523f
VZ
166 /* swab if needed - note that source buffer will be altered */
167 tif->tif_postdecode( tif, (uint8*) buf, tif->tif_scanlinesize );
8414a40c 168
80ed523f 169 status = (*tif->tif_encoderow)(tif, (uint8*) buf,
8414a40c
VZ
170 tif->tif_scanlinesize, sample);
171
172 /* we are now poised at the beginning of the next row */
173 tif->tif_row = row + 1;
174 return (status);
175}
176
177/*
178 * Encode the supplied data and write it to the
179 * specified strip.
180 *
181 * NB: Image length must be setup before writing.
182 */
80ed523f
VZ
183tmsize_t
184TIFFWriteEncodedStrip(TIFF* tif, uint32 strip, void* data, tmsize_t cc)
8414a40c
VZ
185{
186 static const char module[] = "TIFFWriteEncodedStrip";
187 TIFFDirectory *td = &tif->tif_dir;
80ed523f 188 uint16 sample;
8414a40c
VZ
189
190 if (!WRITECHECKSTRIPS(tif, module))
80ed523f 191 return ((tmsize_t) -1);
8414a40c
VZ
192 /*
193 * Check strip array to make sure there's space.
194 * We don't support dynamically growing files that
195 * have data organized in separate bitplanes because
196 * it's too painful. In that case we require that
197 * the imagelength be set properly before the first
198 * write (so that the strips array will be fully
199 * allocated above).
200 */
201 if (strip >= td->td_nstrips) {
202 if (td->td_planarconfig == PLANARCONFIG_SEPARATE) {
80ed523f
VZ
203 TIFFErrorExt(tif->tif_clientdata, module,
204 "Can not grow image by strips when using separate planes");
205 return ((tmsize_t) -1);
8414a40c
VZ
206 }
207 if (!TIFFGrowStrips(tif, 1, module))
80ed523f 208 return ((tmsize_t) -1);
8414a40c 209 td->td_stripsperimage =
80ed523f 210 TIFFhowmany_32(td->td_imagelength, td->td_rowsperstrip);
8414a40c
VZ
211 }
212 /*
213 * Handle delayed allocation of data buffer. This
214 * permits it to be sized according to the directory
215 * info.
216 */
217 if (!BUFFERCHECK(tif))
80ed523f
VZ
218 return ((tmsize_t) -1);
219
220 tif->tif_flags |= TIFF_BUF4WRITE;
8414a40c 221 tif->tif_curstrip = strip;
80ed523f 222
8414a40c
VZ
223 tif->tif_row = (strip % td->td_stripsperimage) * td->td_rowsperstrip;
224 if ((tif->tif_flags & TIFF_CODERSETUP) == 0) {
225 if (!(*tif->tif_setupencode)(tif))
80ed523f 226 return ((tmsize_t) -1);
8414a40c
VZ
227 tif->tif_flags |= TIFF_CODERSETUP;
228 }
8414a40c 229
80ed523f 230 if( td->td_stripbytecount[strip] > 0 )
8414a40c 231 {
80ed523f
VZ
232 /* Make sure that at the first attempt of rewriting the tile, we will have */
233 /* more bytes available in the output buffer than the previous byte count, */
234 /* so that TIFFAppendToStrip() will detect the overflow when it is called the first */
235 /* time if the new compressed tile is bigger than the older one. (GDAL #4771) */
236 if( tif->tif_rawdatasize <= td->td_stripbytecount[strip] )
237 {
238 if( !(TIFFWriteBufferSetup(tif, NULL,
239 (tmsize_t)TIFFroundup_64((uint64)(td->td_stripbytecount[strip] + 1), 1024))) )
240 return ((tmsize_t)(-1));
241 }
242
243 /* Force TIFFAppendToStrip() to consider placing data at end
244 of file. */
8414a40c
VZ
245 tif->tif_curoff = 0;
246 }
80ed523f
VZ
247
248 tif->tif_rawcc = 0;
249 tif->tif_rawcp = tif->tif_rawdata;
250
8414a40c 251 tif->tif_flags &= ~TIFF_POSTENCODE;
80ed523f 252 sample = (uint16)(strip / td->td_stripsperimage);
8414a40c 253 if (!(*tif->tif_preencode)(tif, sample))
80ed523f 254 return ((tmsize_t) -1);
8414a40c
VZ
255
256 /* swab if needed - note that source buffer will be altered */
80ed523f 257 tif->tif_postdecode( tif, (uint8*) data, cc );
8414a40c 258
80ed523f
VZ
259 if (!(*tif->tif_encodestrip)(tif, (uint8*) data, cc, sample))
260 return (0);
8414a40c 261 if (!(*tif->tif_postencode)(tif))
80ed523f 262 return ((tmsize_t) -1);
8414a40c
VZ
263 if (!isFillOrder(tif, td->td_fillorder) &&
264 (tif->tif_flags & TIFF_NOBITREV) == 0)
265 TIFFReverseBits(tif->tif_rawdata, tif->tif_rawcc);
266 if (tif->tif_rawcc > 0 &&
267 !TIFFAppendToStrip(tif, strip, tif->tif_rawdata, tif->tif_rawcc))
80ed523f 268 return ((tmsize_t) -1);
8414a40c
VZ
269 tif->tif_rawcc = 0;
270 tif->tif_rawcp = tif->tif_rawdata;
271 return (cc);
272}
273
274/*
275 * Write the supplied data to the specified strip.
276 *
277 * NB: Image length must be setup before writing.
278 */
80ed523f
VZ
279tmsize_t
280TIFFWriteRawStrip(TIFF* tif, uint32 strip, void* data, tmsize_t cc)
8414a40c
VZ
281{
282 static const char module[] = "TIFFWriteRawStrip";
283 TIFFDirectory *td = &tif->tif_dir;
284
285 if (!WRITECHECKSTRIPS(tif, module))
80ed523f 286 return ((tmsize_t) -1);
8414a40c
VZ
287 /*
288 * Check strip array to make sure there's space.
289 * We don't support dynamically growing files that
290 * have data organized in separate bitplanes because
291 * it's too painful. In that case we require that
292 * the imagelength be set properly before the first
293 * write (so that the strips array will be fully
294 * allocated above).
295 */
296 if (strip >= td->td_nstrips) {
297 if (td->td_planarconfig == PLANARCONFIG_SEPARATE) {
80ed523f
VZ
298 TIFFErrorExt(tif->tif_clientdata, module,
299 "Can not grow image by strips when using separate planes");
300 return ((tmsize_t) -1);
8414a40c
VZ
301 }
302 /*
303 * Watch out for a growing image. The value of
304 * strips/image will initially be 1 (since it
305 * can't be deduced until the imagelength is known).
306 */
307 if (strip >= td->td_stripsperimage)
308 td->td_stripsperimage =
80ed523f 309 TIFFhowmany_32(td->td_imagelength,td->td_rowsperstrip);
8414a40c 310 if (!TIFFGrowStrips(tif, 1, module))
80ed523f 311 return ((tmsize_t) -1);
8414a40c
VZ
312 }
313 tif->tif_curstrip = strip;
314 tif->tif_row = (strip % td->td_stripsperimage) * td->td_rowsperstrip;
80ed523f
VZ
315 return (TIFFAppendToStrip(tif, strip, (uint8*) data, cc) ?
316 cc : (tmsize_t) -1);
8414a40c
VZ
317}
318
319/*
320 * Write and compress a tile of data. The
321 * tile is selected by the (x,y,z,s) coordinates.
322 */
80ed523f
VZ
323tmsize_t
324TIFFWriteTile(TIFF* tif, void* buf, uint32 x, uint32 y, uint32 z, uint16 s)
8414a40c
VZ
325{
326 if (!TIFFCheckTile(tif, x, y, z, s))
80ed523f 327 return ((tmsize_t)(-1));
8414a40c
VZ
328 /*
329 * NB: A tile size of -1 is used instead of tif_tilesize knowing
330 * that TIFFWriteEncodedTile will clamp this to the tile size.
331 * This is done because the tile size may not be defined until
332 * after the output buffer is setup in TIFFWriteBufferSetup.
333 */
334 return (TIFFWriteEncodedTile(tif,
80ed523f 335 TIFFComputeTile(tif, x, y, z, s), buf, (tmsize_t)(-1)));
8414a40c
VZ
336}
337
338/*
339 * Encode the supplied data and write it to the
340 * specified tile. There must be space for the
341 * data. The function clamps individual writes
342 * to a tile to the tile size, but does not (and
343 * can not) check that multiple writes to the same
344 * tile do not write more than tile size data.
345 *
346 * NB: Image length must be setup before writing; this
347 * interface does not support automatically growing
348 * the image on each write (as TIFFWriteScanline does).
349 */
80ed523f
VZ
350tmsize_t
351TIFFWriteEncodedTile(TIFF* tif, uint32 tile, void* data, tmsize_t cc)
8414a40c
VZ
352{
353 static const char module[] = "TIFFWriteEncodedTile";
354 TIFFDirectory *td;
80ed523f 355 uint16 sample;
8414a40c
VZ
356
357 if (!WRITECHECKTILES(tif, module))
80ed523f 358 return ((tmsize_t)(-1));
8414a40c
VZ
359 td = &tif->tif_dir;
360 if (tile >= td->td_nstrips) {
80ed523f
VZ
361 TIFFErrorExt(tif->tif_clientdata, module, "Tile %lu out of range, max %lu",
362 (unsigned long) tile, (unsigned long) td->td_nstrips);
363 return ((tmsize_t)(-1));
8414a40c
VZ
364 }
365 /*
366 * Handle delayed allocation of data buffer. This
367 * permits it to be sized more intelligently (using
368 * directory information).
369 */
370 if (!BUFFERCHECK(tif))
80ed523f 371 return ((tmsize_t)(-1));
8414a40c 372
80ed523f
VZ
373 tif->tif_flags |= TIFF_BUF4WRITE;
374 tif->tif_curtile = tile;
8414a40c 375
80ed523f 376 if( td->td_stripbytecount[tile] > 0 )
8414a40c 377 {
80ed523f
VZ
378 /* Make sure that at the first attempt of rewriting the tile, we will have */
379 /* more bytes available in the output buffer than the previous byte count, */
380 /* so that TIFFAppendToStrip() will detect the overflow when it is called the first */
381 /* time if the new compressed tile is bigger than the older one. (GDAL #4771) */
382 if( tif->tif_rawdatasize <= td->td_stripbytecount[tile] )
383 {
384 if( !(TIFFWriteBufferSetup(tif, NULL,
385 (tmsize_t)TIFFroundup_64((uint64)(td->td_stripbytecount[tile] + 1), 1024))) )
386 return ((tmsize_t)(-1));
387 }
388
389 /* Force TIFFAppendToStrip() to consider placing data at end
390 of file. */
8414a40c
VZ
391 tif->tif_curoff = 0;
392 }
80ed523f
VZ
393
394 tif->tif_rawcc = 0;
395 tif->tif_rawcp = tif->tif_rawdata;
396
8414a40c
VZ
397 /*
398 * Compute tiles per row & per column to compute
399 * current row and column
400 */
80ed523f 401 tif->tif_row = (tile % TIFFhowmany_32(td->td_imagelength, td->td_tilelength))
8414a40c 402 * td->td_tilelength;
80ed523f 403 tif->tif_col = (tile % TIFFhowmany_32(td->td_imagewidth, td->td_tilewidth))
8414a40c
VZ
404 * td->td_tilewidth;
405
406 if ((tif->tif_flags & TIFF_CODERSETUP) == 0) {
407 if (!(*tif->tif_setupencode)(tif))
80ed523f 408 return ((tmsize_t)(-1));
8414a40c
VZ
409 tif->tif_flags |= TIFF_CODERSETUP;
410 }
411 tif->tif_flags &= ~TIFF_POSTENCODE;
80ed523f 412 sample = (uint16)(tile/td->td_stripsperimage);
8414a40c 413 if (!(*tif->tif_preencode)(tif, sample))
80ed523f 414 return ((tmsize_t)(-1));
8414a40c
VZ
415 /*
416 * Clamp write amount to the tile size. This is mostly
417 * done so that callers can pass in some large number
418 * (e.g. -1) and have the tile size used instead.
419 */
420 if ( cc < 1 || cc > tif->tif_tilesize)
421 cc = tif->tif_tilesize;
422
423 /* swab if needed - note that source buffer will be altered */
80ed523f 424 tif->tif_postdecode( tif, (uint8*) data, cc );
8414a40c 425
80ed523f
VZ
426 if (!(*tif->tif_encodetile)(tif, (uint8*) data, cc, sample))
427 return (0);
8414a40c 428 if (!(*tif->tif_postencode)(tif))
80ed523f 429 return ((tmsize_t)(-1));
8414a40c
VZ
430 if (!isFillOrder(tif, td->td_fillorder) &&
431 (tif->tif_flags & TIFF_NOBITREV) == 0)
80ed523f 432 TIFFReverseBits((uint8*)tif->tif_rawdata, tif->tif_rawcc);
8414a40c
VZ
433 if (tif->tif_rawcc > 0 && !TIFFAppendToStrip(tif, tile,
434 tif->tif_rawdata, tif->tif_rawcc))
80ed523f 435 return ((tmsize_t)(-1));
8414a40c
VZ
436 tif->tif_rawcc = 0;
437 tif->tif_rawcp = tif->tif_rawdata;
438 return (cc);
439}
440
441/*
442 * Write the supplied data to the specified strip.
443 * There must be space for the data; we don't check
444 * if strips overlap!
445 *
446 * NB: Image length must be setup before writing; this
447 * interface does not support automatically growing
448 * the image on each write (as TIFFWriteScanline does).
449 */
80ed523f
VZ
450tmsize_t
451TIFFWriteRawTile(TIFF* tif, uint32 tile, void* data, tmsize_t cc)
8414a40c
VZ
452{
453 static const char module[] = "TIFFWriteRawTile";
454
455 if (!WRITECHECKTILES(tif, module))
80ed523f 456 return ((tmsize_t)(-1));
8414a40c 457 if (tile >= tif->tif_dir.td_nstrips) {
80ed523f
VZ
458 TIFFErrorExt(tif->tif_clientdata, module, "Tile %lu out of range, max %lu",
459 (unsigned long) tile,
8414a40c 460 (unsigned long) tif->tif_dir.td_nstrips);
80ed523f 461 return ((tmsize_t)(-1));
8414a40c 462 }
80ed523f
VZ
463 return (TIFFAppendToStrip(tif, tile, (uint8*) data, cc) ?
464 cc : (tmsize_t)(-1));
8414a40c
VZ
465}
466
467#define isUnspecified(tif, f) \
468 (TIFFFieldSet(tif,f) && (tif)->tif_dir.td_imagelength == 0)
469
470int
471TIFFSetupStrips(TIFF* tif)
472{
473 TIFFDirectory* td = &tif->tif_dir;
474
475 if (isTiled(tif))
476 td->td_stripsperimage =
477 isUnspecified(tif, FIELD_TILEDIMENSIONS) ?
478 td->td_samplesperpixel : TIFFNumberOfTiles(tif);
479 else
480 td->td_stripsperimage =
481 isUnspecified(tif, FIELD_ROWSPERSTRIP) ?
482 td->td_samplesperpixel : TIFFNumberOfStrips(tif);
483 td->td_nstrips = td->td_stripsperimage;
484 if (td->td_planarconfig == PLANARCONFIG_SEPARATE)
485 td->td_stripsperimage /= td->td_samplesperpixel;
80ed523f
VZ
486 td->td_stripoffset = (uint64 *)
487 _TIFFmalloc(td->td_nstrips * sizeof (uint64));
488 td->td_stripbytecount = (uint64 *)
489 _TIFFmalloc(td->td_nstrips * sizeof (uint64));
8414a40c
VZ
490 if (td->td_stripoffset == NULL || td->td_stripbytecount == NULL)
491 return (0);
492 /*
493 * Place data at the end-of-file
494 * (by setting offsets to zero).
495 */
80ed523f
VZ
496 _TIFFmemset(td->td_stripoffset, 0, td->td_nstrips*sizeof (uint64));
497 _TIFFmemset(td->td_stripbytecount, 0, td->td_nstrips*sizeof (uint64));
8414a40c
VZ
498 TIFFSetFieldBit(tif, FIELD_STRIPOFFSETS);
499 TIFFSetFieldBit(tif, FIELD_STRIPBYTECOUNTS);
500 return (1);
501}
502#undef isUnspecified
503
504/*
505 * Verify file is writable and that the directory
506 * information is setup properly. In doing the latter
507 * we also "freeze" the state of the directory so
508 * that important information is not changed.
509 */
510int
511TIFFWriteCheck(TIFF* tif, int tiles, const char* module)
512{
513 if (tif->tif_mode == O_RDONLY) {
80ed523f 514 TIFFErrorExt(tif->tif_clientdata, module, "File not open for writing");
8414a40c
VZ
515 return (0);
516 }
517 if (tiles ^ isTiled(tif)) {
80ed523f 518 TIFFErrorExt(tif->tif_clientdata, module, tiles ?
8414a40c
VZ
519 "Can not write tiles to a stripped image" :
520 "Can not write scanlines to a tiled image");
521 return (0);
522 }
80ed523f
VZ
523
524 _TIFFFillStriles( tif );
8414a40c
VZ
525
526 /*
527 * On the first write verify all the required information
528 * has been setup and initialize any data structures that
529 * had to wait until directory information was set.
530 * Note that a lot of our work is assumed to remain valid
531 * because we disallow any of the important parameters
532 * from changing after we start writing (i.e. once
533 * TIFF_BEENWRITING is set, TIFFSetField will only allow
534 * the image's length to be changed).
535 */
536 if (!TIFFFieldSet(tif, FIELD_IMAGEDIMENSIONS)) {
537 TIFFErrorExt(tif->tif_clientdata, module,
80ed523f 538 "Must set \"ImageWidth\" before writing data");
8414a40c
VZ
539 return (0);
540 }
541 if (tif->tif_dir.td_samplesperpixel == 1) {
542 /*
543 * Planarconfiguration is irrelevant in case of single band
544 * images and need not be included. We will set it anyway,
545 * because this field is used in other parts of library even
546 * in the single band case.
547 */
80ed523f
VZ
548 if (!TIFFFieldSet(tif, FIELD_PLANARCONFIG))
549 tif->tif_dir.td_planarconfig = PLANARCONFIG_CONTIG;
8414a40c
VZ
550 } else {
551 if (!TIFFFieldSet(tif, FIELD_PLANARCONFIG)) {
552 TIFFErrorExt(tif->tif_clientdata, module,
80ed523f 553 "Must set \"PlanarConfiguration\" before writing data");
8414a40c
VZ
554 return (0);
555 }
556 }
557 if (tif->tif_dir.td_stripoffset == NULL && !TIFFSetupStrips(tif)) {
558 tif->tif_dir.td_nstrips = 0;
80ed523f
VZ
559 TIFFErrorExt(tif->tif_clientdata, module, "No space for %s arrays",
560 isTiled(tif) ? "tile" : "strip");
8414a40c
VZ
561 return (0);
562 }
80ed523f
VZ
563 if (isTiled(tif))
564 {
565 tif->tif_tilesize = TIFFTileSize(tif);
566 if (tif->tif_tilesize == 0)
567 return (0);
568 }
569 else
570 tif->tif_tilesize = (tmsize_t)(-1);
8414a40c 571 tif->tif_scanlinesize = TIFFScanlineSize(tif);
80ed523f
VZ
572 if (tif->tif_scanlinesize == 0)
573 return (0);
8414a40c
VZ
574 tif->tif_flags |= TIFF_BEENWRITING;
575 return (1);
576}
577
578/*
579 * Setup the raw data buffer used for encoding.
580 */
581int
80ed523f 582TIFFWriteBufferSetup(TIFF* tif, void* bp, tmsize_t size)
8414a40c
VZ
583{
584 static const char module[] = "TIFFWriteBufferSetup";
585
586 if (tif->tif_rawdata) {
587 if (tif->tif_flags & TIFF_MYBUFFER) {
588 _TIFFfree(tif->tif_rawdata);
589 tif->tif_flags &= ~TIFF_MYBUFFER;
590 }
591 tif->tif_rawdata = NULL;
592 }
80ed523f 593 if (size == (tmsize_t)(-1)) {
8414a40c
VZ
594 size = (isTiled(tif) ?
595 tif->tif_tilesize : TIFFStripSize(tif));
596 /*
597 * Make raw data buffer at least 8K
598 */
599 if (size < 8*1024)
600 size = 8*1024;
601 bp = NULL; /* NB: force malloc */
602 }
603 if (bp == NULL) {
604 bp = _TIFFmalloc(size);
605 if (bp == NULL) {
80ed523f 606 TIFFErrorExt(tif->tif_clientdata, module, "No space for output buffer");
8414a40c
VZ
607 return (0);
608 }
609 tif->tif_flags |= TIFF_MYBUFFER;
610 } else
611 tif->tif_flags &= ~TIFF_MYBUFFER;
80ed523f 612 tif->tif_rawdata = (uint8*) bp;
8414a40c
VZ
613 tif->tif_rawdatasize = size;
614 tif->tif_rawcc = 0;
615 tif->tif_rawcp = tif->tif_rawdata;
616 tif->tif_flags |= TIFF_BUFFERSETUP;
617 return (1);
618}
619
620/*
621 * Grow the strip data structures by delta strips.
622 */
623static int
80ed523f 624TIFFGrowStrips(TIFF* tif, uint32 delta, const char* module)
8414a40c 625{
80ed523f
VZ
626 TIFFDirectory *td = &tif->tif_dir;
627 uint64* new_stripoffset;
628 uint64* new_stripbytecount;
8414a40c
VZ
629
630 assert(td->td_planarconfig == PLANARCONFIG_CONTIG);
80ed523f
VZ
631 new_stripoffset = (uint64*)_TIFFrealloc(td->td_stripoffset,
632 (td->td_nstrips + delta) * sizeof (uint64));
633 new_stripbytecount = (uint64*)_TIFFrealloc(td->td_stripbytecount,
634 (td->td_nstrips + delta) * sizeof (uint64));
8414a40c
VZ
635 if (new_stripoffset == NULL || new_stripbytecount == NULL) {
636 if (new_stripoffset)
637 _TIFFfree(new_stripoffset);
638 if (new_stripbytecount)
639 _TIFFfree(new_stripbytecount);
640 td->td_nstrips = 0;
80ed523f 641 TIFFErrorExt(tif->tif_clientdata, module, "No space to expand strip arrays");
8414a40c
VZ
642 return (0);
643 }
644 td->td_stripoffset = new_stripoffset;
645 td->td_stripbytecount = new_stripbytecount;
646 _TIFFmemset(td->td_stripoffset + td->td_nstrips,
80ed523f 647 0, delta*sizeof (uint64));
8414a40c 648 _TIFFmemset(td->td_stripbytecount + td->td_nstrips,
80ed523f 649 0, delta*sizeof (uint64));
8414a40c 650 td->td_nstrips += delta;
80ed523f
VZ
651 tif->tif_flags |= TIFF_DIRTYDIRECT;
652
8414a40c
VZ
653 return (1);
654}
655
656/*
657 * Append the data to the specified strip.
658 */
659static int
80ed523f 660TIFFAppendToStrip(TIFF* tif, uint32 strip, uint8* data, tmsize_t cc)
8414a40c 661{
8414a40c 662 static const char module[] = "TIFFAppendToStrip";
80ed523f
VZ
663 TIFFDirectory *td = &tif->tif_dir;
664 uint64 m;
665 int64 old_byte_count = -1;
8414a40c
VZ
666
667 if (td->td_stripoffset[strip] == 0 || tif->tif_curoff == 0) {
80ed523f
VZ
668 assert(td->td_nstrips > 0);
669
670 if( td->td_stripbytecount[strip] != 0
671 && td->td_stripoffset[strip] != 0
672 && td->td_stripbytecount[strip] >= (uint64) cc )
673 {
674 /*
675 * There is already tile data on disk, and the new tile
676 * data we have will fit in the same space. The only
677 * aspect of this that is risky is that there could be
678 * more data to append to this strip before we are done
679 * depending on how we are getting called.
680 */
681 if (!SeekOK(tif, td->td_stripoffset[strip])) {
682 TIFFErrorExt(tif->tif_clientdata, module,
683 "Seek error at scanline %lu",
684 (unsigned long)tif->tif_row);
685 return (0);
686 }
687 }
688 else
689 {
690 /*
691 * Seek to end of file, and set that as our location to
692 * write this strip.
693 */
694 td->td_stripoffset[strip] = TIFFSeekFile(tif, 0, SEEK_END);
695 tif->tif_flags |= TIFF_DIRTYSTRIP;
696 }
697
698 tif->tif_curoff = td->td_stripoffset[strip];
699
700 /*
701 * We are starting a fresh strip/tile, so set the size to zero.
702 */
703 old_byte_count = td->td_stripbytecount[strip];
704 td->td_stripbytecount[strip] = 0;
8414a40c
VZ
705 }
706
80ed523f
VZ
707 m = tif->tif_curoff+cc;
708 if (!(tif->tif_flags&TIFF_BIGTIFF))
709 m = (uint32)m;
710 if ((m<tif->tif_curoff)||(m<(uint64)cc))
711 {
712 TIFFErrorExt(tif->tif_clientdata, module, "Maximum TIFF file size exceeded");
8414a40c
VZ
713 return (0);
714 }
80ed523f
VZ
715 if (!WriteOK(tif, data, cc)) {
716 TIFFErrorExt(tif->tif_clientdata, module, "Write error at scanline %lu",
717 (unsigned long) tif->tif_row);
718 return (0);
719 }
720 tif->tif_curoff = m;
8414a40c 721 td->td_stripbytecount[strip] += cc;
80ed523f
VZ
722
723 if( (int64) td->td_stripbytecount[strip] != old_byte_count )
724 tif->tif_flags |= TIFF_DIRTYSTRIP;
725
8414a40c
VZ
726 return (1);
727}
728
729/*
730 * Internal version of TIFFFlushData that can be
731 * called by ``encodestrip routines'' w/o concern
732 * for infinite recursion.
733 */
734int
735TIFFFlushData1(TIFF* tif)
736{
80ed523f 737 if (tif->tif_rawcc > 0 && tif->tif_flags & TIFF_BUF4WRITE ) {
8414a40c
VZ
738 if (!isFillOrder(tif, tif->tif_dir.td_fillorder) &&
739 (tif->tif_flags & TIFF_NOBITREV) == 0)
80ed523f 740 TIFFReverseBits((uint8*)tif->tif_rawdata,
8414a40c
VZ
741 tif->tif_rawcc);
742 if (!TIFFAppendToStrip(tif,
743 isTiled(tif) ? tif->tif_curtile : tif->tif_curstrip,
744 tif->tif_rawdata, tif->tif_rawcc))
745 return (0);
746 tif->tif_rawcc = 0;
747 tif->tif_rawcp = tif->tif_rawdata;
748 }
749 return (1);
750}
751
752/*
753 * Set the current write offset. This should only be
754 * used to set the offset to a known previous location
755 * (very carefully), or to 0 so that the next write gets
756 * appended to the end of the file.
757 */
758void
759TIFFSetWriteOffset(TIFF* tif, toff_t off)
760{
761 tif->tif_curoff = off;
762}
763
764/* vim: set ts=8 sts=8 sw=8 noet: */
80ed523f
VZ
765/*
766 * Local Variables:
767 * mode: c
768 * c-basic-offset: 8
769 * fill-column: 78
770 * End:
771 */