FFmpeg coverage

Directory: ../../../ffmpeg/
File: src/libavcodec/exr.c
Date: 2021-09-22 05:58:22
Exec Total Coverage
Lines: 845 1279 66.1%
Branches: 451 762 59.2%
Line Branch Exec Source
1 /*
2 * OpenEXR (.exr) image decoder
3 * Copyright (c) 2006 Industrial Light & Magic, a division of Lucas Digital Ltd. LLC
4 * Copyright (c) 2009 Jimmy Christensen
5 *
6 * B44/B44A, Tile, UINT32 added by Jokyo Images support by CNC - French National Center for Cinema
7 *
8 * This file is part of FFmpeg.
9 *
10 * FFmpeg is free software; you can redistribute it and/or
11 * modify it under the terms of the GNU Lesser General Public
12 * License as published by the Free Software Foundation; either
13 * version 2.1 of the License, or (at your option) any later version.
14 *
15 * FFmpeg is distributed in the hope that it will be useful,
16 * but WITHOUT ANY WARRANTY; without even the implied warranty of
17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
18 * Lesser General Public License for more details.
19 *
20 * You should have received a copy of the GNU Lesser General Public
21 * License along with FFmpeg; if not, write to the Free Software
22 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
23 */
24
25 /**
26 * @file
27 * OpenEXR decoder
28 * @author Jimmy Christensen
29 *
30 * For more information on the OpenEXR format, visit:
31 * http://openexr.com/
32 */
33
34 #include <float.h>
35 #include <zlib.h>
36
37 #include "libavutil/avassert.h"
38 #include "libavutil/common.h"
39 #include "libavutil/imgutils.h"
40 #include "libavutil/intfloat.h"
41 #include "libavutil/avstring.h"
42 #include "libavutil/opt.h"
43 #include "libavutil/color_utils.h"
44
45 #include "avcodec.h"
46 #include "bytestream.h"
47
48 #if HAVE_BIGENDIAN
49 #include "bswapdsp.h"
50 #endif
51
52 #include "exrdsp.h"
53 #include "get_bits.h"
54 #include "internal.h"
55 #include "half2float.h"
56 #include "mathops.h"
57 #include "thread.h"
58
59 enum ExrCompr {
60 EXR_RAW,
61 EXR_RLE,
62 EXR_ZIP1,
63 EXR_ZIP16,
64 EXR_PIZ,
65 EXR_PXR24,
66 EXR_B44,
67 EXR_B44A,
68 EXR_DWAA,
69 EXR_DWAB,
70 EXR_UNKN,
71 };
72
73 enum ExrPixelType {
74 EXR_UINT,
75 EXR_HALF,
76 EXR_FLOAT,
77 EXR_UNKNOWN,
78 };
79
80 enum ExrTileLevelMode {
81 EXR_TILE_LEVEL_ONE,
82 EXR_TILE_LEVEL_MIPMAP,
83 EXR_TILE_LEVEL_RIPMAP,
84 EXR_TILE_LEVEL_UNKNOWN,
85 };
86
87 enum ExrTileLevelRound {
88 EXR_TILE_ROUND_UP,
89 EXR_TILE_ROUND_DOWN,
90 EXR_TILE_ROUND_UNKNOWN,
91 };
92
93 typedef struct HuffEntry {
94 uint8_t len;
95 uint16_t sym;
96 uint32_t code;
97 } HuffEntry;
98
99 typedef struct EXRChannel {
100 int xsub, ysub;
101 enum ExrPixelType pixel_type;
102 } EXRChannel;
103
104 typedef struct EXRTileAttribute {
105 int32_t xSize;
106 int32_t ySize;
107 enum ExrTileLevelMode level_mode;
108 enum ExrTileLevelRound level_round;
109 } EXRTileAttribute;
110
111 typedef struct EXRThreadData {
112 uint8_t *uncompressed_data;
113 int uncompressed_size;
114
115 uint8_t *tmp;
116 int tmp_size;
117
118 uint8_t *bitmap;
119 uint16_t *lut;
120
121 uint8_t *ac_data;
122 unsigned ac_size;
123
124 uint8_t *dc_data;
125 unsigned dc_size;
126
127 uint8_t *rle_data;
128 unsigned rle_size;
129
130 uint8_t *rle_raw_data;
131 unsigned rle_raw_size;
132
133 float block[3][64];
134
135 int ysize, xsize;
136
137 int channel_line_size;
138
139 int run_sym;
140 HuffEntry *he;
141 uint64_t *freq;
142 VLC vlc;
143 } EXRThreadData;
144
145 typedef struct EXRContext {
146 AVClass *class;
147 AVFrame *picture;
148 AVCodecContext *avctx;
149 ExrDSPContext dsp;
150
151 #if HAVE_BIGENDIAN
152 BswapDSPContext bbdsp;
153 #endif
154
155 enum ExrCompr compression;
156 enum ExrPixelType pixel_type;
157 int channel_offsets[4]; // 0 = red, 1 = green, 2 = blue and 3 = alpha
158 const AVPixFmtDescriptor *desc;
159
160 int w, h;
161 uint32_t sar;
162 int32_t xmax, xmin;
163 int32_t ymax, ymin;
164 uint32_t xdelta, ydelta;
165
166 int scan_lines_per_block;
167
168 EXRTileAttribute tile_attr; /* header data attribute of tile */
169 int is_tile; /* 0 if scanline, 1 if tile */
170 int is_multipart;
171 int current_part;
172
173 int is_luma;/* 1 if there is an Y plane */
174
175 GetByteContext gb;
176 const uint8_t *buf;
177 int buf_size;
178
179 EXRChannel *channels;
180 int nb_channels;
181 int current_channel_offset;
182 uint32_t chunk_count;
183
184 EXRThreadData *thread_data;
185
186 const char *layer;
187 int selected_part;
188
189 enum AVColorTransferCharacteristic apply_trc_type;
190 float gamma;
191 union av_intfloat32 gamma_table[65536];
192
193 uint32_t mantissatable[2048];
194 uint32_t exponenttable[64];
195 uint16_t offsettable[64];
196 } EXRContext;
197
198 1838 static int zip_uncompress(EXRContext *s, const uint8_t *src, int compressed_size,
199 int uncompressed_size, EXRThreadData *td)
200 {
201 1838 unsigned long dest_len = uncompressed_size;
202
203
1/2
✓ Branch 1 taken 1838 times.
✗ Branch 2 not taken.
 1838 if (uncompress(td->tmp, &dest_len, src, compressed_size) != Z_OK ||
204
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1838 times.
 1838 dest_len != uncompressed_size)
205 ✗ return AVERROR_INVALIDDATA;
206
207 av_assert1(uncompressed_size % 2 == 0);
208
209 1838 s->dsp.predictor(td->tmp, uncompressed_size);
210 1838 s->dsp.reorder_pixels(td->uncompressed_data, td->tmp, uncompressed_size);
211
212 1838 return 0;
213 }
214
215 1386 static int rle(uint8_t *dst, const uint8_t *src,
216 int compressed_size, int uncompressed_size)
217 {
218 1386 uint8_t *d = dst;
219 1386 const int8_t *s = src;
220 1386 int ssize = compressed_size;
221 1386 int dsize = uncompressed_size;
222 1386 uint8_t *dend = d + dsize;
223 int count;
224
225
2/2
✓ Branch 0 taken 376742 times.
✓ Branch 1 taken 1386 times.
 378128 while (ssize > 0) {
226 376742 count = *s++;
227
228
2/2
✓ Branch 0 taken 186494 times.
✓ Branch 1 taken 190248 times.
 376742 if (count < 0) {
229 186494 count = -count;
230
231
1/2
✓ Branch 0 taken 186494 times.
✗ Branch 1 not taken.
 186494 if ((dsize -= count) < 0 ||
232
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 186494 times.
 186494 (ssize -= count + 1) < 0)
233 ✗ return AVERROR_INVALIDDATA;
234
235
2/2
✓ Branch 0 taken 3276516 times.
✓ Branch 1 taken 186494 times.
 3463010 while (count--)
236 3276516 *d++ = *s++;
237 } else {
238 190248 count++;
239
240
1/2
✓ Branch 0 taken 190248 times.
✗ Branch 1 not taken.
 190248 if ((dsize -= count) < 0 ||
241
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 190248 times.
 190248 (ssize -= 2) < 0)
242 ✗ return AVERROR_INVALIDDATA;
243
244
2/2
✓ Branch 0 taken 3079980 times.
✓ Branch 1 taken 190248 times.
 3270228 while (count--)
245 3079980 *d++ = *s;
246
247 190248 s++;
248 }
249 }
250
251
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1386 times.
 1386 if (dend != d)
252 ✗ return AVERROR_INVALIDDATA;
253
254 1386 return 0;
255 }
256
257 1386 static int rle_uncompress(EXRContext *ctx, const uint8_t *src, int compressed_size,
258 int uncompressed_size, EXRThreadData *td)
259 {
260 1386 rle(td->tmp, src, compressed_size, uncompressed_size);
261
262 av_assert1(uncompressed_size % 2 == 0);
263
264 1386 ctx->dsp.predictor(td->tmp, uncompressed_size);
265 1386 ctx->dsp.reorder_pixels(td->uncompressed_data, td->tmp, uncompressed_size);
266
267 1386 return 0;
268 }
269
270 #define USHORT_RANGE (1 << 16)
271 #define BITMAP_SIZE (1 << 13)
272
273 60 static uint16_t reverse_lut(const uint8_t *bitmap, uint16_t *lut)
274 {
275 60 int i, k = 0;
276
277
2/2
✓ Branch 0 taken 3932160 times.
✓ Branch 1 taken 60 times.
 3932220 for (i = 0; i < USHORT_RANGE; i++)
278
4/4
✓ Branch 0 taken 3932100 times.
✓ Branch 1 taken 60 times.
✓ Branch 2 taken 30710 times.
✓ Branch 3 taken 3901390 times.
 3932160 if ((i == 0) || (bitmap[i >> 3] & (1 << (i & 7))))
279 30770 lut[k++] = i;
280
281 60 i = k - 1;
282
283 60 memset(lut + k, 0, (USHORT_RANGE - k) * 2);
284
285 60 return i;
286 }
287
288 60 static void apply_lut(const uint16_t *lut, uint16_t *dst, int dsize)
289 {
290 int i;
291
292
2/2
✓ Branch 0 taken 1641444 times.
✓ Branch 1 taken 60 times.
 1641504 for (i = 0; i < dsize; ++i)
293 1641444 dst[i] = lut[dst[i]];
294 60 }
295
296 #define HUF_ENCBITS 16 // literal (value) bit length
297 #define HUF_ENCSIZE ((1 << HUF_ENCBITS) + 1) // encoding table size
298
299 60 static void huf_canonical_code_table(uint64_t *freq)
300 {
301 60 uint64_t c, n[59] = { 0 };
302 int i;
303
304
2/2
✓ Branch 0 taken 3932220 times.
✓ Branch 1 taken 60 times.
 3932280 for (i = 0; i < HUF_ENCSIZE; i++)
305 3932220 n[freq[i]] += 1;
306
307 60 c = 0;
308
2/2
✓ Branch 0 taken 3480 times.
✓ Branch 1 taken 60 times.
 3540 for (i = 58; i > 0; --i) {
309 3480 uint64_t nc = ((c + n[i]) >> 1);
310 3480 n[i] = c;
311 3480 c = nc;
312 }
313
314
2/2
✓ Branch 0 taken 3932220 times.
✓ Branch 1 taken 60 times.
 3932280 for (i = 0; i < HUF_ENCSIZE; ++i) {
315 3932220 int l = freq[i];
316
317
2/2
✓ Branch 0 taken 12834 times.
✓ Branch 1 taken 3919386 times.
 3932220 if (l > 0)
318 12834 freq[i] = l | (n[l]++ << 6);
319 }
320 60 }
321
322 #define SHORT_ZEROCODE_RUN 59
323 #define LONG_ZEROCODE_RUN 63
324 #define SHORTEST_LONG_RUN (2 + LONG_ZEROCODE_RUN - SHORT_ZEROCODE_RUN)
325 #define LONGEST_LONG_RUN (255 + SHORTEST_LONG_RUN)
326
327 60 static int huf_unpack_enc_table(GetByteContext *gb,
328 int32_t im, int32_t iM, uint64_t *freq)
329 {
330 GetBitContext gbit;
331 60 int ret = init_get_bits8(&gbit, gb->buffer, bytestream2_get_bytes_left(gb));
332
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 60 times.
 60 if (ret < 0)
333 ✗ return ret;
334
335
2/2
✓ Branch 0 taken 33212 times.
✓ Branch 1 taken 60 times.
 33272 for (; im <= iM; im++) {
336 33212 uint64_t l = freq[im] = get_bits(&gbit, 6);
337
338
2/2
✓ Branch 0 taken 16678 times.
✓ Branch 1 taken 16534 times.
 33212 if (l == LONG_ZEROCODE_RUN) {
339 16678 int zerun = get_bits(&gbit, 8) + SHORTEST_LONG_RUN;
340
341
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 16678 times.
 16678 if (im + zerun > iM + 1)
342 ✗ return AVERROR_INVALIDDATA;
343
344
2/2
✓ Branch 0 taken 3911252 times.
✓ Branch 1 taken 16678 times.
 3927930 while (zerun--)
345 3911252 freq[im++] = 0;
346
347 16678 im--;
348
2/2
✓ Branch 0 taken 2106 times.
✓ Branch 1 taken 14428 times.
 16534 } else if (l >= SHORT_ZEROCODE_RUN) {
349 2106 int zerun = l - SHORT_ZEROCODE_RUN + 2;
350
351
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2106 times.
 2106 if (im + zerun > iM + 1)
352 ✗ return AVERROR_INVALIDDATA;
353
354
2/2
✓ Branch 0 taken 6540 times.
✓ Branch 1 taken 2106 times.
 8646 while (zerun--)
355 6540 freq[im++] = 0;
356
357 2106 im--;
358 }
359 }
360
361 60 bytestream2_skip(gb, (get_bits_count(&gbit) + 7) / 8);
362 60 huf_canonical_code_table(freq);
363
364 60 return 0;
365 }
366
367 60 static int huf_build_dec_table(EXRContext *s,
368 EXRThreadData *td, int im, int iM)
369 {
370 60 int j = 0;
371
372 60 td->run_sym = -1;
373
2/2
✓ Branch 0 taken 3932160 times.
✓ Branch 1 taken 60 times.
 3932220 for (int i = im; i < iM; i++) {
374 3932160 td->he[j].sym = i;
375 3932160 td->he[j].len = td->freq[i] & 63;
376 3932160 td->he[j].code = td->freq[i] >> 6;
377
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3932160 times.
 3932160 if (td->he[j].len > 32) {
378 ✗ avpriv_request_sample(s->avctx, "Too big code length");
379 ✗ return AVERROR_PATCHWELCOME;
380 }
381
2/2
✓ Branch 0 taken 12774 times.
✓ Branch 1 taken 3919386 times.
 3932160 if (td->he[j].len > 0)
382 12774 j++;
383 else
384 3919386 td->run_sym = i;
385 }
386
387
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 60 times.
 60 if (im > 0)
388 ✗ td->run_sym = 0;
389
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 60 times.
 60 else if (iM < 65535)
390 ✗ td->run_sym = 65535;
391
392
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 60 times.
 60 if (td->run_sym == -1) {
393 ✗ avpriv_request_sample(s->avctx, "No place for run symbol");
394 ✗ return AVERROR_PATCHWELCOME;
395 }
396
397 60 td->he[j].sym = td->run_sym;
398 60 td->he[j].len = td->freq[iM] & 63;
399
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 60 times.
 60 if (td->he[j].len > 32) {
400 ✗ avpriv_request_sample(s->avctx, "Too big code length");
401 ✗ return AVERROR_PATCHWELCOME;
402 }
403 60 td->he[j].code = td->freq[iM] >> 6;
404 60 j++;
405
406 60 ff_free_vlc(&td->vlc);
407 60 return ff_init_vlc_sparse(&td->vlc, 12, j,
408 60 &td->he[0].len, sizeof(td->he[0]), sizeof(td->he[0].len),
409 60 &td->he[0].code, sizeof(td->he[0]), sizeof(td->he[0].code),
410 60 &td->he[0].sym, sizeof(td->he[0]), sizeof(td->he[0].sym), 0);
411 }
412
413 60 static int huf_decode(VLC *vlc, GetByteContext *gb, int nbits, int run_sym,
414 int no, uint16_t *out)
415 {
416 GetBitContext gbit;
417 60 int oe = 0;
418
419 60 init_get_bits(&gbit, gb->buffer, nbits);
420
3/4
✓ Branch 1 taken 426962 times.
✓ Branch 2 taken 60 times.
✓ Branch 3 taken 426962 times.
✗ Branch 4 not taken.
 427022 while (get_bits_left(&gbit) > 0 && oe < no) {
421 426962 uint16_t x = get_vlc2(&gbit, vlc->table, 12, 3);
422
423
2/2
✓ Branch 0 taken 6774 times.
✓ Branch 1 taken 420188 times.
 426962 if (x == run_sym) {
424 6774 int run = get_bits(&gbit, 8);
425 uint16_t fill;
426
427
2/4
✓ Branch 0 taken 6774 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 6774 times.
 6774 if (oe == 0 || oe + run > no)
428 ✗ return AVERROR_INVALIDDATA;
429
430 6774 fill = out[oe - 1];
431
432
2/2
✓ Branch 0 taken 1221256 times.
✓ Branch 1 taken 6774 times.
 1228030 while (run-- > 0)
433 1221256 out[oe++] = fill;
434 } else {
435 420188 out[oe++] = x;
436 }
437 }
438
439 60 return 0;
440 }
441
442 60 static int huf_uncompress(EXRContext *s,
443 EXRThreadData *td,
444 GetByteContext *gb,
445 uint16_t *dst, int dst_size)
446 {
447 int32_t im, iM;
448 uint32_t nBits;
449 int ret;
450
451 60 im = bytestream2_get_le32(gb);
452 60 iM = bytestream2_get_le32(gb);
453 60 bytestream2_skip(gb, 4);
454 60 nBits = bytestream2_get_le32(gb);
455
3/6
✓ Branch 0 taken 60 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 60 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 60 times.
✗ Branch 5 not taken.
 60 if (im < 0 || im >= HUF_ENCSIZE ||
456
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 60 times.
 60 iM < 0 || iM >= HUF_ENCSIZE)
457 ✗ return AVERROR_INVALIDDATA;
458
459 60 bytestream2_skip(gb, 4);
460
461
2/2
✓ Branch 0 taken 14 times.
✓ Branch 1 taken 46 times.
 60 if (!td->freq)
462 14 td->freq = av_malloc_array(HUF_ENCSIZE, sizeof(*td->freq));
463
2/2
✓ Branch 0 taken 14 times.
✓ Branch 1 taken 46 times.
 60 if (!td->he)
464 14 td->he = av_calloc(HUF_ENCSIZE, sizeof(*td->he));
465
2/4
✓ Branch 0 taken 60 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 60 times.
 60 if (!td->freq || !td->he) {
466 ✗ ret = AVERROR(ENOMEM);
467 ✗ return ret;
468 }
469
470 60 memset(td->freq, 0, sizeof(*td->freq) * HUF_ENCSIZE);
471
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 60 times.
 60 if ((ret = huf_unpack_enc_table(gb, im, iM, td->freq)) < 0)
472 ✗ return ret;
473
474
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 60 times.
 60 if (nBits > 8 * bytestream2_get_bytes_left(gb)) {
475 ✗ ret = AVERROR_INVALIDDATA;
476 ✗ return ret;
477 }
478
479
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 60 times.
 60 if ((ret = huf_build_dec_table(s, td, im, iM)) < 0)
480 ✗ return ret;
481 60 return huf_decode(&td->vlc, gb, nBits, td->run_sym, dst_size, dst);
482 }
483
484 2183418 static inline void wdec14(uint16_t l, uint16_t h, uint16_t *a, uint16_t *b)
485 {
486 2183418 int16_t ls = l;
487 2183418 int16_t hs = h;
488 2183418 int hi = hs;
489 2183418 int ai = ls + (hi & 1) + (hi >> 1);
490 2183418 int16_t as = ai;
491 2183418 int16_t bs = ai - hi;
492
493 2183418 *a = as;
494 2183418 *b = bs;
495 2183418 }
496
497 #define NBITS 16
498 #define A_OFFSET (1 << (NBITS - 1))
499 #define MOD_MASK ((1 << NBITS) - 1)
500
501 ✗ static inline void wdec16(uint16_t l, uint16_t h, uint16_t *a, uint16_t *b)
502 {
503 ✗ int m = l;
504 ✗ int d = h;
505 ✗ int bb = (m - (d >> 1)) & MOD_MASK;
506 ✗ int aa = (d + bb - A_OFFSET) & MOD_MASK;
507 ✗ *b = bb;
508 ✗ *a = aa;
509 }
510
511 204 static void wav_decode(uint16_t *in, int nx, int ox,
512 int ny, int oy, uint16_t mx)
513 {
514 204 int w14 = (mx < (1 << 14));
515 204 int n = (nx > ny) ? ny : nx;
516 204 int p = 1;
517 int p2;
518
519
2/2
✓ Branch 0 taken 1182 times.
✓ Branch 1 taken 204 times.
 1386 while (p <= n)
520 1182 p <<= 1;
521
522 204 p >>= 1;
523 204 p2 = p;
524 204 p >>= 1;
525
526
2/2
✓ Branch 0 taken 978 times.
✓ Branch 1 taken 204 times.
 1182 while (p >= 1) {
527 978 uint16_t *py = in;
528 978 uint16_t *ey = in + oy * (ny - p2);
529 uint16_t i00, i01, i10, i11;
530 978 int oy1 = oy * p;
531 978 int oy2 = oy * p2;
532 978 int ox1 = ox * p;
533 978 int ox2 = ox * p2;
534
535
2/2
✓ Branch 0 taken 5880 times.
✓ Branch 1 taken 978 times.
 6858 for (; py <= ey; py += oy2) {
536 5880 uint16_t *px = py;
537 5880 uint16_t *ex = py + ox * (nx - p2);
538
539
2/2
✓ Branch 0 taken 545502 times.
✓ Branch 1 taken 5880 times.
 551382 for (; px <= ex; px += ox2) {
540 545502 uint16_t *p01 = px + ox1;
541 545502 uint16_t *p10 = px + oy1;
542 545502 uint16_t *p11 = p10 + ox1;
543
544
1/2
✓ Branch 0 taken 545502 times.
✗ Branch 1 not taken.
 545502 if (w14) {
545 545502 wdec14(*px, *p10, &i00, &i10);
546 545502 wdec14(*p01, *p11, &i01, &i11);
547 545502 wdec14(i00, i01, px, p01);
548 545502 wdec14(i10, i11, p10, p11);
549 } else {
550 ✗ wdec16(*px, *p10, &i00, &i10);
551 ✗ wdec16(*p01, *p11, &i01, &i11);
552 ✗ wdec16(i00, i01, px, p01);
553 ✗ wdec16(i10, i11, p10, p11);
554 }
555 }
556
557
2/2
✓ Branch 0 taken 648 times.
✓ Branch 1 taken 5232 times.
 5880 if (nx & p) {
558 648 uint16_t *p10 = px + oy1;
559
560
1/2
✓ Branch 0 taken 648 times.
✗ Branch 1 not taken.
 648 if (w14)
561 648 wdec14(*px, *p10, &i00, p10);
562 else
563 ✗ wdec16(*px, *p10, &i00, p10);
564
565 648 *px = i00;
566 }
567 }
568
569
2/2
✓ Branch 0 taken 30 times.
✓ Branch 1 taken 948 times.
 978 if (ny & p) {
570 30 uint16_t *px = py;
571 30 uint16_t *ex = py + ox * (nx - p2);
572
573
2/2
✓ Branch 0 taken 762 times.
✓ Branch 1 taken 30 times.
 792 for (; px <= ex; px += ox2) {
574 762 uint16_t *p01 = px + ox1;
575
576
1/2
✓ Branch 0 taken 762 times.
✗ Branch 1 not taken.
 762 if (w14)
577 762 wdec14(*px, *p01, &i00, p01);
578 else
579 ✗ wdec16(*px, *p01, &i00, p01);
580
581 762 *px = i00;
582 }
583 }
584
585 978 p2 = p;
586 978 p >>= 1;
587 }
588 204 }
589
590 60 static int piz_uncompress(EXRContext *s, const uint8_t *src, int ssize,
591 int dsize, EXRThreadData *td)
592 {
593 GetByteContext gb;
594 uint16_t maxval, min_non_zero, max_non_zero;
595 uint16_t *ptr;
596 60 uint16_t *tmp = (uint16_t *)td->tmp;
597 uint16_t *out;
598 uint16_t *in;
599 int ret, i, j;
600 int pixel_half_size;/* 1 for half, 2 for float and uint32 */
601 EXRChannel *channel;
602 int tmp_offset;
603
604
2/2
✓ Branch 0 taken 14 times.
✓ Branch 1 taken 46 times.
 60 if (!td->bitmap)
605 14 td->bitmap = av_malloc(BITMAP_SIZE);
606
2/2
✓ Branch 0 taken 14 times.
✓ Branch 1 taken 46 times.
 60 if (!td->lut)
607 14 td->lut = av_malloc(1 << 17);
608
2/4
✓ Branch 0 taken 60 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 60 times.
 60 if (!td->bitmap || !td->lut) {
609 ✗ av_freep(&td->bitmap);
610 ✗ av_freep(&td->lut);
611 ✗ return AVERROR(ENOMEM);
612 }
613
614 60 bytestream2_init(&gb, src, ssize);
615 60 min_non_zero = bytestream2_get_le16(&gb);
616 60 max_non_zero = bytestream2_get_le16(&gb);
617
618
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 60 times.
 60 if (max_non_zero >= BITMAP_SIZE)
619 ✗ return AVERROR_INVALIDDATA;
620
621 60 memset(td->bitmap, 0, FFMIN(min_non_zero, BITMAP_SIZE));
622
1/2
✓ Branch 0 taken 60 times.
✗ Branch 1 not taken.
 60 if (min_non_zero <= max_non_zero)
623 60 bytestream2_get_buffer(&gb, td->bitmap + min_non_zero,
624 60 max_non_zero - min_non_zero + 1);
625 60 memset(td->bitmap + max_non_zero + 1, 0, BITMAP_SIZE - max_non_zero - 1);
626
627 60 maxval = reverse_lut(td->bitmap, td->lut);
628
629 60 bytestream2_skip(&gb, 4);
630 60 ret = huf_uncompress(s, td, &gb, tmp, dsize / sizeof(uint16_t));
631
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 60 times.
 60 if (ret)
632 ✗ return ret;
633
634 60 ptr = tmp;
635
2/2
✓ Branch 0 taken 180 times.
✓ Branch 1 taken 60 times.
 240 for (i = 0; i < s->nb_channels; i++) {
636 180 channel = &s->channels[i];
637
638
2/2
✓ Branch 0 taken 156 times.
✓ Branch 1 taken 24 times.
 180 if (channel->pixel_type == EXR_HALF)
639 156 pixel_half_size = 1;
640 else
641 24 pixel_half_size = 2;
642
643
2/2
✓ Branch 0 taken 204 times.
✓ Branch 1 taken 180 times.
 384 for (j = 0; j < pixel_half_size; j++)
644 204 wav_decode(ptr + j, td->xsize, pixel_half_size, td->ysize,
645 204 td->xsize * pixel_half_size, maxval);
646 180 ptr += td->xsize * td->ysize * pixel_half_size;
647 }
648
649 60 apply_lut(td->lut, tmp, dsize / sizeof(uint16_t));
650
651 60 out = (uint16_t *)td->uncompressed_data;
652
2/2
✓ Branch 0 taken 1782 times.
✓ Branch 1 taken 60 times.
 1842 for (i = 0; i < td->ysize; i++) {
653 1782 tmp_offset = 0;
654
2/2
✓ Branch 0 taken 5346 times.
✓ Branch 1 taken 1782 times.
 7128 for (j = 0; j < s->nb_channels; j++) {
655 5346 channel = &s->channels[j];
656
2/2
✓ Branch 0 taken 4578 times.
✓ Branch 1 taken 768 times.
 5346 if (channel->pixel_type == EXR_HALF)
657 4578 pixel_half_size = 1;
658 else
659 768 pixel_half_size = 2;
660
661 5346 in = tmp + tmp_offset * td->xsize * td->ysize + i * td->xsize * pixel_half_size;
662 5346 tmp_offset += pixel_half_size;
663
664 #if HAVE_BIGENDIAN
665 s->bbdsp.bswap16_buf(out, in, td->xsize * pixel_half_size);
666 #else
667 5346 memcpy(out, in, td->xsize * 2 * pixel_half_size);
668 #endif
669 5346 out += td->xsize * pixel_half_size;
670 }
671 }
672
673 60 return 0;
674 }
675
676 122 static int pxr24_uncompress(EXRContext *s, const uint8_t *src,
677 int compressed_size, int uncompressed_size,
678 EXRThreadData *td)
679 {
680 122 unsigned long dest_len, expected_len = 0;
681 122 const uint8_t *in = td->tmp;
682 uint8_t *out;
683 int c, i, j;
684
685
2/2
✓ Branch 0 taken 436 times.
✓ Branch 1 taken 122 times.
 558 for (i = 0; i < s->nb_channels; i++) {
686
2/2
✓ Branch 0 taken 62 times.
✓ Branch 1 taken 374 times.
 436 if (s->channels[i].pixel_type == EXR_FLOAT) {
687 62 expected_len += (td->xsize * td->ysize * 3);/* PRX 24 store float in 24 bit instead of 32 */
688
2/2
✓ Branch 0 taken 362 times.
✓ Branch 1 taken 12 times.
 374 } else if (s->channels[i].pixel_type == EXR_HALF) {
689 362 expected_len += (td->xsize * td->ysize * 2);
690 } else {//UINT 32
691 12 expected_len += (td->xsize * td->ysize * 4);
692 }
693 }
694
695 122 dest_len = expected_len;
696
697
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 122 times.
 122 if (uncompress(td->tmp, &dest_len, src, compressed_size) != Z_OK) {
698 ✗ return AVERROR_INVALIDDATA;
699
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 122 times.
 122 } else if (dest_len != expected_len) {
700 ✗ return AVERROR_INVALIDDATA;
701 }
702
703 122 out = td->uncompressed_data;
704
2/2
✓ Branch 0 taken 1778 times.
✓ Branch 1 taken 122 times.
 1900 for (i = 0; i < td->ysize; i++)
705
2/2
✓ Branch 0 taken 5900 times.
✓ Branch 1 taken 1778 times.
 7678 for (c = 0; c < s->nb_channels; c++) {
706 5900 EXRChannel *channel = &s->channels[c];
707 const uint8_t *ptr[4];
708 5900 uint32_t pixel = 0;
709
710
3/4
✓ Branch 0 taken 496 times.
✓ Branch 1 taken 5302 times.
✓ Branch 2 taken 102 times.
✗ Branch 3 not taken.
 5900 switch (channel->pixel_type) {
711 496 case EXR_FLOAT:
712 496 ptr[0] = in;
713 496 ptr[1] = ptr[0] + td->xsize;
714 496 ptr[2] = ptr[1] + td->xsize;
715 496 in = ptr[2] + td->xsize;
716
717
2/2
✓ Branch 0 taken 5952 times.
✓ Branch 1 taken 496 times.
 6448 for (j = 0; j < td->xsize; ++j) {
718 5952 uint32_t diff = ((unsigned)*(ptr[0]++) << 24) |
719 5952 (*(ptr[1]++) << 16) |
720 5952 (*(ptr[2]++) << 8);
721 5952 pixel += diff;
722 5952 bytestream_put_le32(&out, pixel);
723 }
724 496 break;
725 5302 case EXR_HALF:
726 5302 ptr[0] = in;
727 5302 ptr[1] = ptr[0] + td->xsize;
728 5302 in = ptr[1] + td->xsize;
729
2/2
✓ Branch 0 taken 3846078 times.
✓ Branch 1 taken 5302 times.
 3851380 for (j = 0; j < td->xsize; j++) {
730 3846078 uint32_t diff = (*(ptr[0]++) << 8) | *(ptr[1]++);
731
732 3846078 pixel += diff;
733 3846078 bytestream_put_le16(&out, pixel);
734 }
735 5302 break;
736 102 case EXR_UINT:
737 102 ptr[0] = in;
738 102 ptr[1] = ptr[0] + s->xdelta;
739 102 ptr[2] = ptr[1] + s->xdelta;
740 102 ptr[3] = ptr[2] + s->xdelta;
741 102 in = ptr[3] + s->xdelta;
742
743
2/2
✓ Branch 0 taken 1278 times.
✓ Branch 1 taken 102 times.
 1380 for (j = 0; j < s->xdelta; ++j) {
744 1278 uint32_t diff = ((uint32_t)*(ptr[0]++) << 24) |
745 1278 (*(ptr[1]++) << 16) |
746 1278 (*(ptr[2]++) << 8 ) |
747 1278 (*(ptr[3]++));
748 1278 pixel += diff;
749 1278 bytestream_put_le32(&out, pixel);
750 }
751 102 break;
752 ✗ default:
753 ✗ return AVERROR_INVALIDDATA;
754 }
755 }
756
757 122 return 0;
758 }
759
760 118558 static void unpack_14(const uint8_t b[14], uint16_t s[16])
761 {
762 118558 unsigned short shift = (b[ 2] >> 2) & 15;
763 118558 unsigned short bias = (0x20 << shift);
764 int i;
765
766 118558 s[ 0] = (b[0] << 8) | b[1];
767
768 118558 s[ 4] = s[ 0] + ((((b[ 2] << 4) | (b[ 3] >> 4)) & 0x3f) << shift) - bias;
769 118558 s[ 8] = s[ 4] + ((((b[ 3] << 2) | (b[ 4] >> 6)) & 0x3f) << shift) - bias;
770 118558 s[12] = s[ 8] + ((b[ 4] & 0x3f) << shift) - bias;
771
772 118558 s[ 1] = s[ 0] + ((b[ 5] >> 2) << shift) - bias;
773 118558 s[ 5] = s[ 4] + ((((b[ 5] << 4) | (b[ 6] >> 4)) & 0x3f) << shift) - bias;
774 118558 s[ 9] = s[ 8] + ((((b[ 6] << 2) | (b[ 7] >> 6)) & 0x3f) << shift) - bias;
775 118558 s[13] = s[12] + ((b[ 7] & 0x3f) << shift) - bias;
776
777 118558 s[ 2] = s[ 1] + ((b[ 8] >> 2) << shift) - bias;
778 118558 s[ 6] = s[ 5] + ((((b[ 8] << 4) | (b[ 9] >> 4)) & 0x3f) << shift) - bias;
779 118558 s[10] = s[ 9] + ((((b[ 9] << 2) | (b[10] >> 6)) & 0x3f) << shift) - bias;
780 118558 s[14] = s[13] + ((b[10] & 0x3f) << shift) - bias;
781
782 118558 s[ 3] = s[ 2] + ((b[11] >> 2) << shift) - bias;
783 118558 s[ 7] = s[ 6] + ((((b[11] << 4) | (b[12] >> 4)) & 0x3f) << shift) - bias;
784 118558 s[11] = s[10] + ((((b[12] << 2) | (b[13] >> 6)) & 0x3f) << shift) - bias;
785 118558 s[15] = s[14] + ((b[13] & 0x3f) << shift) - bias;
786
787
2/2
✓ Branch 0 taken 1896928 times.
✓ Branch 1 taken 118558 times.
 2015486 for (i = 0; i < 16; ++i) {
788
2/2
✓ Branch 0 taken 1884694 times.
✓ Branch 1 taken 12234 times.
 1896928 if (s[i] & 0x8000)
789 1884694 s[i] &= 0x7fff;
790 else
791 12234 s[i] = ~s[i];
792 }
793 118558 }
794
795 212 static void unpack_3(const uint8_t b[3], uint16_t s[16])
796 {
797 int i;
798
799 212 s[0] = (b[0] << 8) | b[1];
800
801
2/2
✓ Branch 0 taken 210 times.
✓ Branch 1 taken 2 times.
 212 if (s[0] & 0x8000)
802 210 s[0] &= 0x7fff;
803 else
804 2 s[0] = ~s[0];
805
806
2/2
✓ Branch 0 taken 3180 times.
✓ Branch 1 taken 212 times.
 3392 for (i = 1; i < 16; i++)
807 3180 s[i] = s[0];
808 212 }
809
810
811 46 static int b44_uncompress(EXRContext *s, const uint8_t *src, int compressed_size,
812 int uncompressed_size, EXRThreadData *td) {
813 46 const int8_t *sr = src;
814 46 int stay_to_uncompress = compressed_size;
815 int nb_b44_block_w, nb_b44_block_h;
816 int index_tl_x, index_tl_y, index_out, index_tmp;
817 uint16_t tmp_buffer[16]; /* B44 use 4x4 half float pixel */
818 int c, iY, iX, y, x;
819 46 int target_channel_offset = 0;
820
821 /* calc B44 block count */
822 46 nb_b44_block_w = td->xsize / 4;
823
2/2
✓ Branch 0 taken 26 times.
✓ Branch 1 taken 20 times.
 46 if ((td->xsize % 4) != 0)
824 26 nb_b44_block_w++;
825
826 46 nb_b44_block_h = td->ysize / 4;
827
2/2
✓ Branch 0 taken 12 times.
✓ Branch 1 taken 34 times.
 46 if ((td->ysize % 4) != 0)
828 12 nb_b44_block_h++;
829
830
2/2
✓ Branch 0 taken 266 times.
✓ Branch 1 taken 46 times.
 312 for (c = 0; c < s->nb_channels; c++) {
831
2/2
✓ Branch 0 taken 170 times.
✓ Branch 1 taken 96 times.
 266 if (s->channels[c].pixel_type == EXR_HALF) {/* B44 only compress half float data */
832
2/2
✓ Branch 0 taken 748 times.
✓ Branch 1 taken 170 times.
 918 for (iY = 0; iY < nb_b44_block_h; iY++) {
833
2/2
✓ Branch 0 taken 118770 times.
✓ Branch 1 taken 748 times.
 119518 for (iX = 0; iX < nb_b44_block_w; iX++) {/* For each B44 block */
834
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 118770 times.
 118770 if (stay_to_uncompress < 3) {
835 ✗ av_log(s, AV_LOG_ERROR, "Not enough data for B44A block: %d", stay_to_uncompress);
836 ✗ return AVERROR_INVALIDDATA;
837 }
838
839
2/2
✓ Branch 0 taken 212 times.
✓ Branch 1 taken 118558 times.
 118770 if (src[compressed_size - stay_to_uncompress + 2] == 0xfc) { /* B44A block */
840 212 unpack_3(sr, tmp_buffer);
841 212 sr += 3;
842 212 stay_to_uncompress -= 3;
843 } else {/* B44 Block */
844
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 118558 times.
 118558 if (stay_to_uncompress < 14) {
845 ✗ av_log(s, AV_LOG_ERROR, "Not enough data for B44 block: %d", stay_to_uncompress);
846 ✗ return AVERROR_INVALIDDATA;
847 }
848 118558 unpack_14(sr, tmp_buffer);
849 118558 sr += 14;
850 118558 stay_to_uncompress -= 14;
851 }
852
853 /* copy data to uncompress buffer (B44 block can exceed target resolution)*/
854 118770 index_tl_x = iX * 4;
855 118770 index_tl_y = iY * 4;
856
857
4/4
✓ Branch 0 taken 7792 times.
✓ Branch 1 taken 583730 times.
✓ Branch 2 taken 472752 times.
✓ Branch 3 taken 118770 times.
 591522 for (y = index_tl_y; y < FFMIN(index_tl_y + 4, td->ysize); y++) {
858
4/4
✓ Branch 0 taken 8716 times.
✓ Branch 1 taken 2352010 times.
✓ Branch 2 taken 1887974 times.
✓ Branch 3 taken 472752 times.
 2360726 for (x = index_tl_x; x < FFMIN(index_tl_x + 4, td->xsize); x++) {
859 1887974 index_out = target_channel_offset * td->xsize + y * td->channel_line_size + 2 * x;
860 1887974 index_tmp = (y-index_tl_y) * 4 + (x-index_tl_x);
861 1887974 td->uncompressed_data[index_out] = tmp_buffer[index_tmp] & 0xff;
862 1887974 td->uncompressed_data[index_out + 1] = tmp_buffer[index_tmp] >> 8;
863 }
864 }
865 }
866 }
867 170 target_channel_offset += 2;
868 } else {/* Float or UINT 32 channel */
869
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 96 times.
 96 if (stay_to_uncompress < td->ysize * td->xsize * 4) {
870 ✗ av_log(s, AV_LOG_ERROR, "Not enough data for uncompress channel: %d", stay_to_uncompress);
871 ✗ return AVERROR_INVALIDDATA;
872 }
873
874
2/2
✓ Branch 0 taken 800 times.
✓ Branch 1 taken 96 times.
 896 for (y = 0; y < td->ysize; y++) {
875 800 index_out = target_channel_offset * td->xsize + y * td->channel_line_size;
876 800 memcpy(&td->uncompressed_data[index_out], sr, td->xsize * 4);
877 800 sr += td->xsize * 4;
878 }
879 96 target_channel_offset += 4;
880
881 96 stay_to_uncompress -= td->ysize * td->xsize * 4;
882 }
883 }
884
885 46 return 0;
886 }
887
888 ✗ static int ac_uncompress(EXRContext *s, GetByteContext *gb, float *block)
889 {
890 ✗ int ret = 0, n = 1;
891
892 ✗ while (n < 64) {
893 ✗ uint16_t val = bytestream2_get_ne16(gb);
894
895 ✗ if (val == 0xff00) {
896 ✗ n = 64;
897 ✗ } else if ((val >> 8) == 0xff) {
898 ✗ n += val & 0xff;
899 } else {
900 ✗ ret = n;
901 ✗ block[ff_zigzag_direct[n]] = av_int2float(half2float(val,
902 ✗ s->mantissatable,
903 ✗ s->exponenttable,
904 ✗ s->offsettable));
905 ✗ n++;
906 }
907 }
908
909 ✗ return ret;
910 }
911
912 ✗ static void idct_1d(float *blk, int step)
913 {
914 ✗ const float a = .5f * cosf( M_PI / 4.f);
915 ✗ const float b = .5f * cosf( M_PI / 16.f);
916 ✗ const float c = .5f * cosf( M_PI / 8.f);
917 ✗ const float d = .5f * cosf(3.f*M_PI / 16.f);
918 ✗ const float e = .5f * cosf(5.f*M_PI / 16.f);
919 ✗ const float f = .5f * cosf(3.f*M_PI / 8.f);
920 ✗ const float g = .5f * cosf(7.f*M_PI / 16.f);
921
922 float alpha[4], beta[4], theta[4], gamma[4];
923
924 ✗ alpha[0] = c * blk[2 * step];
925 ✗ alpha[1] = f * blk[2 * step];
926 ✗ alpha[2] = c * blk[6 * step];
927 ✗ alpha[3] = f * blk[6 * step];
928
929 ✗ beta[0] = b * blk[1 * step] + d * blk[3 * step] + e * blk[5 * step] + g * blk[7 * step];
930 ✗ beta[1] = d * blk[1 * step] - g * blk[3 * step] - b * blk[5 * step] - e * blk[7 * step];
931 ✗ beta[2] = e * blk[1 * step] - b * blk[3 * step] + g * blk[5 * step] + d * blk[7 * step];
932 ✗ beta[3] = g * blk[1 * step] - e * blk[3 * step] + d * blk[5 * step] - b * blk[7 * step];
933
934 ✗ theta[0] = a * (blk[0 * step] + blk[4 * step]);
935 ✗ theta[3] = a * (blk[0 * step] - blk[4 * step]);
936
937 ✗ theta[1] = alpha[0] + alpha[3];
938 ✗ theta[2] = alpha[1] - alpha[2];
939
940 ✗ gamma[0] = theta[0] + theta[1];
941 ✗ gamma[1] = theta[3] + theta[2];
942 ✗ gamma[2] = theta[3] - theta[2];
943 ✗ gamma[3] = theta[0] - theta[1];
944
945 ✗ blk[0 * step] = gamma[0] + beta[0];
946 ✗ blk[1 * step] = gamma[1] + beta[1];
947 ✗ blk[2 * step] = gamma[2] + beta[2];
948 ✗ blk[3 * step] = gamma[3] + beta[3];
949
950 ✗ blk[4 * step] = gamma[3] - beta[3];
951 ✗ blk[5 * step] = gamma[2] - beta[2];
952 ✗ blk[6 * step] = gamma[1] - beta[1];
953 ✗ blk[7 * step] = gamma[0] - beta[0];
954 }
955
956 ✗ static void dct_inverse(float *block)
957 {
958 ✗ for (int i = 0; i < 8; i++)
959 ✗ idct_1d(block + i, 8);
960
961 ✗ for (int i = 0; i < 8; i++) {
962 ✗ idct_1d(block, 1);
963 ✗ block += 8;
964 }
965 }
966
967 ✗ static void convert(float y, float u, float v,
968 float *b, float *g, float *r)
969 {
970 ✗ *r = y + 1.5747f * v;
971 ✗ *g = y - 0.1873f * u - 0.4682f * v;
972 ✗ *b = y + 1.8556f * u;
973 }
974
975 ✗ static float to_linear(float x, float scale)
976 {
977 ✗ float ax = fabsf(x);
978
979 ✗ if (ax <= 1.f) {
980 ✗ return FFSIGN(x) * powf(ax, 2.2f * scale);
981 } else {
982 ✗ const float log_base = expf(2.2f * scale);
983
984 ✗ return FFSIGN(x) * powf(log_base, ax - 1.f);
985 }
986 }
987
988 ✗ static int dwa_uncompress(EXRContext *s, const uint8_t *src, int compressed_size,
989 int uncompressed_size, EXRThreadData *td)
990 {
991 int64_t version, lo_usize, lo_size;
992 int64_t ac_size, dc_size, rle_usize, rle_csize, rle_raw_size;
993 int64_t ac_count, dc_count, ac_compression;
994 ✗ const int dc_w = td->xsize >> 3;
995 ✗ const int dc_h = td->ysize >> 3;
996 GetByteContext gb, agb;
997 int skip, ret;
998
999 ✗ if (compressed_size <= 88)
1000 ✗ return AVERROR_INVALIDDATA;
1001
1002 ✗ version = AV_RL64(src + 0);
1003 ✗ if (version != 2)
1004 ✗ return AVERROR_INVALIDDATA;
1005
1006 ✗ lo_usize = AV_RL64(src + 8);
1007 ✗ lo_size = AV_RL64(src + 16);
1008 ✗ ac_size = AV_RL64(src + 24);
1009 ✗ dc_size = AV_RL64(src + 32);
1010 ✗ rle_csize = AV_RL64(src + 40);
1011 ✗ rle_usize = AV_RL64(src + 48);
1012 ✗ rle_raw_size = AV_RL64(src + 56);
1013 ✗ ac_count = AV_RL64(src + 64);
1014 ✗ dc_count = AV_RL64(src + 72);
1015 ✗ ac_compression = AV_RL64(src + 80);
1016
1017 ✗ if ( compressed_size < (uint64_t)(lo_size | ac_size | dc_size | rle_csize) || compressed_size < 88LL + lo_size + ac_size + dc_size + rle_csize
1018 ✗ || ac_count > (uint64_t)INT_MAX/2
1019 )
1020 ✗ return AVERROR_INVALIDDATA;
1021
1022 ✗ bytestream2_init(&gb, src + 88, compressed_size - 88);
1023 ✗ skip = bytestream2_get_le16(&gb);
1024 ✗ if (skip < 2)
1025 ✗ return AVERROR_INVALIDDATA;
1026
1027 ✗ bytestream2_skip(&gb, skip - 2);
1028
1029 ✗ if (lo_size > 0) {
1030 ✗ if (lo_usize > uncompressed_size)
1031 ✗ return AVERROR_INVALIDDATA;
1032 ✗ bytestream2_skip(&gb, lo_size);
1033 }
1034
1035 ✗ if (ac_size > 0) {
1036 unsigned long dest_len;
1037 ✗ GetByteContext agb = gb;
1038
1039 ✗ if (ac_count > 3LL * td->xsize * s->scan_lines_per_block)
1040 ✗ return AVERROR_INVALIDDATA;
1041
1042 ✗ dest_len = ac_count * 2LL;
1043
1044 ✗ av_fast_padded_malloc(&td->ac_data, &td->ac_size, dest_len);
1045 ✗ if (!td->ac_data)
1046 ✗ return AVERROR(ENOMEM);
1047
1048 ✗ switch (ac_compression) {
1049 ✗ case 0:
1050 ✗ ret = huf_uncompress(s, td, &agb, (int16_t *)td->ac_data, ac_count);
1051 ✗ if (ret < 0)
1052 ✗ return ret;
1053 ✗ break;
1054 ✗ case 1:
1055 ✗ if (uncompress(td->ac_data, &dest_len, agb.buffer, ac_size) != Z_OK ||
1056 ✗ dest_len != ac_count * 2LL)
1057 ✗ return AVERROR_INVALIDDATA;
1058 ✗ break;
1059 ✗ default:
1060 ✗ return AVERROR_INVALIDDATA;
1061 }
1062
1063 ✗ bytestream2_skip(&gb, ac_size);
1064 }
1065
1066 {
1067 unsigned long dest_len;
1068 ✗ GetByteContext agb = gb;
1069
1070 ✗ if (dc_count != dc_w * dc_h * 3)
1071 ✗ return AVERROR_INVALIDDATA;
1072
1073 ✗ dest_len = dc_count * 2LL;
1074
1075 ✗ av_fast_padded_malloc(&td->dc_data, &td->dc_size, FFALIGN(dest_len, 64) * 2);
1076 ✗ if (!td->dc_data)
1077 ✗ return AVERROR(ENOMEM);
1078
1079 ✗ if (uncompress(td->dc_data + FFALIGN(dest_len, 64), &dest_len, agb.buffer, dc_size) != Z_OK ||
1080 ✗ (dest_len != dc_count * 2LL))
1081 ✗ return AVERROR_INVALIDDATA;
1082
1083 ✗ s->dsp.predictor(td->dc_data + FFALIGN(dest_len, 64), dest_len);
1084 ✗ s->dsp.reorder_pixels(td->dc_data, td->dc_data + FFALIGN(dest_len, 64), dest_len);
1085
1086 ✗ bytestream2_skip(&gb, dc_size);
1087 }
1088
1089 ✗ if (rle_raw_size > 0 && rle_csize > 0 && rle_usize > 0) {
1090 ✗ unsigned long dest_len = rle_usize;
1091
1092 ✗ av_fast_padded_malloc(&td->rle_data, &td->rle_size, rle_usize);
1093 ✗ if (!td->rle_data)
1094 ✗ return AVERROR(ENOMEM);
1095
1096 ✗ av_fast_padded_malloc(&td->rle_raw_data, &td->rle_raw_size, rle_raw_size);
1097 ✗ if (!td->rle_raw_data)
1098 ✗ return AVERROR(ENOMEM);
1099
1100 ✗ if (uncompress(td->rle_data, &dest_len, gb.buffer, rle_csize) != Z_OK ||
1101 ✗ (dest_len != rle_usize))
1102 ✗ return AVERROR_INVALIDDATA;
1103
1104 ✗ ret = rle(td->rle_raw_data, td->rle_data, rle_usize, rle_raw_size);
1105 ✗ if (ret < 0)
1106 ✗ return ret;
1107 ✗ bytestream2_skip(&gb, rle_csize);
1108 }
1109
1110 ✗ bytestream2_init(&agb, td->ac_data, ac_count * 2);
1111
1112 ✗ for (int y = 0; y < td->ysize; y += 8) {
1113 ✗ for (int x = 0; x < td->xsize; x += 8) {
1114 ✗ memset(td->block, 0, sizeof(td->block));
1115
1116 ✗ for (int j = 0; j < 3; j++) {
1117 ✗ float *block = td->block[j];
1118 ✗ const int idx = (x >> 3) + (y >> 3) * dc_w + dc_w * dc_h * j;
1119 ✗ uint16_t *dc = (uint16_t *)td->dc_data;
1120 union av_intfloat32 dc_val;
1121
1122 ✗ dc_val.i = half2float(dc[idx], s->mantissatable,
1123 ✗ s->exponenttable, s->offsettable);
1124
1125 ✗ block[0] = dc_val.f;
1126 ✗ ac_uncompress(s, &agb, block);
1127 ✗ dct_inverse(block);
1128 }
1129
1130 {
1131 ✗ const float scale = s->pixel_type == EXR_FLOAT ? 2.f : 1.f;
1132 ✗ const int o = s->nb_channels == 4;
1133 ✗ float *bo = ((float *)td->uncompressed_data) +
1134 ✗ y * td->xsize * s->nb_channels + td->xsize * (o + 0) + x;
1135 ✗ float *go = ((float *)td->uncompressed_data) +
1136 ✗ y * td->xsize * s->nb_channels + td->xsize * (o + 1) + x;
1137 ✗ float *ro = ((float *)td->uncompressed_data) +
1138 ✗ y * td->xsize * s->nb_channels + td->xsize * (o + 2) + x;
1139 ✗ float *yb = td->block[0];
1140 ✗ float *ub = td->block[1];
1141 ✗ float *vb = td->block[2];
1142
1143 ✗ for (int yy = 0; yy < 8; yy++) {
1144 ✗ for (int xx = 0; xx < 8; xx++) {
1145 ✗ const int idx = xx + yy * 8;
1146
1147 ✗ convert(yb[idx], ub[idx], vb[idx], &bo[xx], &go[xx], &ro[xx]);
1148
1149 ✗ bo[xx] = to_linear(bo[xx], scale);
1150 ✗ go[xx] = to_linear(go[xx], scale);
1151 ✗ ro[xx] = to_linear(ro[xx], scale);
1152 }
1153
1154 ✗ bo += td->xsize * s->nb_channels;
1155 ✗ go += td->xsize * s->nb_channels;
1156 ✗ ro += td->xsize * s->nb_channels;
1157 }
1158 }
1159 }
1160 }
1161
1162 ✗ if (s->nb_channels < 4)
1163 ✗ return 0;
1164
1165 ✗ for (int y = 0; y < td->ysize && td->rle_raw_data; y++) {
1166 ✗ uint32_t *ao = ((uint32_t *)td->uncompressed_data) + y * td->xsize * s->nb_channels;
1167 ✗ uint8_t *ai0 = td->rle_raw_data + y * td->xsize;
1168 ✗ uint8_t *ai1 = td->rle_raw_data + y * td->xsize + rle_raw_size / 2;
1169
1170 ✗ for (int x = 0; x < td->xsize; x++) {
1171 ✗ uint16_t ha = ai0[x] | (ai1[x] << 8);
1172
1173 ✗ ao[x] = half2float(ha, s->mantissatable, s->exponenttable, s->offsettable);
1174 }
1175 }
1176
1177 ✗ return 0;
1178 }
1179
1180 5738 static int decode_block(AVCodecContext *avctx, void *tdata,
1181 int jobnr, int threadnr)
1182 {
1183 5738 EXRContext *s = avctx->priv_data;
1184 5738 AVFrame *const p = s->picture;
1185 5738 EXRThreadData *td = &s->thread_data[threadnr];
1186 5738 const uint8_t *channel_buffer[4] = { 0 };
1187 5738 const uint8_t *buf = s->buf;
1188 uint64_t line_offset, uncompressed_size;
1189 uint8_t *ptr;
1190 uint32_t data_size;
1191 5738 int line, col = 0;
1192 uint64_t tile_x, tile_y, tile_level_x, tile_level_y;
1193 const uint8_t *src;
1194
2/2
✓ Branch 0 taken 22 times.
✓ Branch 1 taken 5716 times.
 5738 int step = s->desc->flags & AV_PIX_FMT_FLAG_FLOAT ? 4 : 2 * s->desc->nb_components;
1195 5738 int bxmin = 0, axmax = 0, window_xoffset = 0;
1196 int window_xmin, window_xmax, window_ymin, window_ymax;
1197 int data_xoffset, data_yoffset, data_window_offset, xsize, ysize;
1198 5738 int i, x, buf_size = s->buf_size;
1199 int c, rgb_channel_count;
1200 5738 float one_gamma = 1.0f / s->gamma;
1201 5738 avpriv_trc_function trc_func = avpriv_get_trc_function_from_trc(s->apply_trc_type);
1202 int ret;
1203
1204 5738 line_offset = AV_RL64(s->gb.buffer + jobnr * 8);
1205
1206
2/2
✓ Branch 0 taken 208 times.
✓ Branch 1 taken 5530 times.
 5738 if (s->is_tile) {
1207
2/4
✓ Branch 0 taken 208 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 208 times.
 208 if (buf_size < 20 || line_offset > buf_size - 20)
1208 ✗ return AVERROR_INVALIDDATA;
1209
1210 208 src = buf + line_offset + 20;
1211
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 208 times.
 208 if (s->is_multipart)
1212 ✗ src += 4;
1213
1214 208 tile_x = AV_RL32(src - 20);
1215 208 tile_y = AV_RL32(src - 16);
1216 208 tile_level_x = AV_RL32(src - 12);
1217 208 tile_level_y = AV_RL32(src - 8);
1218
1219 208 data_size = AV_RL32(src - 4);
1220
2/4
✓ Branch 0 taken 208 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 208 times.
 208 if (data_size <= 0 || data_size > buf_size - line_offset - 20)
1221 ✗ return AVERROR_INVALIDDATA;
1222
1223
2/4
✓ Branch 0 taken 208 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 208 times.
 208 if (tile_level_x || tile_level_y) { /* tile level, is not the full res level */
1224 ✗ avpriv_report_missing_feature(s->avctx, "Subres tile before full res tile");
1225 ✗ return AVERROR_PATCHWELCOME;
1226 }
1227
1228
3/4
✓ Branch 0 taken 140 times.
✓ Branch 1 taken 68 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 140 times.
 208 if (tile_x && s->tile_attr.xSize + (int64_t)FFMAX(s->xmin, 0) >= INT_MAX / tile_x )
1229 ✗ return AVERROR_INVALIDDATA;
1230
3/4
✓ Branch 0 taken 132 times.
✓ Branch 1 taken 76 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 132 times.
 208 if (tile_y && s->tile_attr.ySize + (int64_t)FFMAX(s->ymin, 0) >= INT_MAX / tile_y )
1231 ✗ return AVERROR_INVALIDDATA;
1232
1233 208 line = s->ymin + s->tile_attr.ySize * tile_y;
1234 208 col = s->tile_attr.xSize * tile_x;
1235
1236
2/4
✓ Branch 0 taken 208 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 208 times.
✗ Branch 3 not taken.
 208 if (line < s->ymin || line > s->ymax ||
1237
2/4
✓ Branch 0 taken 208 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 208 times.
 208 s->xmin + col < s->xmin || s->xmin + col > s->xmax)
1238 ✗ return AVERROR_INVALIDDATA;
1239
1240 208 td->ysize = FFMIN(s->tile_attr.ySize, s->ydelta - tile_y * s->tile_attr.ySize);
1241 208 td->xsize = FFMIN(s->tile_attr.xSize, s->xdelta - tile_x * s->tile_attr.xSize);
1242
1243
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 208 times.
 208 if (td->xsize * (uint64_t)s->current_channel_offset > INT_MAX)
1244 ✗ return AVERROR_INVALIDDATA;
1245
1246 208 td->channel_line_size = td->xsize * s->current_channel_offset;/* uncompress size of one line */
1247 208 uncompressed_size = td->channel_line_size * (uint64_t)td->ysize;/* uncompress size of the block */
1248 } else {
1249
2/4
✓ Branch 0 taken 5530 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 5530 times.
 5530 if (buf_size < 8 || line_offset > buf_size - 8)
1250 ✗ return AVERROR_INVALIDDATA;
1251
1252 5530 src = buf + line_offset + 8;
1253
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 5530 times.
 5530 if (s->is_multipart)
1254 ✗ src += 4;
1255 5530 line = AV_RL32(src - 8);
1256
1257
2/4
✓ Branch 0 taken 5530 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 5530 times.
 5530 if (line < s->ymin || line > s->ymax)
1258 ✗ return AVERROR_INVALIDDATA;
1259
1260 5530 data_size = AV_RL32(src - 4);
1261
2/4
✓ Branch 0 taken 5530 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 5530 times.
 5530 if (data_size <= 0 || data_size > buf_size - line_offset - 8)
1262 ✗ return AVERROR_INVALIDDATA;
1263
1264 5530 td->ysize = FFMIN(s->scan_lines_per_block, s->ymax - line + 1); /* s->ydelta - line ?? */
1265 5530 td->xsize = s->xdelta;
1266
1267
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 5530 times.
 5530 if (td->xsize * (uint64_t)s->current_channel_offset > INT_MAX)
1268 ✗ return AVERROR_INVALIDDATA;
1269
1270 5530 td->channel_line_size = td->xsize * s->current_channel_offset;/* uncompress size of one line */
1271 5530 uncompressed_size = td->channel_line_size * (uint64_t)td->ysize;/* uncompress size of the block */
1272
1273
3/4
✓ Branch 0 taken 1390 times.
✓ Branch 1 taken 4140 times.
✓ Branch 2 taken 1390 times.
✗ Branch 3 not taken.
 5530 if ((s->compression == EXR_RAW && (data_size != uncompressed_size ||
1274
1/2
✓ Branch 0 taken 1390 times.
✗ Branch 1 not taken.
 1390 line_offset > buf_size - uncompressed_size)) ||
1275
3/4
✓ Branch 0 taken 4140 times.
✓ Branch 1 taken 1390 times.
✓ Branch 2 taken 4140 times.
✗ Branch 3 not taken.
 5530 (s->compression != EXR_RAW && (data_size > uncompressed_size ||
1276
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4140 times.
 4140 line_offset > buf_size - data_size))) {
1277 ✗ return AVERROR_INVALIDDATA;
1278 }
1279 }
1280
1281 5738 window_xmin = FFMIN(avctx->width, FFMAX(0, s->xmin + col));
1282 5738 window_xmax = FFMIN(avctx->width, FFMAX(0, s->xmin + col + td->xsize));
1283 5738 window_ymin = FFMIN(avctx->height, FFMAX(0, line ));
1284 5738 window_ymax = FFMIN(avctx->height, FFMAX(0, line + td->ysize));
1285 5738 xsize = window_xmax - window_xmin;
1286 5738 ysize = window_ymax - window_ymin;
1287
1288 /* tile or scanline not visible skip decoding */
1289
4/4
✓ Branch 0 taken 5678 times.
✓ Branch 1 taken 60 times.
✓ Branch 2 taken 806 times.
✓ Branch 3 taken 4872 times.
 5738 if (xsize <= 0 || ysize <= 0)
1290 866 return 0;
1291
1292 /* is the first tile or is a scanline */
1293
2/2
✓ Branch 0 taken 4792 times.
✓ Branch 1 taken 80 times.
 4872 if(col == 0) {
1294 4792 window_xmin = 0;
1295 /* pixels to add at the left of the display window */
1296 4792 window_xoffset = FFMAX(0, s->xmin);
1297 /* bytes to add at the left of the display window */
1298 4792 bxmin = window_xoffset * step;
1299 }
1300
1301 /* is the last tile or is a scanline */
1302
2/2
✓ Branch 0 taken 4792 times.
✓ Branch 1 taken 80 times.
 4872 if(col + td->xsize == s->xdelta) {
1303 4792 window_xmax = avctx->width;
1304 /* bytes to add at the right of the display window */
1305 4792 axmax = FFMAX(0, (avctx->width - (s->xmax + 1))) * step;
1306 }
1307
1308
2/4
✓ Branch 0 taken 4872 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 4872 times.
 4872 if (avctx->max_pixels && uncompressed_size > avctx->max_pixels * 16LL)
1309 ✗ return AVERROR_INVALIDDATA;
1310
1311
4/4
✓ Branch 0 taken 1420 times.
✓ Branch 1 taken 3452 times.
✓ Branch 2 taken 26 times.
✓ Branch 3 taken 1394 times.
 4872 if (data_size < uncompressed_size || s->is_tile) { /* td->tmp is use for tile reorganization */
1312 3478 av_fast_padded_malloc(&td->tmp, &td->tmp_size, uncompressed_size);
1313
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3478 times.
 3478 if (!td->tmp)
1314 ✗ return AVERROR(ENOMEM);
1315 }
1316
1317
2/2
✓ Branch 0 taken 3452 times.
✓ Branch 1 taken 1420 times.
 4872 if (data_size < uncompressed_size) {
1318 3452 av_fast_padded_malloc(&td->uncompressed_data,
1319 3452 &td->uncompressed_size, uncompressed_size + 64);/* Force 64 padding for AVX2 reorder_pixels dst */
1320
1321
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3452 times.
 3452 if (!td->uncompressed_data)
1322 ✗ return AVERROR(ENOMEM);
1323
1324 3452 ret = AVERROR_INVALIDDATA;
1325
5/7
✓ Branch 0 taken 1838 times.
✓ Branch 1 taken 60 times.
✓ Branch 2 taken 122 times.
✓ Branch 3 taken 1386 times.
✓ Branch 4 taken 46 times.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
 3452 switch (s->compression) {
1326 1838 case EXR_ZIP1:
1327 case EXR_ZIP16:
1328 1838 ret = zip_uncompress(s, src, data_size, uncompressed_size, td);
1329 1838 break;
1330 60 case EXR_PIZ:
1331 60 ret = piz_uncompress(s, src, data_size, uncompressed_size, td);
1332 60 break;
1333 122 case EXR_PXR24:
1334 122 ret = pxr24_uncompress(s, src, data_size, uncompressed_size, td);
1335 122 break;
1336 1386 case EXR_RLE:
1337 1386 ret = rle_uncompress(s, src, data_size, uncompressed_size, td);
1338 1386 break;
1339 46 case EXR_B44:
1340 case EXR_B44A:
1341 46 ret = b44_uncompress(s, src, data_size, uncompressed_size, td);
1342 46 break;
1343 ✗ case EXR_DWAA:
1344 case EXR_DWAB:
1345 ✗ ret = dwa_uncompress(s, src, data_size, uncompressed_size, td);
1346 ✗ break;
1347 }
1348
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3452 times.
 3452 if (ret < 0) {
1349 ✗ av_log(avctx, AV_LOG_ERROR, "decode_block() failed.\n");
1350 ✗ return ret;
1351 }
1352 3452 src = td->uncompressed_data;
1353 }
1354
1355 /* offsets to crop data outside display window */
1356
4/4
✓ Branch 0 taken 4652 times.
✓ Branch 1 taken 220 times.
✓ Branch 2 taken 4640 times.
✓ Branch 3 taken 232 times.
 4872 data_xoffset = FFABS(FFMIN(0, s->xmin + col)) * (s->pixel_type == EXR_HALF ? 2 : 4);
1357
2/2
✓ Branch 0 taken 4860 times.
✓ Branch 1 taken 12 times.
 4872 data_yoffset = FFABS(FFMIN(0, line));
1358 4872 data_window_offset = (data_yoffset * td->channel_line_size) + data_xoffset;
1359
1360
2/2
✓ Branch 0 taken 4852 times.
✓ Branch 1 taken 20 times.
 4872 if (!s->is_luma) {
1361 4852 channel_buffer[0] = src + (td->xsize * s->channel_offsets[0]) + data_window_offset;
1362 4852 channel_buffer[1] = src + (td->xsize * s->channel_offsets[1]) + data_window_offset;
1363 4852 channel_buffer[2] = src + (td->xsize * s->channel_offsets[2]) + data_window_offset;
1364 4852 rgb_channel_count = 3;
1365 } else { /* put y data in the first channel_buffer */
1366 20 channel_buffer[0] = src + (td->xsize * s->channel_offsets[1]) + data_window_offset;
1367 20 rgb_channel_count = 1;
1368 }
1369
2/2
✓ Branch 0 taken 4258 times.
✓ Branch 1 taken 614 times.
 4872 if (s->channel_offsets[3] >= 0)
1370 4258 channel_buffer[3] = src + (td->xsize * s->channel_offsets[3]) + data_window_offset;
1371
1372
2/2
✓ Branch 0 taken 4852 times.
✓ Branch 1 taken 20 times.
 4872 if (s->desc->flags & AV_PIX_FMT_FLAG_FLOAT) {
1373 /* todo: change this when a floating point pixel format with luma with alpha is implemented */
1374
2/2
✓ Branch 0 taken 594 times.
✓ Branch 1 taken 4258 times.
 4852 int channel_count = s->channel_offsets[3] >= 0 ? 4 : rgb_channel_count;
1375
2/2
✓ Branch 0 taken 20 times.
✓ Branch 1 taken 4832 times.
 4852 if (s->is_luma) {
1376 20 channel_buffer[1] = channel_buffer[0];
1377 20 channel_buffer[2] = channel_buffer[0];
1378 }
1379
1380
2/2
✓ Branch 0 taken 18806 times.
✓ Branch 1 taken 4852 times.
 23658 for (c = 0; c < channel_count; c++) {
1381 18806 int plane = s->desc->comp[c].plane;
1382 18806 ptr = p->data[plane] + window_ymin * p->linesize[plane] + (window_xmin * 4);
1383
1384
2/2
✓ Branch 0 taken 53432 times.
✓ Branch 1 taken 18806 times.
 72238 for (i = 0; i < ysize; i++, ptr += p->linesize[plane]) {
1385 const uint8_t *src;
1386 union av_intfloat32 *ptr_x;
1387
1388 53432 src = channel_buffer[c];
1389 53432 ptr_x = (union av_intfloat32 *)ptr;
1390
1391 // Zero out the start if xmin is not 0
1392 53432 memset(ptr_x, 0, bxmin);
1393 53432 ptr_x += window_xoffset;
1394
1395
2/2
✓ Branch 0 taken 49296 times.
✓ Branch 1 taken 4136 times.
 53432 if (s->pixel_type == EXR_FLOAT ||
1396
1/2
✓ Branch 0 taken 49296 times.
✗ Branch 1 not taken.
 49296 s->compression == EXR_DWAA ||
1397
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 49296 times.
 53432 s->compression == EXR_DWAB) {
1398 // 32-bit
1399 union av_intfloat32 t;
1400
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 4136 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 4136 if (trc_func && c < 3) {
1401 ✗ for (x = 0; x < xsize; x++) {
1402 ✗ t.i = bytestream_get_le32(&src);
1403 ✗ t.f = trc_func(t.f);
1404 ✗ *ptr_x++ = t;
1405 }
1406
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4136 times.
 4136 } else if (one_gamma != 1.f) {
1407 ✗ for (x = 0; x < xsize; x++) {
1408 ✗ t.i = bytestream_get_le32(&src);
1409 ✗ if (t.f > 0.0f && c < 3) /* avoid negative values */
1410 ✗ t.f = powf(t.f, one_gamma);
1411 ✗ *ptr_x++ = t;
1412 }
1413 } else {
1414
2/2
✓ Branch 0 taken 156816 times.
✓ Branch 1 taken 4136 times.
 160952 for (x = 0; x < xsize; x++) {
1415 156816 t.i = bytestream_get_le32(&src);
1416 156816 *ptr_x++ = t;
1417 }
1418 }
1419
1/2
✓ Branch 0 taken 49296 times.
✗ Branch 1 not taken.
 49296 } else if (s->pixel_type == EXR_HALF) {
1420 // 16-bit
1421
3/4
✓ Branch 0 taken 6082 times.
✓ Branch 1 taken 43214 times.
✓ Branch 2 taken 6082 times.
✗ Branch 3 not taken.
 49296 if (c < 3 || !trc_func) {
1422
2/2
✓ Branch 0 taken 21028194 times.
✓ Branch 1 taken 49296 times.
 21077490 for (x = 0; x < xsize; x++) {
1423 21028194 *ptr_x++ = s->gamma_table[bytestream_get_le16(&src)];
1424 }
1425 } else {
1426 ✗ for (x = 0; x < xsize; x++) {
1427 ✗ ptr_x[0].i = half2float(bytestream_get_le16(&src),
1428 ✗ s->mantissatable,
1429 ✗ s->exponenttable,
1430 ✗ s->offsettable);
1431 ✗ ptr_x++;
1432 }
1433 }
1434 }
1435
1436 // Zero out the end if xmax+1 is not w
1437 53432 memset(ptr_x, 0, axmax);
1438 53432 channel_buffer[c] += td->channel_line_size;
1439 }
1440 }
1441 } else {
1442
1443 av_assert1(s->pixel_type == EXR_UINT);
1444 20 ptr = p->data[0] + window_ymin * p->linesize[0] + (window_xmin * s->desc->nb_components * 2);
1445
1446
2/2
✓ Branch 0 taken 350 times.
✓ Branch 1 taken 20 times.
 370 for (i = 0; i < ysize; i++, ptr += p->linesize[0]) {
1447
1448 const uint8_t * a;
1449 const uint8_t *rgb[3];
1450 uint16_t *ptr_x;
1451
1452
2/2
✓ Branch 0 taken 1050 times.
✓ Branch 1 taken 350 times.
 1400 for (c = 0; c < rgb_channel_count; c++) {
1453 1050 rgb[c] = channel_buffer[c];
1454 }
1455
1456
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 350 times.
 350 if (channel_buffer[3])
1457 ✗ a = channel_buffer[3];
1458
1459 350 ptr_x = (uint16_t *) ptr;
1460
1461 // Zero out the start if xmin is not 0
1462 350 memset(ptr_x, 0, bxmin);
1463 350 ptr_x += window_xoffset * s->desc->nb_components;
1464
1465
2/2
✓ Branch 0 taken 10618 times.
✓ Branch 1 taken 350 times.
 10968 for (x = 0; x < xsize; x++) {
1466
2/2
✓ Branch 0 taken 31854 times.
✓ Branch 1 taken 10618 times.
 42472 for (c = 0; c < rgb_channel_count; c++) {
1467 31854 *ptr_x++ = bytestream_get_le32(&rgb[c]) >> 16;
1468 }
1469
1470
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 10618 times.
 10618 if (channel_buffer[3])
1471 ✗ *ptr_x++ = bytestream_get_le32(&a) >> 16;
1472 }
1473
1474 // Zero out the end if xmax+1 is not w
1475 350 memset(ptr_x, 0, axmax);
1476
1477 350 channel_buffer[0] += td->channel_line_size;
1478 350 channel_buffer[1] += td->channel_line_size;
1479 350 channel_buffer[2] += td->channel_line_size;
1480
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 350 times.
 350 if (channel_buffer[3])
1481 ✗ channel_buffer[3] += td->channel_line_size;
1482 }
1483 }
1484
1485 4872 return 0;
1486 }
1487
1488 ✗ static void skip_header_chunk(EXRContext *s)
1489 {
1490 ✗ GetByteContext *gb = &s->gb;
1491
1492 ✗ while (bytestream2_get_bytes_left(gb) > 0) {
1493 ✗ if (!bytestream2_peek_byte(gb))
1494 ✗ break;
1495
1496 // Process unknown variables
1497 ✗ for (int i = 0; i < 2; i++) // value_name and value_type
1498 ✗ while (bytestream2_get_byte(gb) != 0);
1499
1500 // Skip variable length
1501 ✗ bytestream2_skip(gb, bytestream2_get_le32(gb));
1502 }
1503 }
1504
1505 /**
1506 * Check if the variable name corresponds to its data type.
1507 *
1508 * @param s the EXRContext
1509 * @param value_name name of the variable to check
1510 * @param value_type type of the variable to check
1511 * @param minimum_length minimum length of the variable data
1512 *
1513 * @return bytes to read containing variable data
1514 * -1 if variable is not found
1515 * 0 if buffer ended prematurely
1516 */
1517 20886 static int check_header_variable(EXRContext *s,
1518 const char *value_name,
1519 const char *value_type,
1520 unsigned int minimum_length)
1521 {
1522 20886 GetByteContext *gb = &s->gb;
1523 20886 int var_size = -1;
1524
1525
1/2
✓ Branch 1 taken 20886 times.
✗ Branch 2 not taken.
 20886 if (bytestream2_get_bytes_left(gb) >= minimum_length &&
1526
2/2
✓ Branch 0 taken 990 times.
✓ Branch 1 taken 19896 times.
 20886 !strcmp(gb->buffer, value_name)) {
1527 // found value_name, jump to value_type (null terminated strings)
1528 990 gb->buffer += strlen(value_name) + 1;
1529
1/2
✓ Branch 0 taken 990 times.
✗ Branch 1 not taken.
 990 if (!strcmp(gb->buffer, value_type)) {
1530 990 gb->buffer += strlen(value_type) + 1;
1531 990 var_size = bytestream2_get_le32(gb);
1532 // don't go read past boundaries
1533
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 990 times.
 990 if (var_size > bytestream2_get_bytes_left(gb))
1534 ✗ var_size = 0;
1535 } else {
1536 // value_type not found, reset the buffer
1537 ✗ gb->buffer -= strlen(value_name) + 1;
1538 ✗ av_log(s->avctx, AV_LOG_WARNING,
1539 "Unknown data type %s for header variable %s.\n",
1540 value_type, value_name);
1541 }
1542 }
1543
1544 20886 return var_size;
1545 }
1546
1547 148 static int decode_header(EXRContext *s, AVFrame *frame)
1548 {
1549 148 AVDictionary *metadata = NULL;
1550 148 GetByteContext *gb = &s->gb;
1551 int magic_number, version, flags;
1552 148 int layer_match = 0;
1553 int ret;
1554 148 int dup_channels = 0;
1555
1556 148 s->current_channel_offset = 0;
1557 148 s->xmin = ~0;
1558 148 s->xmax = ~0;
1559 148 s->ymin = ~0;
1560 148 s->ymax = ~0;
1561 148 s->xdelta = ~0;
1562 148 s->ydelta = ~0;
1563 148 s->channel_offsets[0] = -1;
1564 148 s->channel_offsets[1] = -1;
1565 148 s->channel_offsets[2] = -1;
1566 148 s->channel_offsets[3] = -1;
1567 148 s->pixel_type = EXR_UNKNOWN;
1568 148 s->compression = EXR_UNKN;
1569 148 s->nb_channels = 0;
1570 148 s->w = 0;
1571 148 s->h = 0;
1572 148 s->tile_attr.xSize = -1;
1573 148 s->tile_attr.ySize = -1;
1574 148 s->is_tile = 0;
1575 148 s->is_multipart = 0;
1576 148 s->is_luma = 0;
1577 148 s->current_part = 0;
1578
1579
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 148 times.
 148 if (bytestream2_get_bytes_left(gb) < 10) {
1580 ✗ av_log(s->avctx, AV_LOG_ERROR, "Header too short to parse.\n");
1581 ✗ return AVERROR_INVALIDDATA;
1582 }
1583
1584 148 magic_number = bytestream2_get_le32(gb);
1585
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 148 times.
 148 if (magic_number != 20000630) {
1586 /* As per documentation of OpenEXR, it is supposed to be
1587 * int 20000630 little-endian */
1588 ✗ av_log(s->avctx, AV_LOG_ERROR, "Wrong magic number %d.\n", magic_number);
1589 ✗ return AVERROR_INVALIDDATA;
1590 }
1591
1592 148 version = bytestream2_get_byte(gb);
1593
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 148 times.
 148 if (version != 2) {
1594 ✗ avpriv_report_missing_feature(s->avctx, "Version %d", version);
1595 ✗ return AVERROR_PATCHWELCOME;
1596 }
1597
1598 148 flags = bytestream2_get_le24(gb);
1599
1600
2/2
✓ Branch 0 taken 44 times.
✓ Branch 1 taken 104 times.
 148 if (flags & 0x02)
1601 44 s->is_tile = 1;
1602
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 148 times.
 148 if (flags & 0x10)
1603 ✗ s->is_multipart = 1;
1604
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 148 times.
 148 if (flags & 0x08) {
1605 ✗ avpriv_report_missing_feature(s->avctx, "deep data");
1606 ✗ return AVERROR_PATCHWELCOME;
1607 }
1608
1609 // Parse the header
1610
1/2
✓ Branch 1 taken 2548 times.
✗ Branch 2 not taken.
 2548 while (bytestream2_get_bytes_left(gb) > 0) {
1611 int var_size;
1612
1613
1/6
✗ Branch 0 not taken.
✓ Branch 1 taken 2548 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
 2548 while (s->is_multipart && s->current_part < s->selected_part &&
1614 ✗ bytestream2_get_bytes_left(gb) > 0) {
1615 ✗ if (bytestream2_peek_byte(gb)) {
1616 ✗ skip_header_chunk(s);
1617 } else {
1618 ✗ bytestream2_skip(gb, 1);
1619 ✗ if (!bytestream2_peek_byte(gb))
1620 ✗ break;
1621 }
1622 ✗ bytestream2_skip(gb, 1);
1623 ✗ s->current_part++;
1624 }
1625
1626
2/2
✓ Branch 1 taken 148 times.
✓ Branch 2 taken 2400 times.
 2548 if (!bytestream2_peek_byte(gb)) {
1627
1/2
✓ Branch 0 taken 148 times.
✗ Branch 1 not taken.
 148 if (!s->is_multipart)
1628 148 break;
1629 ✗ bytestream2_skip(gb, 1);
1630 ✗ if (s->current_part == s->selected_part) {
1631 ✗ while (bytestream2_get_bytes_left(gb) > 0) {
1632 ✗ if (bytestream2_peek_byte(gb)) {
1633 ✗ skip_header_chunk(s);
1634 } else {
1635 ✗ bytestream2_skip(gb, 1);
1636 ✗ if (!bytestream2_peek_byte(gb))
1637 ✗ break;
1638 }
1639 }
1640 }
1641 ✗ if (!bytestream2_peek_byte(gb))
1642 ✗ break;
1643 ✗ s->current_part++;
1644 }
1645
1646
2/2
✓ Branch 1 taken 148 times.
✓ Branch 2 taken 2252 times.
 2400 if ((var_size = check_header_variable(s, "channels",
1647 148 "chlist", 38)) >= 0) {
1648 GetByteContext ch_gb;
1649
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 148 times.
 148 if (!var_size) {
1650 ✗ ret = AVERROR_INVALIDDATA;
1651 ✗ goto fail;
1652 }
1653
1654 148 bytestream2_init(&ch_gb, gb->buffer, var_size);
1655
1656
2/2
✓ Branch 1 taken 764 times.
✓ Branch 2 taken 148 times.
 912 while (bytestream2_get_bytes_left(&ch_gb) >= 19) {
1657 EXRChannel *channel;
1658 enum ExrPixelType current_pixel_type;
1659 764 int channel_index = -1;
1660 int xsub, ysub;
1661
1662
2/2
✓ Branch 0 taken 272 times.
✓ Branch 1 taken 492 times.
 764 if (strcmp(s->layer, "") != 0) {
1663
2/2
✓ Branch 0 taken 150 times.
✓ Branch 1 taken 122 times.
 272 if (strncmp(ch_gb.buffer, s->layer, strlen(s->layer)) == 0) {
1664 150 layer_match = 1;
1665 150 av_log(s->avctx, AV_LOG_INFO,
1666 "Channel match layer : %s.\n", ch_gb.buffer);
1667 150 ch_gb.buffer += strlen(s->layer);
1668
1/2
✓ Branch 0 taken 150 times.
✗ Branch 1 not taken.
 150 if (*ch_gb.buffer == '.')
1669 150 ch_gb.buffer++; /* skip dot if not given */
1670 } else {
1671 122 layer_match = 0;
1672 122 av_log(s->avctx, AV_LOG_INFO,
1673 "Channel doesn't match layer : %s.\n", ch_gb.buffer);
1674 }
1675 } else {
1676 492 layer_match = 1;
1677 }
1678
1679
2/2
✓ Branch 0 taken 642 times.
✓ Branch 1 taken 122 times.
 764 if (layer_match) { /* only search channel if the layer match is valid */
1680
3/4
✓ Branch 1 taken 500 times.
✓ Branch 2 taken 142 times.
✓ Branch 3 taken 500 times.
✗ Branch 4 not taken.
 1142 if (!av_strcasecmp(ch_gb.buffer, "R") ||
1681
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 500 times.
 1000 !av_strcasecmp(ch_gb.buffer, "X") ||
1682 500 !av_strcasecmp(ch_gb.buffer, "U")) {
1683 142 channel_index = 0;
1684 142 s->is_luma = 0;
1685
3/4
✓ Branch 1 taken 358 times.
✓ Branch 2 taken 142 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 358 times.
 858 } else if (!av_strcasecmp(ch_gb.buffer, "G") ||
1686 358 !av_strcasecmp(ch_gb.buffer, "V")) {
1687 142 channel_index = 1;
1688 142 s->is_luma = 0;
1689
2/2
✓ Branch 1 taken 6 times.
✓ Branch 2 taken 352 times.
 358 } else if (!av_strcasecmp(ch_gb.buffer, "Y")) {
1690 6 channel_index = 1;
1691 6 s->is_luma = 1;
1692
3/4
✓ Branch 1 taken 210 times.
✓ Branch 2 taken 142 times.
✓ Branch 3 taken 210 times.
✗ Branch 4 not taken.
 562 } else if (!av_strcasecmp(ch_gb.buffer, "B") ||
1693
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 210 times.
 420 !av_strcasecmp(ch_gb.buffer, "Z") ||
1694 210 !av_strcasecmp(ch_gb.buffer, "W")) {
1695 142 channel_index = 2;
1696 142 s->is_luma = 0;
1697
2/2
✓ Branch 1 taken 58 times.
✓ Branch 2 taken 152 times.
 210 } else if (!av_strcasecmp(ch_gb.buffer, "A")) {
1698 58 channel_index = 3;
1699 } else {
1700 152 av_log(s->avctx, AV_LOG_WARNING,
1701 "Unsupported channel %.256s.\n", ch_gb.buffer);
1702 }
1703 }
1704
1705 /* skip until you get a 0 */
1706
3/4
✓ Branch 1 taken 3960 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 3196 times.
✓ Branch 4 taken 764 times.
 7920 while (bytestream2_get_bytes_left(&ch_gb) > 0 &&
1707 3960 bytestream2_get_byte(&ch_gb))
1708 3196 continue;
1709
1710
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 764 times.
 764 if (bytestream2_get_bytes_left(&ch_gb) < 4) {
1711 ✗ av_log(s->avctx, AV_LOG_ERROR, "Incomplete header.\n");
1712 ✗ ret = AVERROR_INVALIDDATA;
1713 ✗ goto fail;
1714 }
1715
1716 764 current_pixel_type = bytestream2_get_le32(&ch_gb);
1717
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 764 times.
 764 if (current_pixel_type >= EXR_UNKNOWN) {
1718 ✗ avpriv_report_missing_feature(s->avctx, "Pixel type %d",
1719 current_pixel_type);
1720 ✗ ret = AVERROR_PATCHWELCOME;
1721 ✗ goto fail;
1722 }
1723
1724 764 bytestream2_skip(&ch_gb, 4);
1725 764 xsub = bytestream2_get_le32(&ch_gb);
1726 764 ysub = bytestream2_get_le32(&ch_gb);
1727
1728
2/4
✓ Branch 0 taken 764 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 764 times.
 764 if (xsub != 1 || ysub != 1) {
1729 ✗ avpriv_report_missing_feature(s->avctx,
1730 "Subsampling %dx%d",
1731 xsub, ysub);
1732 ✗ ret = AVERROR_PATCHWELCOME;
1733 ✗ goto fail;
1734 }
1735
1736
3/4
✓ Branch 0 taken 490 times.
✓ Branch 1 taken 274 times.
✓ Branch 2 taken 490 times.
✗ Branch 3 not taken.
 764 if (channel_index >= 0 && s->channel_offsets[channel_index] == -1) { /* channel has not been previously assigned */
1737
2/2
✓ Branch 0 taken 342 times.
✓ Branch 1 taken 148 times.
 490 if (s->pixel_type != EXR_UNKNOWN &&
1738
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 342 times.
 342 s->pixel_type != current_pixel_type) {
1739 ✗ av_log(s->avctx, AV_LOG_ERROR,
1740 "RGB channels not of the same depth.\n");
1741 ✗ ret = AVERROR_INVALIDDATA;
1742 ✗ goto fail;
1743 }
1744 490 s->pixel_type = current_pixel_type;
1745 490 s->channel_offsets[channel_index] = s->current_channel_offset;
1746
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 274 times.
 274 } else if (channel_index >= 0) {
1747 ✗ av_log(s->avctx, AV_LOG_WARNING,
1748 "Multiple channels with index %d.\n", channel_index);
1749 ✗ if (++dup_channels > 10) {
1750 ✗ ret = AVERROR_INVALIDDATA;
1751 ✗ goto fail;
1752 }
1753 }
1754
1755 1528 s->channels = av_realloc(s->channels,
1756 764 ++s->nb_channels * sizeof(EXRChannel));
1757
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 764 times.
 764 if (!s->channels) {
1758 ✗ ret = AVERROR(ENOMEM);
1759 ✗ goto fail;
1760 }
1761 764 channel = &s->channels[s->nb_channels - 1];
1762 764 channel->pixel_type = current_pixel_type;
1763 764 channel->xsub = xsub;
1764 764 channel->ysub = ysub;
1765
1766
2/2
✓ Branch 0 taken 404 times.
✓ Branch 1 taken 360 times.
 764 if (current_pixel_type == EXR_HALF) {
1767 404 s->current_channel_offset += 2;
1768 } else {/* Float or UINT32 */
1769 360 s->current_channel_offset += 4;
1770 }
1771 }
1772
1773 /* Check if all channels are set with an offset or if the channels
1774 * are causing an overflow */
1775
2/2
✓ Branch 0 taken 142 times.
✓ Branch 1 taken 6 times.
 148 if (!s->is_luma) {/* if we expected to have at least 3 channels */
1776
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 142 times.
 142 if (FFMIN3(s->channel_offsets[0],
1777 s->channel_offsets[1],
1778 s->channel_offsets[2]) < 0) {
1779 ✗ if (s->channel_offsets[0] < 0)
1780 ✗ av_log(s->avctx, AV_LOG_ERROR, "Missing red channel.\n");
1781 ✗ if (s->channel_offsets[1] < 0)
1782 ✗ av_log(s->avctx, AV_LOG_ERROR, "Missing green channel.\n");
1783 ✗ if (s->channel_offsets[2] < 0)
1784 ✗ av_log(s->avctx, AV_LOG_ERROR, "Missing blue channel.\n");
1785 ✗ ret = AVERROR_INVALIDDATA;
1786 ✗ goto fail;
1787 }
1788 }
1789
1790 // skip one last byte and update main gb
1791 148 gb->buffer = ch_gb.buffer + 1;
1792 148 continue;
1793
2/2
✓ Branch 1 taken 148 times.
✓ Branch 2 taken 2104 times.
 2252 } else if ((var_size = check_header_variable(s, "dataWindow", "box2i",
1794 148 31)) >= 0) {
1795 int xmin, ymin, xmax, ymax;
1796
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 148 times.
 148 if (!var_size) {
1797 ✗ ret = AVERROR_INVALIDDATA;
1798 ✗ goto fail;
1799 }
1800
1801 148 xmin = bytestream2_get_le32(gb);
1802 148 ymin = bytestream2_get_le32(gb);
1803 148 xmax = bytestream2_get_le32(gb);
1804 148 ymax = bytestream2_get_le32(gb);
1805
1806
3/6
✓ Branch 0 taken 148 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 148 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 148 times.
✗ Branch 5 not taken.
 148 if (xmin > xmax || ymin > ymax ||
1807
1/2
✓ Branch 0 taken 148 times.
✗ Branch 1 not taken.
 148 ymax == INT_MAX || xmax == INT_MAX ||
1808
1/2
✓ Branch 0 taken 148 times.
✗ Branch 1 not taken.
 148 (unsigned)xmax - xmin >= INT_MAX ||
1809
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 148 times.
 148 (unsigned)ymax - ymin >= INT_MAX) {
1810 ✗ ret = AVERROR_INVALIDDATA;
1811 ✗ goto fail;
1812 }
1813 148 s->xmin = xmin;
1814 148 s->xmax = xmax;
1815 148 s->ymin = ymin;
1816 148 s->ymax = ymax;
1817 148 s->xdelta = (s->xmax - s->xmin) + 1;
1818 148 s->ydelta = (s->ymax - s->ymin) + 1;
1819
1820 148 continue;
1821
2/2
✓ Branch 1 taken 148 times.
✓ Branch 2 taken 1956 times.
 2104 } else if ((var_size = check_header_variable(s, "displayWindow",
1822 148 "box2i", 34)) >= 0) {
1823 int32_t sx, sy, dx, dy;
1824
1825
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 148 times.
 148 if (!var_size) {
1826 ✗ ret = AVERROR_INVALIDDATA;
1827 ✗ goto fail;
1828 }
1829
1830 148 sx = bytestream2_get_le32(gb);
1831 148 sy = bytestream2_get_le32(gb);
1832 148 dx = bytestream2_get_le32(gb);
1833 148 dy = bytestream2_get_le32(gb);
1834
1835 148 s->w = dx - sx + 1;
1836 148 s->h = dy - sy + 1;
1837
1838 148 continue;
1839
2/2
✓ Branch 1 taken 148 times.
✓ Branch 2 taken 1808 times.
 1956 } else if ((var_size = check_header_variable(s, "lineOrder",
1840 148 "lineOrder", 25)) >= 0) {
1841 int line_order;
1842
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 148 times.
 148 if (!var_size) {
1843 ✗ ret = AVERROR_INVALIDDATA;
1844 ✗ goto fail;
1845 }
1846
1847 148 line_order = bytestream2_get_byte(gb);
1848 148 av_log(s->avctx, AV_LOG_DEBUG, "line order: %d.\n", line_order);
1849
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 148 times.
 148 if (line_order > 2) {
1850 ✗ av_log(s->avctx, AV_LOG_ERROR, "Unknown line order.\n");
1851 ✗ ret = AVERROR_INVALIDDATA;
1852 ✗ goto fail;
1853 }
1854
1855 148 continue;
1856
2/2
✓ Branch 1 taken 148 times.
✓ Branch 2 taken 1660 times.
 1808 } else if ((var_size = check_header_variable(s, "pixelAspectRatio",
1857 "float", 31)) >= 0) {
1858
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 148 times.
 148 if (!var_size) {
1859 ✗ ret = AVERROR_INVALIDDATA;
1860 ✗ goto fail;
1861 }
1862
1863 148 s->sar = bytestream2_get_le32(gb);
1864
1865 148 continue;
1866
2/2
✓ Branch 1 taken 148 times.
✓ Branch 2 taken 1512 times.
 1660 } else if ((var_size = check_header_variable(s, "compression",
1867 "compression", 29)) >= 0) {
1868
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 148 times.
 148 if (!var_size) {
1869 ✗ ret = AVERROR_INVALIDDATA;
1870 ✗ goto fail;
1871 }
1872
1873
1/2
✓ Branch 0 taken 148 times.
✗ Branch 1 not taken.
 148 if (s->compression == EXR_UNKN)
1874 148 s->compression = bytestream2_get_byte(gb);
1875 else {
1876 ✗ bytestream2_skip(gb, 1);
1877 ✗ av_log(s->avctx, AV_LOG_WARNING,
1878 "Found more than one compression attribute.\n");
1879 }
1880
1881 148 continue;
1882
2/2
✓ Branch 1 taken 44 times.
✓ Branch 2 taken 1468 times.
 1512 } else if ((var_size = check_header_variable(s, "tiles",
1883 44 "tiledesc", 22)) >= 0) {
1884 char tileLevel;
1885
1886
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 44 times.
 44 if (!s->is_tile)
1887 ✗ av_log(s->avctx, AV_LOG_WARNING,
1888 "Found tile attribute and scanline flags. Exr will be interpreted as scanline.\n");
1889
1890 44 s->tile_attr.xSize = bytestream2_get_le32(gb);
1891 44 s->tile_attr.ySize = bytestream2_get_le32(gb);
1892
1893 44 tileLevel = bytestream2_get_byte(gb);
1894 44 s->tile_attr.level_mode = tileLevel & 0x0f;
1895 44 s->tile_attr.level_round = (tileLevel >> 4) & 0x0f;
1896
1897
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 44 times.
 44 if (s->tile_attr.level_mode >= EXR_TILE_LEVEL_UNKNOWN) {
1898 ✗ avpriv_report_missing_feature(s->avctx, "Tile level mode %d",
1899 ✗ s->tile_attr.level_mode);
1900 ✗ ret = AVERROR_PATCHWELCOME;
1901 ✗ goto fail;
1902 }
1903
1904
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 44 times.
 44 if (s->tile_attr.level_round >= EXR_TILE_ROUND_UNKNOWN) {
1905 ✗ avpriv_report_missing_feature(s->avctx, "Tile level round %d",
1906 ✗ s->tile_attr.level_round);
1907 ✗ ret = AVERROR_PATCHWELCOME;
1908 ✗ goto fail;
1909 }
1910
1911 44 continue;
1912
2/2
✓ Branch 1 taken 16 times.
✓ Branch 2 taken 1452 times.
 1468 } else if ((var_size = check_header_variable(s, "writer",
1913 16 "string", 1)) >= 0) {
1914 16 uint8_t key[256] = { 0 };
1915
1916 16 bytestream2_get_buffer(gb, key, FFMIN(sizeof(key) - 1, var_size));
1917 16 av_dict_set(&metadata, "writer", key, 0);
1918
1919 16 continue;
1920
2/2
✓ Branch 1 taken 16 times.
✓ Branch 2 taken 1436 times.
 1452 } else if ((var_size = check_header_variable(s, "framesPerSecond",
1921 "rational", 33)) >= 0) {
1922
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 16 times.
 16 if (!var_size) {
1923 ✗ ret = AVERROR_INVALIDDATA;
1924 ✗ goto fail;
1925 }
1926
1927 16 s->avctx->framerate.num = bytestream2_get_le32(gb);
1928 16 s->avctx->framerate.den = bytestream2_get_le32(gb);
1929
1930 16 continue;
1931
2/2
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 1428 times.
 1436 } else if ((var_size = check_header_variable(s, "chunkCount",
1932 "int", 23)) >= 0) {
1933
1934 8 s->chunk_count = bytestream2_get_le32(gb);
1935
1936 8 continue;
1937
2/2
✓ Branch 1 taken 18 times.
✓ Branch 2 taken 1410 times.
 1428 } else if ((var_size = check_header_variable(s, "type",
1938 18 "string", 16)) >= 0) {
1939 18 uint8_t key[256] = { 0 };
1940
1941 18 bytestream2_get_buffer(gb, key, FFMIN(sizeof(key) - 1, var_size));
1942
2/2
✓ Branch 0 taken 8 times.
✓ Branch 1 taken 10 times.
 18 if (strncmp("scanlineimage", key, var_size) &&
1943
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 8 times.
 8 strncmp("tiledimage", key, var_size))
1944 ✗ return AVERROR_PATCHWELCOME;
1945
1946 18 continue;
1947
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 1410 times.
 1410 } else if ((var_size = check_header_variable(s, "preview",
1948 ✗ "preview", 16)) >= 0) {
1949 ✗ uint32_t pw = bytestream2_get_le32(gb);
1950 ✗ uint32_t ph = bytestream2_get_le32(gb);
1951 ✗ int64_t psize = 4LL * pw * ph;
1952
1953 ✗ if (psize >= bytestream2_get_bytes_left(gb))
1954 ✗ return AVERROR_INVALIDDATA;
1955
1956 ✗ bytestream2_skip(gb, psize);
1957
1958 ✗ continue;
1959 }
1960
1961 // Check if there are enough bytes for a header
1962
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 1410 times.
 1410 if (bytestream2_get_bytes_left(gb) <= 9) {
1963 ✗ av_log(s->avctx, AV_LOG_ERROR, "Incomplete header\n");
1964 ✗ ret = AVERROR_INVALIDDATA;
1965 ✗ goto fail;
1966 }
1967
1968 // Process unknown variables
1969 {
1970 1410 uint8_t name[256] = { 0 };
1971 1410 uint8_t type[256] = { 0 };
1972 1410 uint8_t value[256] = { 0 };
1973 1410 int i = 0, size;
1974
1975
2/2
✓ Branch 1 taken 20206 times.
✓ Branch 2 taken 1410 times.
 43232 while (bytestream2_get_bytes_left(gb) > 0 &&
1976
2/4
✓ Branch 0 taken 21616 times.
✗ Branch 1 not taken.
✓ Branch 3 taken 20206 times.
✗ Branch 4 not taken.
 41822 bytestream2_peek_byte(gb) && i < 255) {
1977 20206 name[i++] = bytestream2_get_byte(gb);
1978 }
1979
1980 1410 bytestream2_skip(gb, 1);
1981 1410 i = 0;
1982
2/2
✓ Branch 1 taken 6590 times.
✓ Branch 2 taken 1410 times.
 16000 while (bytestream2_get_bytes_left(gb) > 0 &&
1983
2/4
✓ Branch 0 taken 8000 times.
✗ Branch 1 not taken.
✓ Branch 3 taken 6590 times.
✗ Branch 4 not taken.
 14590 bytestream2_peek_byte(gb) && i < 255) {
1984 6590 type[i++] = bytestream2_get_byte(gb);
1985 }
1986 1410 bytestream2_skip(gb, 1);
1987 1410 size = bytestream2_get_le32(gb);
1988
1989 1410 bytestream2_get_buffer(gb, value, FFMIN(sizeof(value) - 1, size));
1990
2/2
✓ Branch 0 taken 160 times.
✓ Branch 1 taken 1250 times.
 1410 if (!strcmp(type, "string"))
1991 160 av_dict_set(&metadata, name, value, 0);
1992 }
1993 }
1994
1995
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 148 times.
 148 if (s->compression == EXR_UNKN) {
1996 ✗ av_log(s->avctx, AV_LOG_ERROR, "Missing compression attribute.\n");
1997 ✗ ret = AVERROR_INVALIDDATA;
1998 ✗ goto fail;
1999 }
2000
2001
2/2
✓ Branch 0 taken 44 times.
✓ Branch 1 taken 104 times.
 148 if (s->is_tile) {
2002
2/4
✓ Branch 0 taken 44 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 44 times.
 44 if (s->tile_attr.xSize < 1 || s->tile_attr.ySize < 1) {
2003 ✗ av_log(s->avctx, AV_LOG_ERROR, "Invalid tile attribute.\n");
2004 ✗ ret = AVERROR_INVALIDDATA;
2005 ✗ goto fail;
2006 }
2007 }
2008
2009
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 148 times.
 148 if (bytestream2_get_bytes_left(gb) <= 0) {
2010 ✗ av_log(s->avctx, AV_LOG_ERROR, "Incomplete frame.\n");
2011 ✗ ret = AVERROR_INVALIDDATA;
2012 ✗ goto fail;
2013 }
2014
2015 148 frame->metadata = metadata;
2016
2017 // aaand we are done
2018 148 bytestream2_skip(gb, 1);
2019 148 return 0;
2020 ✗ fail:
2021 ✗ av_dict_free(&metadata);
2022 ✗ return ret;
2023 }
2024
2025 148 static int decode_frame(AVCodecContext *avctx, void *data,
2026 int *got_frame, AVPacket *avpkt)
2027 {
2028 148 EXRContext *s = avctx->priv_data;
2029 148 GetByteContext *gb = &s->gb;
2030 148 ThreadFrame frame = { .f = data };
2031 148 AVFrame *picture = data;
2032 uint8_t *ptr;
2033
2034 int i, y, ret, ymax;
2035 int planes;
2036 int out_line_size;
2037 int nb_blocks; /* nb scanline or nb tile */
2038 uint64_t start_offset_table;
2039 uint64_t start_next_scanline;
2040 PutByteContext offset_table_writer;
2041
2042 148 bytestream2_init(gb, avpkt->data, avpkt->size);
2043
2044
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 148 times.
 148 if ((ret = decode_header(s, picture)) < 0)
2045 ✗ return ret;
2046
2047
2/4
✓ Branch 0 taken 148 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 148 times.
 148 if ((s->compression == EXR_DWAA || s->compression == EXR_DWAB) &&
2048 ✗ s->pixel_type == EXR_HALF) {
2049 ✗ s->current_channel_offset *= 2;
2050 ✗ for (int i = 0; i < 4; i++)
2051 ✗ s->channel_offsets[i] *= 2;
2052 }
2053
2054
2/3
✓ Branch 0 taken 138 times.
✓ Branch 1 taken 10 times.
✗ Branch 2 not taken.
 148 switch (s->pixel_type) {
2055 138 case EXR_FLOAT:
2056 case EXR_HALF:
2057
2/2
✓ Branch 0 taken 58 times.
✓ Branch 1 taken 80 times.
 138 if (s->channel_offsets[3] >= 0) {
2058
2/2
✓ Branch 0 taken 56 times.
✓ Branch 1 taken 2 times.
 58 if (!s->is_luma) {
2059 56 avctx->pix_fmt = AV_PIX_FMT_GBRAPF32;
2060 } else {
2061 /* todo: change this when a floating point pixel format with luma with alpha is implemented */
2062 2 avctx->pix_fmt = AV_PIX_FMT_GBRAPF32;
2063 }
2064 } else {
2065
2/2
✓ Branch 0 taken 76 times.
✓ Branch 1 taken 4 times.
 80 if (!s->is_luma) {
2066 76 avctx->pix_fmt = AV_PIX_FMT_GBRPF32;
2067 } else {
2068 4 avctx->pix_fmt = AV_PIX_FMT_GRAYF32;
2069 }
2070 }
2071 138 break;
2072 10 case EXR_UINT:
2073
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 10 times.
 10 if (s->channel_offsets[3] >= 0) {
2074 ✗ if (!s->is_luma) {
2075 ✗ avctx->pix_fmt = AV_PIX_FMT_RGBA64;
2076 } else {
2077 ✗ avctx->pix_fmt = AV_PIX_FMT_YA16;
2078 }
2079 } else {
2080
1/2
✓ Branch 0 taken 10 times.
✗ Branch 1 not taken.
 10 if (!s->is_luma) {
2081 10 avctx->pix_fmt = AV_PIX_FMT_RGB48;
2082 } else {
2083 ✗ avctx->pix_fmt = AV_PIX_FMT_GRAY16;
2084 }
2085 }
2086 10 break;
2087 ✗ default:
2088 ✗ av_log(avctx, AV_LOG_ERROR, "Missing channel list.\n");
2089 ✗ return AVERROR_INVALIDDATA;
2090 }
2091
2092
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 148 times.
 148 if (s->apply_trc_type != AVCOL_TRC_UNSPECIFIED)
2093 ✗ avctx->color_trc = s->apply_trc_type;
2094
2095
3/5
✓ Branch 0 taken 48 times.
✓ Branch 1 taken 50 times.
✓ Branch 2 taken 50 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
 148 switch (s->compression) {
2096 48 case EXR_RAW:
2097 case EXR_RLE:
2098 case EXR_ZIP1:
2099 48 s->scan_lines_per_block = 1;
2100 48 break;
2101 50 case EXR_PXR24:
2102 case EXR_ZIP16:
2103 50 s->scan_lines_per_block = 16;
2104 50 break;
2105 50 case EXR_PIZ:
2106 case EXR_B44:
2107 case EXR_B44A:
2108 case EXR_DWAA:
2109 50 s->scan_lines_per_block = 32;
2110 50 break;
2111 ✗ case EXR_DWAB:
2112 ✗ s->scan_lines_per_block = 256;
2113 ✗ break;
2114 ✗ default:
2115 ✗ avpriv_report_missing_feature(avctx, "Compression %d", s->compression);
2116 ✗ return AVERROR_PATCHWELCOME;
2117 }
2118
2119 /* Verify the xmin, xmax, ymin and ymax before setting the actual image size.
2120 * It's possible for the data window can larger or outside the display window */
2121
2/4
✓ Branch 0 taken 148 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 148 times.
✗ Branch 3 not taken.
 148 if (s->xmin > s->xmax || s->ymin > s->ymax ||
2122
2/4
✓ Branch 0 taken 148 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 148 times.
 148 s->ydelta == 0xFFFFFFFF || s->xdelta == 0xFFFFFFFF) {
2123 ✗ av_log(avctx, AV_LOG_ERROR, "Wrong or missing size information.\n");
2124 ✗ return AVERROR_INVALIDDATA;
2125 }
2126
2127
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 148 times.
 148 if ((ret = ff_set_dimensions(avctx, s->w, s->h)) < 0)
2128 ✗ return ret;
2129
2130 148 ff_set_sar(s->avctx, av_d2q(av_int2float(s->sar), 255));
2131
2132 148 s->desc = av_pix_fmt_desc_get(avctx->pix_fmt);
2133
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 148 times.
 148 if (!s->desc)
2134 ✗ return AVERROR_INVALIDDATA;
2135
2136
2/2
✓ Branch 0 taken 138 times.
✓ Branch 1 taken 10 times.
 148 if (s->desc->flags & AV_PIX_FMT_FLAG_FLOAT) {
2137 138 planes = s->desc->nb_components;
2138 138 out_line_size = avctx->width * 4;
2139 } else {
2140 10 planes = 1;
2141 10 out_line_size = avctx->width * 2 * s->desc->nb_components;
2142 }
2143
2144
2/2
✓ Branch 0 taken 44 times.
✓ Branch 1 taken 104 times.
 148 if (s->is_tile) {
2145 44 nb_blocks = ((s->xdelta + s->tile_attr.xSize - 1) / s->tile_attr.xSize) *
2146 44 ((s->ydelta + s->tile_attr.ySize - 1) / s->tile_attr.ySize);
2147 } else { /* scanline */
2148 104 nb_blocks = (s->ydelta + s->scan_lines_per_block - 1) /
2149 104 s->scan_lines_per_block;
2150 }
2151
2152
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 148 times.
 148 if ((ret = ff_thread_get_buffer(avctx, &frame, 0)) < 0)
2153 ✗ return ret;
2154
2155
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 148 times.
 148 if (bytestream2_get_bytes_left(gb)/8 < nb_blocks)
2156 ✗ return AVERROR_INVALIDDATA;
2157
2158 // check offset table and recreate it if need
2159
4/4
✓ Branch 0 taken 104 times.
✓ Branch 1 taken 44 times.
✓ Branch 3 taken 1 times.
✓ Branch 4 taken 103 times.
 148 if (!s->is_tile && bytestream2_peek_le64(gb) == 0) {
2160 1 av_log(s->avctx, AV_LOG_DEBUG, "recreating invalid scanline offset table\n");
2161
2162 1 start_offset_table = bytestream2_tell(gb);
2163 1 start_next_scanline = start_offset_table + nb_blocks * 8;
2164 1 bytestream2_init_writer(&offset_table_writer, &avpkt->data[start_offset_table], nb_blocks * 8);
2165
2166
2/2
✓ Branch 0 taken 8 times.
✓ Branch 1 taken 1 times.
 9 for (y = 0; y < nb_blocks; y++) {
2167 /* write offset of prev scanline in offset table */
2168 8 bytestream2_put_le64(&offset_table_writer, start_next_scanline);
2169
2170 /* get len of next scanline */
2171 8 bytestream2_seek(gb, start_next_scanline + 4, SEEK_SET);/* skip line number */
2172 8 start_next_scanline += (bytestream2_get_le32(gb) + 8);
2173 }
2174 1 bytestream2_seek(gb, start_offset_table, SEEK_SET);
2175 }
2176
2177 // save pointer we are going to use in decode_block
2178 148 s->buf = avpkt->data;
2179 148 s->buf_size = avpkt->size;
2180
2181 // Zero out the start if ymin is not 0
2182
2/2
✓ Branch 0 taken 474 times.
✓ Branch 1 taken 148 times.
 622 for (i = 0; i < planes; i++) {
2183 474 ptr = picture->data[i];
2184
2/2
✓ Branch 0 taken 480 times.
✓ Branch 1 taken 474 times.
 954 for (y = 0; y < FFMIN(s->ymin, s->h); y++) {
2185 480 memset(ptr, 0, out_line_size);
2186 480 ptr += picture->linesize[i];
2187 }
2188 }
2189
2190 148 s->picture = picture;
2191
2192 148 avctx->execute2(avctx, decode_block, s->thread_data, NULL, nb_blocks);
2193
2194 148 ymax = FFMAX(0, s->ymax + 1);
2195 // Zero out the end if ymax+1 is not h
2196
2/2
✓ Branch 0 taken 8 times.
✓ Branch 1 taken 140 times.
 148 if (ymax < avctx->height)
2197
2/2
✓ Branch 0 taken 24 times.
✓ Branch 1 taken 8 times.
 32 for (i = 0; i < planes; i++) {
2198 24 ptr = picture->data[i] + (ymax * picture->linesize[i]);
2199
2/2
✓ Branch 0 taken 5544 times.
✓ Branch 1 taken 24 times.
 5568 for (y = ymax; y < avctx->height; y++) {
2200 5544 memset(ptr, 0, out_line_size);
2201 5544 ptr += picture->linesize[i];
2202 }
2203 }
2204
2205 148 picture->pict_type = AV_PICTURE_TYPE_I;
2206 148 *got_frame = 1;
2207
2208 148 return avpkt->size;
2209 }
2210
2211 148 static av_cold int decode_init(AVCodecContext *avctx)
2212 {
2213 148 EXRContext *s = avctx->priv_data;
2214 uint32_t i;
2215 union av_intfloat32 t;
2216 148 float one_gamma = 1.0f / s->gamma;
2217 148 avpriv_trc_function trc_func = NULL;
2218
2219 148 half2float_table(s->mantissatable, s->exponenttable, s->offsettable);
2220
2221 148 s->avctx = avctx;
2222
2223 148 ff_exrdsp_init(&s->dsp);
2224
2225 #if HAVE_BIGENDIAN
2226 ff_bswapdsp_init(&s->bbdsp);
2227 #endif
2228
2229 148 trc_func = avpriv_get_trc_function_from_trc(s->apply_trc_type);
2230
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 148 times.
 148 if (trc_func) {
2231 ✗ for (i = 0; i < 65536; ++i) {
2232 ✗ t.i = half2float(i, s->mantissatable, s->exponenttable, s->offsettable);
2233 ✗ t.f = trc_func(t.f);
2234 ✗ s->gamma_table[i] = t;
2235 }
2236 } else {
2237
2/4
✓ Branch 0 taken 148 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 148 times.
✗ Branch 3 not taken.
 148 if (one_gamma > 0.9999f && one_gamma < 1.0001f) {
2238
2/2
✓ Branch 0 taken 9699328 times.
✓ Branch 1 taken 148 times.
 9699476 for (i = 0; i < 65536; ++i) {
2239 9699328 s->gamma_table[i].i = half2float(i, s->mantissatable, s->exponenttable, s->offsettable);
2240 }
2241 } else {
2242 ✗ for (i = 0; i < 65536; ++i) {
2243 ✗ t.i = half2float(i, s->mantissatable, s->exponenttable, s->offsettable);
2244 /* If negative value we reuse half value */
2245 ✗ if (t.f <= 0.0f) {
2246 ✗ s->gamma_table[i] = t;
2247 } else {
2248 ✗ t.f = powf(t.f, one_gamma);
2249 ✗ s->gamma_table[i] = t;
2250 }
2251 }
2252 }
2253 }
2254
2255 // allocate thread data, used for non EXR_RAW compression types
2256 148 s->thread_data = av_calloc(avctx->thread_count, sizeof(*s->thread_data));
2257
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 148 times.
 148 if (!s->thread_data)
2258 ✗ return AVERROR(ENOMEM);
2259
2260 148 return 0;
2261 }
2262
2263 148 static av_cold int decode_end(AVCodecContext *avctx)
2264 {
2265 148 EXRContext *s = avctx->priv_data;
2266 int i;
2267
2/2
✓ Branch 0 taken 148 times.
✓ Branch 1 taken 148 times.
 296 for (i = 0; i < avctx->thread_count; i++) {
2268 148 EXRThreadData *td = &s->thread_data[i];
2269 148 av_freep(&td->uncompressed_data);
2270 148 av_freep(&td->tmp);
2271 148 av_freep(&td->bitmap);
2272 148 av_freep(&td->lut);
2273 148 av_freep(&td->he);
2274 148 av_freep(&td->freq);
2275 148 av_freep(&td->ac_data);
2276 148 av_freep(&td->dc_data);
2277 148 av_freep(&td->rle_data);
2278 148 av_freep(&td->rle_raw_data);
2279 148 ff_free_vlc(&td->vlc);
2280 }
2281
2282 148 av_freep(&s->thread_data);
2283 148 av_freep(&s->channels);
2284
2285 148 return 0;
2286 }
2287
2288 #define OFFSET(x) offsetof(EXRContext, x)
2289 #define VD AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_DECODING_PARAM
2290 static const AVOption options[] = {
2291 { "layer", "Set the decoding layer", OFFSET(layer),
2292 AV_OPT_TYPE_STRING, { .str = "" }, 0, 0, VD },
2293 { "part", "Set the decoding part", OFFSET(selected_part),
2294 AV_OPT_TYPE_INT, { .i64 = 0 }, 0, INT_MAX, VD },
2295 { "gamma", "Set the float gamma value when decoding", OFFSET(gamma),
2296 AV_OPT_TYPE_FLOAT, { .dbl = 1.0f }, 0.001, FLT_MAX, VD },
2297
2298 // XXX: Note the abuse of the enum using AVCOL_TRC_UNSPECIFIED to subsume the existing gamma option
2299 { "apply_trc", "color transfer characteristics to apply to EXR linear input", OFFSET(apply_trc_type),
2300 AV_OPT_TYPE_INT, {.i64 = AVCOL_TRC_UNSPECIFIED }, 1, AVCOL_TRC_NB-1, VD, "apply_trc_type"},
2301 { "bt709", "BT.709", 0,
2302 AV_OPT_TYPE_CONST, {.i64 = AVCOL_TRC_BT709 }, INT_MIN, INT_MAX, VD, "apply_trc_type"},
2303 { "gamma", "gamma", 0,
2304 AV_OPT_TYPE_CONST, {.i64 = AVCOL_TRC_UNSPECIFIED }, INT_MIN, INT_MAX, VD, "apply_trc_type"},
2305 { "gamma22", "BT.470 M", 0,
2306 AV_OPT_TYPE_CONST, {.i64 = AVCOL_TRC_GAMMA22 }, INT_MIN, INT_MAX, VD, "apply_trc_type"},
2307 { "gamma28", "BT.470 BG", 0,
2308 AV_OPT_TYPE_CONST, {.i64 = AVCOL_TRC_GAMMA28 }, INT_MIN, INT_MAX, VD, "apply_trc_type"},
2309 { "smpte170m", "SMPTE 170 M", 0,
2310 AV_OPT_TYPE_CONST, {.i64 = AVCOL_TRC_SMPTE170M }, INT_MIN, INT_MAX, VD, "apply_trc_type"},
2311 { "smpte240m", "SMPTE 240 M", 0,
2312 AV_OPT_TYPE_CONST, {.i64 = AVCOL_TRC_SMPTE240M }, INT_MIN, INT_MAX, VD, "apply_trc_type"},
2313 { "linear", "Linear", 0,
2314 AV_OPT_TYPE_CONST, {.i64 = AVCOL_TRC_LINEAR }, INT_MIN, INT_MAX, VD, "apply_trc_type"},
2315 { "log", "Log", 0,
2316 AV_OPT_TYPE_CONST, {.i64 = AVCOL_TRC_LOG }, INT_MIN, INT_MAX, VD, "apply_trc_type"},
2317 { "log_sqrt", "Log square root", 0,
2318 AV_OPT_TYPE_CONST, {.i64 = AVCOL_TRC_LOG_SQRT }, INT_MIN, INT_MAX, VD, "apply_trc_type"},
2319 { "iec61966_2_4", "IEC 61966-2-4", 0,
2320 AV_OPT_TYPE_CONST, {.i64 = AVCOL_TRC_IEC61966_2_4 }, INT_MIN, INT_MAX, VD, "apply_trc_type"},
2321 { "bt1361", "BT.1361", 0,
2322 AV_OPT_TYPE_CONST, {.i64 = AVCOL_TRC_BT1361_ECG }, INT_MIN, INT_MAX, VD, "apply_trc_type"},
2323 { "iec61966_2_1", "IEC 61966-2-1", 0,
2324 AV_OPT_TYPE_CONST, {.i64 = AVCOL_TRC_IEC61966_2_1 }, INT_MIN, INT_MAX, VD, "apply_trc_type"},
2325 { "bt2020_10bit", "BT.2020 - 10 bit", 0,
2326 AV_OPT_TYPE_CONST, {.i64 = AVCOL_TRC_BT2020_10 }, INT_MIN, INT_MAX, VD, "apply_trc_type"},
2327 { "bt2020_12bit", "BT.2020 - 12 bit", 0,
2328 AV_OPT_TYPE_CONST, {.i64 = AVCOL_TRC_BT2020_12 }, INT_MIN, INT_MAX, VD, "apply_trc_type"},
2329 { "smpte2084", "SMPTE ST 2084", 0,
2330 AV_OPT_TYPE_CONST, {.i64 = AVCOL_TRC_SMPTEST2084 }, INT_MIN, INT_MAX, VD, "apply_trc_type"},
2331 { "smpte428_1", "SMPTE ST 428-1", 0,
2332 AV_OPT_TYPE_CONST, {.i64 = AVCOL_TRC_SMPTEST428_1 }, INT_MIN, INT_MAX, VD, "apply_trc_type"},
2333
2334 { NULL },
2335 };
2336
2337 static const AVClass exr_class = {
2338 .class_name = "EXR",
2339 .item_name = av_default_item_name,
2340 .option = options,
2341 .version = LIBAVUTIL_VERSION_INT,
2342 };
2343
2344 const AVCodec ff_exr_decoder = {
2345 .name = "exr",
2346 .long_name = NULL_IF_CONFIG_SMALL("OpenEXR image"),
2347 .type = AVMEDIA_TYPE_VIDEO,
2348 .id = AV_CODEC_ID_EXR,
2349 .priv_data_size = sizeof(EXRContext),
2350 .init = decode_init,
2351 .close = decode_end,
2352 .decode = decode_frame,
2353 .capabilities = AV_CODEC_CAP_DR1 | AV_CODEC_CAP_FRAME_THREADS |
2354 AV_CODEC_CAP_SLICE_THREADS,
2355 .priv_class = &exr_class,
2356 };
2357