Imported Debian version 2.5.3~trusty1
[deb_ffmpeg.git] / ffmpeg / libavcodec / mss4.c
CommitLineData
2ba45a60
DM
1/*
2 * Microsoft Screen 4 (aka Microsoft Expression Encoder Screen) decoder
3 * Copyright (c) 2012 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 * Microsoft Screen 4 (aka Microsoft Titanium Screen 2,
25 * aka Microsoft Expression Encoder Screen) decoder
26 */
27
28#include "avcodec.h"
29#include "bytestream.h"
30#include "get_bits.h"
31#include "internal.h"
32#include "mss34dsp.h"
33#include "unary.h"
34
35#define HEADER_SIZE 8
36
37enum FrameType {
38 INTRA_FRAME = 0,
39 INTER_FRAME,
40 SKIP_FRAME
41};
42
43enum BlockType {
44 SKIP_BLOCK = 0,
45 DCT_BLOCK,
46 IMAGE_BLOCK,
47};
48
49enum CachePos {
50 LEFT = 0,
51 TOP_LEFT,
52 TOP,
53};
54
55static const uint8_t mss4_dc_vlc_lens[2][16] = {
56 { 0, 1, 5, 1, 1, 1, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0 },
57 { 0, 3, 1, 1, 1, 1, 1, 1, 1, 2, 0, 0, 0, 0, 0, 0 }
58};
59
60static const uint8_t mss4_ac_vlc_lens[2][16] = {
61 { 0, 2, 1, 3, 3, 2, 4, 3, 5, 5, 4, 4, 0, 0, 1, 125 },
62 { 0, 2, 1, 2, 4, 4, 3, 4, 7, 5, 4, 4, 0, 1, 2, 119 }
63};
64
65static const uint8_t mss4_ac_vlc_syms[2][162] = {
66 { 0x01, 0x02, 0x03, 0x00, 0x04, 0x11, 0x05, 0x12,
67 0x21, 0x31, 0x41, 0x06, 0x13, 0x51, 0x61, 0x07,
68 0x22, 0x71, 0x14, 0x32, 0x81, 0x91, 0xA1, 0x08,
69 0x23, 0x42, 0xB1, 0xC1, 0x15, 0x52, 0xD1, 0xF0,
70 0x24, 0x33, 0x62, 0x72, 0x82, 0x09, 0x0A, 0x16,
71 0x17, 0x18, 0x19, 0x1A, 0x25, 0x26, 0x27, 0x28,
72 0x29, 0x2A, 0x34, 0x35, 0x36, 0x37, 0x38, 0x39,
73 0x3A, 0x43, 0x44, 0x45, 0x46, 0x47, 0x48, 0x49,
74 0x4A, 0x53, 0x54, 0x55, 0x56, 0x57, 0x58, 0x59,
75 0x5A, 0x63, 0x64, 0x65, 0x66, 0x67, 0x68, 0x69,
76 0x6A, 0x73, 0x74, 0x75, 0x76, 0x77, 0x78, 0x79,
77 0x7A, 0x83, 0x84, 0x85, 0x86, 0x87, 0x88, 0x89,
78 0x8A, 0x92, 0x93, 0x94, 0x95, 0x96, 0x97, 0x98,
79 0x99, 0x9A, 0xA2, 0xA3, 0xA4, 0xA5, 0xA6, 0xA7,
80 0xA8, 0xA9, 0xAA, 0xB2, 0xB3, 0xB4, 0xB5, 0xB6,
81 0xB7, 0xB8, 0xB9, 0xBA, 0xC2, 0xC3, 0xC4, 0xC5,
82 0xC6, 0xC7, 0xC8, 0xC9, 0xCA, 0xD2, 0xD3, 0xD4,
83 0xD5, 0xD6, 0xD7, 0xD8, 0xD9, 0xDA, 0xE1, 0xE2,
84 0xE3, 0xE4, 0xE5, 0xE6, 0xE7, 0xE8, 0xE9, 0xEA,
85 0xF1, 0xF2, 0xF3, 0xF4, 0xF5, 0xF6, 0xF7, 0xF8,
86 0xF9, 0xFA },
87 { 0x00, 0x01, 0x02, 0x03, 0x11, 0x04, 0x05, 0x21,
88 0x31, 0x06, 0x12, 0x41, 0x51, 0x07, 0x61, 0x71,
89 0x13, 0x22, 0x32, 0x81, 0x08, 0x14, 0x42, 0x91,
90 0xA1, 0xB1, 0xC1, 0x09, 0x23, 0x33, 0x52, 0xF0,
91 0x15, 0x62, 0x72, 0xD1, 0x0A, 0x16, 0x24, 0x34,
92 0xE1, 0x25, 0xF1, 0x17, 0x18, 0x19, 0x1A, 0x26,
93 0x27, 0x28, 0x29, 0x2A, 0x35, 0x36, 0x37, 0x38,
94 0x39, 0x3A, 0x43, 0x44, 0x45, 0x46, 0x47, 0x48,
95 0x49, 0x4A, 0x53, 0x54, 0x55, 0x56, 0x57, 0x58,
96 0x59, 0x5A, 0x63, 0x64, 0x65, 0x66, 0x67, 0x68,
97 0x69, 0x6A, 0x73, 0x74, 0x75, 0x76, 0x77, 0x78,
98 0x79, 0x7A, 0x82, 0x83, 0x84, 0x85, 0x86, 0x87,
99 0x88, 0x89, 0x8A, 0x92, 0x93, 0x94, 0x95, 0x96,
100 0x97, 0x98, 0x99, 0x9A, 0xA2, 0xA3, 0xA4, 0xA5,
101 0xA6, 0xA7, 0xA8, 0xA9, 0xAA, 0xB2, 0xB3, 0xB4,
102 0xB5, 0xB6, 0xB7, 0xB8, 0xB9, 0xBA, 0xC2, 0xC3,
103 0xC4, 0xC5, 0xC6, 0xC7, 0xC8, 0xC9, 0xCA, 0xD2,
104 0xD3, 0xD4, 0xD5, 0xD6, 0xD7, 0xD8, 0xD9, 0xDA,
105 0xE2, 0xE3, 0xE4, 0xE5, 0xE6, 0xE7, 0xE8, 0xE9,
106 0xEA, 0xF2, 0xF3, 0xF4, 0xF5, 0xF6, 0xF7, 0xF8,
107 0xF9, 0xFA }
108};
109
110static const uint8_t vec_len_syms[2][4] = {
111 { 4, 2, 3, 1 },
112 { 4, 1, 2, 3 }
113};
114
115static const uint8_t mss4_vec_entry_vlc_lens[2][16] = {
116 { 0, 2, 2, 3, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 },
117 { 0, 1, 5, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 }
118};
119
120static const uint8_t mss4_vec_entry_vlc_syms[2][9] = {
121 { 0, 7, 6, 5, 8, 4, 3, 1, 2 },
122 { 0, 2, 3, 4, 5, 6, 7, 1, 8 }
123};
124
125#define MAX_ENTRIES 162
126
127typedef struct MSS4Context {
128 AVFrame *pic;
129
130 VLC dc_vlc[2], ac_vlc[2];
131 VLC vec_entry_vlc[2];
132 int block[64];
133 uint8_t imgbuf[3][16 * 16];
134
135 int quality;
136 uint16_t quant_mat[2][64];
137
138 int *prev_dc[3];
139 int dc_stride[3];
140 int dc_cache[4][4];
141
142 int prev_vec[3][4];
143} MSS4Context;
144
145static av_cold int mss4_init_vlc(VLC *vlc, const uint8_t *lens,
146 const uint8_t *syms, int num_syms)
147{
148 uint8_t bits[MAX_ENTRIES];
149 uint16_t codes[MAX_ENTRIES];
150 int i, j;
151 int prefix = 0, max_bits = 0, idx = 0;
152
153 for (i = 0; i < 16; i++) {
154 for (j = 0; j < lens[i]; j++) {
155 bits[idx] = i + 1;
156 codes[idx] = prefix++;
157 max_bits = i + 1;
158 idx++;
159 }
160 prefix <<= 1;
161 }
162
163 return ff_init_vlc_sparse(vlc, FFMIN(max_bits, 9), num_syms, bits, 1, 1,
164 codes, 2, 2, syms, 1, 1, 0);
165}
166
167static av_cold int mss4_init_vlcs(MSS4Context *ctx)
168{
169 int ret, i;
170
171 for (i = 0; i < 2; i++) {
172 ret = mss4_init_vlc(&ctx->dc_vlc[i], mss4_dc_vlc_lens[i], NULL, 12);
173 if (ret)
174 return ret;
175 ret = mss4_init_vlc(&ctx->ac_vlc[i], mss4_ac_vlc_lens[i],
176 mss4_ac_vlc_syms[i], 162);
177 if (ret)
178 return ret;
179 ret = mss4_init_vlc(&ctx->vec_entry_vlc[i], mss4_vec_entry_vlc_lens[i],
180 mss4_vec_entry_vlc_syms[i], 9);
181 if (ret)
182 return ret;
183 }
184 return 0;
185}
186
187static av_cold void mss4_free_vlcs(MSS4Context *ctx)
188{
189 int i;
190
191 for (i = 0; i < 2; i++) {
192 ff_free_vlc(&ctx->dc_vlc[i]);
193 ff_free_vlc(&ctx->ac_vlc[i]);
194 ff_free_vlc(&ctx->vec_entry_vlc[i]);
195 }
196}
197
198/* This function returns values in the range
199 * (-range + 1; -range/2] U [range/2; range - 1)
200 * i.e.
201 * nbits = 0 -> 0
202 * nbits = 1 -> -1, 1
203 * nbits = 2 -> -3, -2, 2, 3
204 */
205static av_always_inline int get_coeff_bits(GetBitContext *gb, int nbits)
206{
207 int val;
208
209 if (!nbits)
210 return 0;
211
212 val = get_bits(gb, nbits);
213 if (val < (1 << (nbits - 1)))
214 val -= (1 << nbits) - 1;
215
216 return val;
217}
218
219static inline int get_coeff(GetBitContext *gb, VLC *vlc)
220{
221 int val = get_vlc2(gb, vlc->table, vlc->bits, 2);
222
223 return get_coeff_bits(gb, val);
224}
225
226static int mss4_decode_dct(GetBitContext *gb, VLC *dc_vlc, VLC *ac_vlc,
227 int *block, int *dc_cache,
228 int bx, int by, uint16_t *quant_mat)
229{
230 int skip, val, pos = 1, zz_pos, dc;
231
232 memset(block, 0, sizeof(*block) * 64);
233
234 dc = get_coeff(gb, dc_vlc);
235 // DC prediction is the same as in MSS3
236 if (by) {
237 if (bx) {
238 int l, tl, t;
239
240 l = dc_cache[LEFT];
241 tl = dc_cache[TOP_LEFT];
242 t = dc_cache[TOP];
243
244 if (FFABS(t - tl) <= FFABS(l - tl))
245 dc += l;
246 else
247 dc += t;
248 } else {
249 dc += dc_cache[TOP];
250 }
251 } else if (bx) {
252 dc += dc_cache[LEFT];
253 }
254 dc_cache[LEFT] = dc;
255 block[0] = dc * quant_mat[0];
256
257 while (pos < 64) {
258 val = get_vlc2(gb, ac_vlc->table, 9, 2);
259 if (!val)
260 return 0;
261 if (val == -1)
262 return -1;
263 if (val == 0xF0) {
264 pos += 16;
265 continue;
266 }
267 skip = val >> 4;
268 val = get_coeff_bits(gb, val & 0xF);
269 pos += skip;
270 if (pos >= 64)
271 return -1;
272
273 zz_pos = ff_zigzag_direct[pos];
274 block[zz_pos] = val * quant_mat[zz_pos];
275 pos++;
276 }
277
278 return pos == 64 ? 0 : -1;
279}
280
281static int mss4_decode_dct_block(MSS4Context *c, GetBitContext *gb,
282 uint8_t *dst[3], int mb_x, int mb_y)
283{
284 int i, j, k, ret;
285 uint8_t *out = dst[0];
286
287 for (j = 0; j < 2; j++) {
288 for (i = 0; i < 2; i++) {
289 int xpos = mb_x * 2 + i;
290 c->dc_cache[j][TOP_LEFT] = c->dc_cache[j][TOP];
291 c->dc_cache[j][TOP] = c->prev_dc[0][mb_x * 2 + i];
292 ret = mss4_decode_dct(gb, c->dc_vlc, c->ac_vlc, c->block,
293 c->dc_cache[j],
294 xpos, mb_y * 2 + j, c->quant_mat[0]);
295 if (ret)
296 return ret;
297 c->prev_dc[0][mb_x * 2 + i] = c->dc_cache[j][LEFT];
298
299 ff_mss34_dct_put(out + xpos * 8, c->pic->linesize[0],
300 c->block);
301 }
302 out += 8 * c->pic->linesize[0];
303 }
304
305 for (i = 1; i < 3; i++) {
306 c->dc_cache[i + 1][TOP_LEFT] = c->dc_cache[i + 1][TOP];
307 c->dc_cache[i + 1][TOP] = c->prev_dc[i][mb_x];
308 ret = mss4_decode_dct(gb, c->dc_vlc + 1, c->ac_vlc + 1,
309 c->block, c->dc_cache[i + 1], mb_x, mb_y,
310 c->quant_mat[1]);
311 if (ret)
312 return ret;
313 c->prev_dc[i][mb_x] = c->dc_cache[i + 1][LEFT];
314
315 ff_mss34_dct_put(c->imgbuf[i], 8, c->block);
316 out = dst[i] + mb_x * 16;
317 // Since the DCT block is coded as YUV420 and the whole frame as YUV444,
318 // we need to scale chroma.
319 for (j = 0; j < 16; j++) {
320 for (k = 0; k < 8; k++)
321 AV_WN16A(out + k * 2, c->imgbuf[i][k + (j & ~1) * 4] * 0x101);
322 out += c->pic->linesize[i];
323 }
324 }
325
326 return 0;
327}
328
329static void read_vec_pos(GetBitContext *gb, int *vec_pos, int *sel_flag,
330 int *sel_len, int *prev)
331{
332 int i, y_flag = 0;
333
334 for (i = 2; i >= 0; i--) {
335 if (!sel_flag[i]) {
336 vec_pos[i] = 0;
337 continue;
338 }
339 if ((!i && !y_flag) || get_bits1(gb)) {
340 if (sel_len[i] > 0) {
341 int pval = prev[i];
342 vec_pos[i] = get_bits(gb, sel_len[i]);
343 if (vec_pos[i] >= pval)
344 vec_pos[i]++;
345 } else {
346 vec_pos[i] = !prev[i];
347 }
348 y_flag = 1;
349 } else {
350 vec_pos[i] = prev[i];
351 }
352 }
353}
354
355static int get_value_cached(GetBitContext *gb, int vec_pos, uint8_t *vec,
356 int vec_size, int component, int shift, int *prev)
357{
358 if (vec_pos < vec_size)
359 return vec[vec_pos];
360 if (!get_bits1(gb))
361 return prev[component];
362 prev[component] = get_bits(gb, 8 - shift) << shift;
363 return prev[component];
364}
365
366#define MKVAL(vals) ((vals)[0] | ((vals)[1] << 3) | ((vals)[2] << 6))
367
368/* Image mode - the hardest to comprehend MSS4 coding mode.
369 *
370 * In this mode all three 16x16 blocks are coded together with a method
371 * remotely similar to the methods employed in MSS1-MSS3.
372 * The idea is that every component has a vector of 1-4 most common symbols
373 * and an escape mode for reading new value from the bitstream. Decoding
374 * consists of retrieving pixel values from the vector or reading new ones
375 * from the bitstream; depending on flags read from the bitstream, these vector
376 * positions can be updated or reused from the state of the previous line
377 * or previous pixel.
378 */
379static int mss4_decode_image_block(MSS4Context *ctx, GetBitContext *gb,
380 uint8_t *picdst[3], int mb_x, int mb_y)
381{
382 uint8_t vec[3][4];
383 int vec_len[3];
384 int sel_len[3], sel_flag[3];
385 int i, j, k, mode, split;
386 int prev_vec1 = 0, prev_split = 0;
387 int vals[3] = { 0 };
388 int prev_pix[3] = { 0 };
389 int prev_mode[16] = { 0 };
390 uint8_t *dst[3];
391
392 const int val_shift = ctx->quality == 100 ? 0 : 2;
393
394 for (i = 0; i < 3; i++)
395 dst[i] = ctx->imgbuf[i];
396
397 for (i = 0; i < 3; i++) {
398 vec_len[i] = vec_len_syms[!!i][get_unary(gb, 0, 3)];
399 for (j = 0; j < vec_len[i]; j++) {
400 vec[i][j] = get_coeff(gb, &ctx->vec_entry_vlc[!!i]);
401 vec[i][j] += ctx->prev_vec[i][j];
402 ctx->prev_vec[i][j] = vec[i][j];
403 }
404 sel_flag[i] = vec_len[i] > 1;
405 sel_len[i] = vec_len[i] > 2 ? vec_len[i] - 2 : 0;
406 }
407
408 for (j = 0; j < 16; j++) {
409 if (get_bits1(gb)) {
410 split = 0;
411 if (get_bits1(gb)) {
412 prev_mode[0] = 0;
413 vals[0] = vals[1] = vals[2] = 0;
414 mode = 2;
415 } else {
416 mode = get_bits1(gb);
417 if (mode)
418 split = get_bits(gb, 4);
419 }
420 for (i = 0; i < 16; i++) {
421 if (mode <= 1) {
422 vals[0] = prev_mode[i] & 7;
423 vals[1] = (prev_mode[i] >> 3) & 7;
424 vals[2] = prev_mode[i] >> 6;
425 if (mode == 1 && i == split) {
426 read_vec_pos(gb, vals, sel_flag, sel_len, vals);
427 }
428 } else if (mode == 2) {
429 if (get_bits1(gb))
430 read_vec_pos(gb, vals, sel_flag, sel_len, vals);
431 }
432 for (k = 0; k < 3; k++)
433 *dst[k]++ = get_value_cached(gb, vals[k], vec[k],
434 vec_len[k], k,
435 val_shift, prev_pix);
436 prev_mode[i] = MKVAL(vals);
437 }
438 } else {
439 if (get_bits1(gb)) {
440 split = get_bits(gb, 4);
441 if (split >= prev_split)
442 split++;
443 prev_split = split;
444 } else {
445 split = prev_split;
446 }
447 if (split) {
448 vals[0] = prev_mode[0] & 7;
449 vals[1] = (prev_mode[0] >> 3) & 7;
450 vals[2] = prev_mode[0] >> 6;
451 for (i = 0; i < 3; i++) {
452 for (k = 0; k < split; k++) {
453 *dst[i]++ = get_value_cached(gb, vals[i], vec[i],
454 vec_len[i], i, val_shift,
455 prev_pix);
456 prev_mode[k] = MKVAL(vals);
457 }
458 }
459 }
460
461 if (split != 16) {
462 vals[0] = prev_vec1 & 7;
463 vals[1] = (prev_vec1 >> 3) & 7;
464 vals[2] = prev_vec1 >> 6;
465 if (get_bits1(gb)) {
466 read_vec_pos(gb, vals, sel_flag, sel_len, vals);
467 prev_vec1 = MKVAL(vals);
468 }
469 for (i = 0; i < 3; i++) {
470 for (k = 0; k < 16 - split; k++) {
471 *dst[i]++ = get_value_cached(gb, vals[i], vec[i],
472 vec_len[i], i, val_shift,
473 prev_pix);
474 prev_mode[split + k] = MKVAL(vals);
475 }
476 }
477 }
478 }
479 }
480
481 for (i = 0; i < 3; i++)
482 for (j = 0; j < 16; j++)
483 memcpy(picdst[i] + mb_x * 16 + j * ctx->pic->linesize[i],
484 ctx->imgbuf[i] + j * 16, 16);
485
486 return 0;
487}
488
489static inline void mss4_update_dc_cache(MSS4Context *c, int mb_x)
490{
491 int i;
492
493 c->dc_cache[0][TOP] = c->prev_dc[0][mb_x * 2 + 1];
494 c->dc_cache[0][LEFT] = 0;
495 c->dc_cache[1][TOP] = 0;
496 c->dc_cache[1][LEFT] = 0;
497
498 for (i = 0; i < 2; i++)
499 c->prev_dc[0][mb_x * 2 + i] = 0;
500
501 for (i = 1; i < 3; i++) {
502 c->dc_cache[i + 1][TOP] = c->prev_dc[i][mb_x];
503 c->dc_cache[i + 1][LEFT] = 0;
504 c->prev_dc[i][mb_x] = 0;
505 }
506}
507
508static int mss4_decode_frame(AVCodecContext *avctx, void *data, int *got_frame,
509 AVPacket *avpkt)
510{
511 const uint8_t *buf = avpkt->data;
512 int buf_size = avpkt->size;
513 MSS4Context *c = avctx->priv_data;
514 GetBitContext gb;
515 GetByteContext bc;
516 uint8_t *dst[3];
517 int width, height, quality, frame_type;
518 int x, y, i, mb_width, mb_height, blk_type;
519 int ret;
520
521 if (buf_size < HEADER_SIZE) {
522 av_log(avctx, AV_LOG_ERROR,
523 "Frame should have at least %d bytes, got %d instead\n",
524 HEADER_SIZE, buf_size);
525 return AVERROR_INVALIDDATA;
526 }
527
528 bytestream2_init(&bc, buf, buf_size);
529 width = bytestream2_get_be16(&bc);
530 height = bytestream2_get_be16(&bc);
531 bytestream2_skip(&bc, 2);
532 quality = bytestream2_get_byte(&bc);
533 frame_type = bytestream2_get_byte(&bc);
534
535 if (width > avctx->width ||
536 height != avctx->height) {
537 av_log(avctx, AV_LOG_ERROR, "Invalid frame dimensions %dx%d\n",
538 width, height);
539 return AVERROR_INVALIDDATA;
540 }
541 if (quality < 1 || quality > 100) {
542 av_log(avctx, AV_LOG_ERROR, "Invalid quality setting %d\n", quality);
543 return AVERROR_INVALIDDATA;
544 }
545 if ((frame_type & ~3) || frame_type == 3) {
546 av_log(avctx, AV_LOG_ERROR, "Invalid frame type %d\n", frame_type);
547 return AVERROR_INVALIDDATA;
548 }
549
550 if (frame_type != SKIP_FRAME && !bytestream2_get_bytes_left(&bc)) {
551 av_log(avctx, AV_LOG_ERROR,
552 "Empty frame found but it is not a skip frame.\n");
553 return AVERROR_INVALIDDATA;
554 }
555
556 if ((ret = ff_reget_buffer(avctx, c->pic)) < 0)
557 return ret;
558 c->pic->key_frame = (frame_type == INTRA_FRAME);
559 c->pic->pict_type = (frame_type == INTRA_FRAME) ? AV_PICTURE_TYPE_I
560 : AV_PICTURE_TYPE_P;
561 if (frame_type == SKIP_FRAME) {
562 *got_frame = 1;
563 if ((ret = av_frame_ref(data, c->pic)) < 0)
564 return ret;
565
566 return buf_size;
567 }
568
569 if (c->quality != quality) {
570 c->quality = quality;
571 for (i = 0; i < 2; i++)
572 ff_mss34_gen_quant_mat(c->quant_mat[i], quality, !i);
573 }
574
575 init_get_bits8(&gb, buf + HEADER_SIZE, (buf_size - HEADER_SIZE));
576
577 mb_width = FFALIGN(width, 16) >> 4;
578 mb_height = FFALIGN(height, 16) >> 4;
579 dst[0] = c->pic->data[0];
580 dst[1] = c->pic->data[1];
581 dst[2] = c->pic->data[2];
582
583 memset(c->prev_vec, 0, sizeof(c->prev_vec));
584 for (y = 0; y < mb_height; y++) {
585 memset(c->dc_cache, 0, sizeof(c->dc_cache));
586 for (x = 0; x < mb_width; x++) {
587 blk_type = decode012(&gb);
588 switch (blk_type) {
589 case DCT_BLOCK:
590 if (mss4_decode_dct_block(c, &gb, dst, x, y) < 0) {
591 av_log(avctx, AV_LOG_ERROR,
592 "Error decoding DCT block %d,%d\n",
593 x, y);
594 return AVERROR_INVALIDDATA;
595 }
596 break;
597 case IMAGE_BLOCK:
598 if (mss4_decode_image_block(c, &gb, dst, x, y) < 0) {
599 av_log(avctx, AV_LOG_ERROR,
600 "Error decoding VQ block %d,%d\n",
601 x, y);
602 return AVERROR_INVALIDDATA;
603 }
604 break;
605 case SKIP_BLOCK:
606 if (frame_type == INTRA_FRAME) {
607 av_log(avctx, AV_LOG_ERROR, "Skip block in intra frame\n");
608 return AVERROR_INVALIDDATA;
609 }
610 break;
611 }
612 if (blk_type != DCT_BLOCK)
613 mss4_update_dc_cache(c, x);
614 }
615 dst[0] += c->pic->linesize[0] * 16;
616 dst[1] += c->pic->linesize[1] * 16;
617 dst[2] += c->pic->linesize[2] * 16;
618 }
619
620 if ((ret = av_frame_ref(data, c->pic)) < 0)
621 return ret;
622
623 *got_frame = 1;
624
625 return buf_size;
626}
627
628static av_cold int mss4_decode_end(AVCodecContext *avctx)
629{
630 MSS4Context * const c = avctx->priv_data;
631 int i;
632
633 av_frame_free(&c->pic);
634 for (i = 0; i < 3; i++)
635 av_freep(&c->prev_dc[i]);
636 mss4_free_vlcs(c);
637
638 return 0;
639}
640
641static av_cold int mss4_decode_init(AVCodecContext *avctx)
642{
643 MSS4Context * const c = avctx->priv_data;
644 int i;
645
646 if (mss4_init_vlcs(c)) {
647 av_log(avctx, AV_LOG_ERROR, "Cannot initialise VLCs\n");
648 mss4_free_vlcs(c);
649 return AVERROR(ENOMEM);
650 }
651 for (i = 0; i < 3; i++) {
652 c->dc_stride[i] = FFALIGN(avctx->width, 16) >> (2 + !!i);
653 c->prev_dc[i] = av_malloc(sizeof(**c->prev_dc) * c->dc_stride[i]);
654 if (!c->prev_dc[i]) {
655 av_log(avctx, AV_LOG_ERROR, "Cannot allocate buffer\n");
656 mss4_free_vlcs(c);
657 return AVERROR(ENOMEM);
658 }
659 }
660
661 c->pic = av_frame_alloc();
662 if (!c->pic) {
663 mss4_decode_end(avctx);
664 return AVERROR(ENOMEM);
665 }
666
667 avctx->pix_fmt = AV_PIX_FMT_YUV444P;
668
669 return 0;
670}
671
672AVCodec ff_mts2_decoder = {
673 .name = "mts2",
674 .long_name = NULL_IF_CONFIG_SMALL("MS Expression Encoder Screen"),
675 .type = AVMEDIA_TYPE_VIDEO,
676 .id = AV_CODEC_ID_MTS2,
677 .priv_data_size = sizeof(MSS4Context),
678 .init = mss4_decode_init,
679 .close = mss4_decode_end,
680 .decode = mss4_decode_frame,
681 .capabilities = CODEC_CAP_DR1,
682};