FFmpeg coverage


Directory: ../../../ffmpeg/
File: src/libavcodec/escape130.c
Date: 2021-09-24 20:55:06
Exec Total Coverage
Lines: 146 158 92.4%
Branches: 47 58 81.0%

Line Branch Exec Source
1 /*
2 * Escape 130 video decoder
3 * Copyright (C) 2008 Eli Friedman (eli.friedman <at> gmail.com)
4 *
5 * This file is part of FFmpeg.
6 *
7 * FFmpeg is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
11 *
12 * FFmpeg is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with FFmpeg; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20 */
21
22 #include "libavutil/attributes.h"
23 #include "libavutil/mem.h"
24
25 #define BITSTREAM_READER_LE
26 #include "avcodec.h"
27 #include "get_bits.h"
28 #include "internal.h"
29
30 typedef struct Escape130Context {
31 uint8_t *old_y_avg;
32
33 uint8_t *new_y, *old_y;
34 uint8_t *new_u, *old_u;
35 uint8_t *new_v, *old_v;
36
37 uint8_t *buf1, *buf2;
38 int linesize[3];
39 } Escape130Context;
40
41 static const uint8_t offset_table[] = { 2, 4, 10, 20 };
42 static const int8_t sign_table[64][4] = {
43 { 0, 0, 0, 0 },
44 { -1, 1, 0, 0 },
45 { 1, -1, 0, 0 },
46 { -1, 0, 1, 0 },
47 { -1, 1, 1, 0 },
48 { 0, -1, 1, 0 },
49 { 1, -1, 1, 0 },
50 { -1, -1, 1, 0 },
51 { 1, 0, -1, 0 },
52 { 0, 1, -1, 0 },
53 { 1, 1, -1, 0 },
54 { -1, 1, -1, 0 },
55 { 1, -1, -1, 0 },
56 { -1, 0, 0, 1 },
57 { -1, 1, 0, 1 },
58 { 0, -1, 0, 1 },
59
60 { 0, 0, 0, 0 },
61 { 1, -1, 0, 1 },
62 { -1, -1, 0, 1 },
63 { -1, 0, 1, 1 },
64 { -1, 1, 1, 1 },
65 { 0, -1, 1, 1 },
66 { 1, -1, 1, 1 },
67 { -1, -1, 1, 1 },
68 { 0, 0, -1, 1 },
69 { 1, 0, -1, 1 },
70 { -1, 0, -1, 1 },
71 { 0, 1, -1, 1 },
72 { 1, 1, -1, 1 },
73 { -1, 1, -1, 1 },
74 { 0, -1, -1, 1 },
75 { 1, -1, -1, 1 },
76
77 { 0, 0, 0, 0 },
78 { -1, -1, -1, 1 },
79 { 1, 0, 0, -1 },
80 { 0, 1, 0, -1 },
81 { 1, 1, 0, -1 },
82 { -1, 1, 0, -1 },
83 { 1, -1, 0, -1 },
84 { 0, 0, 1, -1 },
85 { 1, 0, 1, -1 },
86 { -1, 0, 1, -1 },
87 { 0, 1, 1, -1 },
88 { 1, 1, 1, -1 },
89 { -1, 1, 1, -1 },
90 { 0, -1, 1, -1 },
91 { 1, -1, 1, -1 },
92 { -1, -1, 1, -1 },
93
94 { 0, 0, 0, 0 },
95 { 1, 0, -1, -1 },
96 { 0, 1, -1, -1 },
97 { 1, 1, -1, -1 },
98 { -1, 1, -1, -1 },
99 { 1, -1, -1, -1 }
100 };
101
102 static const int8_t luma_adjust[] = { -4, -3, -2, -1, 1, 2, 3, 4 };
103
104 static const int8_t chroma_adjust[2][8] = {
105 { 1, 1, 0, -1, -1, -1, 0, 1 },
106 { 0, 1, 1, 1, 0, -1, -1, -1 }
107 };
108
109 static const uint8_t chroma_vals[] = {
110 20, 28, 36, 44, 52, 60, 68, 76,
111 84, 92, 100, 106, 112, 116, 120, 124,
112 128, 132, 136, 140, 144, 150, 156, 164,
113 172, 180, 188, 196, 204, 212, 220, 228
114 };
115
116 2 static av_cold int escape130_decode_init(AVCodecContext *avctx)
117 {
118 2 Escape130Context *s = avctx->priv_data;
119 2 avctx->pix_fmt = AV_PIX_FMT_YUV420P;
120
121
2/4
✓ Branch 0 taken 2 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 2 times.
2 if ((avctx->width & 1) || (avctx->height & 1)) {
122 av_log(avctx, AV_LOG_ERROR,
123 "Dimensions should be a multiple of two.\n");
124 return AVERROR_INVALIDDATA;
125 }
126
127 2 s->old_y_avg = av_malloc(avctx->width * avctx->height / 4);
128 2 s->buf1 = av_malloc(avctx->width * avctx->height * 3 / 2);
129 2 s->buf2 = av_malloc(avctx->width * avctx->height * 3 / 2);
130
3/6
✓ Branch 0 taken 2 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
2 if (!s->old_y_avg || !s->buf1 || !s->buf2) {
131 av_log(avctx, AV_LOG_ERROR, "Could not allocate buffer.\n");
132 return AVERROR(ENOMEM);
133 }
134
135 2 s->linesize[0] = avctx->width;
136 2 s->linesize[1] =
137 2 s->linesize[2] = avctx->width / 2;
138
139 2 s->new_y = s->buf1;
140 2 s->new_u = s->new_y + avctx->width * avctx->height;
141 2 s->new_v = s->new_u + avctx->width * avctx->height / 4;
142 2 s->old_y = s->buf2;
143 2 s->old_u = s->old_y + avctx->width * avctx->height;
144 2 s->old_v = s->old_u + avctx->width * avctx->height / 4;
145 2 memset(s->old_y, 0, avctx->width * avctx->height);
146 2 memset(s->old_u, 0x10, avctx->width * avctx->height / 4);
147 2 memset(s->old_v, 0x10, avctx->width * avctx->height / 4);
148
149 2 return 0;
150 }
151
152 2 static av_cold int escape130_decode_close(AVCodecContext *avctx)
153 {
154 2 Escape130Context *s = avctx->priv_data;
155
156 2 av_freep(&s->old_y_avg);
157 2 av_freep(&s->buf1);
158 2 av_freep(&s->buf2);
159
160 2 return 0;
161 }
162
163 1642909 static int decode_skip_count(GetBitContext* gb)
164 {
165 int value;
166
167
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 1642909 times.
1642909 if (get_bits_left(gb) < 1+3)
168 return -1;
169
170 1642909 value = get_bits1(gb);
171
2/2
✓ Branch 0 taken 1274799 times.
✓ Branch 1 taken 368110 times.
1642909 if (value)
172 1274799 return 0;
173
174 368110 value = get_bits(gb, 3);
175
2/2
✓ Branch 0 taken 296591 times.
✓ Branch 1 taken 71519 times.
368110 if (value)
176 296591 return value;
177
178 71519 value = get_bits(gb, 8);
179
2/2
✓ Branch 0 taken 70932 times.
✓ Branch 1 taken 587 times.
71519 if (value)
180 70932 return value + 7;
181
182 587 value = get_bits(gb, 15);
183
1/2
✓ Branch 0 taken 587 times.
✗ Branch 1 not taken.
587 if (value)
184 587 return value + 262;
185
186 return -1;
187 }
188
189 359 static int escape130_decode_frame(AVCodecContext *avctx, void *data,
190 int *got_frame, AVPacket *avpkt)
191 {
192 359 int buf_size = avpkt->size;
193 359 Escape130Context *s = avctx->priv_data;
194 359 AVFrame *pic = data;
195 GetBitContext gb;
196 int ret;
197
198 uint8_t *old_y, *old_cb, *old_cr,
199 *new_y, *new_cb, *new_cr;
200 uint8_t *dstY, *dstU, *dstV;
201 unsigned old_y_stride, old_cb_stride, old_cr_stride,
202 new_y_stride, new_cb_stride, new_cr_stride;
203 359 unsigned total_blocks = avctx->width * avctx->height / 4,
204 359 block_index, block_x = 0;
205 359 unsigned y[4] = { 0 }, cb = 0x10, cr = 0x10;
206 359 int skip = -1, y_avg = 0, i, j;
207 359 uint8_t *ya = s->old_y_avg;
208
209 // first 16 bytes are header; no useful information in here
210
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 359 times.
359 if (buf_size <= 16) {
211 av_log(avctx, AV_LOG_ERROR, "Insufficient frame data\n");
212 return AVERROR_INVALIDDATA;
213 }
214
215
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 359 times.
359 if ((ret = ff_get_buffer(avctx, pic, 0)) < 0)
216 return ret;
217
218
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 359 times.
359 if ((ret = init_get_bits8(&gb, avpkt->data, avpkt->size)) < 0)
219 return ret;
220 359 skip_bits_long(&gb, 16 * 8);
221
222 359 new_y = s->new_y;
223 359 new_cb = s->new_u;
224 359 new_cr = s->new_v;
225 359 new_y_stride = s->linesize[0];
226 359 new_cb_stride = s->linesize[1];
227 359 new_cr_stride = s->linesize[2];
228 359 old_y = s->old_y;
229 359 old_cb = s->old_u;
230 359 old_cr = s->old_v;
231 359 old_y_stride = s->linesize[0];
232 359 old_cb_stride = s->linesize[1];
233 359 old_cr_stride = s->linesize[2];
234
235
2/2
✓ Branch 0 taken 4595200 times.
✓ Branch 1 taken 359 times.
4595559 for (block_index = 0; block_index < total_blocks; block_index++) {
236 // Note that this call will make us skip the rest of the blocks
237 // if the frame ends prematurely.
238
2/2
✓ Branch 0 taken 1642909 times.
✓ Branch 1 taken 2952291 times.
4595200 if (skip == -1)
239 1642909 skip = decode_skip_count(&gb);
240
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4595200 times.
4595200 if (skip == -1) {
241 av_log(avctx, AV_LOG_ERROR, "Error decoding skip value\n");
242 return AVERROR_INVALIDDATA;
243 }
244
245
2/2
✓ Branch 0 taken 2952495 times.
✓ Branch 1 taken 1642705 times.
4595200 if (skip) {
246 2952495 y[0] = old_y[0];
247 2952495 y[1] = old_y[1];
248 2952495 y[2] = old_y[old_y_stride];
249 2952495 y[3] = old_y[old_y_stride + 1];
250 2952495 y_avg = ya[0];
251 2952495 cb = old_cb[0];
252 2952495 cr = old_cr[0];
253 } else {
254
2/2
✓ Branch 1 taken 506697 times.
✓ Branch 2 taken 1136008 times.
1642705 if (get_bits1(&gb)) {
255 506697 unsigned sign_selector = get_bits(&gb, 6);
256 506697 unsigned difference_selector = get_bits(&gb, 2);
257 506697 y_avg = 2 * get_bits(&gb, 5);
258
2/2
✓ Branch 0 taken 2026788 times.
✓ Branch 1 taken 506697 times.
2533485 for (i = 0; i < 4; i++) {
259 2026788 y[i] = av_clip(y_avg + offset_table[difference_selector] *
260 2026788 sign_table[sign_selector][i], 0, 63);
261 }
262
2/2
✓ Branch 1 taken 686938 times.
✓ Branch 2 taken 449070 times.
1136008 } else if (get_bits1(&gb)) {
263
2/2
✓ Branch 1 taken 35948 times.
✓ Branch 2 taken 650990 times.
686938 if (get_bits1(&gb)) {
264 35948 y_avg = get_bits(&gb, 6);
265 } else {
266 650990 unsigned adjust_index = get_bits(&gb, 3);
267 650990 y_avg = (y_avg + luma_adjust[adjust_index]) & 63;
268 }
269
2/2
✓ Branch 0 taken 2747752 times.
✓ Branch 1 taken 686938 times.
3434690 for (i = 0; i < 4; i++)
270 2747752 y[i] = y_avg;
271 }
272
273
2/2
✓ Branch 1 taken 340030 times.
✓ Branch 2 taken 1302675 times.
1642705 if (get_bits1(&gb)) {
274
2/2
✓ Branch 1 taken 9871 times.
✓ Branch 2 taken 330159 times.
340030 if (get_bits1(&gb)) {
275 9871 cb = get_bits(&gb, 5);
276 9871 cr = get_bits(&gb, 5);
277 } else {
278 330159 unsigned adjust_index = get_bits(&gb, 3);
279 330159 cb = (cb + chroma_adjust[0][adjust_index]) & 31;
280 330159 cr = (cr + chroma_adjust[1][adjust_index]) & 31;
281 }
282 }
283 }
284 4595200 *ya++ = y_avg;
285
286 4595200 new_y[0] = y[0];
287 4595200 new_y[1] = y[1];
288 4595200 new_y[new_y_stride] = y[2];
289 4595200 new_y[new_y_stride + 1] = y[3];
290 4595200 *new_cb = cb;
291 4595200 *new_cr = cr;
292
293 4595200 old_y += 2;
294 4595200 old_cb++;
295 4595200 old_cr++;
296 4595200 new_y += 2;
297 4595200 new_cb++;
298 4595200 new_cr++;
299 4595200 block_x++;
300
2/2
✓ Branch 0 taken 28720 times.
✓ Branch 1 taken 4566480 times.
4595200 if (block_x * 2 == avctx->width) {
301 28720 block_x = 0;
302 28720 old_y += old_y_stride * 2 - avctx->width;
303 28720 old_cb += old_cb_stride - avctx->width / 2;
304 28720 old_cr += old_cr_stride - avctx->width / 2;
305 28720 new_y += new_y_stride * 2 - avctx->width;
306 28720 new_cb += new_cb_stride - avctx->width / 2;
307 28720 new_cr += new_cr_stride - avctx->width / 2;
308 }
309
310 4595200 skip--;
311 }
312
313 359 new_y = s->new_y;
314 359 new_cb = s->new_u;
315 359 new_cr = s->new_v;
316 359 dstY = pic->data[0];
317 359 dstU = pic->data[1];
318 359 dstV = pic->data[2];
319
2/2
✓ Branch 0 taken 57440 times.
✓ Branch 1 taken 359 times.
57799 for (j = 0; j < avctx->height; j++) {
320
2/2
✓ Branch 0 taken 18380800 times.
✓ Branch 1 taken 57440 times.
18438240 for (i = 0; i < avctx->width; i++)
321 18380800 dstY[i] = new_y[i] << 2;
322 57440 dstY += pic->linesize[0];
323 57440 new_y += new_y_stride;
324 }
325
2/2
✓ Branch 0 taken 28720 times.
✓ Branch 1 taken 359 times.
29079 for (j = 0; j < avctx->height / 2; j++) {
326
2/2
✓ Branch 0 taken 4595200 times.
✓ Branch 1 taken 28720 times.
4623920 for (i = 0; i < avctx->width / 2; i++) {
327 4595200 dstU[i] = chroma_vals[new_cb[i]];
328 4595200 dstV[i] = chroma_vals[new_cr[i]];
329 }
330 28720 dstU += pic->linesize[1];
331 28720 dstV += pic->linesize[2];
332 28720 new_cb += new_cb_stride;
333 28720 new_cr += new_cr_stride;
334 }
335
336 ff_dlog(avctx, "Frame data: provided %d bytes, used %d bytes\n",
337 buf_size, get_bits_count(&gb) >> 3);
338
339 359 FFSWAP(uint8_t*, s->old_y, s->new_y);
340 359 FFSWAP(uint8_t*, s->old_u, s->new_u);
341 359 FFSWAP(uint8_t*, s->old_v, s->new_v);
342
343 359 *got_frame = 1;
344
345 359 return buf_size;
346 }
347
348 const AVCodec ff_escape130_decoder = {
349 .name = "escape130",
350 .long_name = NULL_IF_CONFIG_SMALL("Escape 130"),
351 .type = AVMEDIA_TYPE_VIDEO,
352 .id = AV_CODEC_ID_ESCAPE130,
353 .priv_data_size = sizeof(Escape130Context),
354 .init = escape130_decode_init,
355 .close = escape130_decode_close,
356 .decode = escape130_decode_frame,
357 .capabilities = AV_CODEC_CAP_DR1,
358 .caps_internal = FF_CODEC_CAP_INIT_THREADSAFE | FF_CODEC_CAP_INIT_CLEANUP,
359 };
360