FFmpeg coverage


Directory: ../../../ffmpeg/
File: src/libavcodec/qpeg.c
Date: 2022-07-07 01:21:54
Exec Total Coverage
Lines: 163 189 86.2%
Branches: 92 120 76.7%

Line Branch Exec Source
1 /*
2 * QPEG codec
3 * Copyright (c) 2004 Konstantin Shishkov
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 /**
23 * @file
24 * QPEG codec.
25 */
26
27 #include "avcodec.h"
28 #include "bytestream.h"
29 #include "codec_internal.h"
30 #include "decode.h"
31 #include "internal.h"
32
33 typedef struct QpegContext{
34 AVCodecContext *avctx;
35 AVFrame *ref;
36 uint32_t pal[256];
37 GetByteContext buffer;
38 } QpegContext;
39
40 1 static void qpeg_decode_intra(QpegContext *qctx, uint8_t *dst,
41 int stride, int width, int height)
42 {
43 int i;
44 int code;
45 int c0, c1;
46 int run, copy;
47 1 int filled = 0;
48 int rows_to_go;
49
50 1 rows_to_go = height;
51 1 height--;
52 1 dst = dst + height * stride;
53
54
3/4
✓ Branch 1 taken 10407 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 10406 times.
✓ Branch 4 taken 1 times.
10407 while ((bytestream2_get_bytes_left(&qctx->buffer) > 0) && (rows_to_go > 0)) {
55 10406 code = bytestream2_get_byte(&qctx->buffer);
56 10406 run = copy = 0;
57
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 10406 times.
10406 if(code == 0xFC) /* end-of-picture code */
58 break;
59
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 10406 times.
10406 if(code >= 0xF8) { /* very long run */
60 c0 = bytestream2_get_byte(&qctx->buffer);
61 c1 = bytestream2_get_byte(&qctx->buffer);
62 run = ((code & 0x7) << 16) + (c0 << 8) + c1 + 2;
63
2/2
✓ Branch 0 taken 21 times.
✓ Branch 1 taken 10385 times.
10406 } else if (code >= 0xF0) { /* long run */
64 21 c0 = bytestream2_get_byte(&qctx->buffer);
65 21 run = ((code & 0xF) << 8) + c0 + 2;
66
2/2
✓ Branch 0 taken 5687 times.
✓ Branch 1 taken 4698 times.
10385 } else if (code >= 0xE0) { /* short run */
67 5687 run = (code & 0x1F) + 2;
68
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4698 times.
4698 } else if (code >= 0xC0) { /* very long copy */
69 c0 = bytestream2_get_byte(&qctx->buffer);
70 c1 = bytestream2_get_byte(&qctx->buffer);
71 copy = ((code & 0x3F) << 16) + (c0 << 8) + c1 + 1;
72
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4698 times.
4698 } else if (code >= 0x80) { /* long copy */
73 c0 = bytestream2_get_byte(&qctx->buffer);
74 copy = ((code & 0x7F) << 8) + c0 + 1;
75 } else { /* short copy */
76 4698 copy = code + 1;
77 }
78
79 /* perform actual run or copy */
80
2/2
✓ Branch 0 taken 5708 times.
✓ Branch 1 taken 4698 times.
10406 if(run) {
81 int p;
82
83 5708 p = bytestream2_get_byte(&qctx->buffer);
84
2/2
✓ Branch 0 taken 5708 times.
✓ Branch 1 taken 5708 times.
11416 for(i = 0; i < run; i++) {
85 5708 int step = FFMIN(run - i, width - filled);
86 5708 memset(dst+filled, p, step);
87 5708 filled += step;
88 5708 i += step - 1;
89
2/2
✓ Branch 0 taken 30 times.
✓ Branch 1 taken 5678 times.
5708 if (filled >= width) {
90 30 filled = 0;
91 30 dst -= stride;
92 30 rows_to_go--;
93
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 30 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
30 while (run - i > width && rows_to_go > 0) {
94 memset(dst, p, width);
95 dst -= stride;
96 rows_to_go--;
97 i += width;
98 }
99
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 30 times.
30 if(rows_to_go <= 0)
100 break;
101 }
102 }
103 } else {
104
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 4698 times.
4698 if (bytestream2_get_bytes_left(&qctx->buffer) < copy)
105 copy = bytestream2_get_bytes_left(&qctx->buffer);
106
2/2
✓ Branch 0 taken 4891 times.
✓ Branch 1 taken 4697 times.
9588 while (copy > 0) {
107 4891 int step = FFMIN(copy, width - filled);
108 4891 bytestream2_get_bufferu(&qctx->buffer, dst + filled, step);
109 4891 filled += step;
110 4891 copy -= step;
111
2/2
✓ Branch 0 taken 210 times.
✓ Branch 1 taken 4681 times.
4891 if (filled >= width) {
112 210 filled = 0;
113 210 dst -= stride;
114 210 rows_to_go--;
115
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 209 times.
210 if(rows_to_go <= 0)
116 1 break;
117 }
118 }
119 }
120 }
121 1 }
122
123 static const uint8_t qpeg_table_h[16] =
124 { 0x00, 0x20, 0x20, 0x20, 0x18, 0x10, 0x10, 0x20, 0x10, 0x08, 0x18, 0x08, 0x08, 0x18, 0x10, 0x04};
125 static const uint8_t qpeg_table_w[16] =
126 { 0x00, 0x20, 0x18, 0x08, 0x18, 0x10, 0x20, 0x10, 0x08, 0x10, 0x20, 0x20, 0x08, 0x10, 0x18, 0x04};
127
128 /* Decodes delta frames */
129 99 static void av_noinline qpeg_decode_inter(QpegContext *qctx, uint8_t *dst,
130 int stride, int width, int height,
131 int delta, const uint8_t *ctable,
132 uint8_t *refdata)
133 {
134 int i, j;
135 int code;
136 99 int filled = 0;
137 int orig_height;
138
139
1/2
✓ Branch 0 taken 99 times.
✗ Branch 1 not taken.
99 if (refdata) {
140 /* copy prev frame */
141
2/2
✓ Branch 0 taken 23760 times.
✓ Branch 1 taken 99 times.
23859 for (i = 0; i < height; i++)
142 23760 memcpy(dst + (i * stride), refdata + (i * stride), width);
143 } else {
144 refdata = dst;
145 }
146
147 99 orig_height = height;
148 99 height--;
149 99 dst = dst + height * stride;
150
151
3/4
✓ Branch 1 taken 731786 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 731779 times.
✓ Branch 4 taken 7 times.
731786 while ((bytestream2_get_bytes_left(&qctx->buffer) > 0) && (height >= 0)) {
152 731779 code = bytestream2_get_byte(&qctx->buffer);
153
154
1/2
✓ Branch 0 taken 731779 times.
✗ Branch 1 not taken.
731779 if(delta) {
155 /* motion compensation */
156
4/4
✓ Branch 1 taken 751094 times.
✓ Branch 2 taken 92 times.
✓ Branch 3 taken 19407 times.
✓ Branch 4 taken 731687 times.
751186 while(bytestream2_get_bytes_left(&qctx->buffer) > 0 && (code & 0xF0) == 0xF0) {
157
1/2
✓ Branch 0 taken 19407 times.
✗ Branch 1 not taken.
19407 if(delta == 1) {
158 int me_idx;
159 int me_w, me_h, me_x, me_y;
160 uint8_t *me_plane;
161 int corr, val;
162
163 /* get block size by index */
164 19407 me_idx = code & 0xF;
165 19407 me_w = qpeg_table_w[me_idx];
166 19407 me_h = qpeg_table_h[me_idx];
167
168 /* extract motion vector */
169 19407 corr = bytestream2_get_byte(&qctx->buffer);
170
171 19407 val = corr >> 4;
172
2/2
✓ Branch 0 taken 8637 times.
✓ Branch 1 taken 10770 times.
19407 if(val > 7)
173 8637 val -= 16;
174 19407 me_x = val;
175
176 19407 val = corr & 0xF;
177
2/2
✓ Branch 0 taken 9623 times.
✓ Branch 1 taken 9784 times.
19407 if(val > 7)
178 9623 val -= 16;
179 19407 me_y = val;
180
181 /* check motion vector */
182
2/4
✓ Branch 0 taken 19407 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 19407 times.
✗ Branch 3 not taken.
19407 if ((me_x + filled < 0) || (me_x + me_w + filled > width) ||
183
2/4
✓ Branch 0 taken 19407 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 19407 times.
✗ Branch 3 not taken.
19407 (height - me_y - me_h < 0) || (height - me_y >= orig_height) ||
184
2/4
✓ Branch 0 taken 19407 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 19407 times.
19407 (filled + me_w > width) || (height - me_h < 0))
185 av_log(qctx->avctx, AV_LOG_ERROR, "Bogus motion vector (%i,%i), block size %ix%i at %i,%i\n",
186 me_x, me_y, me_w, me_h, filled, height);
187 else {
188 /* do motion compensation */
189 19407 me_plane = refdata + (filled + me_x) + (height - me_y) * stride;
190
2/2
✓ Branch 0 taken 158312 times.
✓ Branch 1 taken 19407 times.
177719 for(j = 0; j < me_h; j++) {
191
2/2
✓ Branch 0 taken 1296640 times.
✓ Branch 1 taken 158312 times.
1454952 for(i = 0; i < me_w; i++)
192 1296640 dst[filled + i - (j * stride)] = me_plane[i - (j * stride)];
193 }
194 }
195 }
196 19407 code = bytestream2_get_byte(&qctx->buffer);
197 }
198 }
199
200
2/2
✓ Branch 0 taken 92 times.
✓ Branch 1 taken 731687 times.
731779 if(code == 0xE0) /* end-of-picture code */
201 92 break;
202
2/2
✓ Branch 0 taken 96361 times.
✓ Branch 1 taken 635326 times.
731687 if(code > 0xE0) { /* run code: 0xE1..0xFF */
203 int p;
204
205 96361 code &= 0x1F;
206 96361 p = bytestream2_get_byte(&qctx->buffer);
207
2/2
✓ Branch 0 taken 263167 times.
✓ Branch 1 taken 96361 times.
359528 for(i = 0; i <= code; i++) {
208 263167 dst[filled++] = p;
209
2/2
✓ Branch 0 taken 72 times.
✓ Branch 1 taken 263095 times.
263167 if(filled >= width) {
210 72 filled = 0;
211 72 dst -= stride;
212 72 height--;
213
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 72 times.
72 if (height < 0)
214 break;
215 }
216 }
217
2/2
✓ Branch 0 taken 214471 times.
✓ Branch 1 taken 420855 times.
635326 } else if(code >= 0xC0) { /* copy code: 0xC0..0xDF */
218 214471 code &= 0x1F;
219
220
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 214471 times.
214471 if(code + 1 > bytestream2_get_bytes_left(&qctx->buffer))
221 break;
222
223
2/2
✓ Branch 0 taken 743443 times.
✓ Branch 1 taken 214471 times.
957914 for(i = 0; i <= code; i++) {
224 743443 dst[filled++] = bytestream2_get_byte(&qctx->buffer);
225
2/2
✓ Branch 0 taken 16 times.
✓ Branch 1 taken 743427 times.
743443 if(filled >= width) {
226 16 filled = 0;
227 16 dst -= stride;
228 16 height--;
229
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 16 times.
16 if (height < 0)
230 break;
231 }
232 }
233
2/2
✓ Branch 0 taken 249130 times.
✓ Branch 1 taken 171725 times.
420855 } else if(code >= 0x80) { /* skip code: 0x80..0xBF */
234 int skip;
235
236 249130 code &= 0x3F;
237 /* codes 0x80 and 0x81 are actually escape codes,
238 skip value minus constant is in the next byte */
239
2/2
✓ Branch 0 taken 29191 times.
✓ Branch 1 taken 219939 times.
249130 if(!code)
240 29191 skip = bytestream2_get_byte(&qctx->buffer) + 64;
241
2/2
✓ Branch 0 taken 273 times.
✓ Branch 1 taken 219666 times.
219939 else if(code == 1)
242 273 skip = bytestream2_get_byte(&qctx->buffer) + 320;
243 else
244 219666 skip = code;
245 249130 filled += skip;
246
2/2
✓ Branch 0 taken 23398 times.
✓ Branch 1 taken 249130 times.
272528 while( filled >= width) {
247 23398 filled -= width;
248 23398 dst -= stride;
249 23398 height--;
250
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 23398 times.
23398 if(height < 0)
251 break;
252 }
253 } else {
254 /* zero code treated as one-pixel skip */
255
2/2
✓ Branch 0 taken 124870 times.
✓ Branch 1 taken 46855 times.
171725 if(code) {
256 124870 dst[filled++] = ctable[code & 0x7F];
257 }
258 else
259 46855 filled++;
260
2/2
✓ Branch 0 taken 182 times.
✓ Branch 1 taken 171543 times.
171725 if(filled >= width) {
261 182 filled = 0;
262 182 dst -= stride;
263 182 height--;
264 }
265 }
266 }
267 99 }
268
269 100 static int decode_frame(AVCodecContext *avctx, AVFrame *p,
270 int *got_frame, AVPacket *avpkt)
271 {
272 uint8_t ctable[128];
273 100 QpegContext * const a = avctx->priv_data;
274 100 AVFrame * const ref = a->ref;
275 uint8_t* outdata;
276 int delta, intra, ret;
277
278
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 100 times.
100 if (avpkt->size < 0x86) {
279 av_log(avctx, AV_LOG_ERROR, "Packet is too small\n");
280 return AVERROR_INVALIDDATA;
281 }
282
283 100 bytestream2_init(&a->buffer, avpkt->data, avpkt->size);
284
285
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 100 times.
100 if ((ret = ff_get_buffer(avctx, p, AV_GET_BUFFER_FLAG_REF)) < 0)
286 return ret;
287 100 outdata = p->data[0];
288 100 bytestream2_skip(&a->buffer, 4);
289 100 bytestream2_get_buffer(&a->buffer, ctable, 128);
290 100 bytestream2_skip(&a->buffer, 1);
291
292 100 delta = bytestream2_get_byte(&a->buffer);
293 100 intra = delta == 0x10;
294
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 99 times.
100 if (intra) {
295 1 qpeg_decode_intra(a, outdata, p->linesize[0], avctx->width, avctx->height);
296 } else {
297 99 qpeg_decode_inter(a, outdata, p->linesize[0], avctx->width, avctx->height, delta, ctable, ref->data[0]);
298 }
299
300 /* make the palette available on the way out */
301 100 p->palette_has_changed = ff_copy_palette(a->pal, avpkt, avctx);
302 100 memcpy(p->data[1], a->pal, AVPALETTE_SIZE);
303
304 100 av_frame_unref(ref);
305
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 100 times.
100 if ((ret = av_frame_ref(ref, p)) < 0)
306 return ret;
307
308 100 p->key_frame = intra;
309
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 99 times.
100 p->pict_type = intra ? AV_PICTURE_TYPE_I : AV_PICTURE_TYPE_P;
310
311 100 *got_frame = 1;
312
313 100 return avpkt->size;
314 }
315
316 2 static void decode_flush(AVCodecContext *avctx){
317 2 QpegContext * const a = avctx->priv_data;
318 int i, pal_size;
319 const uint8_t *pal_src;
320
321 2 av_frame_unref(a->ref);
322
323 2 pal_size = FFMIN(1024U, avctx->extradata_size);
324 2 pal_src = avctx->extradata + avctx->extradata_size - pal_size;
325
326
2/2
✓ Branch 0 taken 512 times.
✓ Branch 1 taken 2 times.
514 for (i=0; i<pal_size/4; i++)
327 512 a->pal[i] = 0xFFU<<24 | AV_RL32(pal_src+4*i);
328 2 }
329
330 2 static av_cold int decode_end(AVCodecContext *avctx)
331 {
332 2 QpegContext * const a = avctx->priv_data;
333
334 2 av_frame_free(&a->ref);
335
336 2 return 0;
337 }
338
339 2 static av_cold int decode_init(AVCodecContext *avctx){
340 2 QpegContext * const a = avctx->priv_data;
341
342 2 a->avctx = avctx;
343 2 avctx->pix_fmt= AV_PIX_FMT_PAL8;
344
345 2 a->ref = av_frame_alloc();
346
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
2 if (!a->ref)
347 return AVERROR(ENOMEM);
348
349 2 decode_flush(avctx);
350
351 2 return 0;
352 }
353
354 const FFCodec ff_qpeg_decoder = {
355 .p.name = "qpeg",
356 .p.long_name = NULL_IF_CONFIG_SMALL("Q-team QPEG"),
357 .p.type = AVMEDIA_TYPE_VIDEO,
358 .p.id = AV_CODEC_ID_QPEG,
359 .priv_data_size = sizeof(QpegContext),
360 .init = decode_init,
361 .close = decode_end,
362 FF_CODEC_DECODE_CB(decode_frame),
363 .flush = decode_flush,
364 .p.capabilities = AV_CODEC_CAP_DR1,
365 .caps_internal = FF_CODEC_CAP_INIT_THREADSAFE |
366 FF_CODEC_CAP_INIT_CLEANUP,
367 };
368