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