Line | Branch | Exec | Source |
---|---|---|---|
1 | /* | ||
2 | * Go2Webinar / Go2Meeting decoder | ||
3 | * Copyright (c) 2012 Konstantin Shishkov | ||
4 | * Copyright (c) 2013 Maxim Poliakovski | ||
5 | * | ||
6 | * This file is part of FFmpeg. | ||
7 | * | ||
8 | * FFmpeg is free software; you can redistribute it and/or | ||
9 | * modify it under the terms of the GNU Lesser General Public | ||
10 | * License as published by the Free Software Foundation; either | ||
11 | * version 2.1 of the License, or (at your option) any later version. | ||
12 | * | ||
13 | * FFmpeg is distributed in the hope that it will be useful, | ||
14 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
15 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | ||
16 | * Lesser General Public License for more details. | ||
17 | * | ||
18 | * You should have received a copy of the GNU Lesser General Public | ||
19 | * License along with FFmpeg; if not, write to the Free Software | ||
20 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | ||
21 | */ | ||
22 | |||
23 | /** | ||
24 | * @file | ||
25 | * Go2Webinar / Go2Meeting decoder | ||
26 | */ | ||
27 | |||
28 | #include <inttypes.h> | ||
29 | #include <zlib.h> | ||
30 | |||
31 | #include "libavutil/imgutils.h" | ||
32 | #include "libavutil/mem_internal.h" | ||
33 | |||
34 | #include "avcodec.h" | ||
35 | #include "blockdsp.h" | ||
36 | #include "bytestream.h" | ||
37 | #include "codec_internal.h" | ||
38 | #include "decode.h" | ||
39 | #include "elsdec.h" | ||
40 | #include "get_bits.h" | ||
41 | #include "idctdsp.h" | ||
42 | #include "jpegtables.h" | ||
43 | #include "mjpegdec.h" | ||
44 | |||
45 | #define EPIC_PIX_STACK_SIZE 1024 | ||
46 | #define EPIC_PIX_STACK_MAX (EPIC_PIX_STACK_SIZE - 1) | ||
47 | |||
48 | enum ChunkType { | ||
49 | DISPLAY_INFO = 0xC8, | ||
50 | TILE_DATA, | ||
51 | CURSOR_POS, | ||
52 | CURSOR_SHAPE, | ||
53 | CHUNK_CC, | ||
54 | CHUNK_CD | ||
55 | }; | ||
56 | |||
57 | enum Compression { | ||
58 | COMPR_EPIC_J_B = 2, | ||
59 | COMPR_KEMPF_J_B, | ||
60 | }; | ||
61 | |||
62 | /* These tables are already permuted according to ff_zigzag_direct */ | ||
63 | static const uint8_t luma_quant[64] = { | ||
64 | 8, 6, 6, 7, 6, 5, 8, 7, | ||
65 | 7, 7, 9, 9, 8, 10, 12, 20, | ||
66 | 13, 12, 11, 11, 12, 25, 18, 19, | ||
67 | 15, 20, 29, 26, 31, 30, 29, 26, | ||
68 | 28, 28, 32, 36, 46, 39, 32, 34, | ||
69 | 44, 35, 28, 28, 40, 55, 41, 44, | ||
70 | 48, 49, 52, 52, 52, 31, 39, 57, | ||
71 | 61, 56, 50, 60, 46, 51, 52, 50, | ||
72 | }; | ||
73 | |||
74 | static const uint8_t chroma_quant[64] = { | ||
75 | 9, 9, 9, 12, 11, 12, 24, 13, | ||
76 | 13, 24, 50, 33, 28, 33, 50, 50, | ||
77 | 50, 50, 50, 50, 50, 50, 50, 50, | ||
78 | 50, 50, 50, 50, 50, 50, 50, 50, | ||
79 | 50, 50, 50, 50, 50, 50, 50, 50, | ||
80 | 50, 50, 50, 50, 50, 50, 50, 50, | ||
81 | 50, 50, 50, 50, 50, 50, 50, 50, | ||
82 | 50, 50, 50, 50, 50, 50, 50, 50, | ||
83 | }; | ||
84 | |||
85 | typedef struct ePICPixListElem { | ||
86 | struct ePICPixListElem *next; | ||
87 | uint32_t pixel; | ||
88 | uint8_t rung; | ||
89 | } ePICPixListElem; | ||
90 | |||
91 | typedef struct ePICPixHashElem { | ||
92 | uint32_t pix_id; | ||
93 | struct ePICPixListElem *list; | ||
94 | } ePICPixHashElem; | ||
95 | |||
96 | #define EPIC_HASH_SIZE 256 | ||
97 | typedef struct ePICPixHash { | ||
98 | ePICPixHashElem *bucket[EPIC_HASH_SIZE]; | ||
99 | int bucket_size[EPIC_HASH_SIZE]; | ||
100 | int bucket_fill[EPIC_HASH_SIZE]; | ||
101 | } ePICPixHash; | ||
102 | |||
103 | typedef struct ePICContext { | ||
104 | ElsDecCtx els_ctx; | ||
105 | int next_run_pos; | ||
106 | ElsUnsignedRung unsigned_rung; | ||
107 | uint8_t W_flag_rung; | ||
108 | uint8_t N_flag_rung; | ||
109 | uint8_t W_ctx_rung[256]; | ||
110 | uint8_t N_ctx_rung[512]; | ||
111 | uint8_t nw_pred_rung[256]; | ||
112 | uint8_t ne_pred_rung[256]; | ||
113 | uint8_t prev_row_rung[14]; | ||
114 | uint8_t runlen_zeroes[14]; | ||
115 | uint8_t runlen_one; | ||
116 | int stack_pos; | ||
117 | uint32_t stack[EPIC_PIX_STACK_SIZE]; | ||
118 | ePICPixHash hash; | ||
119 | } ePICContext; | ||
120 | |||
121 | typedef struct JPGContext { | ||
122 | BlockDSPContext bdsp; | ||
123 | IDCTDSPContext idsp; | ||
124 | uint8_t permutated_scantable[64]; | ||
125 | |||
126 | VLC dc_vlc[2], ac_vlc[2]; | ||
127 | int prev_dc[3]; | ||
128 | DECLARE_ALIGNED(32, int16_t, block)[6][64]; | ||
129 | |||
130 | uint8_t *buf; | ||
131 | } JPGContext; | ||
132 | |||
133 | typedef struct G2MContext { | ||
134 | ePICContext ec; | ||
135 | JPGContext jc; | ||
136 | |||
137 | int version; | ||
138 | |||
139 | int compression; | ||
140 | int width, height, bpp; | ||
141 | int orig_width, orig_height; | ||
142 | int tile_width, tile_height; | ||
143 | int tiles_x, tiles_y, tile_x, tile_y; | ||
144 | |||
145 | int got_header; | ||
146 | |||
147 | uint8_t *framebuf; | ||
148 | int framebuf_stride; | ||
149 | unsigned int framebuf_allocated; | ||
150 | |||
151 | uint8_t *synth_tile, *jpeg_tile, *epic_buf, *epic_buf_base; | ||
152 | int tile_stride, epic_buf_stride, old_tile_w, old_tile_h; | ||
153 | int swapuv; | ||
154 | |||
155 | uint8_t *kempf_buf, *kempf_flags; | ||
156 | |||
157 | uint8_t *cursor; | ||
158 | int cursor_stride; | ||
159 | int cursor_fmt; | ||
160 | int cursor_w, cursor_h, cursor_x, cursor_y; | ||
161 | int cursor_hot_x, cursor_hot_y; | ||
162 | } G2MContext; | ||
163 | |||
164 | 6 | static av_cold int jpg_init(AVCodecContext *avctx, JPGContext *c) | |
165 | { | ||
166 | int ret; | ||
167 | |||
168 | 6 | ret = ff_mjpeg_build_vlc(&c->dc_vlc[0], ff_mjpeg_bits_dc_luminance, | |
169 | ff_mjpeg_val_dc, 0, avctx); | ||
170 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
|
6 | if (ret) |
171 | ✗ | return ret; | |
172 | 6 | ret = ff_mjpeg_build_vlc(&c->dc_vlc[1], ff_mjpeg_bits_dc_chrominance, | |
173 | ff_mjpeg_val_dc, 0, avctx); | ||
174 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
|
6 | if (ret) |
175 | ✗ | return ret; | |
176 | 6 | ret = ff_mjpeg_build_vlc(&c->ac_vlc[0], ff_mjpeg_bits_ac_luminance, | |
177 | ff_mjpeg_val_ac_luminance, 1, avctx); | ||
178 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
|
6 | if (ret) |
179 | ✗ | return ret; | |
180 | 6 | ret = ff_mjpeg_build_vlc(&c->ac_vlc[1], ff_mjpeg_bits_ac_chrominance, | |
181 | ff_mjpeg_val_ac_chrominance, 1, avctx); | ||
182 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
|
6 | if (ret) |
183 | ✗ | return ret; | |
184 | |||
185 | 6 | ff_blockdsp_init(&c->bdsp); | |
186 | 6 | ff_idctdsp_init(&c->idsp, avctx); | |
187 | 6 | ff_permute_scantable(c->permutated_scantable, ff_zigzag_direct, | |
188 | 6 | c->idsp.idct_permutation); | |
189 | |||
190 | 6 | return 0; | |
191 | } | ||
192 | |||
193 | 6 | static av_cold void jpg_free_context(JPGContext *ctx) | |
194 | { | ||
195 | int i; | ||
196 | |||
197 |
2/2✓ Branch 0 taken 12 times.
✓ Branch 1 taken 6 times.
|
18 | for (i = 0; i < 2; i++) { |
198 | 12 | ff_vlc_free(&ctx->dc_vlc[i]); | |
199 | 12 | ff_vlc_free(&ctx->ac_vlc[i]); | |
200 | } | ||
201 | |||
202 | 6 | av_freep(&ctx->buf); | |
203 | 6 | } | |
204 | |||
205 | 159 | static void jpg_unescape(const uint8_t *src, int src_size, | |
206 | uint8_t *dst, int *dst_size) | ||
207 | { | ||
208 | 159 | const uint8_t *src_end = src + src_size; | |
209 | 159 | uint8_t *dst_start = dst; | |
210 | |||
211 |
2/2✓ Branch 0 taken 155059 times.
✓ Branch 1 taken 159 times.
|
155218 | while (src < src_end) { |
212 | 155059 | uint8_t x = *src++; | |
213 | |||
214 | 155059 | *dst++ = x; | |
215 | |||
216 |
3/4✓ Branch 0 taken 893 times.
✓ Branch 1 taken 154166 times.
✓ Branch 2 taken 893 times.
✗ Branch 3 not taken.
|
155059 | if (x == 0xFF && !*src) |
217 | 893 | src++; | |
218 | } | ||
219 | 159 | *dst_size = dst - dst_start; | |
220 | 159 | } | |
221 | |||
222 | 38284 | static int jpg_decode_block(JPGContext *c, GetBitContext *gb, | |
223 | int plane, int16_t *block) | ||
224 | { | ||
225 | int dc, val, pos; | ||
226 | 38284 | const int is_chroma = !!plane; | |
227 |
2/2✓ Branch 0 taken 13158 times.
✓ Branch 1 taken 25126 times.
|
38284 | const uint8_t *qmat = is_chroma ? chroma_quant : luma_quant; |
228 | |||
229 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 38284 times.
|
38284 | if (get_bits_left(gb) < 1) |
230 | ✗ | return AVERROR_INVALIDDATA; | |
231 | |||
232 | 38284 | c->bdsp.clear_block(block); | |
233 | 38284 | dc = get_vlc2(gb, c->dc_vlc[is_chroma].table, 9, 2); | |
234 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 38284 times.
|
38284 | if (dc < 0) |
235 | ✗ | return AVERROR_INVALIDDATA; | |
236 |
2/2✓ Branch 0 taken 22214 times.
✓ Branch 1 taken 16070 times.
|
38284 | if (dc) |
237 | 22214 | dc = get_xbits(gb, dc); | |
238 | 38284 | dc = dc * qmat[0] + c->prev_dc[plane]; | |
239 | 38284 | block[0] = dc; | |
240 | 38284 | c->prev_dc[plane] = dc; | |
241 | |||
242 | 38284 | pos = 0; | |
243 |
2/2✓ Branch 0 taken 224976 times.
✓ Branch 1 taken 382 times.
|
225358 | while (pos < 63) { |
244 | 224976 | val = get_vlc2(gb, c->ac_vlc[is_chroma].table, 9, 2); | |
245 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 224976 times.
|
224976 | if (val < 0) |
246 | ✗ | return AVERROR_INVALIDDATA; | |
247 | 224976 | pos += val >> 4; | |
248 | 224976 | val &= 0xF; | |
249 |
2/2✓ Branch 0 taken 37902 times.
✓ Branch 1 taken 187074 times.
|
224976 | if (pos > 63) |
250 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 37902 times.
|
37902 | return val ? AVERROR_INVALIDDATA : 0; |
251 |
2/2✓ Branch 0 taken 186680 times.
✓ Branch 1 taken 394 times.
|
187074 | if (val) { |
252 | 186680 | int nbits = val; | |
253 | |||
254 | 186680 | val = get_xbits(gb, nbits); | |
255 | 186680 | val *= qmat[pos]; | |
256 | 186680 | block[c->permutated_scantable[pos]] = val; | |
257 | } | ||
258 | } | ||
259 | 382 | return 0; | |
260 | } | ||
261 | |||
262 | 1684224 | static inline void yuv2rgb(uint8_t *out, int ridx, int Y, int U, int V) | |
263 | { | ||
264 | 1684224 | out[ridx] = av_clip_uint8(Y + (91881 * V + 32768 >> 16)); | |
265 | 1684224 | out[1] = av_clip_uint8(Y + (-22554 * U - 46802 * V + 32768 >> 16)); | |
266 | 1684224 | out[2 - ridx] = av_clip_uint8(Y + (116130 * U + 32768 >> 16)); | |
267 | 1684224 | } | |
268 | |||
269 | 159 | static int jpg_decode_data(JPGContext *c, int width, int height, | |
270 | const uint8_t *src, int src_size, | ||
271 | uint8_t *dst, int dst_stride, | ||
272 | const uint8_t *mask, int mask_stride, int num_mbs, | ||
273 | int swapuv) | ||
274 | { | ||
275 | GetBitContext gb; | ||
276 | int mb_w, mb_h, mb_x, mb_y, i, j; | ||
277 | int bx, by; | ||
278 | int unesc_size; | ||
279 | int ret; | ||
280 |
2/2✓ Branch 0 taken 50 times.
✓ Branch 1 taken 109 times.
|
159 | const int ridx = swapuv ? 2 : 0; |
281 | |||
282 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 159 times.
|
159 | if ((ret = av_reallocp(&c->buf, |
283 | 159 | src_size + AV_INPUT_BUFFER_PADDING_SIZE)) < 0) | |
284 | ✗ | return ret; | |
285 | 159 | jpg_unescape(src, src_size, c->buf, &unesc_size); | |
286 | 159 | memset(c->buf + unesc_size, 0, AV_INPUT_BUFFER_PADDING_SIZE); | |
287 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 159 times.
|
159 | if((ret = init_get_bits8(&gb, c->buf, unesc_size)) < 0) |
288 | ✗ | return ret; | |
289 | |||
290 | 159 | width = FFALIGN(width, 16); | |
291 | 159 | mb_w = width >> 4; | |
292 | 159 | mb_h = (height + 15) >> 4; | |
293 | |||
294 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 159 times.
|
159 | if (!num_mbs) |
295 | ✗ | num_mbs = mb_w * mb_h * 4; | |
296 | |||
297 |
2/2✓ Branch 0 taken 477 times.
✓ Branch 1 taken 159 times.
|
636 | for (i = 0; i < 3; i++) |
298 | 477 | c->prev_dc[i] = 1024; | |
299 | 159 | bx = | |
300 | 159 | by = 0; | |
301 | 159 | c->bdsp.clear_blocks(c->block[0]); | |
302 |
1/2✓ Branch 0 taken 1069 times.
✗ Branch 1 not taken.
|
1069 | for (mb_y = 0; mb_y < mb_h; mb_y++) { |
303 |
2/2✓ Branch 0 taken 11016 times.
✓ Branch 1 taken 910 times.
|
11926 | for (mb_x = 0; mb_x < mb_w; mb_x++) { |
304 |
5/6✓ Branch 0 taken 11016 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 4671 times.
✓ Branch 3 taken 6345 times.
✓ Branch 4 taken 4574 times.
✓ Branch 5 taken 97 times.
|
11016 | if (mask && !mask[mb_x * 2] && !mask[mb_x * 2 + 1] && |
305 |
2/2✓ Branch 0 taken 4458 times.
✓ Branch 1 taken 116 times.
|
4574 | !mask[mb_x * 2 + mask_stride] && |
306 |
2/2✓ Branch 0 taken 4437 times.
✓ Branch 1 taken 21 times.
|
4458 | !mask[mb_x * 2 + 1 + mask_stride]) { |
307 | 4437 | bx += 16; | |
308 | 4437 | continue; | |
309 | } | ||
310 |
2/2✓ Branch 0 taken 13158 times.
✓ Branch 1 taken 6579 times.
|
19737 | for (j = 0; j < 2; j++) { |
311 |
2/2✓ Branch 0 taken 26316 times.
✓ Branch 1 taken 13158 times.
|
39474 | for (i = 0; i < 2; i++) { |
312 |
3/4✓ Branch 0 taken 26316 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 1190 times.
✓ Branch 3 taken 25126 times.
|
26316 | if (mask && !mask[mb_x * 2 + i + j * mask_stride]) |
313 | 1190 | continue; | |
314 | 25126 | num_mbs--; | |
315 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 25126 times.
|
25126 | if ((ret = jpg_decode_block(c, &gb, 0, |
316 | 25126 | c->block[i + j * 2])) != 0) | |
317 | ✗ | return ret; | |
318 | 25126 | c->idsp.idct(c->block[i + j * 2]); | |
319 | } | ||
320 | } | ||
321 |
2/2✓ Branch 0 taken 13158 times.
✓ Branch 1 taken 6579 times.
|
19737 | for (i = 1; i < 3; i++) { |
322 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 13158 times.
|
13158 | if ((ret = jpg_decode_block(c, &gb, i, c->block[i + 3])) != 0) |
323 | ✗ | return ret; | |
324 | 13158 | c->idsp.idct(c->block[i + 3]); | |
325 | } | ||
326 | |||
327 |
2/2✓ Branch 0 taken 105264 times.
✓ Branch 1 taken 6579 times.
|
111843 | for (j = 0; j < 16; j++) { |
328 | 105264 | uint8_t *out = dst + bx * 3 + (by + j) * dst_stride; | |
329 |
2/2✓ Branch 0 taken 1684224 times.
✓ Branch 1 taken 105264 times.
|
1789488 | for (i = 0; i < 16; i++) { |
330 | int Y, U, V; | ||
331 | |||
332 | 1684224 | Y = c->block[(j >> 3) * 2 + (i >> 3)][(i & 7) + (j & 7) * 8]; | |
333 | 1684224 | U = c->block[4][(i >> 1) + (j >> 1) * 8] - 128; | |
334 | 1684224 | V = c->block[5][(i >> 1) + (j >> 1) * 8] - 128; | |
335 | 1684224 | yuv2rgb(out + i * 3, ridx, Y, U, V); | |
336 | } | ||
337 | } | ||
338 | |||
339 |
2/2✓ Branch 0 taken 159 times.
✓ Branch 1 taken 6420 times.
|
6579 | if (!num_mbs) |
340 | 159 | return 0; | |
341 | 6420 | bx += 16; | |
342 | } | ||
343 | 910 | bx = 0; | |
344 | 910 | by += 16; | |
345 |
1/2✓ Branch 0 taken 910 times.
✗ Branch 1 not taken.
|
910 | if (mask) |
346 | 910 | mask += mask_stride * 2; | |
347 | } | ||
348 | |||
349 | ✗ | return 0; | |
350 | } | ||
351 | |||
352 | #define LOAD_NEIGHBOURS(x) \ | ||
353 | W = curr_row[(x) - 1]; \ | ||
354 | N = above_row[(x)]; \ | ||
355 | WW = curr_row[(x) - 2]; \ | ||
356 | NW = above_row[(x) - 1]; \ | ||
357 | NE = above_row[(x) + 1]; \ | ||
358 | NN = above2_row[(x)]; \ | ||
359 | NNW = above2_row[(x) - 1]; \ | ||
360 | NWW = above_row[(x) - 2]; \ | ||
361 | NNE = above2_row[(x) + 1] | ||
362 | |||
363 | #define UPDATE_NEIGHBOURS(x) \ | ||
364 | NNW = NN; \ | ||
365 | NN = NNE; \ | ||
366 | NWW = NW; \ | ||
367 | NW = N; \ | ||
368 | N = NE; \ | ||
369 | NE = above_row[(x) + 1]; \ | ||
370 | NNE = above2_row[(x) + 1] | ||
371 | |||
372 | #define R_shift 16 | ||
373 | #define G_shift 8 | ||
374 | #define B_shift 0 | ||
375 | |||
376 | /* improved djb2 hash from http://www.cse.yorku.ca/~oz/hash.html */ | ||
377 | 97781 | static int djb2_hash(uint32_t key) | |
378 | { | ||
379 | 97781 | uint32_t h = 5381; | |
380 | |||
381 | 97781 | h = (h * 33) ^ ((key >> 24) & 0xFF); // xxx: probably not needed at all | |
382 | 97781 | h = (h * 33) ^ ((key >> 16) & 0xFF); | |
383 | 97781 | h = (h * 33) ^ ((key >> 8) & 0xFF); | |
384 | 97781 | h = (h * 33) ^ (key & 0xFF); | |
385 | |||
386 | 97781 | return h & (EPIC_HASH_SIZE - 1); | |
387 | } | ||
388 | |||
389 | 136 | static void epic_hash_init(ePICPixHash *hash) | |
390 | { | ||
391 | 136 | memset(hash, 0, sizeof(*hash)); | |
392 | 136 | } | |
393 | |||
394 | 97247 | static ePICPixHashElem *epic_hash_find(const ePICPixHash *hash, uint32_t key) | |
395 | { | ||
396 | 97247 | int i, idx = djb2_hash(key); | |
397 | 97247 | ePICPixHashElem *bucket = hash->bucket[idx]; | |
398 | |||
399 |
2/2✓ Branch 0 taken 94232 times.
✓ Branch 1 taken 3067 times.
|
97299 | for (i = 0; i < hash->bucket_fill[idx]; i++) |
400 |
2/2✓ Branch 0 taken 94180 times.
✓ Branch 1 taken 52 times.
|
94232 | if (bucket[i].pix_id == key) |
401 | 94180 | return &bucket[i]; | |
402 | |||
403 | 3067 | return NULL; | |
404 | } | ||
405 | |||
406 | 534 | static ePICPixHashElem *epic_hash_add(ePICPixHash *hash, uint32_t key) | |
407 | { | ||
408 | ePICPixHashElem *bucket, *ret; | ||
409 | 534 | int idx = djb2_hash(key); | |
410 | |||
411 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 534 times.
|
534 | if (hash->bucket_size[idx] > INT_MAX / sizeof(**hash->bucket)) |
412 | ✗ | return NULL; | |
413 | |||
414 |
2/2✓ Branch 0 taken 531 times.
✓ Branch 1 taken 3 times.
|
534 | if (!(hash->bucket_fill[idx] < hash->bucket_size[idx])) { |
415 | 531 | int new_size = hash->bucket_size[idx] + 16; | |
416 | 531 | bucket = av_realloc(hash->bucket[idx], new_size * sizeof(*bucket)); | |
417 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 531 times.
|
531 | if (!bucket) |
418 | ✗ | return NULL; | |
419 | 531 | hash->bucket[idx] = bucket; | |
420 | 531 | hash->bucket_size[idx] = new_size; | |
421 | } | ||
422 | |||
423 | 534 | ret = &hash->bucket[idx][hash->bucket_fill[idx]++]; | |
424 | 534 | memset(ret, 0, sizeof(*ret)); | |
425 | 534 | ret->pix_id = key; | |
426 | 534 | return ret; | |
427 | } | ||
428 | |||
429 | 1658 | static int epic_add_pixel_to_cache(ePICPixHash *hash, uint32_t key, uint32_t pix) | |
430 | { | ||
431 | ePICPixListElem *new_elem; | ||
432 | 1658 | ePICPixHashElem *hash_elem = epic_hash_find(hash, key); | |
433 | |||
434 |
2/2✓ Branch 0 taken 534 times.
✓ Branch 1 taken 1124 times.
|
1658 | if (!hash_elem) { |
435 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 534 times.
|
534 | if (!(hash_elem = epic_hash_add(hash, key))) |
436 | ✗ | return AVERROR(ENOMEM); | |
437 | } | ||
438 | |||
439 | 1658 | new_elem = av_mallocz(sizeof(*new_elem)); | |
440 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1658 times.
|
1658 | if (!new_elem) |
441 | ✗ | return AVERROR(ENOMEM); | |
442 | |||
443 | 1658 | new_elem->pixel = pix; | |
444 | 1658 | new_elem->next = hash_elem->list; | |
445 | 1658 | hash_elem->list = new_elem; | |
446 | |||
447 | 1658 | return 0; | |
448 | } | ||
449 | |||
450 | 41955 | static inline int epic_cache_entries_for_pixel(const ePICPixHash *hash, | |
451 | uint32_t pix) | ||
452 | { | ||
453 | 41955 | ePICPixHashElem *hash_elem = epic_hash_find(hash, pix); | |
454 | |||
455 |
3/4✓ Branch 0 taken 39956 times.
✓ Branch 1 taken 1999 times.
✓ Branch 2 taken 39956 times.
✗ Branch 3 not taken.
|
41955 | if (hash_elem != NULL && hash_elem->list != NULL) |
456 | 39956 | return 1; | |
457 | |||
458 | 1999 | return 0; | |
459 | } | ||
460 | |||
461 | 136 | static void epic_free_pixel_cache(ePICPixHash *hash) | |
462 | { | ||
463 | int i, j; | ||
464 | |||
465 |
2/2✓ Branch 0 taken 34816 times.
✓ Branch 1 taken 136 times.
|
34952 | for (i = 0; i < EPIC_HASH_SIZE; i++) { |
466 |
2/2✓ Branch 0 taken 534 times.
✓ Branch 1 taken 34816 times.
|
35350 | for (j = 0; j < hash->bucket_fill[i]; j++) { |
467 | 534 | ePICPixListElem *list_elem = hash->bucket[i][j].list; | |
468 |
2/2✓ Branch 0 taken 1658 times.
✓ Branch 1 taken 534 times.
|
2192 | while (list_elem) { |
469 | 1658 | ePICPixListElem *tmp = list_elem->next; | |
470 | 1658 | av_free(list_elem); | |
471 | 1658 | list_elem = tmp; | |
472 | } | ||
473 | } | ||
474 | 34816 | av_freep(&hash->bucket[i]); | |
475 | 34816 | hash->bucket_size[i] = | |
476 | 34816 | hash->bucket_fill[i] = 0; | |
477 | } | ||
478 | 136 | } | |
479 | |||
480 | 510904 | static inline int is_pixel_on_stack(const ePICContext *dc, uint32_t pix) | |
481 | { | ||
482 | int i; | ||
483 | |||
484 |
2/2✓ Branch 0 taken 806184 times.
✓ Branch 1 taken 357504 times.
|
1163688 | for (i = 0; i < dc->stack_pos; i++) |
485 |
2/2✓ Branch 0 taken 153400 times.
✓ Branch 1 taken 652784 times.
|
806184 | if (dc->stack[i] == pix) |
486 | 153400 | break; | |
487 | |||
488 | 510904 | return i != dc->stack_pos; | |
489 | } | ||
490 | |||
491 | #define TOSIGNED(val) (((val) >> 1) ^ -((val) & 1)) | ||
492 | |||
493 | 9558 | static inline int epic_decode_component_pred(ePICContext *dc, | |
494 | int N, int W, int NW) | ||
495 | { | ||
496 | 9558 | unsigned delta = ff_els_decode_unsigned(&dc->els_ctx, &dc->unsigned_rung); | |
497 | 9558 | return mid_pred(N, N + W - NW, W) - TOSIGNED(delta); | |
498 | } | ||
499 | |||
500 | 4739 | static uint32_t epic_decode_pixel_pred(ePICContext *dc, int x, int y, | |
501 | const uint32_t *curr_row, | ||
502 | const uint32_t *above_row) | ||
503 | { | ||
504 | uint32_t N, W, NW, pred; | ||
505 | unsigned delta; | ||
506 | int GN, GW, GNW, R, G, B; | ||
507 | |||
508 |
4/4✓ Branch 0 taken 3657 times.
✓ Branch 1 taken 1082 times.
✓ Branch 2 taken 3186 times.
✓ Branch 3 taken 471 times.
|
4739 | if (x && y) { |
509 | 3186 | W = curr_row[x - 1]; | |
510 | 3186 | N = above_row[x]; | |
511 | 3186 | NW = above_row[x - 1]; | |
512 | |||
513 | 3186 | GN = (N >> G_shift) & 0xFF; | |
514 | 3186 | GW = (W >> G_shift) & 0xFF; | |
515 | 3186 | GNW = (NW >> G_shift) & 0xFF; | |
516 | |||
517 | 3186 | G = epic_decode_component_pred(dc, GN, GW, GNW); | |
518 | |||
519 | 6372 | R = G + epic_decode_component_pred(dc, | |
520 | 3186 | ((N >> R_shift) & 0xFF) - GN, | |
521 | 3186 | ((W >> R_shift) & 0xFF) - GW, | |
522 | 3186 | ((NW >> R_shift) & 0xFF) - GNW); | |
523 | |||
524 | 3186 | B = G + epic_decode_component_pred(dc, | |
525 | 3186 | ((N >> B_shift) & 0xFF) - GN, | |
526 | 3186 | ((W >> B_shift) & 0xFF) - GW, | |
527 | 3186 | ((NW >> B_shift) & 0xFF) - GNW); | |
528 | } else { | ||
529 |
2/2✓ Branch 0 taken 471 times.
✓ Branch 1 taken 1082 times.
|
1553 | if (x) |
530 | 471 | pred = curr_row[x - 1]; | |
531 | else | ||
532 | 1082 | pred = above_row[x]; | |
533 | |||
534 | 1553 | delta = ff_els_decode_unsigned(&dc->els_ctx, &dc->unsigned_rung); | |
535 | 1553 | R = ((pred >> R_shift) & 0xFF) - TOSIGNED(delta); | |
536 | |||
537 | 1553 | delta = ff_els_decode_unsigned(&dc->els_ctx, &dc->unsigned_rung); | |
538 | 1553 | G = ((pred >> G_shift) & 0xFF) - TOSIGNED(delta); | |
539 | |||
540 | 1553 | delta = ff_els_decode_unsigned(&dc->els_ctx, &dc->unsigned_rung); | |
541 | 1553 | B = ((pred >> B_shift) & 0xFF) - TOSIGNED(delta); | |
542 | } | ||
543 | |||
544 |
6/12✓ Branch 0 taken 4739 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 4739 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 4739 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 4739 times.
✗ Branch 7 not taken.
✓ Branch 8 taken 4739 times.
✗ Branch 9 not taken.
✗ Branch 10 not taken.
✓ Branch 11 taken 4739 times.
|
4739 | if (R<0 || G<0 || B<0 || R > 255 || G > 255 || B > 255) { |
545 | ✗ | avpriv_request_sample(NULL, "RGB %d %d %d (out of range)", R, G, B); | |
546 | ✗ | return 0; | |
547 | } | ||
548 | |||
549 | 4739 | return (R << R_shift) | (G << G_shift) | (B << B_shift); | |
550 | } | ||
551 | |||
552 | 99449 | static int epic_predict_pixel(ePICContext *dc, uint8_t *rung, | |
553 | uint32_t *pPix, uint32_t pix) | ||
554 | { | ||
555 |
2/2✓ Branch 1 taken 91084 times.
✓ Branch 2 taken 8365 times.
|
99449 | if (!ff_els_decode_bit(&dc->els_ctx, rung)) { |
556 | 91084 | *pPix = pix; | |
557 | 91084 | return 1; | |
558 | } | ||
559 | 8365 | dc->stack[dc->stack_pos++ & EPIC_PIX_STACK_MAX] = pix; | |
560 | 8365 | return 0; | |
561 | } | ||
562 | |||
563 | 95756 | static int epic_handle_edges(ePICContext *dc, int x, int y, | |
564 | const uint32_t *curr_row, | ||
565 | const uint32_t *above_row, uint32_t *pPix) | ||
566 | { | ||
567 | uint32_t pix; | ||
568 | |||
569 |
4/4✓ Branch 0 taken 17184 times.
✓ Branch 1 taken 78572 times.
✓ Branch 2 taken 136 times.
✓ Branch 3 taken 17048 times.
|
95756 | if (!x && !y) { /* special case: top-left pixel */ |
570 | /* the top-left pixel is coded independently with 3 unsigned numbers */ | ||
571 | 136 | *pPix = (ff_els_decode_unsigned(&dc->els_ctx, &dc->unsigned_rung) << R_shift) | | |
572 | 136 | (ff_els_decode_unsigned(&dc->els_ctx, &dc->unsigned_rung) << G_shift) | | |
573 | 136 | (ff_els_decode_unsigned(&dc->els_ctx, &dc->unsigned_rung) << B_shift); | |
574 | 136 | return 1; | |
575 | } | ||
576 | |||
577 |
2/2✓ Branch 0 taken 78572 times.
✓ Branch 1 taken 17048 times.
|
95620 | if (x) { /* predict from W first */ |
578 | 78572 | pix = curr_row[x - 1]; | |
579 |
2/2✓ Branch 1 taken 72049 times.
✓ Branch 2 taken 6523 times.
|
78572 | if (epic_predict_pixel(dc, &dc->W_flag_rung, pPix, pix)) |
580 | 72049 | return 1; | |
581 | } | ||
582 | |||
583 |
2/2✓ Branch 0 taken 22080 times.
✓ Branch 1 taken 1491 times.
|
23571 | if (y) { /* then try to predict from N */ |
584 | 22080 | pix = above_row[x]; | |
585 |
4/4✓ Branch 0 taken 5032 times.
✓ Branch 1 taken 17048 times.
✓ Branch 2 taken 3829 times.
✓ Branch 3 taken 1203 times.
|
22080 | if (!dc->stack_pos || dc->stack[0] != pix) { |
586 |
2/2✓ Branch 1 taken 19035 times.
✓ Branch 2 taken 1842 times.
|
20877 | if (epic_predict_pixel(dc, &dc->N_flag_rung, pPix, pix)) |
587 | 19035 | return 1; | |
588 | } | ||
589 | } | ||
590 | |||
591 | 4536 | return 0; | |
592 | } | ||
593 | |||
594 | 218240 | static int epic_decode_run_length(ePICContext *dc, int x, int y, int tile_width, | |
595 | const uint32_t *curr_row, | ||
596 | const uint32_t *above_row, | ||
597 | const uint32_t *above2_row, | ||
598 | uint32_t *pPix, int *pRun) | ||
599 | { | ||
600 | 218240 | int idx, got_pixel = 0, WWneW, old_WWneW = 0; | |
601 | uint32_t W, WW, N, NN, NW, NE, NWW, NNW, NNE; | ||
602 | |||
603 | 218240 | *pRun = 0; | |
604 | |||
605 | 218240 | LOAD_NEIGHBOURS(x); | |
606 | |||
607 |
2/2✓ Branch 0 taken 199455 times.
✓ Branch 1 taken 18785 times.
|
218240 | if (dc->next_run_pos == x) { |
608 | /* can't reuse W for the new pixel in this case */ | ||
609 | 199455 | WWneW = 1; | |
610 | } else { | ||
611 |
2/2✓ Branch 0 taken 3742 times.
✓ Branch 1 taken 15043 times.
|
18785 | idx = (WW != W) << 7 | |
612 |
2/2✓ Branch 0 taken 1686 times.
✓ Branch 1 taken 17099 times.
|
18785 | (NW != W) << 6 | |
613 |
2/2✓ Branch 0 taken 1703 times.
✓ Branch 1 taken 17082 times.
|
18785 | (N != NE) << 5 | |
614 |
2/2✓ Branch 0 taken 2227 times.
✓ Branch 1 taken 16558 times.
|
18785 | (NW != N) << 4 | |
615 |
2/2✓ Branch 0 taken 2155 times.
✓ Branch 1 taken 16630 times.
|
18785 | (NWW != NW) << 3 | |
616 |
2/2✓ Branch 0 taken 1969 times.
✓ Branch 1 taken 16816 times.
|
18785 | (NNE != NE) << 2 | |
617 |
2/2✓ Branch 0 taken 1883 times.
✓ Branch 1 taken 16902 times.
|
18785 | (NN != N) << 1 | |
618 | 18785 | (NNW != NW); | |
619 | 18785 | WWneW = ff_els_decode_bit(&dc->els_ctx, &dc->W_ctx_rung[idx]); | |
620 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18785 times.
|
18785 | if (WWneW < 0) |
621 | ✗ | return WWneW; | |
622 | } | ||
623 | |||
624 |
2/2✓ Branch 0 taken 201691 times.
✓ Branch 1 taken 16549 times.
|
218240 | if (WWneW) |
625 | 201691 | dc->stack[dc->stack_pos++ & EPIC_PIX_STACK_MAX] = W; | |
626 | else { | ||
627 | 16549 | *pPix = W; | |
628 | 16549 | got_pixel = 1; | |
629 | } | ||
630 | |||
631 | do { | ||
632 | 503572 | int NWneW = 1; | |
633 |
2/2✓ Branch 0 taken 237740 times.
✓ Branch 1 taken 265832 times.
|
503572 | if (got_pixel) // pixel value already known (derived from either W or N) |
634 | 237740 | NWneW = *pPix != N; | |
635 | else { // pixel value is unknown and will be decoded later | ||
636 |
2/2✓ Branch 0 taken 201691 times.
✓ Branch 1 taken 64141 times.
|
265832 | NWneW = *pRun ? NWneW : NW != W; |
637 | |||
638 | /* TODO: RFC this mess! */ | ||
639 |
4/5✓ Branch 0 taken 126306 times.
✓ Branch 1 taken 139526 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 176548 times.
✓ Branch 4 taken 89284 times.
|
265832 | switch (((NW != N) << 2) | (NWneW << 1) | WWneW) { |
640 | ✗ | case 0: | |
641 | ✗ | break; // do nothing here | |
642 | 176548 | case 3: | |
643 | case 5: | ||
644 | case 6: | ||
645 | case 7: | ||
646 |
2/2✓ Branch 1 taken 152558 times.
✓ Branch 2 taken 23990 times.
|
176548 | if (!is_pixel_on_stack(dc, N)) { |
647 | 457674 | idx = WWneW << 8 | | |
648 |
4/4✓ Branch 0 taken 131415 times.
✓ Branch 1 taken 21143 times.
✓ Branch 2 taken 73563 times.
✓ Branch 3 taken 57852 times.
|
152558 | (*pRun ? old_WWneW : WW != W) << 7 | |
649 | 305116 | NWneW << 6 | | |
650 |
2/2✓ Branch 0 taken 56675 times.
✓ Branch 1 taken 95883 times.
|
152558 | (N != NE) << 5 | |
651 |
2/2✓ Branch 0 taken 102316 times.
✓ Branch 1 taken 50242 times.
|
152558 | (NW != N) << 4 | |
652 |
2/2✓ Branch 0 taken 63177 times.
✓ Branch 1 taken 89381 times.
|
152558 | (NWW != NW) << 3 | |
653 |
2/2✓ Branch 0 taken 40781 times.
✓ Branch 1 taken 111777 times.
|
152558 | (NNE != NE) << 2 | |
654 |
2/2✓ Branch 0 taken 48988 times.
✓ Branch 1 taken 103570 times.
|
152558 | (NN != N) << 1 | |
655 | 152558 | (NNW != NW); | |
656 |
2/2✓ Branch 1 taken 129319 times.
✓ Branch 2 taken 23239 times.
|
152558 | if (!ff_els_decode_bit(&dc->els_ctx, &dc->N_ctx_rung[idx])) { |
657 | 129319 | NWneW = 0; | |
658 | 129319 | *pPix = N; | |
659 | 129319 | got_pixel = 1; | |
660 | 129319 | break; | |
661 | } | ||
662 | } | ||
663 | /* fall through */ | ||
664 | default: | ||
665 | 136513 | NWneW = 1; | |
666 | 136513 | old_WWneW = WWneW; | |
667 |
2/2✓ Branch 1 taken 23239 times.
✓ Branch 2 taken 113274 times.
|
136513 | if (!is_pixel_on_stack(dc, N)) |
668 | 23239 | dc->stack[dc->stack_pos++ & EPIC_PIX_STACK_MAX] = N; | |
669 | } | ||
670 | } | ||
671 | |||
672 | 503572 | (*pRun)++; | |
673 |
2/2✓ Branch 0 taken 3378 times.
✓ Branch 1 taken 500194 times.
|
503572 | if (x + *pRun >= tile_width - 1) |
674 | 3378 | break; | |
675 | |||
676 | 500194 | UPDATE_NEIGHBOURS(x + *pRun); | |
677 | |||
678 |
6/6✓ Branch 0 taken 233276 times.
✓ Branch 1 taken 266918 times.
✓ Branch 2 taken 123216 times.
✓ Branch 3 taken 110060 times.
✓ Branch 4 taken 99184 times.
✓ Branch 5 taken 24032 times.
|
500194 | if (!NWneW && NW == N && N == NE) { |
679 | int pos, run, rle; | ||
680 | 99184 | int start_pos = x + *pRun; | |
681 | |||
682 | /* scan for a run of pix in the line above */ | ||
683 | 99184 | uint32_t pix = above_row[start_pos + 1]; | |
684 |
2/2✓ Branch 0 taken 2496269 times.
✓ Branch 1 taken 15422 times.
|
2511691 | for (pos = start_pos + 2; pos < tile_width; pos++) |
685 |
2/2✓ Branch 0 taken 83762 times.
✓ Branch 1 taken 2412507 times.
|
2496269 | if (!(above_row[pos] == pix)) |
686 | 83762 | break; | |
687 | 99184 | run = pos - start_pos - 1; | |
688 | 99184 | idx = av_ceil_log2(run); | |
689 |
2/2✓ Branch 1 taken 66738 times.
✓ Branch 2 taken 32446 times.
|
99184 | if (ff_els_decode_bit(&dc->els_ctx, &dc->prev_row_rung[idx])) |
690 | 66738 | *pRun += run; | |
691 | else { | ||
692 | int flag; | ||
693 | /* run-length is coded as plain binary number of idx - 1 bits */ | ||
694 |
2/2✓ Branch 0 taken 99317 times.
✓ Branch 1 taken 32446 times.
|
131763 | for (pos = idx - 1, rle = 0, flag = 0; pos >= 0; pos--) { |
695 |
4/4✓ Branch 0 taken 94613 times.
✓ Branch 1 taken 4704 times.
✓ Branch 2 taken 33844 times.
✓ Branch 3 taken 60769 times.
|
193930 | if ((1 << pos) + rle < run && |
696 |
2/2✓ Branch 0 taken 25102 times.
✓ Branch 1 taken 69511 times.
|
94613 | ff_els_decode_bit(&dc->els_ctx, |
697 | flag ? &dc->runlen_one | ||
698 | : &dc->runlen_zeroes[pos])) { | ||
699 | 33844 | flag = 1; | |
700 | 33844 | rle |= 1 << pos; | |
701 | } | ||
702 | } | ||
703 | 32446 | *pRun += rle; | |
704 | 32446 | break; // return immediately | |
705 | } | ||
706 |
2/2✓ Branch 0 taken 13519 times.
✓ Branch 1 taken 53219 times.
|
66738 | if (x + *pRun >= tile_width - 1) |
707 | 13519 | break; | |
708 | |||
709 | 53219 | LOAD_NEIGHBOURS(x + *pRun); | |
710 | 53219 | WWneW = 0; | |
711 | 53219 | NWneW = 0; | |
712 | } | ||
713 | |||
714 | 908458 | idx = WWneW << 7 | | |
715 | 908458 | NWneW << 6 | | |
716 |
2/2✓ Branch 0 taken 201628 times.
✓ Branch 1 taken 252601 times.
|
454229 | (N != NE) << 5 | |
717 |
2/2✓ Branch 0 taken 201195 times.
✓ Branch 1 taken 253034 times.
|
454229 | (NW != N) << 4 | |
718 |
2/2✓ Branch 0 taken 144163 times.
✓ Branch 1 taken 310066 times.
|
454229 | (NWW != NW) << 3 | |
719 |
2/2✓ Branch 0 taken 185477 times.
✓ Branch 1 taken 268752 times.
|
454229 | (NNE != NE) << 2 | |
720 |
2/2✓ Branch 0 taken 187945 times.
✓ Branch 1 taken 266284 times.
|
454229 | (NN != N) << 1 | |
721 | 454229 | (NNW != NW); | |
722 | 454229 | WWneW = ff_els_decode_bit(&dc->els_ctx, &dc->W_ctx_rung[idx]); | |
723 |
2/2✓ Branch 0 taken 285332 times.
✓ Branch 1 taken 168897 times.
|
454229 | } while (!WWneW); |
724 | |||
725 | 218240 | dc->next_run_pos = x + *pRun; | |
726 | 218240 | return got_pixel; | |
727 | } | ||
728 | |||
729 | 36049 | static int epic_predict_pixel2(ePICContext *dc, uint8_t *rung, | |
730 | uint32_t *pPix, uint32_t pix) | ||
731 | { | ||
732 |
2/2✓ Branch 1 taken 22192 times.
✓ Branch 2 taken 13857 times.
|
36049 | if (ff_els_decode_bit(&dc->els_ctx, rung)) { |
733 | 22192 | *pPix = pix; | |
734 | 22192 | return 1; | |
735 | } | ||
736 | 13857 | dc->stack[dc->stack_pos++ & EPIC_PIX_STACK_MAX] = pix; | |
737 | 13857 | return 0; | |
738 | } | ||
739 | |||
740 | 76908 | static int epic_predict_from_NW_NE(ePICContext *dc, int x, int y, int run, | |
741 | int tile_width, const uint32_t *curr_row, | ||
742 | const uint32_t *above_row, uint32_t *pPix) | ||
743 | { | ||
744 | int pos; | ||
745 | |||
746 | /* try to reuse the NW pixel first */ | ||
747 |
4/4✓ Branch 0 taken 75465 times.
✓ Branch 1 taken 1443 times.
✓ Branch 2 taken 73974 times.
✓ Branch 3 taken 1491 times.
|
76908 | if (x && y) { |
748 | 73974 | uint32_t NW = above_row[x - 1]; | |
749 |
6/6✓ Branch 0 taken 26314 times.
✓ Branch 1 taken 47660 times.
✓ Branch 2 taken 20304 times.
✓ Branch 3 taken 6010 times.
✓ Branch 5 taken 20189 times.
✓ Branch 6 taken 115 times.
|
73974 | if (NW != curr_row[x - 1] && NW != above_row[x] && !is_pixel_on_stack(dc, NW)) { |
750 |
2/2✓ Branch 1 taken 12191 times.
✓ Branch 2 taken 7998 times.
|
20189 | if (epic_predict_pixel2(dc, &dc->nw_pred_rung[NW & 0xFF], pPix, NW)) |
751 | 12191 | return 1; | |
752 | } | ||
753 | } | ||
754 | |||
755 | /* try to reuse the NE[x + run, y] pixel */ | ||
756 | 64717 | pos = x + run - 1; | |
757 |
4/4✓ Branch 0 taken 64303 times.
✓ Branch 1 taken 414 times.
✓ Branch 2 taken 62826 times.
✓ Branch 3 taken 1477 times.
|
64717 | if (pos < tile_width - 1 && y) { |
758 | 62826 | uint32_t NE = above_row[pos + 1]; | |
759 |
4/4✓ Branch 0 taken 21217 times.
✓ Branch 1 taken 41609 times.
✓ Branch 3 taken 15860 times.
✓ Branch 4 taken 5357 times.
|
62826 | if (NE != above_row[pos] && !is_pixel_on_stack(dc, NE)) { |
760 |
2/2✓ Branch 1 taken 10001 times.
✓ Branch 2 taken 5859 times.
|
15860 | if (epic_predict_pixel2(dc, &dc->ne_pred_rung[NE & 0xFF], pPix, NE)) |
761 | 10001 | return 1; | |
762 | } | ||
763 | } | ||
764 | |||
765 | 54716 | return 0; | |
766 | } | ||
767 | |||
768 | 53634 | static int epic_decode_from_cache(ePICContext *dc, uint32_t W, uint32_t *pPix) | |
769 | { | ||
770 | 53634 | ePICPixListElem *list, *prev = NULL; | |
771 | 53634 | ePICPixHashElem *hash_elem = epic_hash_find(&dc->hash, W); | |
772 | |||
773 |
3/4✓ Branch 0 taken 53100 times.
✓ Branch 1 taken 534 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 53100 times.
|
53634 | if (!hash_elem || !hash_elem->list) |
774 | 534 | return 0; | |
775 | |||
776 | 53100 | list = hash_elem->list; | |
777 |
2/2✓ Branch 0 taken 153582 times.
✓ Branch 1 taken 1124 times.
|
154706 | while (list) { |
778 |
2/2✓ Branch 1 taken 142918 times.
✓ Branch 2 taken 10664 times.
|
153582 | if (!is_pixel_on_stack(dc, list->pixel)) { |
779 |
2/2✓ Branch 1 taken 51976 times.
✓ Branch 2 taken 90942 times.
|
142918 | if (ff_els_decode_bit(&dc->els_ctx, &list->rung)) { |
780 | 51976 | *pPix = list->pixel; | |
781 |
2/2✓ Branch 0 taken 36609 times.
✓ Branch 1 taken 15367 times.
|
51976 | if (list != hash_elem->list) { |
782 | 36609 | prev->next = list->next; | |
783 | 36609 | list->next = hash_elem->list; | |
784 | 36609 | hash_elem->list = list; | |
785 | } | ||
786 | 51976 | return 1; | |
787 | } | ||
788 | 90942 | dc->stack[dc->stack_pos++ & EPIC_PIX_STACK_MAX] = list->pixel; | |
789 | } | ||
790 | 101606 | prev = list; | |
791 | 101606 | list = list->next; | |
792 | } | ||
793 | |||
794 | 1124 | return 0; | |
795 | } | ||
796 | |||
797 | 136 | static int epic_decode_tile(ePICContext *dc, uint8_t *out, int tile_height, | |
798 | int tile_width, int stride) | ||
799 | { | ||
800 | int x, y; | ||
801 | uint32_t pix; | ||
802 | 136 | uint32_t *curr_row = NULL, *above_row = NULL, *above2_row; | |
803 | |||
804 |
2/2✓ Branch 0 taken 17184 times.
✓ Branch 1 taken 136 times.
|
17320 | for (y = 0; y < tile_height; y++, out += stride) { |
805 | 17184 | above2_row = above_row; | |
806 | 17184 | above_row = curr_row; | |
807 | 17184 | curr_row = (uint32_t *) out; | |
808 | |||
809 |
2/2✓ Branch 0 taken 315995 times.
✓ Branch 1 taken 17184 times.
|
333179 | for (x = 0, dc->next_run_pos = 0; x < tile_width;) { |
810 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 315995 times.
|
315995 | if (dc->els_ctx.err) |
811 | ✗ | return AVERROR_INVALIDDATA; // bail out in the case of ELS overflow | |
812 | |||
813 | 315995 | pix = curr_row[x - 1]; // get W pixel | |
814 | |||
815 |
4/4✓ Branch 0 taken 293467 times.
✓ Branch 1 taken 22528 times.
✓ Branch 2 taken 259371 times.
✓ Branch 3 taken 34096 times.
|
315995 | if (y >= 1 && x >= 2 && |
816 |
4/4✓ Branch 0 taken 101977 times.
✓ Branch 1 taken 157394 times.
✓ Branch 2 taken 59168 times.
✓ Branch 3 taken 42809 times.
|
259371 | pix != curr_row[x - 2] && pix != above_row[x - 1] && |
817 |
6/6✓ Branch 0 taken 49937 times.
✓ Branch 1 taken 9231 times.
✓ Branch 2 taken 41955 times.
✓ Branch 3 taken 7982 times.
✓ Branch 4 taken 1999 times.
✓ Branch 5 taken 39956 times.
|
101123 | pix != above_row[x - 2] && pix != above_row[x] && |
818 | 41955 | !epic_cache_entries_for_pixel(&dc->hash, pix)) { | |
819 | 1999 | curr_row[x] = epic_decode_pixel_pred(dc, x, y, curr_row, above_row); | |
820 | 1999 | x++; | |
821 | } else { | ||
822 | int got_pixel, run; | ||
823 | 313996 | dc->stack_pos = 0; // empty stack | |
824 | |||
825 |
6/6✓ Branch 0 taken 268956 times.
✓ Branch 1 taken 45040 times.
✓ Branch 2 taken 235132 times.
✓ Branch 3 taken 33824 times.
✓ Branch 4 taken 16892 times.
✓ Branch 5 taken 218240 times.
|
313996 | if (y < 2 || x < 2 || x == tile_width - 1) { |
826 | 95756 | run = 1; | |
827 | 95756 | got_pixel = epic_handle_edges(dc, x, y, curr_row, above_row, &pix); | |
828 | } else { | ||
829 | 218240 | got_pixel = epic_decode_run_length(dc, x, y, tile_width, | |
830 | curr_row, above_row, | ||
831 | above2_row, &pix, &run); | ||
832 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 218240 times.
|
218240 | if (got_pixel < 0) |
833 | ✗ | return got_pixel; | |
834 | } | ||
835 | |||
836 |
4/4✓ Branch 0 taken 76908 times.
✓ Branch 1 taken 237088 times.
✓ Branch 3 taken 54716 times.
✓ Branch 4 taken 22192 times.
|
313996 | if (!got_pixel && !epic_predict_from_NW_NE(dc, x, y, run, |
837 | tile_width, curr_row, | ||
838 | above_row, &pix)) { | ||
839 | 54716 | uint32_t ref_pix = curr_row[x - 1]; | |
840 |
4/4✓ Branch 0 taken 53634 times.
✓ Branch 1 taken 1082 times.
✓ Branch 3 taken 1658 times.
✓ Branch 4 taken 51976 times.
|
54716 | if (!x || !epic_decode_from_cache(dc, ref_pix, &pix)) { |
841 | 2740 | pix = epic_decode_pixel_pred(dc, x, y, curr_row, above_row); | |
842 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 2740 times.
|
2740 | if (is_pixel_on_stack(dc, pix)) |
843 | ✗ | return AVERROR_INVALIDDATA; | |
844 | |||
845 |
2/2✓ Branch 0 taken 1658 times.
✓ Branch 1 taken 1082 times.
|
2740 | if (x) { |
846 | 1658 | int ret = epic_add_pixel_to_cache(&dc->hash, | |
847 | ref_pix, | ||
848 | pix); | ||
849 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1658 times.
|
1658 | if (ret) |
850 | ✗ | return ret; | |
851 | } | ||
852 | } | ||
853 | } | ||
854 |
2/2✓ Branch 0 taken 2845745 times.
✓ Branch 1 taken 313996 times.
|
3159741 | for (; run > 0; x++, run--) |
855 | 2845745 | curr_row[x] = pix; | |
856 | } | ||
857 | } | ||
858 | } | ||
859 | |||
860 | 136 | return 0; | |
861 | } | ||
862 | |||
863 | 136 | static int epic_jb_decode_tile(G2MContext *c, int tile_x, int tile_y, | |
864 | const uint8_t *src, size_t src_size, | ||
865 | AVCodecContext *avctx) | ||
866 | { | ||
867 | 136 | uint8_t prefix, mask = 0x80; | |
868 | int extrabytes, tile_width, tile_height, awidth, aheight; | ||
869 | size_t els_dsize; | ||
870 | uint8_t *dst; | ||
871 | |||
872 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 136 times.
|
136 | if (!src_size) |
873 | ✗ | return 0; | |
874 | |||
875 | /* get data size of the ELS partition as unsigned variable-length integer */ | ||
876 | 136 | prefix = *src++; | |
877 | 136 | src_size--; | |
878 |
3/4✓ Branch 0 taken 88 times.
✓ Branch 1 taken 136 times.
✓ Branch 2 taken 88 times.
✗ Branch 3 not taken.
|
224 | for (extrabytes = 0; (prefix & mask) && (extrabytes < 7); extrabytes++) |
879 | 88 | mask >>= 1; | |
880 |
2/4✓ Branch 0 taken 136 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 136 times.
|
136 | if (extrabytes > 3 || src_size < extrabytes) { |
881 | ✗ | av_log(avctx, AV_LOG_ERROR, "ePIC: invalid data size VLI\n"); | |
882 | ✗ | return AVERROR_INVALIDDATA; | |
883 | } | ||
884 | |||
885 | 136 | els_dsize = prefix & ((0x80 >> extrabytes) - 1); // mask out the length prefix | |
886 |
2/2✓ Branch 0 taken 88 times.
✓ Branch 1 taken 136 times.
|
224 | while (extrabytes-- > 0) { |
887 | 88 | els_dsize = (els_dsize << 8) | *src++; | |
888 | 88 | src_size--; | |
889 | } | ||
890 | |||
891 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 136 times.
|
136 | if (src_size < els_dsize) { |
892 | ✗ | av_log(avctx, AV_LOG_ERROR, "ePIC: data too short, needed %"SIZE_SPECIFIER", got %"SIZE_SPECIFIER"\n", | |
893 | els_dsize, src_size); | ||
894 | ✗ | return AVERROR_INVALIDDATA; | |
895 | } | ||
896 | |||
897 | 136 | tile_width = FFMIN(c->width - tile_x * c->tile_width, c->tile_width); | |
898 | 136 | tile_height = FFMIN(c->height - tile_y * c->tile_height, c->tile_height); | |
899 | 136 | awidth = FFALIGN(tile_width, 16); | |
900 | 136 | aheight = FFALIGN(tile_height, 16); | |
901 | |||
902 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 136 times.
|
136 | if (tile_width > (1 << FF_ARRAY_ELEMS(c->ec.prev_row_rung))) { |
903 | ✗ | avpriv_request_sample(avctx, "large tile width"); | |
904 | ✗ | return AVERROR_INVALIDDATA; | |
905 | } | ||
906 | |||
907 |
1/2✓ Branch 0 taken 136 times.
✗ Branch 1 not taken.
|
136 | if (els_dsize) { |
908 | int ret, i, j, k; | ||
909 | uint8_t tr_r, tr_g, tr_b, *buf; | ||
910 | uint32_t *in; | ||
911 | /* ELS decoder initializations */ | ||
912 | 136 | memset(&c->ec, 0, sizeof(c->ec)); | |
913 | 136 | ff_els_decoder_init(&c->ec.els_ctx, src, els_dsize); | |
914 | 136 | epic_hash_init(&c->ec.hash); | |
915 | |||
916 | /* decode transparent pixel value */ | ||
917 | 136 | tr_r = ff_els_decode_unsigned(&c->ec.els_ctx, &c->ec.unsigned_rung); | |
918 | 136 | tr_g = ff_els_decode_unsigned(&c->ec.els_ctx, &c->ec.unsigned_rung); | |
919 | 136 | tr_b = ff_els_decode_unsigned(&c->ec.els_ctx, &c->ec.unsigned_rung); | |
920 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 136 times.
|
136 | if (c->ec.els_ctx.err != 0) { |
921 | ✗ | av_log(avctx, AV_LOG_ERROR, | |
922 | "ePIC: couldn't decode transparency pixel!\n"); | ||
923 | ✗ | ff_els_decoder_uninit(&c->ec.unsigned_rung); | |
924 | ✗ | return AVERROR_INVALIDDATA; | |
925 | } | ||
926 | |||
927 | 136 | ret = epic_decode_tile(&c->ec, c->epic_buf, tile_height, tile_width, | |
928 | c->epic_buf_stride); | ||
929 | |||
930 | 136 | epic_free_pixel_cache(&c->ec.hash); | |
931 | 136 | ff_els_decoder_uninit(&c->ec.unsigned_rung); | |
932 | |||
933 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 136 times.
|
136 | if (ret) { |
934 | ✗ | av_log(avctx, AV_LOG_ERROR, | |
935 | "ePIC: tile decoding failed, frame=%"PRId64", tile_x=%d, tile_y=%d\n", | ||
936 | avctx->frame_num, tile_x, tile_y); | ||
937 | ✗ | return AVERROR_INVALIDDATA; | |
938 | } | ||
939 | |||
940 | 136 | buf = c->epic_buf; | |
941 | 136 | dst = c->framebuf + tile_x * c->tile_width * 3 + | |
942 | 136 | tile_y * c->tile_height * c->framebuf_stride; | |
943 | |||
944 |
2/2✓ Branch 0 taken 17184 times.
✓ Branch 1 taken 136 times.
|
17320 | for (j = 0; j < tile_height; j++) { |
945 | 17184 | uint8_t *out = dst; | |
946 | 17184 | in = (uint32_t *) buf; | |
947 |
2/2✓ Branch 0 taken 2847744 times.
✓ Branch 1 taken 17184 times.
|
2864928 | for (i = 0; i < tile_width; i++) { |
948 | 2847744 | out[0] = (in[i] >> R_shift) & 0xFF; | |
949 | 2847744 | out[1] = (in[i] >> G_shift) & 0xFF; | |
950 | 2847744 | out[2] = (in[i] >> B_shift) & 0xFF; | |
951 | 2847744 | out += 3; | |
952 | } | ||
953 | 17184 | buf += c->epic_buf_stride; | |
954 | 17184 | dst += c->framebuf_stride; | |
955 | } | ||
956 | |||
957 |
2/2✓ Branch 0 taken 92 times.
✓ Branch 1 taken 44 times.
|
136 | if (src_size > els_dsize) { |
958 | uint8_t *jpg; | ||
959 | uint32_t tr; | ||
960 | 92 | int bstride = FFALIGN(tile_width, 16) >> 3; | |
961 | 92 | int nblocks = 0; | |
962 | 92 | int estride = c->epic_buf_stride >> 2; | |
963 | |||
964 | 92 | src += els_dsize; | |
965 | 92 | src_size -= els_dsize; | |
966 | |||
967 | 92 | in = (uint32_t *) c->epic_buf; | |
968 | 92 | tr = (tr_r << R_shift) | (tr_g << G_shift) | (tr_b << B_shift); | |
969 | |||
970 | 92 | memset(c->kempf_flags, 0, | |
971 | 92 | (aheight >> 3) * bstride * sizeof(*c->kempf_flags)); | |
972 |
2/2✓ Branch 0 taken 1451 times.
✓ Branch 1 taken 92 times.
|
1543 | for (j = 0; j < tile_height; j += 8) { |
973 |
2/2✓ Branch 0 taken 31154 times.
✓ Branch 1 taken 1451 times.
|
32605 | for (i = 0; i < tile_width; i += 8) { |
974 | 31154 | c->kempf_flags[(i >> 3) + (j >> 3) * bstride] = 0; | |
975 |
2/2✓ Branch 0 taken 1096594 times.
✓ Branch 1 taken 16412 times.
|
1113006 | for (k = 0; k < 8 * 8; k++) { |
976 |
2/2✓ Branch 0 taken 14742 times.
✓ Branch 1 taken 1081852 times.
|
1096594 | if (in[i + (k & 7) + (k >> 3) * estride] == tr) { |
977 | 14742 | c->kempf_flags[(i >> 3) + (j >> 3) * bstride] = 1; | |
978 | 14742 | nblocks++; | |
979 | 14742 | break; | |
980 | } | ||
981 | } | ||
982 | } | ||
983 | 1451 | in += 8 * estride; | |
984 | } | ||
985 | |||
986 | 92 | memset(c->jpeg_tile, 0, c->tile_stride * aheight); | |
987 | 92 | jpg_decode_data(&c->jc, awidth, aheight, src, src_size, | |
988 | c->jpeg_tile, c->tile_stride, | ||
989 | 92 | c->kempf_flags, bstride, nblocks, c->swapuv); | |
990 | |||
991 | 92 | in = (uint32_t *) c->epic_buf; | |
992 | 92 | dst = c->framebuf + tile_x * c->tile_width * 3 + | |
993 | 92 | tile_y * c->tile_height * c->framebuf_stride; | |
994 | 92 | jpg = c->jpeg_tile; | |
995 |
2/2✓ Branch 0 taken 11580 times.
✓ Branch 1 taken 92 times.
|
11672 | for (j = 0; j < tile_height; j++) { |
996 |
2/2✓ Branch 0 taken 1988928 times.
✓ Branch 1 taken 11580 times.
|
2000508 | for (i = 0; i < tile_width; i++) |
997 |
2/2✓ Branch 0 taken 717442 times.
✓ Branch 1 taken 1271486 times.
|
1988928 | if (in[i] == tr) |
998 | 717442 | memcpy(dst + i * 3, jpg + i * 3, 3); | |
999 | 11580 | in += c->epic_buf_stride >> 2; | |
1000 | 11580 | dst += c->framebuf_stride; | |
1001 | 11580 | jpg += c->tile_stride; | |
1002 | } | ||
1003 | } | ||
1004 | } else { | ||
1005 | ✗ | dst = c->framebuf + tile_x * c->tile_width * 3 + | |
1006 | ✗ | tile_y * c->tile_height * c->framebuf_stride; | |
1007 | ✗ | return jpg_decode_data(&c->jc, tile_width, tile_height, src, src_size, | |
1008 | dst, c->framebuf_stride, NULL, 0, 0, c->swapuv); | ||
1009 | } | ||
1010 | |||
1011 | 136 | return 0; | |
1012 | } | ||
1013 | |||
1014 | 81 | static int kempf_restore_buf(const uint8_t *src, int len, | |
1015 | uint8_t *dst, int stride, | ||
1016 | const uint8_t *jpeg_tile, int tile_stride, | ||
1017 | int width, int height, | ||
1018 | const uint8_t *pal, int npal, int tidx) | ||
1019 | { | ||
1020 | GetBitContext gb; | ||
1021 | int i, j, nb, col; | ||
1022 | int ret; | ||
1023 | 81 | int align_width = FFALIGN(width, 16); | |
1024 | |||
1025 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 81 times.
|
81 | if ((ret = init_get_bits8(&gb, src, len)) < 0) |
1026 | ✗ | return ret; | |
1027 | |||
1028 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 81 times.
|
81 | if (npal <= 2) nb = 1; |
1029 |
2/2✓ Branch 0 taken 15 times.
✓ Branch 1 taken 66 times.
|
81 | else if (npal <= 4) nb = 2; |
1030 |
2/2✓ Branch 0 taken 37 times.
✓ Branch 1 taken 29 times.
|
66 | else if (npal <= 16) nb = 4; |
1031 | 29 | else nb = 8; | |
1032 | |||
1033 |
4/4✓ Branch 0 taken 8576 times.
✓ Branch 1 taken 1792 times.
✓ Branch 2 taken 10368 times.
✓ Branch 3 taken 81 times.
|
10449 | for (j = 0; j < height; j++, dst += stride, jpeg_tile = FF_PTR_ADD(jpeg_tile, tile_stride)) { |
1034 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 10368 times.
|
10368 | if (get_bits(&gb, 8)) |
1035 | ✗ | continue; | |
1036 |
2/2✓ Branch 0 taken 1644544 times.
✓ Branch 1 taken 10368 times.
|
1654912 | for (i = 0; i < width; i++) { |
1037 | 1644544 | col = get_bits(&gb, nb); | |
1038 |
2/2✓ Branch 0 taken 1323579 times.
✓ Branch 1 taken 320965 times.
|
1644544 | if (col != tidx) |
1039 | 1323579 | memcpy(dst + i * 3, pal + col * 3, 3); | |
1040 | else | ||
1041 | 320965 | memcpy(dst + i * 3, jpeg_tile + i * 3, 3); | |
1042 | } | ||
1043 | 10368 | skip_bits_long(&gb, nb * (align_width - width)); | |
1044 | } | ||
1045 | |||
1046 | 81 | return 0; | |
1047 | } | ||
1048 | |||
1049 | 101 | static int kempf_decode_tile(G2MContext *c, int tile_x, int tile_y, | |
1050 | const uint8_t *src, int src_size) | ||
1051 | { | ||
1052 | int width, height; | ||
1053 | 101 | int hdr, zsize, npal, tidx = -1, ret; | |
1054 | int i, j; | ||
1055 | 101 | const uint8_t *src_end = src + src_size; | |
1056 | uint8_t pal[768], transp[3]; | ||
1057 | 101 | uLongf dlen = (c->tile_width + 1) * c->tile_height; | |
1058 | int sub_type; | ||
1059 | int nblocks, cblocks, bstride; | ||
1060 | int bits, bitbuf, coded; | ||
1061 | 101 | uint8_t *dst = c->framebuf + tile_x * c->tile_width * 3 + | |
1062 | 101 | tile_y * c->tile_height * c->framebuf_stride; | |
1063 | |||
1064 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 101 times.
|
101 | if (src_size < 2) |
1065 | ✗ | return AVERROR_INVALIDDATA; | |
1066 | |||
1067 | 101 | width = FFMIN(c->width - tile_x * c->tile_width, c->tile_width); | |
1068 | 101 | height = FFMIN(c->height - tile_y * c->tile_height, c->tile_height); | |
1069 | |||
1070 | 101 | hdr = *src++; | |
1071 | 101 | sub_type = hdr >> 5; | |
1072 |
2/2✓ Branch 0 taken 20 times.
✓ Branch 1 taken 81 times.
|
101 | if (sub_type == 0) { |
1073 | int j; | ||
1074 | 20 | memcpy(transp, src, 3); | |
1075 | 20 | src += 3; | |
1076 |
2/2✓ Branch 0 taken 2560 times.
✓ Branch 1 taken 20 times.
|
2580 | for (j = 0; j < height; j++, dst += c->framebuf_stride) |
1077 |
2/2✓ Branch 0 taken 401408 times.
✓ Branch 1 taken 2560 times.
|
403968 | for (i = 0; i < width; i++) |
1078 | 401408 | memcpy(dst + i * 3, transp, 3); | |
1079 | 20 | return 0; | |
1080 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 81 times.
|
81 | } else if (sub_type == 1) { |
1081 | ✗ | return jpg_decode_data(&c->jc, width, height, src, src_end - src, | |
1082 | dst, c->framebuf_stride, NULL, 0, 0, 0); | ||
1083 | } | ||
1084 | |||
1085 |
2/2✓ Branch 0 taken 67 times.
✓ Branch 1 taken 14 times.
|
81 | if (sub_type != 2) { |
1086 | 67 | memcpy(transp, src, 3); | |
1087 | 67 | src += 3; | |
1088 | } | ||
1089 | 81 | npal = *src++ + 1; | |
1090 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 81 times.
|
81 | if (src_end - src < npal * 3) |
1091 | ✗ | return AVERROR_INVALIDDATA; | |
1092 | 81 | memcpy(pal, src, npal * 3); | |
1093 | 81 | src += npal * 3; | |
1094 |
2/2✓ Branch 0 taken 67 times.
✓ Branch 1 taken 14 times.
|
81 | if (sub_type != 2) { |
1095 |
1/2✓ Branch 0 taken 134 times.
✗ Branch 1 not taken.
|
134 | for (i = 0; i < npal; i++) { |
1096 |
2/2✓ Branch 0 taken 67 times.
✓ Branch 1 taken 67 times.
|
134 | if (!memcmp(pal + i * 3, transp, 3)) { |
1097 | 67 | tidx = i; | |
1098 | 67 | break; | |
1099 | } | ||
1100 | } | ||
1101 | } | ||
1102 | |||
1103 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 81 times.
|
81 | if (src_end - src < 2) |
1104 | ✗ | return 0; | |
1105 | 81 | zsize = (src[0] << 8) | src[1]; | |
1106 | 81 | src += 2; | |
1107 | |||
1108 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 81 times.
|
81 | if (src_end - src < zsize + (sub_type != 2)) |
1109 | ✗ | return AVERROR_INVALIDDATA; | |
1110 | |||
1111 | 81 | ret = uncompress(c->kempf_buf, &dlen, src, zsize); | |
1112 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 81 times.
|
81 | if (ret) |
1113 | ✗ | return AVERROR_INVALIDDATA; | |
1114 | 81 | src += zsize; | |
1115 | |||
1116 |
2/2✓ Branch 0 taken 14 times.
✓ Branch 1 taken 67 times.
|
81 | if (sub_type == 2) { |
1117 | 14 | kempf_restore_buf(c->kempf_buf, dlen, dst, c->framebuf_stride, | |
1118 | NULL, 0, width, height, pal, npal, tidx); | ||
1119 | 14 | return 0; | |
1120 | } | ||
1121 | |||
1122 | 67 | nblocks = *src++ + 1; | |
1123 | 67 | cblocks = 0; | |
1124 | 67 | bstride = FFALIGN(width, 16) >> 3; | |
1125 | // blocks are coded LSB and we need normal bitreader for JPEG data | ||
1126 | 67 | bits = 0; | |
1127 |
2/2✓ Branch 0 taken 536 times.
✓ Branch 1 taken 67 times.
|
603 | for (i = 0; i < (FFALIGN(height, 16) >> 4); i++) { |
1128 |
2/2✓ Branch 0 taken 5448 times.
✓ Branch 1 taken 536 times.
|
5984 | for (j = 0; j < (FFALIGN(width, 16) >> 4); j++) { |
1129 |
2/2✓ Branch 0 taken 681 times.
✓ Branch 1 taken 4767 times.
|
5448 | if (!bits) { |
1130 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 681 times.
|
681 | if (src >= src_end) |
1131 | ✗ | return AVERROR_INVALIDDATA; | |
1132 | 681 | bitbuf = *src++; | |
1133 | 681 | bits = 8; | |
1134 | } | ||
1135 | 5448 | coded = bitbuf & 1; | |
1136 | 5448 | bits--; | |
1137 | 5448 | bitbuf >>= 1; | |
1138 | 5448 | cblocks += coded; | |
1139 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 5448 times.
|
5448 | if (cblocks > nblocks) |
1140 | ✗ | return AVERROR_INVALIDDATA; | |
1141 | 5448 | c->kempf_flags[j * 2 + i * 2 * bstride] = | |
1142 | 5448 | c->kempf_flags[j * 2 + 1 + i * 2 * bstride] = | |
1143 | 5448 | c->kempf_flags[j * 2 + (i * 2 + 1) * bstride] = | |
1144 | 5448 | c->kempf_flags[j * 2 + 1 + (i * 2 + 1) * bstride] = coded; | |
1145 | } | ||
1146 | } | ||
1147 | |||
1148 | 67 | memset(c->jpeg_tile, 0, c->tile_stride * height); | |
1149 | 67 | jpg_decode_data(&c->jc, width, height, src, src_end - src, | |
1150 | c->jpeg_tile, c->tile_stride, | ||
1151 | 67 | c->kempf_flags, bstride, nblocks * 4, 0); | |
1152 | |||
1153 | 67 | kempf_restore_buf(c->kempf_buf, dlen, dst, c->framebuf_stride, | |
1154 | 67 | c->jpeg_tile, c->tile_stride, | |
1155 | width, height, pal, npal, tidx); | ||
1156 | |||
1157 | 67 | return 0; | |
1158 | } | ||
1159 | |||
1160 | 4 | static int g2m_init_buffers(G2MContext *c) | |
1161 | { | ||
1162 | int aligned_height; | ||
1163 | |||
1164 | 4 | c->framebuf_stride = FFALIGN(c->width + 15, 16) * 3; | |
1165 | 4 | aligned_height = c->height + 15; | |
1166 | |||
1167 | 4 | av_fast_mallocz(&c->framebuf, &c->framebuf_allocated, c->framebuf_stride * aligned_height); | |
1168 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
|
4 | if (!c->framebuf) |
1169 | ✗ | return AVERROR(ENOMEM); | |
1170 | |||
1171 |
3/4✓ Branch 0 taken 1 times.
✓ Branch 1 taken 3 times.
✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
|
4 | if (!c->synth_tile || !c->jpeg_tile || |
1172 |
2/4✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
|
1 | (c->compression == 2 && !c->epic_buf_base) || |
1173 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
|
1 | c->old_tile_w < c->tile_width || |
1174 | ✗ | c->old_tile_h < c->tile_height) { | |
1175 | 4 | c->tile_stride = FFALIGN(c->tile_width, 16) * 3; | |
1176 | 4 | c->epic_buf_stride = FFALIGN(c->tile_width * 4, 16); | |
1177 | 4 | aligned_height = FFALIGN(c->tile_height, 16); | |
1178 | 4 | av_freep(&c->synth_tile); | |
1179 | 4 | av_freep(&c->jpeg_tile); | |
1180 | 4 | av_freep(&c->kempf_buf); | |
1181 | 4 | av_freep(&c->kempf_flags); | |
1182 | 4 | av_freep(&c->epic_buf_base); | |
1183 | 4 | c->epic_buf = NULL; | |
1184 | 4 | c->synth_tile = av_mallocz(c->tile_stride * aligned_height); | |
1185 | 4 | c->jpeg_tile = av_mallocz(c->tile_stride * aligned_height); | |
1186 | 4 | c->kempf_buf = av_mallocz((c->tile_width + 1) * aligned_height + | |
1187 | AV_INPUT_BUFFER_PADDING_SIZE); | ||
1188 | 4 | c->kempf_flags = av_mallocz(c->tile_width * aligned_height); | |
1189 |
2/4✓ Branch 0 taken 4 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 4 times.
✗ Branch 3 not taken.
|
4 | if (!c->synth_tile || !c->jpeg_tile || |
1190 |
2/4✓ Branch 0 taken 4 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 4 times.
|
4 | !c->kempf_buf || !c->kempf_flags) |
1191 | ✗ | return AVERROR(ENOMEM); | |
1192 |
2/2✓ Branch 0 taken 3 times.
✓ Branch 1 taken 1 times.
|
4 | if (c->compression == 2) { |
1193 | 3 | c->epic_buf_base = av_mallocz(c->epic_buf_stride * aligned_height + 4); | |
1194 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 3 times.
|
3 | if (!c->epic_buf_base) |
1195 | ✗ | return AVERROR(ENOMEM); | |
1196 | 3 | c->epic_buf = c->epic_buf_base + 4; | |
1197 | } | ||
1198 | } | ||
1199 | |||
1200 | 4 | return 0; | |
1201 | } | ||
1202 | |||
1203 | 14 | static int g2m_load_cursor(AVCodecContext *avctx, G2MContext *c, | |
1204 | GetByteContext *gb) | ||
1205 | { | ||
1206 | int i, j, k; | ||
1207 | uint8_t *dst; | ||
1208 | uint32_t bits; | ||
1209 | uint32_t cur_size, cursor_w, cursor_h, cursor_stride; | ||
1210 | uint32_t cursor_hot_x, cursor_hot_y; | ||
1211 | int cursor_fmt, err; | ||
1212 | |||
1213 | 14 | cur_size = bytestream2_get_be32(gb); | |
1214 | 14 | cursor_w = bytestream2_get_byte(gb); | |
1215 | 14 | cursor_h = bytestream2_get_byte(gb); | |
1216 | 14 | cursor_hot_x = bytestream2_get_byte(gb); | |
1217 | 14 | cursor_hot_y = bytestream2_get_byte(gb); | |
1218 | 14 | cursor_fmt = bytestream2_get_byte(gb); | |
1219 | |||
1220 |
4/4✓ Branch 0 taken 13 times.
✓ Branch 1 taken 1 times.
✓ Branch 2 taken 13 times.
✓ Branch 3 taken 1 times.
|
14 | cursor_stride = FFALIGN(cursor_w, cursor_fmt==1 ? 32 : 1) * 4; |
1221 | |||
1222 |
3/6✓ Branch 0 taken 14 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 14 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 14 times.
✗ Branch 5 not taken.
|
14 | if (cursor_w < 1 || cursor_w > 256 || |
1223 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 14 times.
|
14 | cursor_h < 1 || cursor_h > 256) { |
1224 | ✗ | av_log(avctx, AV_LOG_ERROR, "Invalid cursor dimensions %"PRIu32"x%"PRIu32"\n", | |
1225 | cursor_w, cursor_h); | ||
1226 | ✗ | return AVERROR_INVALIDDATA; | |
1227 | } | ||
1228 |
2/4✓ Branch 0 taken 14 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 14 times.
|
14 | if (cursor_hot_x > cursor_w || cursor_hot_y > cursor_h) { |
1229 | ✗ | av_log(avctx, AV_LOG_WARNING, "Invalid hotspot position %"PRIu32",%"PRIu32"\n", | |
1230 | cursor_hot_x, cursor_hot_y); | ||
1231 | ✗ | cursor_hot_x = FFMIN(cursor_hot_x, cursor_w - 1); | |
1232 | ✗ | cursor_hot_y = FFMIN(cursor_hot_y, cursor_h - 1); | |
1233 | } | ||
1234 |
1/2✓ Branch 1 taken 14 times.
✗ Branch 2 not taken.
|
14 | if (cur_size - 9 > bytestream2_get_bytes_left(gb) || |
1235 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 14 times.
|
14 | c->cursor_w * c->cursor_h / 4 > cur_size) { |
1236 | ✗ | av_log(avctx, AV_LOG_ERROR, "Invalid cursor data size %"PRIu32"/%u\n", | |
1237 | cur_size, bytestream2_get_bytes_left(gb)); | ||
1238 | ✗ | return AVERROR_INVALIDDATA; | |
1239 | } | ||
1240 |
3/4✓ Branch 0 taken 1 times.
✓ Branch 1 taken 13 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 1 times.
|
14 | if (cursor_fmt != 1 && cursor_fmt != 32) { |
1241 | ✗ | avpriv_report_missing_feature(avctx, "Cursor format %d", | |
1242 | cursor_fmt); | ||
1243 | ✗ | return AVERROR_PATCHWELCOME; | |
1244 | } | ||
1245 | |||
1246 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 14 times.
|
14 | if ((err = av_reallocp(&c->cursor, cursor_stride * cursor_h)) < 0) { |
1247 | ✗ | av_log(avctx, AV_LOG_ERROR, "Cannot allocate cursor buffer\n"); | |
1248 | ✗ | return err; | |
1249 | } | ||
1250 | |||
1251 | 14 | c->cursor_w = cursor_w; | |
1252 | 14 | c->cursor_h = cursor_h; | |
1253 | 14 | c->cursor_hot_x = cursor_hot_x; | |
1254 | 14 | c->cursor_hot_y = cursor_hot_y; | |
1255 | 14 | c->cursor_fmt = cursor_fmt; | |
1256 | 14 | c->cursor_stride = cursor_stride; | |
1257 | |||
1258 | 14 | dst = c->cursor; | |
1259 |
2/3✓ Branch 0 taken 13 times.
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
|
14 | switch (c->cursor_fmt) { |
1260 | 13 | case 1: // old monochrome | |
1261 |
2/2✓ Branch 0 taken 416 times.
✓ Branch 1 taken 13 times.
|
429 | for (j = 0; j < c->cursor_h; j++) { |
1262 |
2/2✓ Branch 0 taken 416 times.
✓ Branch 1 taken 416 times.
|
832 | for (i = 0; i < c->cursor_w; i += 32) { |
1263 | 416 | bits = bytestream2_get_be32(gb); | |
1264 |
2/2✓ Branch 0 taken 13312 times.
✓ Branch 1 taken 416 times.
|
13728 | for (k = 0; k < 32; k++) { |
1265 | 13312 | dst[0] = !!(bits & 0x80000000); | |
1266 | 13312 | dst += 4; | |
1267 | 13312 | bits <<= 1; | |
1268 | } | ||
1269 | } | ||
1270 | } | ||
1271 | |||
1272 | 13 | dst = c->cursor; | |
1273 |
2/2✓ Branch 0 taken 416 times.
✓ Branch 1 taken 13 times.
|
429 | for (j = 0; j < c->cursor_h; j++) { |
1274 |
2/2✓ Branch 0 taken 416 times.
✓ Branch 1 taken 416 times.
|
832 | for (i = 0; i < c->cursor_w; i += 32) { |
1275 | 416 | bits = bytestream2_get_be32(gb); | |
1276 |
2/2✓ Branch 0 taken 13312 times.
✓ Branch 1 taken 416 times.
|
13728 | for (k = 0; k < 32; k++) { |
1277 | 13312 | int mask_bit = !!(bits & 0x80000000); | |
1278 |
3/3✓ Branch 0 taken 824 times.
✓ Branch 1 taken 920 times.
✓ Branch 2 taken 11568 times.
|
13312 | switch (dst[0] * 2 + mask_bit) { |
1279 | 824 | case 0: | |
1280 | 824 | dst[0] = 0xFF; | |
1281 | 824 | dst[1] = 0x00; | |
1282 | 824 | dst[2] = 0x00; | |
1283 | 824 | dst[3] = 0x00; | |
1284 | 824 | break; | |
1285 | 920 | case 1: | |
1286 | 920 | dst[0] = 0xFF; | |
1287 | 920 | dst[1] = 0xFF; | |
1288 | 920 | dst[2] = 0xFF; | |
1289 | 920 | dst[3] = 0xFF; | |
1290 | 920 | break; | |
1291 | 11568 | default: | |
1292 | 11568 | dst[0] = 0x00; | |
1293 | 11568 | dst[1] = 0x00; | |
1294 | 11568 | dst[2] = 0x00; | |
1295 | 11568 | dst[3] = 0x00; | |
1296 | } | ||
1297 | 13312 | dst += 4; | |
1298 | 13312 | bits <<= 1; | |
1299 | } | ||
1300 | } | ||
1301 | } | ||
1302 | 13 | break; | |
1303 | 1 | case 32: // full colour | |
1304 | /* skip monochrome version of the cursor and decode RGBA instead */ | ||
1305 | 1 | bytestream2_skip(gb, c->cursor_h * (FFALIGN(c->cursor_w, 32) >> 3)); | |
1306 |
2/2✓ Branch 0 taken 32 times.
✓ Branch 1 taken 1 times.
|
33 | for (j = 0; j < c->cursor_h; j++) { |
1307 |
2/2✓ Branch 0 taken 1024 times.
✓ Branch 1 taken 32 times.
|
1056 | for (i = 0; i < c->cursor_w; i++) { |
1308 | 1024 | int val = bytestream2_get_be32(gb); | |
1309 | 1024 | *dst++ = val >> 0; | |
1310 | 1024 | *dst++ = val >> 8; | |
1311 | 1024 | *dst++ = val >> 16; | |
1312 | 1024 | *dst++ = val >> 24; | |
1313 | } | ||
1314 | } | ||
1315 | 1 | break; | |
1316 | ✗ | default: | |
1317 | ✗ | return AVERROR_PATCHWELCOME; | |
1318 | } | ||
1319 | 14 | return 0; | |
1320 | } | ||
1321 | |||
1322 | #define APPLY_ALPHA(src, new, alpha) \ | ||
1323 | src = (src * (256 - alpha) + new * alpha) >> 8 | ||
1324 | |||
1325 | 247 | static void g2m_paint_cursor(G2MContext *c, uint8_t *dst, int stride) | |
1326 | { | ||
1327 | int i, j; | ||
1328 | int x, y, w, h; | ||
1329 | const uint8_t *cursor; | ||
1330 | |||
1331 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 247 times.
|
247 | if (!c->cursor) |
1332 | ✗ | return; | |
1333 | |||
1334 | 247 | x = c->cursor_x - c->cursor_hot_x; | |
1335 | 247 | y = c->cursor_y - c->cursor_hot_y; | |
1336 | |||
1337 | 247 | cursor = c->cursor; | |
1338 | 247 | w = c->cursor_w; | |
1339 | 247 | h = c->cursor_h; | |
1340 | |||
1341 |
2/2✓ Branch 0 taken 52 times.
✓ Branch 1 taken 195 times.
|
247 | if (x + w > c->width) |
1342 | 52 | w = c->width - x; | |
1343 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 247 times.
|
247 | if (y + h > c->height) |
1344 | ✗ | h = c->height - y; | |
1345 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 247 times.
|
247 | if (x < 0) { |
1346 | ✗ | w += x; | |
1347 | ✗ | cursor += -x * 4; | |
1348 | } else { | ||
1349 | 247 | dst += x * 3; | |
1350 | } | ||
1351 | |||
1352 |
2/2✓ Branch 0 taken 3 times.
✓ Branch 1 taken 244 times.
|
247 | if (y < 0) |
1353 | 3 | h += y; | |
1354 |
2/4✓ Branch 0 taken 247 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 247 times.
|
247 | if (w < 0 || h < 0) |
1355 | ✗ | return; | |
1356 |
2/2✓ Branch 0 taken 3 times.
✓ Branch 1 taken 244 times.
|
247 | if (y < 0) { |
1357 | 3 | cursor += -y * c->cursor_stride; | |
1358 | } else { | ||
1359 | 244 | dst += y * stride; | |
1360 | } | ||
1361 | |||
1362 |
2/2✓ Branch 0 taken 7886 times.
✓ Branch 1 taken 247 times.
|
8133 | for (j = 0; j < h; j++) { |
1363 |
2/2✓ Branch 0 taken 229088 times.
✓ Branch 1 taken 7886 times.
|
236974 | for (i = 0; i < w; i++) { |
1364 | 229088 | uint8_t alpha = cursor[i * 4]; | |
1365 | 229088 | APPLY_ALPHA(dst[i * 3 + 0], cursor[i * 4 + 1], alpha); | |
1366 | 229088 | APPLY_ALPHA(dst[i * 3 + 1], cursor[i * 4 + 2], alpha); | |
1367 | 229088 | APPLY_ALPHA(dst[i * 3 + 2], cursor[i * 4 + 3], alpha); | |
1368 | } | ||
1369 | 7886 | dst += stride; | |
1370 | 7886 | cursor += c->cursor_stride; | |
1371 | } | ||
1372 | } | ||
1373 | |||
1374 | 247 | static int g2m_decode_frame(AVCodecContext *avctx, AVFrame *pic, | |
1375 | int *got_picture_ptr, AVPacket *avpkt) | ||
1376 | { | ||
1377 | 247 | const uint8_t *buf = avpkt->data; | |
1378 | 247 | int buf_size = avpkt->size; | |
1379 | 247 | G2MContext *c = avctx->priv_data; | |
1380 | GetByteContext bc, tbc; | ||
1381 | int magic; | ||
1382 | 247 | int got_header = 0; | |
1383 | uint32_t chunk_size, r_mask, g_mask, b_mask; | ||
1384 | int chunk_type, chunk_start; | ||
1385 | int i; | ||
1386 | int ret; | ||
1387 | |||
1388 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 247 times.
|
247 | if (buf_size < 12) { |
1389 | ✗ | av_log(avctx, AV_LOG_ERROR, | |
1390 | "Frame should have at least 12 bytes, got %d instead\n", | ||
1391 | buf_size); | ||
1392 | ✗ | return AVERROR_INVALIDDATA; | |
1393 | } | ||
1394 | |||
1395 | 247 | bytestream2_init(&bc, buf, buf_size); | |
1396 | |||
1397 | 247 | magic = bytestream2_get_be32(&bc); | |
1398 |
1/2✓ Branch 0 taken 247 times.
✗ Branch 1 not taken.
|
247 | if ((magic & ~0xF) != MKBETAG('G', '2', 'M', '0') || |
1399 |
2/4✓ Branch 0 taken 247 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 247 times.
|
247 | (magic & 0xF) < 2 || (magic & 0xF) > 5) { |
1400 | ✗ | av_log(avctx, AV_LOG_ERROR, "Wrong magic %08X\n", magic); | |
1401 | ✗ | return AVERROR_INVALIDDATA; | |
1402 | } | ||
1403 | |||
1404 | 247 | c->swapuv = magic == MKBETAG('G', '2', 'M', '2'); | |
1405 | |||
1406 |
2/2✓ Branch 1 taken 514 times.
✓ Branch 2 taken 247 times.
|
761 | while (bytestream2_get_bytes_left(&bc) > 5) { |
1407 | 514 | chunk_size = bytestream2_get_le32(&bc) - 1; | |
1408 | 514 | chunk_type = bytestream2_get_byte(&bc); | |
1409 | 514 | chunk_start = bytestream2_tell(&bc); | |
1410 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 514 times.
|
514 | if (chunk_size > bytestream2_get_bytes_left(&bc)) { |
1411 | ✗ | av_log(avctx, AV_LOG_ERROR, "Invalid chunk size %"PRIu32" type %02X\n", | |
1412 | chunk_size, chunk_type); | ||
1413 | ✗ | break; | |
1414 | } | ||
1415 |
5/6✓ Branch 0 taken 4 times.
✓ Branch 1 taken 237 times.
✓ Branch 2 taken 245 times.
✓ Branch 3 taken 14 times.
✓ Branch 4 taken 14 times.
✗ Branch 5 not taken.
|
514 | switch (chunk_type) { |
1416 | 4 | case DISPLAY_INFO: | |
1417 | 4 | got_header = | |
1418 | 4 | c->got_header = 0; | |
1419 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
|
4 | if (chunk_size < 21) { |
1420 | ✗ | av_log(avctx, AV_LOG_ERROR, "Invalid display info size %"PRIu32"\n", | |
1421 | chunk_size); | ||
1422 | ✗ | break; | |
1423 | } | ||
1424 | 4 | c->width = bytestream2_get_be32(&bc); | |
1425 | 4 | c->height = bytestream2_get_be32(&bc); | |
1426 |
2/4✓ Branch 0 taken 4 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 4 times.
|
4 | if (c->width < 16 || c->height < 16) { |
1427 | ✗ | av_log(avctx, AV_LOG_ERROR, | |
1428 | "Invalid frame dimensions %dx%d\n", | ||
1429 | c->width, c->height); | ||
1430 | ✗ | ret = AVERROR_INVALIDDATA; | |
1431 | ✗ | goto header_fail; | |
1432 | } | ||
1433 |
2/4✓ Branch 0 taken 4 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 4 times.
|
4 | if (c->width != avctx->width || c->height != avctx->height) { |
1434 | ✗ | ret = ff_set_dimensions(avctx, c->width, c->height); | |
1435 | ✗ | if (ret < 0) | |
1436 | ✗ | goto header_fail; | |
1437 | } | ||
1438 | 4 | c->compression = bytestream2_get_be32(&bc); | |
1439 |
3/4✓ Branch 0 taken 1 times.
✓ Branch 1 taken 3 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 1 times.
|
4 | if (c->compression != 2 && c->compression != 3) { |
1440 | ✗ | avpriv_report_missing_feature(avctx, "Compression method %d", | |
1441 | c->compression); | ||
1442 | ✗ | ret = AVERROR_PATCHWELCOME; | |
1443 | ✗ | goto header_fail; | |
1444 | } | ||
1445 | 4 | c->tile_width = bytestream2_get_be32(&bc); | |
1446 | 4 | c->tile_height = bytestream2_get_be32(&bc); | |
1447 |
2/4✓ Branch 0 taken 4 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 4 times.
✗ Branch 3 not taken.
|
4 | if (c->tile_width <= 0 || c->tile_height <= 0 || |
1448 |
1/2✓ Branch 0 taken 4 times.
✗ Branch 1 not taken.
|
4 | ((c->tile_width | c->tile_height) & 0xF) || |
1449 |
2/4✓ Branch 0 taken 4 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 4 times.
|
8 | c->tile_width * (uint64_t)c->tile_height >= INT_MAX / 4 || |
1450 | 4 | av_image_check_size2(c->tile_width, c->tile_height, avctx->max_pixels, avctx->pix_fmt, 0, avctx) < 0 | |
1451 | ) { | ||
1452 | ✗ | av_log(avctx, AV_LOG_ERROR, | |
1453 | "Invalid tile dimensions %dx%d\n", | ||
1454 | c->tile_width, c->tile_height); | ||
1455 | ✗ | ret = AVERROR_INVALIDDATA; | |
1456 | ✗ | goto header_fail; | |
1457 | } | ||
1458 | 4 | c->tiles_x = (c->width + c->tile_width - 1) / c->tile_width; | |
1459 | 4 | c->tiles_y = (c->height + c->tile_height - 1) / c->tile_height; | |
1460 | 4 | c->bpp = bytestream2_get_byte(&bc); | |
1461 |
1/2✓ Branch 0 taken 4 times.
✗ Branch 1 not taken.
|
4 | if (c->bpp == 32) { |
1462 |
1/2✓ Branch 1 taken 4 times.
✗ Branch 2 not taken.
|
4 | if (bytestream2_get_bytes_left(&bc) < 16 || |
1463 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
|
4 | (chunk_size - 21) < 16) { |
1464 | ✗ | av_log(avctx, AV_LOG_ERROR, | |
1465 | "Display info: missing bitmasks!\n"); | ||
1466 | ✗ | ret = AVERROR_INVALIDDATA; | |
1467 | ✗ | goto header_fail; | |
1468 | } | ||
1469 | 4 | r_mask = bytestream2_get_be32(&bc); | |
1470 | 4 | g_mask = bytestream2_get_be32(&bc); | |
1471 | 4 | b_mask = bytestream2_get_be32(&bc); | |
1472 |
3/6✓ Branch 0 taken 4 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 4 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 4 times.
|
4 | if (r_mask != 0xFF0000 || g_mask != 0xFF00 || b_mask != 0xFF) { |
1473 | ✗ | avpriv_report_missing_feature(avctx, | |
1474 | "Bitmasks: R=%"PRIX32", G=%"PRIX32", B=%"PRIX32, | ||
1475 | r_mask, g_mask, b_mask); | ||
1476 | ✗ | ret = AVERROR_PATCHWELCOME; | |
1477 | ✗ | goto header_fail; | |
1478 | } | ||
1479 | } else { | ||
1480 | ✗ | avpriv_request_sample(avctx, "bpp=%d", c->bpp); | |
1481 | ✗ | ret = AVERROR_PATCHWELCOME; | |
1482 | ✗ | goto header_fail; | |
1483 | } | ||
1484 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 4 times.
|
4 | if (g2m_init_buffers(c)) { |
1485 | ✗ | ret = AVERROR(ENOMEM); | |
1486 | ✗ | goto header_fail; | |
1487 | } | ||
1488 | 4 | got_header = 1; | |
1489 | 4 | break; | |
1490 | 237 | case TILE_DATA: | |
1491 |
2/4✓ Branch 0 taken 237 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 237 times.
|
237 | if (!c->tiles_x || !c->tiles_y) { |
1492 | ✗ | av_log(avctx, AV_LOG_WARNING, | |
1493 | "No display info - skipping tile\n"); | ||
1494 | ✗ | break; | |
1495 | } | ||
1496 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 237 times.
|
237 | if (chunk_size < 2) { |
1497 | ✗ | av_log(avctx, AV_LOG_ERROR, "Invalid tile data size %"PRIu32"\n", | |
1498 | chunk_size); | ||
1499 | ✗ | break; | |
1500 | } | ||
1501 | 237 | c->tile_x = bytestream2_get_byte(&bc); | |
1502 | 237 | c->tile_y = bytestream2_get_byte(&bc); | |
1503 |
2/4✓ Branch 0 taken 237 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 237 times.
|
237 | if (c->tile_x >= c->tiles_x || c->tile_y >= c->tiles_y) { |
1504 | ✗ | av_log(avctx, AV_LOG_ERROR, | |
1505 | "Invalid tile pos %d,%d (in %dx%d grid)\n", | ||
1506 | c->tile_x, c->tile_y, c->tiles_x, c->tiles_y); | ||
1507 | ✗ | break; | |
1508 | } | ||
1509 | 237 | ret = 0; | |
1510 |
2/3✓ Branch 0 taken 136 times.
✓ Branch 1 taken 101 times.
✗ Branch 2 not taken.
|
237 | switch (c->compression) { |
1511 | 136 | case COMPR_EPIC_J_B: | |
1512 | 272 | ret = epic_jb_decode_tile(c, c->tile_x, c->tile_y, | |
1513 | 136 | buf + bytestream2_tell(&bc), | |
1514 | 136 | chunk_size - 2, avctx); | |
1515 | 136 | break; | |
1516 | 101 | case COMPR_KEMPF_J_B: | |
1517 | 202 | ret = kempf_decode_tile(c, c->tile_x, c->tile_y, | |
1518 | 101 | buf + bytestream2_tell(&bc), | |
1519 | 101 | chunk_size - 2); | |
1520 | 101 | break; | |
1521 | } | ||
1522 |
1/4✗ Branch 0 not taken.
✓ Branch 1 taken 237 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
|
237 | if (ret && c->framebuf) |
1523 | ✗ | av_log(avctx, AV_LOG_ERROR, "Error decoding tile %d,%d\n", | |
1524 | c->tile_x, c->tile_y); | ||
1525 | 237 | break; | |
1526 | 245 | case CURSOR_POS: | |
1527 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 245 times.
|
245 | if (chunk_size < 5) { |
1528 | ✗ | av_log(avctx, AV_LOG_ERROR, "Invalid cursor pos size %"PRIu32"\n", | |
1529 | chunk_size); | ||
1530 | ✗ | break; | |
1531 | } | ||
1532 | 245 | c->cursor_x = bytestream2_get_be16(&bc); | |
1533 | 245 | c->cursor_y = bytestream2_get_be16(&bc); | |
1534 | 245 | break; | |
1535 | 14 | case CURSOR_SHAPE: | |
1536 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 14 times.
|
14 | if (chunk_size < 8) { |
1537 | ✗ | av_log(avctx, AV_LOG_ERROR, "Invalid cursor data size %"PRIu32"\n", | |
1538 | chunk_size); | ||
1539 | ✗ | break; | |
1540 | } | ||
1541 | 14 | bytestream2_init(&tbc, buf + bytestream2_tell(&bc), | |
1542 | 14 | chunk_size - 4); | |
1543 | 14 | g2m_load_cursor(avctx, c, &tbc); | |
1544 | 14 | break; | |
1545 | 14 | case CHUNK_CC: | |
1546 | case CHUNK_CD: | ||
1547 | 14 | break; | |
1548 | ✗ | default: | |
1549 | ✗ | av_log(avctx, AV_LOG_WARNING, "Skipping chunk type %02d\n", | |
1550 | chunk_type); | ||
1551 | } | ||
1552 | |||
1553 | /* navigate to next chunk */ | ||
1554 | 514 | bytestream2_skip(&bc, chunk_start + chunk_size - bytestream2_tell(&bc)); | |
1555 | } | ||
1556 |
2/2✓ Branch 0 taken 4 times.
✓ Branch 1 taken 243 times.
|
247 | if (got_header) |
1557 | 4 | c->got_header = 1; | |
1558 | |||
1559 |
3/6✓ Branch 0 taken 247 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 247 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 247 times.
✗ Branch 5 not taken.
|
247 | if (c->width && c->height && c->framebuf) { |
1560 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 247 times.
|
247 | if ((ret = ff_get_buffer(avctx, pic, 0)) < 0) |
1561 | ✗ | return ret; | |
1562 | |||
1563 |
2/2✓ Branch 0 taken 4 times.
✓ Branch 1 taken 243 times.
|
247 | if (got_header) |
1564 | 4 | pic->flags |= AV_FRAME_FLAG_KEY; | |
1565 | else | ||
1566 | 243 | pic->flags &= ~AV_FRAME_FLAG_KEY; | |
1567 |
2/2✓ Branch 0 taken 4 times.
✓ Branch 1 taken 243 times.
|
247 | pic->pict_type = got_header ? AV_PICTURE_TYPE_I : AV_PICTURE_TYPE_P; |
1568 | |||
1569 |
2/2✓ Branch 0 taken 210316 times.
✓ Branch 1 taken 247 times.
|
210563 | for (i = 0; i < avctx->height; i++) |
1570 | 210316 | memcpy(pic->data[0] + i * pic->linesize[0], | |
1571 | 210316 | c->framebuf + i * c->framebuf_stride, | |
1572 | 210316 | c->width * 3); | |
1573 | 247 | g2m_paint_cursor(c, pic->data[0], pic->linesize[0]); | |
1574 | |||
1575 | 247 | *got_picture_ptr = 1; | |
1576 | } | ||
1577 | |||
1578 | 247 | return buf_size; | |
1579 | |||
1580 | ✗ | header_fail: | |
1581 | ✗ | c->width = | |
1582 | ✗ | c->height = 0; | |
1583 | ✗ | c->tiles_x = | |
1584 | ✗ | c->tiles_y = 0; | |
1585 | ✗ | c->tile_width = | |
1586 | ✗ | c->tile_height = 0; | |
1587 | ✗ | return ret; | |
1588 | } | ||
1589 | |||
1590 | 6 | static av_cold int g2m_decode_init(AVCodecContext *avctx) | |
1591 | { | ||
1592 | 6 | G2MContext *const c = avctx->priv_data; | |
1593 | int ret; | ||
1594 | |||
1595 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 6 times.
|
6 | if ((ret = jpg_init(avctx, &c->jc)) != 0) { |
1596 | ✗ | av_log(avctx, AV_LOG_ERROR, "Cannot initialise VLCs\n"); | |
1597 | ✗ | return AVERROR(ENOMEM); | |
1598 | } | ||
1599 | |||
1600 | 6 | avctx->pix_fmt = AV_PIX_FMT_RGB24; | |
1601 | |||
1602 | // store original sizes and check against those if resize happens | ||
1603 | 6 | c->orig_width = avctx->width; | |
1604 | 6 | c->orig_height = avctx->height; | |
1605 | |||
1606 | 6 | return 0; | |
1607 | } | ||
1608 | |||
1609 | 6 | static av_cold int g2m_decode_end(AVCodecContext *avctx) | |
1610 | { | ||
1611 | 6 | G2MContext *const c = avctx->priv_data; | |
1612 | |||
1613 | 6 | jpg_free_context(&c->jc); | |
1614 | |||
1615 | 6 | av_freep(&c->epic_buf_base); | |
1616 | 6 | c->epic_buf = NULL; | |
1617 | 6 | av_freep(&c->kempf_buf); | |
1618 | 6 | av_freep(&c->kempf_flags); | |
1619 | 6 | av_freep(&c->synth_tile); | |
1620 | 6 | av_freep(&c->jpeg_tile); | |
1621 | 6 | av_freep(&c->cursor); | |
1622 | 6 | av_freep(&c->framebuf); | |
1623 | 6 | c->framebuf_allocated = 0; | |
1624 | |||
1625 | 6 | return 0; | |
1626 | } | ||
1627 | |||
1628 | const FFCodec ff_g2m_decoder = { | ||
1629 | .p.name = "g2m", | ||
1630 | CODEC_LONG_NAME("Go2Meeting"), | ||
1631 | .p.type = AVMEDIA_TYPE_VIDEO, | ||
1632 | .p.id = AV_CODEC_ID_G2M, | ||
1633 | .priv_data_size = sizeof(G2MContext), | ||
1634 | .init = g2m_decode_init, | ||
1635 | .close = g2m_decode_end, | ||
1636 | FF_CODEC_DECODE_CB(g2m_decode_frame), | ||
1637 | .p.capabilities = AV_CODEC_CAP_DR1, | ||
1638 | .caps_internal = FF_CODEC_CAP_INIT_CLEANUP, | ||
1639 | }; | ||
1640 |