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