1 /*
2  * Copyright (c) 1997 Greg Ward Larson
3  * Copyright (c) 1997 Silicon Graphics, Inc.
4  *
5  * Permission to use, copy, modify, distribute, and sell this software and
6  * its documentation for any purpose is hereby granted without fee, provided
7  * that (i) the above copyright notices and this permission notice appear in
8  * all copies of the software and related documentation, and (ii) the names of
9  * Sam Leffler, Greg Larson and Silicon Graphics may not be used in any
10  * advertising or publicity relating to the software without the specific,
11  * prior written permission of Sam Leffler, Greg Larson and Silicon Graphics.
12  *
13  * THE SOFTWARE IS PROVIDED "AS-IS" AND WITHOUT WARRANTY OF ANY KIND,
14  * EXPRESS, IMPLIED OR OTHERWISE, INCLUDING WITHOUT LIMITATION, ANY
15  * WARRANTY OF MERCHANTABILITY OR FITNESS FOR A PARTICULAR PURPOSE.
16  *
17  * IN NO EVENT SHALL SAM LEFFLER, GREG LARSON OR SILICON GRAPHICS BE LIABLE
18  * FOR ANY SPECIAL, INCIDENTAL, INDIRECT OR CONSEQUENTIAL DAMAGES OF ANY KIND,
19  * OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS,
20  * WHETHER OR NOT ADVISED OF THE POSSIBILITY OF DAMAGE, AND ON ANY THEORY OF
21  * LIABILITY, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE
22  * OF THIS SOFTWARE.
23  */
24 
25 #include "tiffiop.h"
26 #ifdef LOGLUV_SUPPORT
27 
28 /*
29  * TIFF Library.
30  * LogLuv compression support for high dynamic range images.
31  *
32  * Contributed by Greg Larson.
33  *
34  * LogLuv image support uses the TIFF library to store 16 or 10-bit
35  * log luminance values with 8 bits each of u and v or a 14-bit index.
36  *
37  * The codec can take as input and produce as output 32-bit IEEE float values
38  * as well as 16-bit integer values.  A 16-bit luminance is interpreted
39  * as a sign bit followed by a 15-bit integer that is converted
40  * to and from a linear magnitude using the transformation:
41  *
42  *	L = 2^( (Le+.5)/256 - 64 )		# real from 15-bit
43  *
44  *	Le = floor( 256*(log2(L) + 64) )	# 15-bit from real
45  *
46  * The actual conversion to world luminance units in candelas per sq. meter
47  * requires an additional multiplier, which is stored in the TIFFTAG_STONITS.
48  * This value is usually set such that a reasonable exposure comes from
49  * clamping decoded luminances above 1 to 1 in the displayed image.
50  *
51  * The 16-bit values for u and v may be converted to real values by dividing
52  * each by 32768.  (This allows for negative values, which aren't useful as
53  * far as we know, but are left in case of future improvements in human
54  * color vision.)
55  *
56  * Conversion from (u,v), which is actually the CIE (u',v') system for
57  * you color scientists, is accomplished by the following transformation:
58  *
59  *	u = 4*x / (-2*x + 12*y + 3)
60  *	v = 9*y / (-2*x + 12*y + 3)
61  *
62  *	x = 9*u / (6*u - 16*v + 12)
63  *	y = 4*v / (6*u - 16*v + 12)
64  *
65  * This process is greatly simplified by passing 32-bit IEEE floats
66  * for each of three CIE XYZ coordinates.  The codec then takes care
67  * of conversion to and from LogLuv, though the application is still
68  * responsible for interpreting the TIFFTAG_STONITS calibration factor.
69  *
70  * By definition, a CIE XYZ vector of [1 1 1] corresponds to a neutral white
71  * point of (x,y)=(1/3,1/3).  However, most color systems assume some other
72  * white point, such as D65, and an absolute color conversion to XYZ then
73  * to another color space with a different white point may introduce an
74  * unwanted color cast to the image.  It is often desirable, therefore, to
75  * perform a white point conversion that maps the input white to [1 1 1]
76  * in XYZ, then record the original white point using the TIFFTAG_WHITEPOINT
77  * tag value.  A decoder that demands absolute color calibration may use
78  * this white point tag to get back the original colors, but usually it
79  * will be ignored and the new white point will be used instead that
80  * matches the output color space.
81  *
82  * Pixel information is compressed into one of two basic encodings, depending
83  * on the setting of the compression tag, which is one of COMPRESSION_SGILOG
84  * or COMPRESSION_SGILOG24.  For COMPRESSION_SGILOG, greyscale data is
85  * stored as:
86  *
87  *	 1       15
88  *	|-+---------------|
89  *
90  * COMPRESSION_SGILOG color data is stored as:
91  *
92  *	 1       15           8        8
93  *	|-+---------------|--------+--------|
94  *	 S       Le           ue       ve
95  *
96  * For the 24-bit COMPRESSION_SGILOG24 color format, the data is stored as:
97  *
98  *	     10           14
99  *	|----------|--------------|
100  *	     Le'          Ce
101  *
102  * There is no sign bit in the 24-bit case, and the (u,v) chromaticity is
103  * encoded as an index for optimal color resolution.  The 10 log bits are
104  * defined by the following conversions:
105  *
106  *	L = 2^((Le'+.5)/64 - 12)		# real from 10-bit
107  *
108  *	Le' = floor( 64*(log2(L) + 12) )	# 10-bit from real
109  *
110  * The 10 bits of the smaller format may be converted into the 15 bits of
111  * the larger format by multiplying by 4 and adding 13314.  Obviously,
112  * a smaller range of magnitudes is covered (about 5 orders of magnitude
113  * instead of 38), and the lack of a sign bit means that negative luminances
114  * are not allowed.  (Well, they aren't allowed in the real world, either,
115  * but they are useful for certain types of image processing.)
116  *
117  * The desired user format is controlled by the setting the internal
118  * pseudo tag TIFFTAG_SGILOGDATAFMT to one of:
119  *  SGILOGDATAFMT_FLOAT       = IEEE 32-bit float XYZ values
120  *  SGILOGDATAFMT_16BIT	      = 16-bit integer encodings of logL, u and v
121  * Raw data i/o is also possible using:
122  *  SGILOGDATAFMT_RAW         = 32-bit unsigned integer with encoded pixel
123  * In addition, the following decoding is provided for ease of display:
124  *  SGILOGDATAFMT_8BIT        = 8-bit default RGB gamma-corrected values
125  *
126  * For grayscale images, we provide the following data formats:
127  *  SGILOGDATAFMT_FLOAT       = IEEE 32-bit float Y values
128  *  SGILOGDATAFMT_16BIT       = 16-bit integer w/ encoded luminance
129  *  SGILOGDATAFMT_8BIT        = 8-bit gray monitor values
130  *
131  * Note that the COMPRESSION_SGILOG applies a simple run-length encoding
132  * scheme by separating the logL, u and v bytes for each row and applying
133  * a PackBits type of compression.  Since the 24-bit encoding is not
134  * adaptive, the 32-bit color format takes less space in many cases.
135  *
136  * Further control is provided over the conversion from higher-resolution
137  * formats to final encoded values through the pseudo tag
138  * TIFFTAG_SGILOGENCODE:
139  *  SGILOGENCODE_NODITHER     = do not dither encoded values
140  *  SGILOGENCODE_RANDITHER    = apply random dithering during encoding
141  *
142  * The default value of this tag is SGILOGENCODE_NODITHER for
143  * COMPRESSION_SGILOG to maximize run-length encoding and
144  * SGILOGENCODE_RANDITHER for COMPRESSION_SGILOG24 to turn
145  * quantization errors into noise.
146  */
147 
148 #include <stdio.h>
149 #include <stdlib.h>
150 #include <math.h>
151 
152 /*
153  * State block for each open TIFF
154  * file using LogLuv compression/decompression.
155  */
156 typedef struct logLuvState LogLuvState;
157 
158 struct logLuvState {
159         int                     encoder_state;  /* 1 if encoder correctly initialized */
160 	int                     user_datafmt;   /* user data format */
161 	int                     encode_meth;    /* encoding method */
162 	int                     pixel_size;     /* bytes per pixel */
163 
164 	uint8*                  tbuf;           /* translation buffer */
165 	tmsize_t                tbuflen;        /* buffer length */
166 	void (*tfunc)(LogLuvState*, uint8*, tmsize_t);
167 
168 	TIFFVSetMethod          vgetparent;     /* super-class method */
169 	TIFFVSetMethod          vsetparent;     /* super-class method */
170 };
171 
172 #define DecoderState(tif)	((LogLuvState*) (tif)->tif_data)
173 #define EncoderState(tif)	((LogLuvState*) (tif)->tif_data)
174 
175 #define SGILOGDATAFMT_UNKNOWN -1
176 
177 #define MINRUN 4 /* minimum run length */
178 
179 /*
180  * Decode a string of 16-bit gray pixels.
181  */
182 static int
LogL16Decode(TIFF * tif,uint8 * op,tmsize_t occ,uint16 s)183 LogL16Decode(TIFF* tif, uint8* op, tmsize_t occ, uint16 s)
184 {
185 	static const char module[] = "LogL16Decode";
186 	LogLuvState* sp = DecoderState(tif);
187 	int shft;
188 	tmsize_t i;
189 	tmsize_t npixels;
190 	unsigned char* bp;
191 	int16* tp;
192 	int16 b;
193 	tmsize_t cc;
194 	int rc;
195 
196 	assert(s == 0);
197 	assert(sp != NULL);
198 
199 	npixels = occ / sp->pixel_size;
200 
201 	if (sp->user_datafmt == SGILOGDATAFMT_16BIT)
202 		tp = (int16*) op;
203 	else {
204 		if(sp->tbuflen < npixels) {
205 			TIFFErrorExt(tif->tif_clientdata, module,
206 						 "Translation buffer too short");
207 			return (0);
208 		}
209 		tp = (int16*) sp->tbuf;
210 	}
211 	_TIFFmemset((void*) tp, 0, npixels*sizeof (tp[0]));
212 
213 	bp = (unsigned char*) tif->tif_rawcp;
214 	cc = tif->tif_rawcc;
215 	/* get each byte string */
216 	for (shft = 8; shft >= 0; shft -=8) {
217 		for (i = 0; i < npixels && cc > 0; ) {
218 			if (*bp >= 128) {		/* run */
219 				if( cc < 2 )
220 					break;
221 				rc = *bp++ + (2-128);
222 				b = (int16)(*bp++ << shft);
223 				cc -= 2;
224 				while (rc-- && i < npixels)
225 					tp[i++] |= b;
226 			} else {			/* non-run */
227 				rc = *bp++;		/* nul is noop */
228 				while (--cc && rc-- && i < npixels)
229 					tp[i++] |= (int16)*bp++ << shft;
230 			}
231 		}
232 		if (i != npixels) {
233 #if defined(__WIN32__) && (defined(_MSC_VER) || defined(__MINGW32__))
234 			TIFFErrorExt(tif->tif_clientdata, module,
235 			    "Not enough data at row %lu (short %I64d pixels)",
236 				     (unsigned long) tif->tif_row,
237 				     (unsigned __int64) (npixels - i));
238 #else
239 			TIFFErrorExt(tif->tif_clientdata, module,
240 			    "Not enough data at row %lu (short %llu pixels)",
241 				     (unsigned long) tif->tif_row,
242 				     (unsigned long long) (npixels - i));
243 #endif
244 			tif->tif_rawcp = (uint8*) bp;
245 			tif->tif_rawcc = cc;
246 			return (0);
247 		}
248 	}
249 	(*sp->tfunc)(sp, op, npixels);
250 	tif->tif_rawcp = (uint8*) bp;
251 	tif->tif_rawcc = cc;
252 	return (1);
253 }
254 
255 /*
256  * Decode a string of 24-bit pixels.
257  */
258 static int
LogLuvDecode24(TIFF * tif,uint8 * op,tmsize_t occ,uint16 s)259 LogLuvDecode24(TIFF* tif, uint8* op, tmsize_t occ, uint16 s)
260 {
261 	static const char module[] = "LogLuvDecode24";
262 	LogLuvState* sp = DecoderState(tif);
263 	tmsize_t cc;
264 	tmsize_t i;
265 	tmsize_t npixels;
266 	unsigned char* bp;
267 	uint32* tp;
268 
269 	assert(s == 0);
270 	assert(sp != NULL);
271 
272 	npixels = occ / sp->pixel_size;
273 
274 	if (sp->user_datafmt == SGILOGDATAFMT_RAW)
275 		tp = (uint32 *)op;
276 	else {
277 		if(sp->tbuflen < npixels) {
278 			TIFFErrorExt(tif->tif_clientdata, module,
279 						 "Translation buffer too short");
280 			return (0);
281 		}
282 		tp = (uint32 *) sp->tbuf;
283 	}
284 	/* copy to array of uint32 */
285 	bp = (unsigned char*) tif->tif_rawcp;
286 	cc = tif->tif_rawcc;
287 	for (i = 0; i < npixels && cc >= 3; i++) {
288 		tp[i] = bp[0] << 16 | bp[1] << 8 | bp[2];
289 		bp += 3;
290 		cc -= 3;
291 	}
292 	tif->tif_rawcp = (uint8*) bp;
293 	tif->tif_rawcc = cc;
294 	if (i != npixels) {
295 #if defined(__WIN32__) && (defined(_MSC_VER) || defined(__MINGW32__))
296 		TIFFErrorExt(tif->tif_clientdata, module,
297 			"Not enough data at row %lu (short %I64d pixels)",
298 			     (unsigned long) tif->tif_row,
299 			     (unsigned __int64) (npixels - i));
300 #else
301 		TIFFErrorExt(tif->tif_clientdata, module,
302 			"Not enough data at row %lu (short %llu pixels)",
303 			     (unsigned long) tif->tif_row,
304 			     (unsigned long long) (npixels - i));
305 #endif
306 		return (0);
307 	}
308 	(*sp->tfunc)(sp, op, npixels);
309 	return (1);
310 }
311 
312 /*
313  * Decode a string of 32-bit pixels.
314  */
315 static int
LogLuvDecode32(TIFF * tif,uint8 * op,tmsize_t occ,uint16 s)316 LogLuvDecode32(TIFF* tif, uint8* op, tmsize_t occ, uint16 s)
317 {
318 	static const char module[] = "LogLuvDecode32";
319 	LogLuvState* sp;
320 	int shft;
321 	tmsize_t i;
322 	tmsize_t npixels;
323 	unsigned char* bp;
324 	uint32* tp;
325 	uint32 b;
326 	tmsize_t cc;
327 	int rc;
328 
329 	assert(s == 0);
330 	sp = DecoderState(tif);
331 	assert(sp != NULL);
332 
333 	npixels = occ / sp->pixel_size;
334 
335 	if (sp->user_datafmt == SGILOGDATAFMT_RAW)
336 		tp = (uint32*) op;
337 	else {
338 		if(sp->tbuflen < npixels) {
339 			TIFFErrorExt(tif->tif_clientdata, module,
340 						 "Translation buffer too short");
341 			return (0);
342 		}
343 		tp = (uint32*) sp->tbuf;
344 	}
345 	_TIFFmemset((void*) tp, 0, npixels*sizeof (tp[0]));
346 
347 	bp = (unsigned char*) tif->tif_rawcp;
348 	cc = tif->tif_rawcc;
349 	/* get each byte string */
350 	for (shft = 24; shft >= 0; shft -=8) {
351 		for (i = 0; i < npixels && cc > 0; ) {
352 			if (*bp >= 128) {		/* run */
353 				if( cc < 2 )
354 					break;
355 				rc = *bp++ + (2-128);
356 				b = (uint32)*bp++ << shft;
357 				cc -= 2;
358 				while (rc-- && i < npixels)
359 					tp[i++] |= b;
360 			} else {			/* non-run */
361 				rc = *bp++;		/* nul is noop */
362 				while (--cc && rc-- && i < npixels)
363 					tp[i++] |= (uint32)*bp++ << shft;
364 			}
365 		}
366 		if (i != npixels) {
367 #if defined(__WIN32__) && (defined(_MSC_VER) || defined(__MINGW32__))
368 			TIFFErrorExt(tif->tif_clientdata, module,
369 			"Not enough data at row %lu (short %I64d pixels)",
370 				     (unsigned long) tif->tif_row,
371 				     (unsigned __int64) (npixels - i));
372 #else
373 			TIFFErrorExt(tif->tif_clientdata, module,
374 			"Not enough data at row %lu (short %llu pixels)",
375 				     (unsigned long) tif->tif_row,
376 				     (unsigned long long) (npixels - i));
377 #endif
378 			tif->tif_rawcp = (uint8*) bp;
379 			tif->tif_rawcc = cc;
380 			return (0);
381 		}
382 	}
383 	(*sp->tfunc)(sp, op, npixels);
384 	tif->tif_rawcp = (uint8*) bp;
385 	tif->tif_rawcc = cc;
386 	return (1);
387 }
388 
389 /*
390  * Decode a strip of pixels.  We break it into rows to
391  * maintain synchrony with the encode algorithm, which
392  * is row by row.
393  */
394 static int
LogLuvDecodeStrip(TIFF * tif,uint8 * bp,tmsize_t cc,uint16 s)395 LogLuvDecodeStrip(TIFF* tif, uint8* bp, tmsize_t cc, uint16 s)
396 {
397 	tmsize_t rowlen = TIFFScanlineSize(tif);
398 
399         if (rowlen == 0)
400                 return 0;
401 
402 	assert(cc%rowlen == 0);
403 	while (cc && (*tif->tif_decoderow)(tif, bp, rowlen, s)) {
404 		bp += rowlen;
405 		cc -= rowlen;
406 	}
407 	return (cc == 0);
408 }
409 
410 /*
411  * Decode a tile of pixels.  We break it into rows to
412  * maintain synchrony with the encode algorithm, which
413  * is row by row.
414  */
415 static int
LogLuvDecodeTile(TIFF * tif,uint8 * bp,tmsize_t cc,uint16 s)416 LogLuvDecodeTile(TIFF* tif, uint8* bp, tmsize_t cc, uint16 s)
417 {
418 	tmsize_t rowlen = TIFFTileRowSize(tif);
419 
420         if (rowlen == 0)
421                 return 0;
422 
423 	assert(cc%rowlen == 0);
424 	while (cc && (*tif->tif_decoderow)(tif, bp, rowlen, s)) {
425 		bp += rowlen;
426 		cc -= rowlen;
427 	}
428 	return (cc == 0);
429 }
430 
431 /*
432  * Encode a row of 16-bit pixels.
433  */
434 static int
LogL16Encode(TIFF * tif,uint8 * bp,tmsize_t cc,uint16 s)435 LogL16Encode(TIFF* tif, uint8* bp, tmsize_t cc, uint16 s)
436 {
437 	static const char module[] = "LogL16Encode";
438 	LogLuvState* sp = EncoderState(tif);
439 	int shft;
440 	tmsize_t i;
441 	tmsize_t j;
442 	tmsize_t npixels;
443 	uint8* op;
444 	int16* tp;
445 	int16 b;
446 	tmsize_t occ;
447 	int rc=0, mask;
448 	tmsize_t beg;
449 
450 	assert(s == 0);
451 	assert(sp != NULL);
452 	npixels = cc / sp->pixel_size;
453 
454 	if (sp->user_datafmt == SGILOGDATAFMT_16BIT)
455 		tp = (int16*) bp;
456 	else {
457 		tp = (int16*) sp->tbuf;
458 		if(sp->tbuflen < npixels) {
459 			TIFFErrorExt(tif->tif_clientdata, module,
460 						 "Translation buffer too short");
461 			return (0);
462 		}
463 		(*sp->tfunc)(sp, bp, npixels);
464 	}
465 	/* compress each byte string */
466 	op = tif->tif_rawcp;
467 	occ = tif->tif_rawdatasize - tif->tif_rawcc;
468 	for (shft = 8; shft >= 0; shft -=8) {
469 		for (i = 0; i < npixels; i += rc) {
470 			if (occ < 4) {
471 				tif->tif_rawcp = op;
472 				tif->tif_rawcc = tif->tif_rawdatasize - occ;
473 				if (!TIFFFlushData1(tif))
474 					return (0);
475 				op = tif->tif_rawcp;
476 				occ = tif->tif_rawdatasize - tif->tif_rawcc;
477 			}
478 			mask = 0xff << shft;		/* find next run */
479 			for (beg = i; beg < npixels; beg += rc) {
480 				b = (int16) (tp[beg] & mask);
481 				rc = 1;
482 				while (rc < 127+2 && beg+rc < npixels &&
483 				    (tp[beg+rc] & mask) == b)
484 					rc++;
485 				if (rc >= MINRUN)
486 					break;		/* long enough */
487 			}
488 			if (beg-i > 1 && beg-i < MINRUN) {
489 				b = (int16) (tp[i] & mask);/*check short run */
490 				j = i+1;
491 				while ((tp[j++] & mask) == b)
492 					if (j == beg) {
493 						*op++ = (uint8)(128-2+j-i);
494 						*op++ = (uint8)(b >> shft);
495 						occ -= 2;
496 						i = beg;
497 						break;
498 					}
499 			}
500 			while (i < beg) {		/* write out non-run */
501 				if ((j = beg-i) > 127) j = 127;
502 				if (occ < j+3) {
503 					tif->tif_rawcp = op;
504 					tif->tif_rawcc = tif->tif_rawdatasize - occ;
505 					if (!TIFFFlushData1(tif))
506 						return (0);
507 					op = tif->tif_rawcp;
508 					occ = tif->tif_rawdatasize - tif->tif_rawcc;
509 				}
510 				*op++ = (uint8) j; occ--;
511 				while (j--) {
512 					*op++ = (uint8) (tp[i++] >> shft & 0xff);
513 					occ--;
514 				}
515 			}
516 			if (rc >= MINRUN) {		/* write out run */
517 				*op++ = (uint8) (128-2+rc);
518 				*op++ = (uint8) (tp[beg] >> shft & 0xff);
519 				occ -= 2;
520 			} else
521 				rc = 0;
522 		}
523 	}
524 	tif->tif_rawcp = op;
525 	tif->tif_rawcc = tif->tif_rawdatasize - occ;
526 
527 	return (1);
528 }
529 
530 /*
531  * Encode a row of 24-bit pixels.
532  */
533 static int
LogLuvEncode24(TIFF * tif,uint8 * bp,tmsize_t cc,uint16 s)534 LogLuvEncode24(TIFF* tif, uint8* bp, tmsize_t cc, uint16 s)
535 {
536 	static const char module[] = "LogLuvEncode24";
537 	LogLuvState* sp = EncoderState(tif);
538 	tmsize_t i;
539 	tmsize_t npixels;
540 	tmsize_t occ;
541 	uint8* op;
542 	uint32* tp;
543 
544 	assert(s == 0);
545 	assert(sp != NULL);
546 	npixels = cc / sp->pixel_size;
547 
548 	if (sp->user_datafmt == SGILOGDATAFMT_RAW)
549 		tp = (uint32*) bp;
550 	else {
551 		tp = (uint32*) sp->tbuf;
552 		if(sp->tbuflen < npixels) {
553 			TIFFErrorExt(tif->tif_clientdata, module,
554 						 "Translation buffer too short");
555 			return (0);
556 		}
557 		(*sp->tfunc)(sp, bp, npixels);
558 	}
559 	/* write out encoded pixels */
560 	op = tif->tif_rawcp;
561 	occ = tif->tif_rawdatasize - tif->tif_rawcc;
562 	for (i = npixels; i--; ) {
563 		if (occ < 3) {
564 			tif->tif_rawcp = op;
565 			tif->tif_rawcc = tif->tif_rawdatasize - occ;
566 			if (!TIFFFlushData1(tif))
567 				return (0);
568 			op = tif->tif_rawcp;
569 			occ = tif->tif_rawdatasize - tif->tif_rawcc;
570 		}
571 		*op++ = (uint8)(*tp >> 16);
572 		*op++ = (uint8)(*tp >> 8 & 0xff);
573 		*op++ = (uint8)(*tp++ & 0xff);
574 		occ -= 3;
575 	}
576 	tif->tif_rawcp = op;
577 	tif->tif_rawcc = tif->tif_rawdatasize - occ;
578 
579 	return (1);
580 }
581 
582 /*
583  * Encode a row of 32-bit pixels.
584  */
585 static int
LogLuvEncode32(TIFF * tif,uint8 * bp,tmsize_t cc,uint16 s)586 LogLuvEncode32(TIFF* tif, uint8* bp, tmsize_t cc, uint16 s)
587 {
588 	static const char module[] = "LogLuvEncode32";
589 	LogLuvState* sp = EncoderState(tif);
590 	int shft;
591 	tmsize_t i;
592 	tmsize_t j;
593 	tmsize_t npixels;
594 	uint8* op;
595 	uint32* tp;
596 	uint32 b;
597 	tmsize_t occ;
598 	int rc=0, mask;
599 	tmsize_t beg;
600 
601 	assert(s == 0);
602 	assert(sp != NULL);
603 
604 	npixels = cc / sp->pixel_size;
605 
606 	if (sp->user_datafmt == SGILOGDATAFMT_RAW)
607 		tp = (uint32*) bp;
608 	else {
609 		tp = (uint32*) sp->tbuf;
610 		if(sp->tbuflen < npixels) {
611 			TIFFErrorExt(tif->tif_clientdata, module,
612 						 "Translation buffer too short");
613 			return (0);
614 		}
615 		(*sp->tfunc)(sp, bp, npixels);
616 	}
617 	/* compress each byte string */
618 	op = tif->tif_rawcp;
619 	occ = tif->tif_rawdatasize - tif->tif_rawcc;
620 	for (shft = 24; shft >= 0; shft -=8) {
621 		for (i = 0; i < npixels; i += rc) {
622 			if (occ < 4) {
623 				tif->tif_rawcp = op;
624 				tif->tif_rawcc = tif->tif_rawdatasize - occ;
625 				if (!TIFFFlushData1(tif))
626 					return (0);
627 				op = tif->tif_rawcp;
628 				occ = tif->tif_rawdatasize - tif->tif_rawcc;
629 			}
630 			mask = 0xff << shft;		/* find next run */
631 			for (beg = i; beg < npixels; beg += rc) {
632 				b = tp[beg] & mask;
633 				rc = 1;
634 				while (rc < 127+2 && beg+rc < npixels &&
635 						(tp[beg+rc] & mask) == b)
636 					rc++;
637 				if (rc >= MINRUN)
638 					break;		/* long enough */
639 			}
640 			if (beg-i > 1 && beg-i < MINRUN) {
641 				b = tp[i] & mask;	/* check short run */
642 				j = i+1;
643 				while ((tp[j++] & mask) == b)
644 					if (j == beg) {
645 						*op++ = (uint8)(128-2+j-i);
646 						*op++ = (uint8)(b >> shft);
647 						occ -= 2;
648 						i = beg;
649 						break;
650 					}
651 			}
652 			while (i < beg) {		/* write out non-run */
653 				if ((j = beg-i) > 127) j = 127;
654 				if (occ < j+3) {
655 					tif->tif_rawcp = op;
656 					tif->tif_rawcc = tif->tif_rawdatasize - occ;
657 					if (!TIFFFlushData1(tif))
658 						return (0);
659 					op = tif->tif_rawcp;
660 					occ = tif->tif_rawdatasize - tif->tif_rawcc;
661 				}
662 				*op++ = (uint8) j; occ--;
663 				while (j--) {
664 					*op++ = (uint8)(tp[i++] >> shft & 0xff);
665 					occ--;
666 				}
667 			}
668 			if (rc >= MINRUN) {		/* write out run */
669 				*op++ = (uint8) (128-2+rc);
670 				*op++ = (uint8)(tp[beg] >> shft & 0xff);
671 				occ -= 2;
672 			} else
673 				rc = 0;
674 		}
675 	}
676 	tif->tif_rawcp = op;
677 	tif->tif_rawcc = tif->tif_rawdatasize - occ;
678 
679 	return (1);
680 }
681 
682 /*
683  * Encode a strip of pixels.  We break it into rows to
684  * avoid encoding runs across row boundaries.
685  */
686 static int
LogLuvEncodeStrip(TIFF * tif,uint8 * bp,tmsize_t cc,uint16 s)687 LogLuvEncodeStrip(TIFF* tif, uint8* bp, tmsize_t cc, uint16 s)
688 {
689 	tmsize_t rowlen = TIFFScanlineSize(tif);
690 
691         if (rowlen == 0)
692                 return 0;
693 
694 	assert(cc%rowlen == 0);
695 	while (cc && (*tif->tif_encoderow)(tif, bp, rowlen, s) == 1) {
696 		bp += rowlen;
697 		cc -= rowlen;
698 	}
699 	return (cc == 0);
700 }
701 
702 /*
703  * Encode a tile of pixels.  We break it into rows to
704  * avoid encoding runs across row boundaries.
705  */
706 static int
LogLuvEncodeTile(TIFF * tif,uint8 * bp,tmsize_t cc,uint16 s)707 LogLuvEncodeTile(TIFF* tif, uint8* bp, tmsize_t cc, uint16 s)
708 {
709 	tmsize_t rowlen = TIFFTileRowSize(tif);
710 
711         if (rowlen == 0)
712                 return 0;
713 
714 	assert(cc%rowlen == 0);
715 	while (cc && (*tif->tif_encoderow)(tif, bp, rowlen, s) == 1) {
716 		bp += rowlen;
717 		cc -= rowlen;
718 	}
719 	return (cc == 0);
720 }
721 
722 /*
723  * Encode/Decode functions for converting to and from user formats.
724  */
725 
726 #include "uvcode.h"
727 
728 #ifndef UVSCALE
729 #define U_NEU		0.210526316
730 #define V_NEU		0.473684211
731 #define UVSCALE		410.
732 #endif
733 
734 #ifndef	M_LN2
735 #define M_LN2		0.69314718055994530942
736 #endif
737 #ifndef M_PI
738 #define M_PI		3.14159265358979323846
739 #endif
740 #undef log2 /* Conflict with C'99 function */
741 #define log2(x)		((1./M_LN2)*log(x))
742 #undef exp2  /* Conflict with C'99 function */
743 #define exp2(x)		exp(M_LN2*(x))
744 
745 #define itrunc(x,m)	((m)==SGILOGENCODE_NODITHER ? \
746 				(int)(x) : \
747 				(int)((x) + rand()*(1./RAND_MAX) - .5))
748 
749 #if !LOGLUV_PUBLIC
750 static
751 #endif
752 double
LogL16toY(int p16)753 LogL16toY(int p16)		/* compute luminance from 16-bit LogL */
754 {
755 	int	Le = p16 & 0x7fff;
756 	double	Y;
757 
758 	if (!Le)
759 		return (0.);
760 	Y = exp(M_LN2/256.*(Le+.5) - M_LN2*64.);
761 	return (!(p16 & 0x8000) ? Y : -Y);
762 }
763 
764 #if !LOGLUV_PUBLIC
765 static
766 #endif
767 int
LogL16fromY(double Y,int em)768 LogL16fromY(double Y, int em)	/* get 16-bit LogL from Y */
769 {
770 	if (Y >= 1.8371976e19)
771 		return (0x7fff);
772 	if (Y <= -1.8371976e19)
773 		return (0xffff);
774 	if (Y > 5.4136769e-20)
775 		return itrunc(256.*(log2(Y) + 64.), em);
776 	if (Y < -5.4136769e-20)
777 		return (~0x7fff | itrunc(256.*(log2(-Y) + 64.), em));
778 	return (0);
779 }
780 
781 static void
L16toY(LogLuvState * sp,uint8 * op,tmsize_t n)782 L16toY(LogLuvState* sp, uint8* op, tmsize_t n)
783 {
784 	int16* l16 = (int16*) sp->tbuf;
785 	float* yp = (float*) op;
786 
787 	while (n-- > 0)
788 		*yp++ = (float)LogL16toY(*l16++);
789 }
790 
791 static void
L16toGry(LogLuvState * sp,uint8 * op,tmsize_t n)792 L16toGry(LogLuvState* sp, uint8* op, tmsize_t n)
793 {
794 	int16* l16 = (int16*) sp->tbuf;
795 	uint8* gp = (uint8*) op;
796 
797 	while (n-- > 0) {
798 		double Y = LogL16toY(*l16++);
799 		*gp++ = (uint8) ((Y <= 0.) ? 0 : (Y >= 1.) ? 255 : (int)(256.*sqrt(Y)));
800 	}
801 }
802 
803 static void
L16fromY(LogLuvState * sp,uint8 * op,tmsize_t n)804 L16fromY(LogLuvState* sp, uint8* op, tmsize_t n)
805 {
806 	int16* l16 = (int16*) sp->tbuf;
807 	float* yp = (float*) op;
808 
809 	while (n-- > 0)
810 		*l16++ = (int16) (LogL16fromY(*yp++, sp->encode_meth));
811 }
812 
813 #if !LOGLUV_PUBLIC
814 static
815 #endif
816 void
XYZtoRGB24(float xyz[3],uint8 rgb[3])817 XYZtoRGB24(float xyz[3], uint8 rgb[3])
818 {
819 	double	r, g, b;
820 					/* assume CCIR-709 primaries */
821 	r =  2.690*xyz[0] + -1.276*xyz[1] + -0.414*xyz[2];
822 	g = -1.022*xyz[0] +  1.978*xyz[1] +  0.044*xyz[2];
823 	b =  0.061*xyz[0] + -0.224*xyz[1] +  1.163*xyz[2];
824 					/* assume 2.0 gamma for speed */
825 	/* could use integer sqrt approx., but this is probably faster */
826 	rgb[0] = (uint8)((r<=0.) ? 0 : (r >= 1.) ? 255 : (int)(256.*sqrt(r)));
827 	rgb[1] = (uint8)((g<=0.) ? 0 : (g >= 1.) ? 255 : (int)(256.*sqrt(g)));
828 	rgb[2] = (uint8)((b<=0.) ? 0 : (b >= 1.) ? 255 : (int)(256.*sqrt(b)));
829 }
830 
831 #if !LOGLUV_PUBLIC
832 static
833 #endif
834 double
LogL10toY(int p10)835 LogL10toY(int p10)		/* compute luminance from 10-bit LogL */
836 {
837 	if (p10 == 0)
838 		return (0.);
839 	return (exp(M_LN2/64.*(p10+.5) - M_LN2*12.));
840 }
841 
842 #if !LOGLUV_PUBLIC
843 static
844 #endif
845 int
LogL10fromY(double Y,int em)846 LogL10fromY(double Y, int em)	/* get 10-bit LogL from Y */
847 {
848 	if (Y >= 15.742)
849 		return (0x3ff);
850 	else if (Y <= .00024283)
851 		return (0);
852 	else
853 		return itrunc(64.*(log2(Y) + 12.), em);
854 }
855 
856 #define NANGLES		100
857 #define uv2ang(u, v)	( (NANGLES*.499999999/M_PI) \
858 				* atan2((v)-V_NEU,(u)-U_NEU) + .5*NANGLES )
859 
860 static int
oog_encode(double u,double v)861 oog_encode(double u, double v)		/* encode out-of-gamut chroma */
862 {
863 	static int	oog_table[NANGLES];
864 	static int	initialized = 0;
865 	register int	i;
866 
867 	if (!initialized) {		/* set up perimeter table */
868 		double	eps[NANGLES], ua, va, ang, epsa;
869 		int	ui, vi, ustep;
870 		for (i = NANGLES; i--; )
871 			eps[i] = 2.;
872 		for (vi = UV_NVS; vi--; ) {
873 			va = UV_VSTART + (vi+.5)*UV_SQSIZ;
874 			ustep = uv_row[vi].nus-1;
875 			if (vi == UV_NVS-1 || vi == 0 || ustep <= 0)
876 				ustep = 1;
877 			for (ui = uv_row[vi].nus-1; ui >= 0; ui -= ustep) {
878 				ua = uv_row[vi].ustart + (ui+.5)*UV_SQSIZ;
879 				ang = uv2ang(ua, va);
880 				i = (int) ang;
881 				epsa = fabs(ang - (i+.5));
882 				if (epsa < eps[i]) {
883 					oog_table[i] = uv_row[vi].ncum + ui;
884 					eps[i] = epsa;
885 				}
886 			}
887 		}
888 		for (i = NANGLES; i--; )	/* fill any holes */
889 			if (eps[i] > 1.5) {
890 				int	i1, i2;
891 				for (i1 = 1; i1 < NANGLES/2; i1++)
892 					if (eps[(i+i1)%NANGLES] < 1.5)
893 						break;
894 				for (i2 = 1; i2 < NANGLES/2; i2++)
895 					if (eps[(i+NANGLES-i2)%NANGLES] < 1.5)
896 						break;
897 				if (i1 < i2)
898 					oog_table[i] =
899 						oog_table[(i+i1)%NANGLES];
900 				else
901 					oog_table[i] =
902 						oog_table[(i+NANGLES-i2)%NANGLES];
903 			}
904 		initialized = 1;
905 	}
906 	i = (int) uv2ang(u, v);		/* look up hue angle */
907 	return (oog_table[i]);
908 }
909 
910 #undef uv2ang
911 #undef NANGLES
912 
913 #if !LOGLUV_PUBLIC
914 static
915 #endif
916 int
uv_encode(double u,double v,int em)917 uv_encode(double u, double v, int em)	/* encode (u',v') coordinates */
918 {
919 	register int	vi, ui;
920 
921 	if (v < UV_VSTART)
922 		return oog_encode(u, v);
923 	vi = itrunc((v - UV_VSTART)*(1./UV_SQSIZ), em);
924 	if (vi >= UV_NVS)
925 		return oog_encode(u, v);
926 	if (u < uv_row[vi].ustart)
927 		return oog_encode(u, v);
928 	ui = itrunc((u - uv_row[vi].ustart)*(1./UV_SQSIZ), em);
929 	if (ui >= uv_row[vi].nus)
930 		return oog_encode(u, v);
931 
932 	return (uv_row[vi].ncum + ui);
933 }
934 
935 #if !LOGLUV_PUBLIC
936 static
937 #endif
938 int
uv_decode(double * up,double * vp,int c)939 uv_decode(double *up, double *vp, int c)	/* decode (u',v') index */
940 {
941 	int	upper, lower;
942 	register int	ui, vi;
943 
944 	if (c < 0 || c >= UV_NDIVS)
945 		return (-1);
946 	lower = 0;				/* binary search */
947 	upper = UV_NVS;
948 	while (upper - lower > 1) {
949 		vi = (lower + upper) >> 1;
950 		ui = c - uv_row[vi].ncum;
951 		if (ui > 0)
952 			lower = vi;
953 		else if (ui < 0)
954 			upper = vi;
955 		else {
956 			lower = vi;
957 			break;
958 		}
959 	}
960 	vi = lower;
961 	ui = c - uv_row[vi].ncum;
962 	*up = uv_row[vi].ustart + (ui+.5)*UV_SQSIZ;
963 	*vp = UV_VSTART + (vi+.5)*UV_SQSIZ;
964 	return (0);
965 }
966 
967 #if !LOGLUV_PUBLIC
968 static
969 #endif
970 void
LogLuv24toXYZ(uint32 p,float XYZ[3])971 LogLuv24toXYZ(uint32 p, float XYZ[3])
972 {
973 	int	Ce;
974 	double	L, u, v, s, x, y;
975 					/* decode luminance */
976 	L = LogL10toY(p>>14 & 0x3ff);
977 	if (L <= 0.) {
978 		XYZ[0] = XYZ[1] = XYZ[2] = 0.;
979 		return;
980 	}
981 					/* decode color */
982 	Ce = p & 0x3fff;
983 	if (uv_decode(&u, &v, Ce) < 0) {
984 		u = U_NEU; v = V_NEU;
985 	}
986 	s = 1./(6.*u - 16.*v + 12.);
987 	x = 9.*u * s;
988 	y = 4.*v * s;
989 					/* convert to XYZ */
990 	XYZ[0] = (float)(x/y * L);
991 	XYZ[1] = (float)L;
992 	XYZ[2] = (float)((1.-x-y)/y * L);
993 }
994 
995 #if !LOGLUV_PUBLIC
996 static
997 #endif
998 uint32
LogLuv24fromXYZ(float XYZ[3],int em)999 LogLuv24fromXYZ(float XYZ[3], int em)
1000 {
1001 	int	Le, Ce;
1002 	double	u, v, s;
1003 					/* encode luminance */
1004 	Le = LogL10fromY(XYZ[1], em);
1005 					/* encode color */
1006 	s = XYZ[0] + 15.*XYZ[1] + 3.*XYZ[2];
1007 	if (!Le || s <= 0.) {
1008 		u = U_NEU;
1009 		v = V_NEU;
1010 	} else {
1011 		u = 4.*XYZ[0] / s;
1012 		v = 9.*XYZ[1] / s;
1013 	}
1014 	Ce = uv_encode(u, v, em);
1015 	if (Ce < 0)			/* never happens */
1016 		Ce = uv_encode(U_NEU, V_NEU, SGILOGENCODE_NODITHER);
1017 					/* combine encodings */
1018 	return (Le << 14 | Ce);
1019 }
1020 
1021 static void
Luv24toXYZ(LogLuvState * sp,uint8 * op,tmsize_t n)1022 Luv24toXYZ(LogLuvState* sp, uint8* op, tmsize_t n)
1023 {
1024 	uint32* luv = (uint32*) sp->tbuf;
1025 	float* xyz = (float*) op;
1026 
1027 	while (n-- > 0) {
1028 		LogLuv24toXYZ(*luv, xyz);
1029 		xyz += 3;
1030 		luv++;
1031 	}
1032 }
1033 
1034 static void
Luv24toLuv48(LogLuvState * sp,uint8 * op,tmsize_t n)1035 Luv24toLuv48(LogLuvState* sp, uint8* op, tmsize_t n)
1036 {
1037 	uint32* luv = (uint32*) sp->tbuf;
1038 	int16* luv3 = (int16*) op;
1039 
1040 	while (n-- > 0) {
1041 		double u, v;
1042 
1043 		*luv3++ = (int16)((*luv >> 12 & 0xffd) + 13314);
1044 		if (uv_decode(&u, &v, *luv&0x3fff) < 0) {
1045 			u = U_NEU;
1046 			v = V_NEU;
1047 		}
1048 		*luv3++ = (int16)(u * (1L<<15));
1049 		*luv3++ = (int16)(v * (1L<<15));
1050 		luv++;
1051 	}
1052 }
1053 
1054 static void
Luv24toRGB(LogLuvState * sp,uint8 * op,tmsize_t n)1055 Luv24toRGB(LogLuvState* sp, uint8* op, tmsize_t n)
1056 {
1057 	uint32* luv = (uint32*) sp->tbuf;
1058 	uint8* rgb = (uint8*) op;
1059 
1060 	while (n-- > 0) {
1061 		float xyz[3];
1062 
1063 		LogLuv24toXYZ(*luv++, xyz);
1064 		XYZtoRGB24(xyz, rgb);
1065 		rgb += 3;
1066 	}
1067 }
1068 
1069 static void
Luv24fromXYZ(LogLuvState * sp,uint8 * op,tmsize_t n)1070 Luv24fromXYZ(LogLuvState* sp, uint8* op, tmsize_t n)
1071 {
1072 	uint32* luv = (uint32*) sp->tbuf;
1073 	float* xyz = (float*) op;
1074 
1075 	while (n-- > 0) {
1076 		*luv++ = LogLuv24fromXYZ(xyz, sp->encode_meth);
1077 		xyz += 3;
1078 	}
1079 }
1080 
1081 static void
Luv24fromLuv48(LogLuvState * sp,uint8 * op,tmsize_t n)1082 Luv24fromLuv48(LogLuvState* sp, uint8* op, tmsize_t n)
1083 {
1084 	uint32* luv = (uint32*) sp->tbuf;
1085 	int16* luv3 = (int16*) op;
1086 
1087 	while (n-- > 0) {
1088 		int Le, Ce;
1089 
1090 		if (luv3[0] <= 0)
1091 			Le = 0;
1092 		else if (luv3[0] >= (1<<12)+3314)
1093 			Le = (1<<10) - 1;
1094 		else if (sp->encode_meth == SGILOGENCODE_NODITHER)
1095 			Le = (luv3[0]-3314) >> 2;
1096 		else
1097 			Le = itrunc(.25*(luv3[0]-3314.), sp->encode_meth);
1098 
1099 		Ce = uv_encode((luv3[1]+.5)/(1<<15), (luv3[2]+.5)/(1<<15),
1100 					sp->encode_meth);
1101 		if (Ce < 0)	/* never happens */
1102 			Ce = uv_encode(U_NEU, V_NEU, SGILOGENCODE_NODITHER);
1103 		*luv++ = (uint32)Le << 14 | Ce;
1104 		luv3 += 3;
1105 	}
1106 }
1107 
1108 #if !LOGLUV_PUBLIC
1109 static
1110 #endif
1111 void
LogLuv32toXYZ(uint32 p,float XYZ[3])1112 LogLuv32toXYZ(uint32 p, float XYZ[3])
1113 {
1114 	double	L, u, v, s, x, y;
1115 					/* decode luminance */
1116 	L = LogL16toY((int)p >> 16);
1117 	if (L <= 0.) {
1118 		XYZ[0] = XYZ[1] = XYZ[2] = 0.;
1119 		return;
1120 	}
1121 					/* decode color */
1122 	u = 1./UVSCALE * ((p>>8 & 0xff) + .5);
1123 	v = 1./UVSCALE * ((p & 0xff) + .5);
1124 	s = 1./(6.*u - 16.*v + 12.);
1125 	x = 9.*u * s;
1126 	y = 4.*v * s;
1127 					/* convert to XYZ */
1128 	XYZ[0] = (float)(x/y * L);
1129 	XYZ[1] = (float)L;
1130 	XYZ[2] = (float)((1.-x-y)/y * L);
1131 }
1132 
1133 #if !LOGLUV_PUBLIC
1134 static
1135 #endif
1136 uint32
LogLuv32fromXYZ(float XYZ[3],int em)1137 LogLuv32fromXYZ(float XYZ[3], int em)
1138 {
1139 	unsigned int	Le, ue, ve;
1140 	double	u, v, s;
1141 					/* encode luminance */
1142 	Le = (unsigned int)LogL16fromY(XYZ[1], em);
1143 					/* encode color */
1144 	s = XYZ[0] + 15.*XYZ[1] + 3.*XYZ[2];
1145 	if (!Le || s <= 0.) {
1146 		u = U_NEU;
1147 		v = V_NEU;
1148 	} else {
1149 		u = 4.*XYZ[0] / s;
1150 		v = 9.*XYZ[1] / s;
1151 	}
1152 	if (u <= 0.) ue = 0;
1153 	else ue = itrunc(UVSCALE*u, em);
1154 	if (ue > 255) ue = 255;
1155 	if (v <= 0.) ve = 0;
1156 	else ve = itrunc(UVSCALE*v, em);
1157 	if (ve > 255) ve = 255;
1158 					/* combine encodings */
1159 	return (Le << 16 | ue << 8 | ve);
1160 }
1161 
1162 static void
Luv32toXYZ(LogLuvState * sp,uint8 * op,tmsize_t n)1163 Luv32toXYZ(LogLuvState* sp, uint8* op, tmsize_t n)
1164 {
1165 	uint32* luv = (uint32*) sp->tbuf;
1166 	float* xyz = (float*) op;
1167 
1168 	while (n-- > 0) {
1169 		LogLuv32toXYZ(*luv++, xyz);
1170 		xyz += 3;
1171 	}
1172 }
1173 
1174 static void
Luv32toLuv48(LogLuvState * sp,uint8 * op,tmsize_t n)1175 Luv32toLuv48(LogLuvState* sp, uint8* op, tmsize_t n)
1176 {
1177 	uint32* luv = (uint32*) sp->tbuf;
1178 	int16* luv3 = (int16*) op;
1179 
1180 	while (n-- > 0) {
1181 		double u, v;
1182 
1183 		*luv3++ = (int16)(*luv >> 16);
1184 		u = 1./UVSCALE * ((*luv>>8 & 0xff) + .5);
1185 		v = 1./UVSCALE * ((*luv & 0xff) + .5);
1186 		*luv3++ = (int16)(u * (1L<<15));
1187 		*luv3++ = (int16)(v * (1L<<15));
1188 		luv++;
1189 	}
1190 }
1191 
1192 static void
Luv32toRGB(LogLuvState * sp,uint8 * op,tmsize_t n)1193 Luv32toRGB(LogLuvState* sp, uint8* op, tmsize_t n)
1194 {
1195 	uint32* luv = (uint32*) sp->tbuf;
1196 	uint8* rgb = (uint8*) op;
1197 
1198 	while (n-- > 0) {
1199 		float xyz[3];
1200 
1201 		LogLuv32toXYZ(*luv++, xyz);
1202 		XYZtoRGB24(xyz, rgb);
1203 		rgb += 3;
1204 	}
1205 }
1206 
1207 static void
Luv32fromXYZ(LogLuvState * sp,uint8 * op,tmsize_t n)1208 Luv32fromXYZ(LogLuvState* sp, uint8* op, tmsize_t n)
1209 {
1210 	uint32* luv = (uint32*) sp->tbuf;
1211 	float* xyz = (float*) op;
1212 
1213 	while (n-- > 0) {
1214 		*luv++ = LogLuv32fromXYZ(xyz, sp->encode_meth);
1215 		xyz += 3;
1216 	}
1217 }
1218 
1219 static void
Luv32fromLuv48(LogLuvState * sp,uint8 * op,tmsize_t n)1220 Luv32fromLuv48(LogLuvState* sp, uint8* op, tmsize_t n)
1221 {
1222 	uint32* luv = (uint32*) sp->tbuf;
1223 	int16* luv3 = (int16*) op;
1224 
1225 	if (sp->encode_meth == SGILOGENCODE_NODITHER) {
1226 		while (n-- > 0) {
1227 			*luv++ = (uint32)luv3[0] << 16 |
1228 				(luv3[1]*(uint32)(UVSCALE+.5) >> 7 & 0xff00) |
1229 				(luv3[2]*(uint32)(UVSCALE+.5) >> 15 & 0xff);
1230 			luv3 += 3;
1231 		}
1232 		return;
1233 	}
1234 	while (n-- > 0) {
1235 		*luv++ = (uint32)luv3[0] << 16 |
1236 	(itrunc(luv3[1]*(UVSCALE/(1<<15)), sp->encode_meth) << 8 & 0xff00) |
1237 		(itrunc(luv3[2]*(UVSCALE/(1<<15)), sp->encode_meth) & 0xff);
1238 		luv3 += 3;
1239 	}
1240 }
1241 
1242 static void
_logLuvNop(LogLuvState * sp,uint8 * op,tmsize_t n)1243 _logLuvNop(LogLuvState* sp, uint8* op, tmsize_t n)
1244 {
1245 	(void) sp; (void) op; (void) n;
1246 }
1247 
1248 static int
LogL16GuessDataFmt(TIFFDirectory * td)1249 LogL16GuessDataFmt(TIFFDirectory *td)
1250 {
1251 #define	PACK(s,b,f)	(((b)<<6)|((s)<<3)|(f))
1252 	switch (PACK(td->td_samplesperpixel, td->td_bitspersample, td->td_sampleformat)) {
1253 	case PACK(1, 32, SAMPLEFORMAT_IEEEFP):
1254 		return (SGILOGDATAFMT_FLOAT);
1255 	case PACK(1, 16, SAMPLEFORMAT_VOID):
1256 	case PACK(1, 16, SAMPLEFORMAT_INT):
1257 	case PACK(1, 16, SAMPLEFORMAT_UINT):
1258 		return (SGILOGDATAFMT_16BIT);
1259 	case PACK(1,  8, SAMPLEFORMAT_VOID):
1260 	case PACK(1,  8, SAMPLEFORMAT_UINT):
1261 		return (SGILOGDATAFMT_8BIT);
1262 	}
1263 #undef PACK
1264 	return (SGILOGDATAFMT_UNKNOWN);
1265 }
1266 
1267 
1268 #define TIFF_SIZE_T_MAX ((size_t) ~ ((size_t)0))
1269 #define TIFF_TMSIZE_T_MAX (tmsize_t)(TIFF_SIZE_T_MAX >> 1)
1270 
1271 static tmsize_t
multiply_ms(tmsize_t m1,tmsize_t m2)1272 multiply_ms(tmsize_t m1, tmsize_t m2)
1273 {
1274         if( m1 == 0 || m2 > TIFF_TMSIZE_T_MAX / m1 )
1275             return 0;
1276         return m1 * m2;
1277 }
1278 
1279 static int
LogL16InitState(TIFF * tif)1280 LogL16InitState(TIFF* tif)
1281 {
1282 	static const char module[] = "LogL16InitState";
1283 	TIFFDirectory *td = &tif->tif_dir;
1284 	LogLuvState* sp = DecoderState(tif);
1285 
1286 	assert(sp != NULL);
1287 	assert(td->td_photometric == PHOTOMETRIC_LOGL);
1288 
1289 	if( td->td_samplesperpixel != 1 )
1290 	{
1291 		TIFFErrorExt(tif->tif_clientdata, module,
1292 		             "Sorry, can not handle LogL image with %s=%d",
1293 			     "Samples/pixel", td->td_samplesperpixel);
1294 		return 0;
1295 	}
1296 
1297 	/* for some reason, we can't do this in TIFFInitLogL16 */
1298 	if (sp->user_datafmt == SGILOGDATAFMT_UNKNOWN)
1299 		sp->user_datafmt = LogL16GuessDataFmt(td);
1300 	switch (sp->user_datafmt) {
1301 	case SGILOGDATAFMT_FLOAT:
1302 		sp->pixel_size = sizeof (float);
1303 		break;
1304 	case SGILOGDATAFMT_16BIT:
1305 		sp->pixel_size = sizeof (int16);
1306 		break;
1307 	case SGILOGDATAFMT_8BIT:
1308 		sp->pixel_size = sizeof (uint8);
1309 		break;
1310 	default:
1311 		TIFFErrorExt(tif->tif_clientdata, module,
1312 		    "No support for converting user data format to LogL");
1313 		return (0);
1314 	}
1315         if( isTiled(tif) )
1316             sp->tbuflen = multiply_ms(td->td_tilewidth, td->td_tilelength);
1317         else if( td->td_rowsperstrip < td->td_imagelength )
1318             sp->tbuflen = multiply_ms(td->td_imagewidth, td->td_rowsperstrip);
1319         else
1320             sp->tbuflen = multiply_ms(td->td_imagewidth, td->td_imagelength);
1321 	if (multiply_ms(sp->tbuflen, sizeof (int16)) == 0 ||
1322 	    (sp->tbuf = (uint8*) _TIFFmalloc(sp->tbuflen * sizeof (int16))) == NULL) {
1323 		TIFFErrorExt(tif->tif_clientdata, module, "No space for SGILog translation buffer");
1324 		return (0);
1325 	}
1326 	return (1);
1327 }
1328 
1329 static int
LogLuvGuessDataFmt(TIFFDirectory * td)1330 LogLuvGuessDataFmt(TIFFDirectory *td)
1331 {
1332 	int guess;
1333 
1334 	/*
1335 	 * If the user didn't tell us their datafmt,
1336 	 * take our best guess from the bitspersample.
1337 	 */
1338 #define	PACK(a,b)	(((a)<<3)|(b))
1339 	switch (PACK(td->td_bitspersample, td->td_sampleformat)) {
1340 	case PACK(32, SAMPLEFORMAT_IEEEFP):
1341 		guess = SGILOGDATAFMT_FLOAT;
1342 		break;
1343 	case PACK(32, SAMPLEFORMAT_VOID):
1344 	case PACK(32, SAMPLEFORMAT_UINT):
1345 	case PACK(32, SAMPLEFORMAT_INT):
1346 		guess = SGILOGDATAFMT_RAW;
1347 		break;
1348 	case PACK(16, SAMPLEFORMAT_VOID):
1349 	case PACK(16, SAMPLEFORMAT_INT):
1350 	case PACK(16, SAMPLEFORMAT_UINT):
1351 		guess = SGILOGDATAFMT_16BIT;
1352 		break;
1353 	case PACK( 8, SAMPLEFORMAT_VOID):
1354 	case PACK( 8, SAMPLEFORMAT_UINT):
1355 		guess = SGILOGDATAFMT_8BIT;
1356 		break;
1357 	default:
1358 		guess = SGILOGDATAFMT_UNKNOWN;
1359 		break;
1360 #undef PACK
1361 	}
1362 	/*
1363 	 * Double-check samples per pixel.
1364 	 */
1365 	switch (td->td_samplesperpixel) {
1366 	case 1:
1367 		if (guess != SGILOGDATAFMT_RAW)
1368 			guess = SGILOGDATAFMT_UNKNOWN;
1369 		break;
1370 	case 3:
1371 		if (guess == SGILOGDATAFMT_RAW)
1372 			guess = SGILOGDATAFMT_UNKNOWN;
1373 		break;
1374 	default:
1375 		guess = SGILOGDATAFMT_UNKNOWN;
1376 		break;
1377 	}
1378 	return (guess);
1379 }
1380 
1381 static int
LogLuvInitState(TIFF * tif)1382 LogLuvInitState(TIFF* tif)
1383 {
1384 	static const char module[] = "LogLuvInitState";
1385 	TIFFDirectory* td = &tif->tif_dir;
1386 	LogLuvState* sp = DecoderState(tif);
1387 
1388 	assert(sp != NULL);
1389 	assert(td->td_photometric == PHOTOMETRIC_LOGLUV);
1390 
1391 	/* for some reason, we can't do this in TIFFInitLogLuv */
1392 	if (td->td_planarconfig != PLANARCONFIG_CONTIG) {
1393 		TIFFErrorExt(tif->tif_clientdata, module,
1394 		    "SGILog compression cannot handle non-contiguous data");
1395 		return (0);
1396 	}
1397 	if (sp->user_datafmt == SGILOGDATAFMT_UNKNOWN)
1398 		sp->user_datafmt = LogLuvGuessDataFmt(td);
1399 	switch (sp->user_datafmt) {
1400 	case SGILOGDATAFMT_FLOAT:
1401 		sp->pixel_size = 3*sizeof (float);
1402 		break;
1403 	case SGILOGDATAFMT_16BIT:
1404 		sp->pixel_size = 3*sizeof (int16);
1405 		break;
1406 	case SGILOGDATAFMT_RAW:
1407 		sp->pixel_size = sizeof (uint32);
1408 		break;
1409 	case SGILOGDATAFMT_8BIT:
1410 		sp->pixel_size = 3*sizeof (uint8);
1411 		break;
1412 	default:
1413 		TIFFErrorExt(tif->tif_clientdata, module,
1414 		    "No support for converting user data format to LogLuv");
1415 		return (0);
1416 	}
1417         if( isTiled(tif) )
1418             sp->tbuflen = multiply_ms(td->td_tilewidth, td->td_tilelength);
1419         else if( td->td_rowsperstrip < td->td_imagelength )
1420             sp->tbuflen = multiply_ms(td->td_imagewidth, td->td_rowsperstrip);
1421         else
1422             sp->tbuflen = multiply_ms(td->td_imagewidth, td->td_imagelength);
1423 	if (multiply_ms(sp->tbuflen, sizeof (uint32)) == 0 ||
1424 	    (sp->tbuf = (uint8*) _TIFFmalloc(sp->tbuflen * sizeof (uint32))) == NULL) {
1425 		TIFFErrorExt(tif->tif_clientdata, module, "No space for SGILog translation buffer");
1426 		return (0);
1427 	}
1428 	return (1);
1429 }
1430 
1431 static int
LogLuvFixupTags(TIFF * tif)1432 LogLuvFixupTags(TIFF* tif)
1433 {
1434 	(void) tif;
1435 	return (1);
1436 }
1437 
1438 static int
LogLuvSetupDecode(TIFF * tif)1439 LogLuvSetupDecode(TIFF* tif)
1440 {
1441 	static const char module[] = "LogLuvSetupDecode";
1442 	LogLuvState* sp = DecoderState(tif);
1443 	TIFFDirectory* td = &tif->tif_dir;
1444 
1445 	tif->tif_postdecode = _TIFFNoPostDecode;
1446 	switch (td->td_photometric) {
1447 	case PHOTOMETRIC_LOGLUV:
1448 		if (!LogLuvInitState(tif))
1449 			break;
1450 		if (td->td_compression == COMPRESSION_SGILOG24) {
1451 			tif->tif_decoderow = LogLuvDecode24;
1452 			switch (sp->user_datafmt) {
1453 			case SGILOGDATAFMT_FLOAT:
1454 				sp->tfunc = Luv24toXYZ;
1455 				break;
1456 			case SGILOGDATAFMT_16BIT:
1457 				sp->tfunc = Luv24toLuv48;
1458 				break;
1459 			case SGILOGDATAFMT_8BIT:
1460 				sp->tfunc = Luv24toRGB;
1461 				break;
1462 			}
1463 		} else {
1464 			tif->tif_decoderow = LogLuvDecode32;
1465 			switch (sp->user_datafmt) {
1466 			case SGILOGDATAFMT_FLOAT:
1467 				sp->tfunc = Luv32toXYZ;
1468 				break;
1469 			case SGILOGDATAFMT_16BIT:
1470 				sp->tfunc = Luv32toLuv48;
1471 				break;
1472 			case SGILOGDATAFMT_8BIT:
1473 				sp->tfunc = Luv32toRGB;
1474 				break;
1475 			}
1476 		}
1477 		return (1);
1478 	case PHOTOMETRIC_LOGL:
1479 		if (!LogL16InitState(tif))
1480 			break;
1481 		tif->tif_decoderow = LogL16Decode;
1482 		switch (sp->user_datafmt) {
1483 		case SGILOGDATAFMT_FLOAT:
1484 			sp->tfunc = L16toY;
1485 			break;
1486 		case SGILOGDATAFMT_8BIT:
1487 			sp->tfunc = L16toGry;
1488 			break;
1489 		}
1490 		return (1);
1491 	default:
1492 		TIFFErrorExt(tif->tif_clientdata, module,
1493 		    "Inappropriate photometric interpretation %d for SGILog compression; %s",
1494 		    td->td_photometric, "must be either LogLUV or LogL");
1495 		break;
1496 	}
1497 	return (0);
1498 }
1499 
1500 static int
LogLuvSetupEncode(TIFF * tif)1501 LogLuvSetupEncode(TIFF* tif)
1502 {
1503 	static const char module[] = "LogLuvSetupEncode";
1504 	LogLuvState* sp = EncoderState(tif);
1505 	TIFFDirectory* td = &tif->tif_dir;
1506 
1507 	switch (td->td_photometric) {
1508 	case PHOTOMETRIC_LOGLUV:
1509 		if (!LogLuvInitState(tif))
1510 			break;
1511 		if (td->td_compression == COMPRESSION_SGILOG24) {
1512 			tif->tif_encoderow = LogLuvEncode24;
1513 			switch (sp->user_datafmt) {
1514 			case SGILOGDATAFMT_FLOAT:
1515 				sp->tfunc = Luv24fromXYZ;
1516 				break;
1517 			case SGILOGDATAFMT_16BIT:
1518 				sp->tfunc = Luv24fromLuv48;
1519 				break;
1520 			case SGILOGDATAFMT_RAW:
1521 				break;
1522 			default:
1523 				goto notsupported;
1524 			}
1525 		} else {
1526 			tif->tif_encoderow = LogLuvEncode32;
1527 			switch (sp->user_datafmt) {
1528 			case SGILOGDATAFMT_FLOAT:
1529 				sp->tfunc = Luv32fromXYZ;
1530 				break;
1531 			case SGILOGDATAFMT_16BIT:
1532 				sp->tfunc = Luv32fromLuv48;
1533 				break;
1534 			case SGILOGDATAFMT_RAW:
1535 				break;
1536 			default:
1537 				goto notsupported;
1538 			}
1539 		}
1540 		break;
1541 	case PHOTOMETRIC_LOGL:
1542 		if (!LogL16InitState(tif))
1543 			break;
1544 		tif->tif_encoderow = LogL16Encode;
1545 		switch (sp->user_datafmt) {
1546 		case SGILOGDATAFMT_FLOAT:
1547 			sp->tfunc = L16fromY;
1548 			break;
1549 		case SGILOGDATAFMT_16BIT:
1550 			break;
1551 		default:
1552 			goto notsupported;
1553 		}
1554 		break;
1555 	default:
1556 		TIFFErrorExt(tif->tif_clientdata, module,
1557 		    "Inappropriate photometric interpretation %d for SGILog compression; %s",
1558 		    td->td_photometric, "must be either LogLUV or LogL");
1559 		break;
1560 	}
1561 	sp->encoder_state = 1;
1562 	return (1);
1563 notsupported:
1564 	TIFFErrorExt(tif->tif_clientdata, module,
1565 	    "SGILog compression supported only for %s, or raw data",
1566 	    td->td_photometric == PHOTOMETRIC_LOGL ? "Y, L" : "XYZ, Luv");
1567 	return (0);
1568 }
1569 
1570 static void
LogLuvClose(TIFF * tif)1571 LogLuvClose(TIFF* tif)
1572 {
1573         LogLuvState* sp = (LogLuvState*) tif->tif_data;
1574 	TIFFDirectory *td = &tif->tif_dir;
1575 
1576 	assert(sp != 0);
1577 	/*
1578 	 * For consistency, we always want to write out the same
1579 	 * bitspersample and sampleformat for our TIFF file,
1580 	 * regardless of the data format being used by the application.
1581 	 * Since this routine is called after tags have been set but
1582 	 * before they have been recorded in the file, we reset them here.
1583          * Note: this is really a nasty approach. See PixarLogClose
1584 	 */
1585         if( sp->encoder_state )
1586         {
1587             /* See PixarLogClose. Might avoid issues with tags whose size depends
1588              * on those below, but not completely sure this is enough. */
1589             td->td_samplesperpixel =
1590                 (td->td_photometric == PHOTOMETRIC_LOGL) ? 1 : 3;
1591             td->td_bitspersample = 16;
1592             td->td_sampleformat = SAMPLEFORMAT_INT;
1593         }
1594 }
1595 
1596 static void
LogLuvCleanup(TIFF * tif)1597 LogLuvCleanup(TIFF* tif)
1598 {
1599 	LogLuvState* sp = (LogLuvState *)tif->tif_data;
1600 
1601 	assert(sp != 0);
1602 
1603 	tif->tif_tagmethods.vgetfield = sp->vgetparent;
1604 	tif->tif_tagmethods.vsetfield = sp->vsetparent;
1605 
1606 	if (sp->tbuf)
1607 		_TIFFfree(sp->tbuf);
1608 	_TIFFfree(sp);
1609 	tif->tif_data = NULL;
1610 
1611 	_TIFFSetDefaultCompressionState(tif);
1612 }
1613 
1614 static int
LogLuvVSetField(TIFF * tif,uint32 tag,va_list ap)1615 LogLuvVSetField(TIFF* tif, uint32 tag, va_list ap)
1616 {
1617 	static const char module[] = "LogLuvVSetField";
1618 	LogLuvState* sp = DecoderState(tif);
1619 	int bps, fmt;
1620 
1621 	switch (tag) {
1622 	case TIFFTAG_SGILOGDATAFMT:
1623 		sp->user_datafmt = (int) va_arg(ap, int);
1624 		/*
1625 		 * Tweak the TIFF header so that the rest of libtiff knows what
1626 		 * size of data will be passed between app and library, and
1627 		 * assume that the app knows what it is doing and is not
1628 		 * confused by these header manipulations...
1629 		 */
1630 		switch (sp->user_datafmt) {
1631 		case SGILOGDATAFMT_FLOAT:
1632 			bps = 32;
1633 			fmt = SAMPLEFORMAT_IEEEFP;
1634 			break;
1635 		case SGILOGDATAFMT_16BIT:
1636 			bps = 16;
1637 			fmt = SAMPLEFORMAT_INT;
1638 			break;
1639 		case SGILOGDATAFMT_RAW:
1640 			bps = 32;
1641 			fmt = SAMPLEFORMAT_UINT;
1642 			TIFFSetField(tif, TIFFTAG_SAMPLESPERPIXEL, 1);
1643 			break;
1644 		case SGILOGDATAFMT_8BIT:
1645 			bps = 8;
1646 			fmt = SAMPLEFORMAT_UINT;
1647 			break;
1648 		default:
1649 			TIFFErrorExt(tif->tif_clientdata, tif->tif_name,
1650 			    "Unknown data format %d for LogLuv compression",
1651 			    sp->user_datafmt);
1652 			return (0);
1653 		}
1654 		TIFFSetField(tif, TIFFTAG_BITSPERSAMPLE, bps);
1655 		TIFFSetField(tif, TIFFTAG_SAMPLEFORMAT, fmt);
1656 		/*
1657 		 * Must recalculate sizes should bits/sample change.
1658 		 */
1659 		tif->tif_tilesize = isTiled(tif) ? TIFFTileSize(tif) : (tmsize_t) -1;
1660 		tif->tif_scanlinesize = TIFFScanlineSize(tif);
1661 		return (1);
1662 	case TIFFTAG_SGILOGENCODE:
1663 		sp->encode_meth = (int) va_arg(ap, int);
1664 		if (sp->encode_meth != SGILOGENCODE_NODITHER &&
1665 		    sp->encode_meth != SGILOGENCODE_RANDITHER) {
1666 			TIFFErrorExt(tif->tif_clientdata, module,
1667 			    "Unknown encoding %d for LogLuv compression",
1668 			    sp->encode_meth);
1669 			return (0);
1670 		}
1671 		return (1);
1672 	default:
1673 		return (*sp->vsetparent)(tif, tag, ap);
1674 	}
1675 }
1676 
1677 static int
LogLuvVGetField(TIFF * tif,uint32 tag,va_list ap)1678 LogLuvVGetField(TIFF* tif, uint32 tag, va_list ap)
1679 {
1680 	LogLuvState *sp = (LogLuvState *)tif->tif_data;
1681 
1682 	switch (tag) {
1683 	case TIFFTAG_SGILOGDATAFMT:
1684 		*va_arg(ap, int*) = sp->user_datafmt;
1685 		return (1);
1686 	default:
1687 		return (*sp->vgetparent)(tif, tag, ap);
1688 	}
1689 }
1690 
1691 static const TIFFField LogLuvFields[] = {
1692     { TIFFTAG_SGILOGDATAFMT, 0, 0, TIFF_SHORT, 0, TIFF_SETGET_INT, TIFF_SETGET_UNDEFINED, FIELD_PSEUDO, TRUE, FALSE, "SGILogDataFmt", NULL},
1693     { TIFFTAG_SGILOGENCODE, 0, 0, TIFF_SHORT, 0, TIFF_SETGET_INT, TIFF_SETGET_UNDEFINED, FIELD_PSEUDO, TRUE, FALSE, "SGILogEncode", NULL}
1694 };
1695 
1696 int
TIFFInitSGILog(TIFF * tif,int scheme)1697 TIFFInitSGILog(TIFF* tif, int scheme)
1698 {
1699 	static const char module[] = "TIFFInitSGILog";
1700 	LogLuvState* sp;
1701 
1702 	assert(scheme == COMPRESSION_SGILOG24 || scheme == COMPRESSION_SGILOG);
1703 
1704 	/*
1705 	 * Merge codec-specific tag information.
1706 	 */
1707 	if (!_TIFFMergeFields(tif, LogLuvFields,
1708 			      TIFFArrayCount(LogLuvFields))) {
1709 		TIFFErrorExt(tif->tif_clientdata, module,
1710 		    "Merging SGILog codec-specific tags failed");
1711 		return 0;
1712 	}
1713 
1714 	/*
1715 	 * Allocate state block so tag methods have storage to record values.
1716 	 */
1717 	tif->tif_data = (uint8*) _TIFFmalloc(sizeof (LogLuvState));
1718 	if (tif->tif_data == NULL)
1719 		goto bad;
1720 	sp = (LogLuvState*) tif->tif_data;
1721 	_TIFFmemset((void*)sp, 0, sizeof (*sp));
1722 	sp->user_datafmt = SGILOGDATAFMT_UNKNOWN;
1723 	sp->encode_meth = (scheme == COMPRESSION_SGILOG24) ?
1724 	    SGILOGENCODE_RANDITHER : SGILOGENCODE_NODITHER;
1725 	sp->tfunc = _logLuvNop;
1726 
1727 	/*
1728 	 * Install codec methods.
1729 	 * NB: tif_decoderow & tif_encoderow are filled
1730 	 *     in at setup time.
1731 	 */
1732 	tif->tif_fixuptags = LogLuvFixupTags;
1733 	tif->tif_setupdecode = LogLuvSetupDecode;
1734 	tif->tif_decodestrip = LogLuvDecodeStrip;
1735 	tif->tif_decodetile = LogLuvDecodeTile;
1736 	tif->tif_setupencode = LogLuvSetupEncode;
1737 	tif->tif_encodestrip = LogLuvEncodeStrip;
1738 	tif->tif_encodetile = LogLuvEncodeTile;
1739 	tif->tif_close = LogLuvClose;
1740 	tif->tif_cleanup = LogLuvCleanup;
1741 
1742 	/*
1743 	 * Override parent get/set field methods.
1744 	 */
1745 	sp->vgetparent = tif->tif_tagmethods.vgetfield;
1746 	tif->tif_tagmethods.vgetfield = LogLuvVGetField;   /* hook for codec tags */
1747 	sp->vsetparent = tif->tif_tagmethods.vsetfield;
1748 	tif->tif_tagmethods.vsetfield = LogLuvVSetField;   /* hook for codec tags */
1749 
1750 	return (1);
1751 bad:
1752 	TIFFErrorExt(tif->tif_clientdata, module,
1753 		     "%s: No space for LogLuv state block", tif->tif_name);
1754 	return (0);
1755 }
1756 #endif /* LOGLUV_SUPPORT */
1757 
1758 /* vim: set ts=8 sts=8 sw=8 noet: */
1759 /*
1760  * Local Variables:
1761  * mode: c
1762  * c-basic-offset: 8
1763  * fill-column: 78
1764  * End:
1765  */
1766