FFmpeg coverage

Directory: ../../../ffmpeg/
File: src/libavcodec/svq1dec.c
Date: 2024-11-20 23:03:26
Exec Total Coverage
Lines: 297 356 83.4%
Functions: 12 14 85.7%
Branches: 171 229 74.7%
Line Branch Exec Source
1 /*
2 * SVQ1 decoder
3 * ported to MPlayer by Arpi <arpi@thot.banki.hu>
4 * ported to libavcodec by Nick Kurshev <nickols_k@mail.ru>
5 *
6 * Copyright (c) 2002 The Xine project
7 * Copyright (c) 2002 The FFmpeg project
8 *
9 * SVQ1 Encoder (c) 2004 Mike Melanson <melanson@pcisys.net>
10 *
11 * This file is part of FFmpeg.
12 *
13 * FFmpeg is free software; you can redistribute it and/or
14 * modify it under the terms of the GNU Lesser General Public
15 * License as published by the Free Software Foundation; either
16 * version 2.1 of the License, or (at your option) any later version.
17 *
18 * FFmpeg is distributed in the hope that it will be useful,
19 * but WITHOUT ANY WARRANTY; without even the implied warranty of
20 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
21 * Lesser General Public License for more details.
22 *
23 * You should have received a copy of the GNU Lesser General Public
24 * License along with FFmpeg; if not, write to the Free Software
25 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
26 */
27
28 /**
29 * @file
30 * Sorenson Vector Quantizer #1 (SVQ1) video codec.
31 * For more information of the SVQ1 algorithm, visit:
32 * http://www.pcisys.net/~melanson/codecs/
33 */
34
35 #include "libavutil/crc.h"
36 #include "libavutil/mem.h"
37 #include "libavutil/thread.h"
38
39 #include "avcodec.h"
40 #include "codec_internal.h"
41 #include "decode.h"
42 #include "get_bits.h"
43 #include "h263data.h"
44 #include "hpeldsp.h"
45 #include "mathops.h"
46 #include "svq1.h"
47
48 #define SVQ1_BLOCK_TYPE_VLC_BITS 3
49 static VLCElem svq1_block_type[8];
50 static VLCElem svq1_motion_component[176];
51 static const VLCElem *svq1_intra_multistage[6];
52 static const VLCElem *svq1_inter_multistage[6];
53 static VLCElem svq1_intra_mean[632];
54 static VLCElem svq1_inter_mean[1434];
55
56 /* motion vector (prediction) */
57 typedef struct svq1_pmv_s {
58 int x;
59 int y;
60 } svq1_pmv;
61
62 typedef struct SVQ1Context {
63 HpelDSPContext hdsp;
64 GetBitContext gb;
65 AVFrame *prev;
66
67 uint8_t *pkt_swapped;
68 int pkt_swapped_allocated;
69
70 svq1_pmv *pmv;
71 int pmv_allocated;
72
73 int width;
74 int height;
75 int frame_code;
76 int nonref; // 1 if the current frame won't be referenced
77
78 int last_tempref;
79 } SVQ1Context;
80
81 static const uint8_t string_table[256] = {
82 0x00, 0xD5, 0x7F, 0xAA, 0xFE, 0x2B, 0x81, 0x54,
83 0x29, 0xFC, 0x56, 0x83, 0xD7, 0x02, 0xA8, 0x7D,
84 0x52, 0x87, 0x2D, 0xF8, 0xAC, 0x79, 0xD3, 0x06,
85 0x7B, 0xAE, 0x04, 0xD1, 0x85, 0x50, 0xFA, 0x2F,
86 0xA4, 0x71, 0xDB, 0x0E, 0x5A, 0x8F, 0x25, 0xF0,
87 0x8D, 0x58, 0xF2, 0x27, 0x73, 0xA6, 0x0C, 0xD9,
88 0xF6, 0x23, 0x89, 0x5C, 0x08, 0xDD, 0x77, 0xA2,
89 0xDF, 0x0A, 0xA0, 0x75, 0x21, 0xF4, 0x5E, 0x8B,
90 0x9D, 0x48, 0xE2, 0x37, 0x63, 0xB6, 0x1C, 0xC9,
91 0xB4, 0x61, 0xCB, 0x1E, 0x4A, 0x9F, 0x35, 0xE0,
92 0xCF, 0x1A, 0xB0, 0x65, 0x31, 0xE4, 0x4E, 0x9B,
93 0xE6, 0x33, 0x99, 0x4C, 0x18, 0xCD, 0x67, 0xB2,
94 0x39, 0xEC, 0x46, 0x93, 0xC7, 0x12, 0xB8, 0x6D,
95 0x10, 0xC5, 0x6F, 0xBA, 0xEE, 0x3B, 0x91, 0x44,
96 0x6B, 0xBE, 0x14, 0xC1, 0x95, 0x40, 0xEA, 0x3F,
97 0x42, 0x97, 0x3D, 0xE8, 0xBC, 0x69, 0xC3, 0x16,
98 0xEF, 0x3A, 0x90, 0x45, 0x11, 0xC4, 0x6E, 0xBB,
99 0xC6, 0x13, 0xB9, 0x6C, 0x38, 0xED, 0x47, 0x92,
100 0xBD, 0x68, 0xC2, 0x17, 0x43, 0x96, 0x3C, 0xE9,
101 0x94, 0x41, 0xEB, 0x3E, 0x6A, 0xBF, 0x15, 0xC0,
102 0x4B, 0x9E, 0x34, 0xE1, 0xB5, 0x60, 0xCA, 0x1F,
103 0x62, 0xB7, 0x1D, 0xC8, 0x9C, 0x49, 0xE3, 0x36,
104 0x19, 0xCC, 0x66, 0xB3, 0xE7, 0x32, 0x98, 0x4D,
105 0x30, 0xE5, 0x4F, 0x9A, 0xCE, 0x1B, 0xB1, 0x64,
106 0x72, 0xA7, 0x0D, 0xD8, 0x8C, 0x59, 0xF3, 0x26,
107 0x5B, 0x8E, 0x24, 0xF1, 0xA5, 0x70, 0xDA, 0x0F,
108 0x20, 0xF5, 0x5F, 0x8A, 0xDE, 0x0B, 0xA1, 0x74,
109 0x09, 0xDC, 0x76, 0xA3, 0xF7, 0x22, 0x88, 0x5D,
110 0xD6, 0x03, 0xA9, 0x7C, 0x28, 0xFD, 0x57, 0x82,
111 0xFF, 0x2A, 0x80, 0x55, 0x01, 0xD4, 0x7E, 0xAB,
112 0x84, 0x51, 0xFB, 0x2E, 0x7A, 0xAF, 0x05, 0xD0,
113 0xAD, 0x78, 0xD2, 0x07, 0x53, 0x86, 0x2C, 0xF9
114 };
115
116 #define SVQ1_PROCESS_VECTOR() \
117 for (; level > 0; i++) { \
118 /* process next depth */ \
119 if (i == m) { \
120 m = n; \
121 if (--level == 0) \
122 break; \
123 } \
124 /* divide block if next bit set */ \
125 if (!get_bits1(bitbuf)) \
126 break; \
127 /* add child nodes */ \
128 list[n++] = list[i]; \
129 list[n++] = list[i] + (((level & 1) ? pitch : 1) << ((level >> 1) + 1));\
130 }
131
132 #define SVQ1_ADD_CODEBOOK() \
133 /* add codebook entries to vector */ \
134 for (j = 0; j < stages; j++) { \
135 n3 = codebook[entries[j]] ^ 0x80808080; \
136 n1 += (n3 & 0xFF00FF00) >> 8; \
137 n2 += n3 & 0x00FF00FF; \
138 } \
139 \
140 /* clip to [0..255] */ \
141 if (n1 & 0xFF00FF00) { \
142 n3 = (n1 >> 15 & 0x00010001 | 0x01000100) - 0x00010001; \
143 n1 += 0x7F007F00; \
144 n1 |= (~n1 >> 15 & 0x00010001 | 0x01000100) - 0x00010001; \
145 n1 &= n3 & 0x00FF00FF; \
146 } \
147 \
148 if (n2 & 0xFF00FF00) { \
149 n3 = (n2 >> 15 & 0x00010001 | 0x01000100) - 0x00010001; \
150 n2 += 0x7F007F00; \
151 n2 |= (~n2 >> 15 & 0x00010001 | 0x01000100) - 0x00010001; \
152 n2 &= n3 & 0x00FF00FF; \
153 }
154
155 #define SVQ1_CALC_CODEBOOK_ENTRIES(cbook) \
156 codebook = (const uint32_t *)cbook[level]; \
157 if (stages > 0) \
158 bit_cache = get_bits(bitbuf, 4 * stages); \
159 /* calculate codebook entries for this vector */ \
160 for (j = 0; j < stages; j++) { \
161 entries[j] = (((bit_cache >> (4 * (stages - j - 1))) & 0xF) + \
162 16 * j) << (level + 1); \
163 } \
164 mean -= stages * 128; \
165 n4 = (mean << 16) + mean;
166
167 12349 static int svq1_decode_block_intra(GetBitContext *bitbuf, uint8_t *pixels,
168 ptrdiff_t pitch)
169 {
170 uint32_t bit_cache;
171 uint8_t *list[63];
172 uint32_t *dst;
173 const uint32_t *codebook;
174 int entries[6];
175 int i, j, m, n;
176 int stages;
177 unsigned mean;
178 unsigned x, y, width, height, level;
179 uint32_t n1, n2, n3, n4;
180
181 /* initialize list for breadth first processing of vectors */
182 12349 list[0] = pixels;
183
184 /* recursively process vector */
185
2/2
✓ Branch 0 taken 243731 times.
✓ Branch 1 taken 12349 times.
 256080 for (i = 0, m = 1, n = 1, level = 5; i < n; i++) {
186
10/10
✓ Branch 0 taken 58388 times.
✓ Branch 1 taken 261290 times.
✓ Branch 2 taken 11197 times.
✓ Branch 3 taken 47191 times.
✓ Branch 5 taken 77099 times.
✓ Branch 6 taken 231382 times.
✓ Branch 7 taken 139085 times.
✓ Branch 8 taken 92297 times.
✓ Branch 9 taken 319678 times.
✓ Branch 10 taken 155435 times.
 475113 SVQ1_PROCESS_VECTOR();
187
188 /* destination address and vector size */
189 243731 dst = (uint32_t *)list[i];
190 243731 width = 1 << ((4 + level) / 2);
191 243731 height = 1 << ((3 + level) / 2);
192
193 /* get number of stages (-1 skips vector, 0 for mean only) */
194 243731 stages = get_vlc2(bitbuf, svq1_intra_multistage[level], 4, 2) - 1;
195
196
2/2
✓ Branch 0 taken 371 times.
✓ Branch 1 taken 243360 times.
 243731 if (stages == -1) {
197
2/2
✓ Branch 0 taken 4122 times.
✓ Branch 1 taken 371 times.
 4493 for (y = 0; y < height; y++)
198 4122 memset(&dst[y * (pitch / 4)], 0, width);
199 371 continue; /* skip vector */
200 }
201
202
3/4
✓ Branch 0 taken 183609 times.
✓ Branch 1 taken 59751 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 183609 times.
 243360 if ((stages > 0 && level >= 4)) {
203 ff_dlog(NULL,
204 "Error (svq1_decode_block_intra): invalid vector: stages=%i level=%i\n",
205 stages, level);
206 ✗ return AVERROR_INVALIDDATA; /* invalid vector */
207 }
208
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 243360 times.
 243360 av_assert0(stages >= 0);
209
210 243360 mean = get_vlc2(bitbuf, svq1_intra_mean, 8, 3);
211
212
2/2
✓ Branch 0 taken 59751 times.
✓ Branch 1 taken 183609 times.
 243360 if (stages == 0) {
213
2/2
✓ Branch 0 taken 158440 times.
✓ Branch 1 taken 59751 times.
 218191 for (y = 0; y < height; y++)
214 158440 memset(&dst[y * (pitch / 4)], mean, width);
215 } else {
216
3/4
✓ Branch 0 taken 183609 times.
✗ Branch 1 not taken.
✓ Branch 3 taken 949129 times.
✓ Branch 4 taken 183609 times.
 1132738 SVQ1_CALC_CODEBOOK_ENTRIES(ff_svq1_intra_codebooks);
217
218
2/2
✓ Branch 0 taken 498458 times.
✓ Branch 1 taken 183609 times.
 682067 for (y = 0; y < height; y++) {
219
2/2
✓ Branch 0 taken 580850 times.
✓ Branch 1 taken 498458 times.
 1079308 for (x = 0; x < width / 4; x++, codebook++) {
220 580850 n1 = n4;
221 580850 n2 = n4;
222
6/6
✓ Branch 0 taken 2927974 times.
✓ Branch 1 taken 580850 times.
✓ Branch 2 taken 221 times.
✓ Branch 3 taken 580629 times.
✓ Branch 4 taken 350 times.
✓ Branch 5 taken 580500 times.
 3508824 SVQ1_ADD_CODEBOOK()
223 /* store result */
224 580850 dst[x] = n1 << 8 | n2;
225 }
226 498458 dst += pitch / 4;
227 }
228 }
229 }
230
231 12349 return 0;
232 }
233
234 65833 static int svq1_decode_block_non_intra(GetBitContext *bitbuf, uint8_t *pixels,
235 ptrdiff_t pitch, int buggy)
236 {
237 uint32_t bit_cache;
238 uint8_t *list[63];
239 uint32_t *dst;
240 const uint32_t *codebook;
241 int entries[6];
242 int i, j, m, n;
243 int stages;
244 unsigned mean;
245 int x, y, width, height, level;
246 uint32_t n1, n2, n3, n4;
247
248 /* initialize list for breadth first processing of vectors */
249 65833 list[0] = pixels;
250
251 /* recursively process vector */
252
2/2
✓ Branch 0 taken 1235087 times.
✓ Branch 1 taken 65833 times.
 1300920 for (i = 0, m = 1, n = 1, level = 5; i < n; i++) {
253
10/10
✓ Branch 0 taken 311449 times.
✓ Branch 1 taken 1260126 times.
✓ Branch 2 taken 60606 times.
✓ Branch 3 taken 250843 times.
✓ Branch 5 taken 341715 times.
✓ Branch 6 taken 1169254 times.
✓ Branch 7 taken 714294 times.
✓ Branch 8 taken 454960 times.
✓ Branch 9 taken 1571575 times.
✓ Branch 10 taken 832766 times.
 2404341 SVQ1_PROCESS_VECTOR();
254
255 /* destination address and vector size */
256 1235087 dst = (uint32_t *)list[i];
257 1235087 width = 1 << ((4 + level) / 2);
258 1235087 height = 1 << ((3 + level) / 2);
259
260 /* get number of stages (-1 skips vector, 0 for mean only) */
261 1235087 stages = get_vlc2(bitbuf, svq1_inter_multistage[level], 3, 2) - 1;
262
263
2/2
✓ Branch 0 taken 30272 times.
✓ Branch 1 taken 1204815 times.
 1235087 if (stages == -1)
264 30272 continue; /* skip vector */
265
266
3/4
✓ Branch 0 taken 466794 times.
✓ Branch 1 taken 738021 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 466794 times.
 1204815 if ((stages > 0 && level >= 4)) {
267 ff_dlog(NULL,
268 "Error (svq1_decode_block_non_intra): invalid vector: stages=%i level=%i\n",
269 stages, level);
270 ✗ return AVERROR_INVALIDDATA; /* invalid vector */
271 }
272
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1204815 times.
 1204815 av_assert0(stages >= 0);
273
274 1204815 mean = get_vlc2(bitbuf, svq1_inter_mean, 9, 3) - 256;
275
276
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1204815 times.
 1204815 if (buggy) {
277 ✗ if (mean == -128)
278 ✗ mean = 128;
279 ✗ else if (mean == 128)
280 ✗ mean = -128;
281 }
282
283
4/4
✓ Branch 0 taken 466794 times.
✓ Branch 1 taken 738021 times.
✓ Branch 3 taken 2169968 times.
✓ Branch 4 taken 1204815 times.
 3374783 SVQ1_CALC_CODEBOOK_ENTRIES(ff_svq1_inter_codebooks);
284
285
2/2
✓ Branch 0 taken 3231412 times.
✓ Branch 1 taken 1204815 times.
 4436227 for (y = 0; y < height; y++) {
286
2/2
✓ Branch 0 taken 4014664 times.
✓ Branch 1 taken 3231412 times.
 7246076 for (x = 0; x < width / 4; x++) {
287 4014664 n3 = dst[x];
288 /* add mean value to vector */
289 4014664 n1 = n4 + ((n3 & 0xFF00FF00) >> 8);
290 4014664 n2 = n4 + (n3 & 0x00FF00FF);
291
6/6
✓ Branch 0 taken 5364768 times.
✓ Branch 1 taken 4014664 times.
✓ Branch 2 taken 8297 times.
✓ Branch 3 taken 4006367 times.
✓ Branch 4 taken 8590 times.
✓ Branch 5 taken 4006074 times.
 9379432 SVQ1_ADD_CODEBOOK()
292 /* store result */
293 4014664 dst[x] = n1 << 8 | n2;
294
2/2
✓ Branch 0 taken 3643944 times.
✓ Branch 1 taken 370720 times.
 4014664 if (codebook != NULL)
295 3643944 codebook++;
296 }
297 3231412 dst += pitch / 4;
298 }
299 }
300 65833 return 0;
301 }
302
303 79966 static int svq1_decode_motion_vector(GetBitContext *bitbuf, svq1_pmv *mv,
304 svq1_pmv **pmv)
305 {
306 int diff;
307 int i;
308
309
2/2
✓ Branch 0 taken 159932 times.
✓ Branch 1 taken 79966 times.
 239898 for (i = 0; i < 2; i++) {
310 /* get motion code */
311 159932 diff = get_vlc2(bitbuf, svq1_motion_component, 7, 2);
312
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 159932 times.
 159932 if (diff < 0)
313 ✗ return AVERROR_INVALIDDATA;
314
2/2
✓ Branch 0 taken 79270 times.
✓ Branch 1 taken 80662 times.
 159932 else if (diff) {
315
2/2
✓ Branch 1 taken 34424 times.
✓ Branch 2 taken 44846 times.
 79270 if (get_bits1(bitbuf))
316 34424 diff = -diff;
317 }
318
319 /* add median of motion vector predictors and clip result */
320
2/2
✓ Branch 0 taken 79966 times.
✓ Branch 1 taken 79966 times.
 159932 if (i == 1)
321 79966 mv->y = sign_extend(diff + mid_pred(pmv[0]->y, pmv[1]->y, pmv[2]->y), 6);
322 else
323 79966 mv->x = sign_extend(diff + mid_pred(pmv[0]->x, pmv[1]->x, pmv[2]->x), 6);
324 }
325
326 79966 return 0;
327 }
328
329 8382 static void svq1_skip_block(uint8_t *current, uint8_t *previous,
330 ptrdiff_t pitch, int x, int y)
331 {
332 uint8_t *src;
333 uint8_t *dst;
334 int i;
335
336 8382 src = &previous[x + y * pitch];
337 8382 dst = current;
338
339
2/2
✓ Branch 0 taken 134112 times.
✓ Branch 1 taken 8382 times.
 142494 for (i = 0; i < 16; i++) {
340 134112 memcpy(dst, src, 16);
341 134112 src += pitch;
342 134112 dst += pitch;
343 }
344 8382 }
345
346 61122 static int svq1_motion_inter_block(HpelDSPContext *hdsp, GetBitContext *bitbuf,
347 uint8_t *current, uint8_t *previous,
348 ptrdiff_t pitch, svq1_pmv *motion, int x, int y,
349 int width, int height)
350 {
351 uint8_t *src;
352 uint8_t *dst;
353 svq1_pmv mv;
354 svq1_pmv *pmv[3];
355 int result;
356
357 /* predict and decode motion vector */
358 61122 pmv[0] = &motion[0];
359
2/2
✓ Branch 0 taken 5031 times.
✓ Branch 1 taken 56091 times.
 61122 if (y == 0) {
360 5031 pmv[1] =
361 5031 pmv[2] = pmv[0];
362 } else {
363 56091 pmv[1] = &motion[x / 8 + 2];
364 56091 pmv[2] = &motion[x / 8 + 4];
365 }
366
367 61122 result = svq1_decode_motion_vector(bitbuf, &mv, pmv);
368
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 61122 times.
 61122 if (result)
369 ✗ return result;
370
371 61122 motion[0].x =
372 61122 motion[x / 8 + 2].x =
373 61122 motion[x / 8 + 3].x = mv.x;
374 61122 motion[0].y =
375 61122 motion[x / 8 + 2].y =
376 61122 motion[x / 8 + 3].y = mv.y;
377
378 61122 mv.x = av_clip(mv.x, -2 * x, 2 * (width - x - 16));
379 61122 mv.y = av_clip(mv.y, -2 * y, 2 * (height - y - 16));
380
381 61122 src = &previous[(x + (mv.x >> 1)) + (y + (mv.y >> 1)) * pitch];
382 61122 dst = current;
383
384 61122 hdsp->put_pixels_tab[0][(mv.y & 1) << 1 | (mv.x & 1)](dst, src, pitch, 16);
385
386 61122 return 0;
387 }
388
389 4711 static int svq1_motion_inter_4v_block(HpelDSPContext *hdsp, GetBitContext *bitbuf,
390 uint8_t *current, uint8_t *previous,
391 ptrdiff_t pitch, svq1_pmv *motion, int x, int y,
392 int width, int height)
393 {
394 uint8_t *src;
395 uint8_t *dst;
396 svq1_pmv mv;
397 svq1_pmv *pmv[4];
398 int i, result;
399
400 /* predict and decode motion vector (0) */
401 4711 pmv[0] = &motion[0];
402
2/2
✓ Branch 0 taken 734 times.
✓ Branch 1 taken 3977 times.
 4711 if (y == 0) {
403 734 pmv[1] =
404 734 pmv[2] = pmv[0];
405 } else {
406 3977 pmv[1] = &motion[(x / 8) + 2];
407 3977 pmv[2] = &motion[(x / 8) + 4];
408 }
409
410 4711 result = svq1_decode_motion_vector(bitbuf, &mv, pmv);
411
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4711 times.
 4711 if (result)
412 ✗ return result;
413
414 /* predict and decode motion vector (1) */
415 4711 pmv[0] = &mv;
416
2/2
✓ Branch 0 taken 734 times.
✓ Branch 1 taken 3977 times.
 4711 if (y == 0) {
417 734 pmv[1] =
418 734 pmv[2] = pmv[0];
419 } else {
420 3977 pmv[1] = &motion[(x / 8) + 3];
421 }
422 4711 result = svq1_decode_motion_vector(bitbuf, &motion[0], pmv);
423
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4711 times.
 4711 if (result)
424 ✗ return result;
425
426 /* predict and decode motion vector (2) */
427 4711 pmv[1] = &motion[0];
428 4711 pmv[2] = &motion[(x / 8) + 1];
429
430 4711 result = svq1_decode_motion_vector(bitbuf, &motion[(x / 8) + 2], pmv);
431
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4711 times.
 4711 if (result)
432 ✗ return result;
433
434 /* predict and decode motion vector (3) */
435 4711 pmv[2] = &motion[(x / 8) + 2];
436 4711 pmv[3] = &motion[(x / 8) + 3];
437
438 4711 result = svq1_decode_motion_vector(bitbuf, pmv[3], pmv);
439
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4711 times.
 4711 if (result)
440 ✗ return result;
441
442 /* form predictions */
443
2/2
✓ Branch 0 taken 18844 times.
✓ Branch 1 taken 4711 times.
 23555 for (i = 0; i < 4; i++) {
444 18844 int mvx = pmv[i]->x + (i & 1) * 16;
445 18844 int mvy = pmv[i]->y + (i >> 1) * 16;
446
447 // FIXME: clipping or padding?
448 18844 mvx = av_clip(mvx, -2 * x, 2 * (width - x - 8));
449 18844 mvy = av_clip(mvy, -2 * y, 2 * (height - y - 8));
450
451 18844 src = &previous[(x + (mvx >> 1)) + (y + (mvy >> 1)) * pitch];
452 18844 dst = current;
453
454 18844 hdsp->put_pixels_tab[1][((mvy & 1) << 1) | (mvx & 1)](dst, src, pitch, 8);
455
456 /* select next block */
457
2/2
✓ Branch 0 taken 9422 times.
✓ Branch 1 taken 9422 times.
 18844 if (i & 1)
458 9422 current += 8 * (pitch - 1);
459 else
460 9422 current += 8;
461 }
462
463 4711 return 0;
464 }
465
466 78031 static int svq1_decode_delta_block(AVCodecContext *avctx, HpelDSPContext *hdsp,
467 GetBitContext *bitbuf,
468 uint8_t *current, uint8_t *previous,
469 ptrdiff_t pitch, svq1_pmv *motion, int x, int y,
470 int width, int height, int buggy)
471 {
472 uint32_t block_type;
473 78031 int result = 0;
474
475 /* get block type */
476 78031 block_type = get_vlc2(bitbuf, svq1_block_type,
477 SVQ1_BLOCK_TYPE_VLC_BITS, 1);
478
479 /* reset motion vectors */
480
4/4
✓ Branch 0 taken 69649 times.
✓ Branch 1 taken 8382 times.
✓ Branch 2 taken 3816 times.
✓ Branch 3 taken 65833 times.
 78031 if (block_type == SVQ1_BLOCK_SKIP || block_type == SVQ1_BLOCK_INTRA) {
481 12198 motion[0].x =
482 12198 motion[0].y =
483 12198 motion[x / 8 + 2].x =
484 12198 motion[x / 8 + 2].y =
485 12198 motion[x / 8 + 3].x =
486 12198 motion[x / 8 + 3].y = 0;
487 }
488
489
4/5
✓ Branch 0 taken 8382 times.
✓ Branch 1 taken 61122 times.
✓ Branch 2 taken 4711 times.
✓ Branch 3 taken 3816 times.
✗ Branch 4 not taken.
 78031 switch (block_type) {
490 8382 case SVQ1_BLOCK_SKIP:
491 8382 svq1_skip_block(current, previous, pitch, x, y);
492 8382 break;
493
494 61122 case SVQ1_BLOCK_INTER:
495 61122 result = svq1_motion_inter_block(hdsp, bitbuf, current, previous,
496 pitch, motion, x, y, width, height);
497
498
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 61122 times.
 61122 if (result != 0) {
499 ff_dlog(avctx, "Error in svq1_motion_inter_block %i\n", result);
500 ✗ break;
501 }
502 61122 result = svq1_decode_block_non_intra(bitbuf, current, pitch, buggy);
503 61122 break;
504
505 4711 case SVQ1_BLOCK_INTER_4V:
506 4711 result = svq1_motion_inter_4v_block(hdsp, bitbuf, current, previous,
507 pitch, motion, x, y, width, height);
508
509
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4711 times.
 4711 if (result != 0) {
510 ff_dlog(avctx, "Error in svq1_motion_inter_4v_block %i\n", result);
511 ✗ break;
512 }
513 4711 result = svq1_decode_block_non_intra(bitbuf, current, pitch, buggy);
514 4711 break;
515
516 3816 case SVQ1_BLOCK_INTRA:
517 3816 result = svq1_decode_block_intra(bitbuf, current, pitch);
518 3816 break;
519 }
520
521 78031 return result;
522 }
523
524 ✗ static void svq1_parse_string(GetBitContext *bitbuf, uint8_t out[257])
525 {
526 uint8_t seed;
527 int i;
528
529 ✗ out[0] = get_bits(bitbuf, 8);
530 ✗ seed = string_table[out[0]];
531
532 ✗ for (i = 1; i <= out[0]; i++) {
533 ✗ out[i] = get_bits(bitbuf, 8) ^ seed;
534 ✗ seed = string_table[out[i] ^ seed];
535 }
536 ✗ out[i] = 0;
537 ✗ }
538
539 368 static int svq1_decode_frame_header(AVCodecContext *avctx, AVFrame *frame, int * buggy)
540 {
541 368 SVQ1Context *s = avctx->priv_data;
542 368 GetBitContext *bitbuf = &s->gb;
543 int frame_size_code;
544 368 int width = s->width;
545 368 int height = s->height;
546 int tempref;
547
548 368 tempref = get_bits(bitbuf, 8); /* temporal_reference */
549
5/6
✓ Branch 0 taken 202 times.
✓ Branch 1 taken 166 times.
✓ Branch 2 taken 196 times.
✓ Branch 3 taken 6 times.
✗ Branch 4 not taken.
✓ Branch 5 taken 196 times.
 368 *buggy = tempref == 0 && s->last_tempref == 0 && avctx->extradata_size == 0;
550 368 s->last_tempref = tempref;
551
552 /* frame type */
553 368 s->nonref = 0;
554
2/4
✓ Branch 1 taken 36 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 332 times.
✗ Branch 4 not taken.
 368 switch (get_bits(bitbuf, 2)) {
555 36 case 0:
556 36 frame->pict_type = AV_PICTURE_TYPE_I;
557 36 break;
558 ✗ case 2:
559 ✗ s->nonref = 1;
560 332 case 1:
561 332 frame->pict_type = AV_PICTURE_TYPE_P;
562 332 break;
563 ✗ default:
564 ✗ av_log(avctx, AV_LOG_ERROR, "Invalid frame type.\n");
565 ✗ return AVERROR_INVALIDDATA;
566 }
567
568
2/2
✓ Branch 0 taken 36 times.
✓ Branch 1 taken 332 times.
 368 if (frame->pict_type == AV_PICTURE_TYPE_I) {
569 /* unknown fields */
570
3/4
✓ Branch 0 taken 35 times.
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 35 times.
 36 if (s->frame_code == 0x50 || s->frame_code == 0x60) {
571 1 int csum = get_bits(bitbuf, 16);
572
573 1 csum = av_bswap16(av_crc(av_crc_get_table(AV_CRC_16_CCITT), av_bswap16(csum), bitbuf->buffer, bitbuf->size_in_bits >> 3));
574
575 ff_dlog(avctx, "%s checksum (%02x) for packet data\n",
576 (csum == 0) ? "correct" : "incorrect", csum);
577 }
578
579
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 36 times.
 36 if ((s->frame_code ^ 0x10) >= 0x50) {
580 uint8_t msg[257];
581
582 ✗ svq1_parse_string(bitbuf, msg);
583
584 ✗ av_log(avctx, AV_LOG_INFO,
585 "embedded message:\n%s\n", ((char *)msg) + 1);
586 }
587
588 36 skip_bits(bitbuf, 2);
589 36 skip_bits(bitbuf, 2);
590 36 skip_bits1(bitbuf);
591
592 /* load frame size */
593 36 frame_size_code = get_bits(bitbuf, 3);
594
595
2/2
✓ Branch 0 taken 6 times.
✓ Branch 1 taken 30 times.
 36 if (frame_size_code == 7) {
596 /* load width, height (12 bits each) */
597 6 width = get_bits(bitbuf, 12);
598 6 height = get_bits(bitbuf, 12);
599
600
2/4
✓ Branch 0 taken 6 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 6 times.
 6 if (!width || !height)
601 ✗ return AVERROR_INVALIDDATA;
602 } else {
603 /* get width, height from table */
604 30 width = ff_svq1_frame_size_table[frame_size_code][0];
605 30 height = ff_svq1_frame_size_table[frame_size_code][1];
606 }
607 }
608
609 /* unknown fields */
610
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 368 times.
 368 if (get_bits1(bitbuf)) {
611 ✗ skip_bits1(bitbuf); /* use packet checksum if (1) */
612 ✗ skip_bits1(bitbuf); /* component checksums after image data if (1) */
613
614 ✗ if (get_bits(bitbuf, 2) != 0)
615 ✗ return AVERROR_INVALIDDATA;
616 }
617
618
2/2
✓ Branch 1 taken 168 times.
✓ Branch 2 taken 200 times.
 368 if (get_bits1(bitbuf)) {
619 168 skip_bits1(bitbuf);
620 168 skip_bits(bitbuf, 4);
621 168 skip_bits1(bitbuf);
622 168 skip_bits(bitbuf, 2);
623
624
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 168 times.
 168 if (skip_1stop_8data_bits(bitbuf) < 0)
625 ✗ return AVERROR_INVALIDDATA;
626 }
627
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 368 times.
 368 if (get_bits_left(bitbuf) <= 0)
628 ✗ return AVERROR_INVALIDDATA;
629
630 368 s->width = width;
631 368 s->height = height;
632 368 return 0;
633 }
634
635 368 static int svq1_decode_frame(AVCodecContext *avctx, AVFrame *cur,
636 int *got_frame, AVPacket *avpkt)
637 {
638 368 const uint8_t *buf = avpkt->data;
639 368 int buf_size = avpkt->size;
640 368 SVQ1Context *s = avctx->priv_data;
641 uint8_t *current;
642 int result, i, x, y, width, height, buggy;
643 int ret;
644
645 /* initialize bit buffer */
646 368 ret = init_get_bits8(&s->gb, buf, buf_size);
647
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 368 times.
 368 if (ret < 0)
648 ✗ return ret;
649
650 /* decode frame header */
651 368 s->frame_code = get_bits(&s->gb, 22);
652
653
2/4
✓ Branch 0 taken 368 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 368 times.
 368 if ((s->frame_code & ~0x70) || !(s->frame_code & 0x60))
654 ✗ return AVERROR_INVALIDDATA;
655
656 /* swap some header bytes (why?) */
657
2/2
✓ Branch 0 taken 13 times.
✓ Branch 1 taken 355 times.
 368 if (s->frame_code != 0x20) {
658 uint32_t *src;
659
660
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 13 times.
 13 if (buf_size < 9 * 4) {
661 ✗ av_log(avctx, AV_LOG_ERROR, "Input packet too small\n");
662 ✗ return AVERROR_INVALIDDATA;
663 }
664
665 13 av_fast_padded_malloc(&s->pkt_swapped,
666 13 &s->pkt_swapped_allocated,
667 buf_size);
668
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 13 times.
 13 if (!s->pkt_swapped)
669 ✗ return AVERROR(ENOMEM);
670
671 13 memcpy(s->pkt_swapped, buf, buf_size);
672 13 buf = s->pkt_swapped;
673 13 init_get_bits(&s->gb, buf, buf_size * 8);
674 13 skip_bits(&s->gb, 22);
675
676 13 src = (uint32_t *)(s->pkt_swapped + 4);
677
678
2/2
✓ Branch 0 taken 52 times.
✓ Branch 1 taken 13 times.
 65 for (i = 0; i < 4; i++)
679 52 src[i] = ((src[i] << 16) | (src[i] >> 16)) ^ src[7 - i];
680 }
681
682 368 result = svq1_decode_frame_header(avctx, cur, &buggy);
683
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 368 times.
 368 if (result != 0) {
684 ff_dlog(avctx, "Error in svq1_decode_frame_header %i\n", result);
685 ✗ return result;
686 }
687
688 368 result = ff_set_dimensions(avctx, s->width, s->height);
689
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 368 times.
 368 if (result < 0)
690 ✗ return result;
691
692
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 368 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 368 if ((avctx->skip_frame >= AVDISCARD_NONREF && s->nonref) ||
693
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 368 times.
 368 (avctx->skip_frame >= AVDISCARD_NONKEY &&
694 ✗ cur->pict_type != AV_PICTURE_TYPE_I) ||
695
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 368 times.
 368 avctx->skip_frame >= AVDISCARD_ALL)
696 ✗ return buf_size;
697
698 368 result = ff_get_buffer(avctx, cur, s->nonref ? 0 : AV_GET_BUFFER_FLAG_REF);
699
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 368 times.
 368 if (result < 0)
700 ✗ return result;
701
702 368 av_fast_padded_malloc(&s->pmv, &s->pmv_allocated, (FFALIGN(s->width, 16) / 8 + 3) * sizeof(*s->pmv));
703
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 368 times.
 368 if (!s->pmv)
704 ✗ return AVERROR(ENOMEM);
705
706 /* decode y, u and v components */
707
2/2
✓ Branch 0 taken 1104 times.
✓ Branch 1 taken 368 times.
 1472 for (i = 0; i < 3; i++) {
708 1104 int linesize = cur->linesize[i];
709
2/2
✓ Branch 0 taken 368 times.
✓ Branch 1 taken 736 times.
 1104 if (i == 0) {
710 368 width = FFALIGN(s->width, 16);
711 368 height = FFALIGN(s->height, 16);
712 } else {
713
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 736 times.
 736 if (avctx->flags & AV_CODEC_FLAG_GRAY)
714 ✗ break;
715 736 width = FFALIGN(s->width / 4, 16);
716 736 height = FFALIGN(s->height / 4, 16);
717 }
718
719 1104 current = cur->data[i];
720
721
2/2
✓ Branch 0 taken 108 times.
✓ Branch 1 taken 996 times.
 1104 if (cur->pict_type == AV_PICTURE_TYPE_I) {
722 /* keyframe */
723
2/2
✓ Branch 0 taken 643 times.
✓ Branch 1 taken 108 times.
 751 for (y = 0; y < height; y += 16) {
724
2/2
✓ Branch 0 taken 8533 times.
✓ Branch 1 taken 643 times.
 9176 for (x = 0; x < width; x += 16) {
725 8533 result = svq1_decode_block_intra(&s->gb, &current[x],
726 linesize);
727
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 8533 times.
 8533 if (result) {
728 ✗ av_log(avctx, AV_LOG_ERROR,
729 "Error in svq1_decode_block %i (keyframe)\n",
730 result);
731 ✗ return result;
732 }
733 }
734 643 current += 16 * linesize;
735 }
736 } else {
737 /* delta frame */
738 996 uint8_t *previous = s->prev->data[i];
739
1/2
✓ Branch 0 taken 996 times.
✗ Branch 1 not taken.
 996 if (!previous ||
740
2/4
✓ Branch 0 taken 996 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 996 times.
 996 s->prev->width != s->width || s->prev->height != s->height) {
741 ✗ av_log(avctx, AV_LOG_ERROR, "Missing reference frame.\n");
742 ✗ return AVERROR_INVALIDDATA;
743 }
744
745 996 memset(s->pmv, 0, ((width / 8) + 3) * sizeof(svq1_pmv));
746
747
2/2
✓ Branch 0 taken 5901 times.
✓ Branch 1 taken 996 times.
 6897 for (y = 0; y < height; y += 16) {
748
2/2
✓ Branch 0 taken 78031 times.
✓ Branch 1 taken 5901 times.
 83932 for (x = 0; x < width; x += 16) {
749 78031 result = svq1_decode_delta_block(avctx, &s->hdsp,
750 &s->gb, &current[x],
751 previous, linesize,
752 s->pmv, x, y, width, height, buggy);
753
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 78031 times.
 78031 if (result != 0) {
754 ff_dlog(avctx,
755 "Error in svq1_decode_delta_block %i\n",
756 result);
757 ✗ return result;
758 }
759 }
760
761 5901 s->pmv[0].x =
762 5901 s->pmv[0].y = 0;
763
764 5901 current += 16 * linesize;
765 }
766 }
767 }
768
769
1/2
✓ Branch 0 taken 368 times.
✗ Branch 1 not taken.
 368 if (!s->nonref) {
770 368 result = av_frame_replace(s->prev, cur);
771
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 368 times.
 368 if (result < 0)
772 ✗ return result;
773 }
774
775 368 *got_frame = 1;
776 368 result = buf_size;
777
778 368 return result;
779 }
780
781 9 static av_cold void svq1_static_init(void)
782 {
783 static VLCElem table[196];
784 9 VLCInitState state = VLC_INIT_STATE(table);
785
786 9 VLC_INIT_STATIC_TABLE(svq1_block_type, SVQ1_BLOCK_TYPE_VLC_BITS, 4,
787 &ff_svq1_block_type_vlc[0][1], 2, 1,
788 &ff_svq1_block_type_vlc[0][0], 2, 1, 0);
789
790 9 VLC_INIT_STATIC_TABLE(svq1_motion_component, 7, 33,
791 &ff_mvtab[0][1], 2, 1,
792 &ff_mvtab[0][0], 2, 1, 0);
793
794
2/2
✓ Branch 0 taken 54 times.
✓ Branch 1 taken 9 times.
 63 for (int i = 0; i < 6; i++) {
795 54 svq1_intra_multistage[i] =
796 54 ff_vlc_init_tables(&state, 4, 8,
797 54 &ff_svq1_intra_multistage_vlc[i][0][1], 2, 1,
798 54 &ff_svq1_intra_multistage_vlc[i][0][0], 2, 1, 0);
799 54 svq1_inter_multistage[i] =
800 54 ff_vlc_init_tables(&state, 3, 8,
801 54 &ff_svq1_inter_multistage_vlc[i][0][1], 2, 1,
802 54 &ff_svq1_inter_multistage_vlc[i][0][0], 2, 1, 0);
803 }
804
805 9 VLC_INIT_STATIC_TABLE(svq1_intra_mean, 8, 256,
806 &ff_svq1_intra_mean_vlc[0][1], 4, 2,
807 &ff_svq1_intra_mean_vlc[0][0], 4, 2, 0);
808
809 9 VLC_INIT_STATIC_TABLE(svq1_inter_mean, 9, 512,
810 &ff_svq1_inter_mean_vlc[0][1], 4, 2,
811 &ff_svq1_inter_mean_vlc[0][0], 4, 2, 0);
812 9 }
813
814 15 static av_cold int svq1_decode_init(AVCodecContext *avctx)
815 {
816 static AVOnce init_static_once = AV_ONCE_INIT;
817 15 SVQ1Context *s = avctx->priv_data;
818
819 15 s->prev = av_frame_alloc();
820
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 15 times.
 15 if (!s->prev)
821 ✗ return AVERROR(ENOMEM);
822
823 15 s->width = avctx->width + 3 & ~3;
824 15 s->height = avctx->height + 3 & ~3;
825 15 avctx->pix_fmt = AV_PIX_FMT_YUV410P;
826
827 15 ff_hpeldsp_init(&s->hdsp, avctx->flags);
828
829 15 ff_thread_once(&init_static_once, svq1_static_init);
830
831 15 s->last_tempref = 0xFF;
832
833 15 return 0;
834 }
835
836 15 static av_cold int svq1_decode_end(AVCodecContext *avctx)
837 {
838 15 SVQ1Context *s = avctx->priv_data;
839
840 15 av_frame_free(&s->prev);
841 15 av_freep(&s->pkt_swapped);
842 15 s->pkt_swapped_allocated = 0;
843 15 av_freep(&s->pmv);
844 15 s->pmv_allocated = 0;
845
846 15 return 0;
847 }
848
849 ✗ static void svq1_flush(AVCodecContext *avctx)
850 {
851 ✗ SVQ1Context *s = avctx->priv_data;
852
853 ✗ av_frame_unref(s->prev);
854 ✗ }
855
856 const FFCodec ff_svq1_decoder = {
857 .p.name = "svq1",
858 CODEC_LONG_NAME("Sorenson Vector Quantizer 1 / Sorenson Video 1 / SVQ1"),
859 .p.type = AVMEDIA_TYPE_VIDEO,
860 .p.id = AV_CODEC_ID_SVQ1,
861 .priv_data_size = sizeof(SVQ1Context),
862 .init = svq1_decode_init,
863 .close = svq1_decode_end,
864 FF_CODEC_DECODE_CB(svq1_decode_frame),
865 .p.capabilities = AV_CODEC_CAP_DR1,
866 .flush = svq1_flush,
867 };
868