1 /*
2 * Microsoft Screen 2 (aka Windows Media Video V9 Screen) decoder
3 *
4 * This file is part of FFmpeg.
5 *
6 * FFmpeg is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
10 *
11 * FFmpeg is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with FFmpeg; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19 */
20
21 /**
22 * @file
23 * Microsoft Screen 2 (aka Windows Media Video V9 Screen) decoder
24 */
25
36
46
48 {
49 while ((c->
high >> 15) - (c->
low >> 15) < 2) {
50 if ((c->
low ^ c->
high) & 0x10000) {
54 }
55 c->
high = (uint16_t)c->
high << 8 | 0xFF;
57 c->
low = (uint16_t)c->
low << 8;
58 }
59 }
60
62
63 /* L. Stuiver and A. Moffat: "Piecewise Integer Mapping for Arithmetic Coding."
64 * In Proc. 8th Data Compression Conference (DCC '98), pp. 3-12, Mar. 1998 */
65
67 {
68 int split = (n << 1) - range;
69
70 if (value > split)
71 return split + (value - split >> 1);
72 else
74 }
75
77 int low,
int high,
int n)
78 {
79 int split = (n << 1) - range;
80
81 if (high > split)
82 c->
high = split + (high - split << 1);
83 else
85
87
88 if (low > split)
89 c->
low += split + (low - split << 1);
90 else
92 }
93
95 {
96 int range = c->
high - c->
low + 1;
99
100 if (n << scale > range)
101 scale--;
102
103 n <<= scale;
104
106
108
110
112 }
113
115 {
116 int range = c->
high - c->
low + 1,
n = *probs;
119
120 if (
n << scale > range)
121 scale--;
122
124
126 while (probs[++i] >
val) ;
127
129 probs[i] << scale, probs[i - 1] << scale,
n);
130
131 return i;
132 }
133
135
137 {
138 int diff = (c->high >> 16) - (c->low >> 16);
141
142 while (!(diff & 0x80)) {
143 bits++;
144 diff <<= 1;
145 }
146
147 return (bits + bp + 7 >> 3) + ((c->low >> 16) + 1 == c->high >> 16);
148 }
149
151 {
154 c->
value = bytestream2_get_be24(gB);
158 }
159
161 {
162 int i, ncol;
164
166 return 0;
167
168 ncol = *buf++;
169 if (ncol > ctx->
free_colours || buf_size < 2 + ncol * 3)
171 for (i = 0; i < ncol; i++)
173
174 return 1 + ncol * 3;
175 }
176
178 int keyframe,
int w,
int h)
179 {
180 int last_symbol = 0, repeat = 0, prev_avail = 0;
181
182 if (!keyframe) {
183 int x, y, endx, endy, t;
184
185 #define READ_PAIR(a, b) \
186 a = bytestream2_get_byte(gB) << 4; \
187 t = bytestream2_get_byte(gB); \
188 a |= t >> 4; \
189 b = (t & 0xF) << 8; \
190 b |= bytestream2_get_byte(gB); \
191
194
195 if (endx >= w || endy >= h || x > endx || y > endy)
197 dst += x + stride * y;
198 w = endx - x + 1;
199 h = endy - y + 1;
200 if (y)
201 prev_avail = 1;
202 }
203
204 do {
205 uint16_t *p = dst;
206 do {
207 if (repeat-- < 1) {
208 int b = bytestream2_get_byte(gB);
209 if (b < 128)
210 last_symbol = b << 8 | bytestream2_get_byte(gB);
211 else if (b > 129) {
212 repeat = 0;
213 while (b-- > 130) {
214 if (repeat >= (INT_MAX >> 8) - 1) {
217 }
218 repeat = (repeat << 8) + bytestream2_get_byte(gB) + 1;
219 }
220 if (last_symbol == -2) {
221 int skip =
FFMIN((
unsigned)repeat, dst + w - p);
222 repeat -= skip;
223 p += skip;
224 }
225 } else
226 last_symbol = 127 -
b;
227 }
228 if (last_symbol >= 0)
229 *p = last_symbol;
230 else if (last_symbol == -1 && prev_avail)
232 } while (++p < dst + w);
234 prev_avail = 1;
235 } while (--h);
236
237 return 0;
238 }
239
241 uint8_t *rgb_dst,
int rgb_stride, uint32_t *pal,
242 int keyframe,
int kf_slipt,
int slice,
int w,
int h)
243 {
245 uint32_t codes[270];
247
248 int current_length = 0, read_codes = 0, next_code = 0, current_codes = 0;
249 int remaining_codes, surplus_codes, i;
250
251 const int alphabet_size = 270 - keyframe;
252
253 int last_symbol = 0, repeat = 0, prev_avail = 0;
254
255 if (!keyframe) {
256 int x, y, clipw, cliph;
257
262
263 if (x + clipw > w || y + cliph > h)
265 pal_dst += pal_stride * y + x;
266 rgb_dst += rgb_stride * y + x * 3;
267 w = clipw;
268 h = cliph;
269 if (y)
270 prev_avail = 1;
271 } else {
272 if (slice > 0) {
273 pal_dst += pal_stride * kf_slipt;
274 rgb_dst += rgb_stride * kf_slipt;
275 prev_avail = 1;
276 h -= kf_slipt;
277 } else
278 h = kf_slipt;
279 }
280
281 /* read explicit codes */
282 do {
283 while (current_codes--) {
285 if (symbol >= 204 - keyframe)
286 symbol += 14 - keyframe;
287 else if (symbol > 189)
288 symbol =
get_bits1(gb) + (symbol << 1) - 190;
289 if (bits[symbol])
291 bits[symbol] = current_length;
292 codes[symbol] = next_code++;
293 read_codes++;
294 }
295 current_length++;
296 next_code <<= 1;
297 remaining_codes = (1 << current_length) - next_code;
298 current_codes =
get_bits(gb, av_ceil_log2(remaining_codes + 1));
299 if (current_length > 22 || current_codes > remaining_codes)
301 } while (current_codes != remaining_codes);
302
303 remaining_codes = alphabet_size - read_codes;
304
305 /* determine the minimum length to fit the rest of the alphabet */
306 while ((surplus_codes = (2 << current_length) -
307 (next_code << 1) - remaining_codes) < 0) {
308 current_length++;
309 next_code <<= 1;
310 }
311
312 /* add the rest of the symbols lexicographically */
313 for (i = 0; i < alphabet_size; i++)
314 if (!bits[i]) {
315 if (surplus_codes-- == 0) {
316 current_length++;
317 next_code <<= 1;
318 }
319 bits[i] = current_length;
320 codes[i] = next_code++;
321 }
322
323 if (next_code != 1 << current_length)
325
326 if ((i =
init_vlc(&vlc, 9, alphabet_size, bits, 1, 1, codes, 4, 4, 0)) < 0)
327 return i;
328
329 /* frame decode */
330 do {
333 do {
334 if (repeat-- < 1) {
336 if (b < 256)
338 else if (b < 268) {
339 b -= 256;
340 if (b == 11)
342
343 if (!b)
344 repeat = 0;
345 else
347
348 repeat += (1 <<
b) - 1;
349
350 if (last_symbol == -2) {
351 int skip =
FFMIN(repeat, pal_dst + w - pp);
352 repeat -= skip;
353 pp += skip;
354 rp += skip * 3;
355 }
356 } else
357 last_symbol = 267 -
b;
358 }
359 if (last_symbol >= 0) {
360 *pp = last_symbol;
362 } else if (last_symbol == -1 && prev_avail) {
363 *pp = *(pp - pal_stride);
364 memcpy(rp, rp - rgb_stride, 3);
365 }
366 rp += 3;
367 } while (++pp < pal_dst + w);
368 pal_dst += pal_stride;
369 rgb_dst += rgb_stride;
370 prev_avail = 1;
371 } while (--h);
372
374 return 0;
375 }
376
378 int x,
int y,
int w,
int h,
int wmv9_mask)
379 {
385 int ret;
386
388
390 return ret;
391
393
397 }
398
402 }
403
405
409 return ret;
410 }
411
413
414 v->
bits = buf_size * 8;
415
422
424
426
428
430
437 "Asymmetric WMV9 rectangle subsampling");
438
440
441 if (wmv9_mask != -1)
448 w, h);
449 else
454 w, h);
455
457
458 return 0;
459 }
460
464
465 #define MAX_WMV9_RECTANGLES 20
466 #define ARITH2_PADDING 2
467
470 {
472 int buf_size = avpkt->
size;
479
480 int keyframe, has_wmv9, has_mv, is_rle, is_555, ret;
481
483 int used_rects = 0, i, implicit_rect = 0,
av_uninit(wmv9_mask);
484
487
489 return ret;
490
504 else
506 } else
508 } else {
509 if (keyframe)
511 }
512 } else
514
518
522
523 if (buf_size < 1)
525
528
532
533 if (has_wmv9) {
536
537 implicit_rect = !arith2_get_bit(&acoder);
538
539 while (arith2_get_bit(&acoder)) {
542 r = &wmv9rects[used_rects];
543 if (!used_rects)
545 else
547 wmv9rects[used_rects - 1].
x) +
548 wmv9rects[used_rects - 1].
x;
552 used_rects++;
553 }
554
555 if (implicit_rect && used_rects) {
558 }
559
560 if (implicit_rect) {
563 wmv9rects[0].
w = avctx->
width;
564 wmv9rects[0].
h = avctx->
height;
565
566 used_rects = 1;
567 }
568 for (i = 0; i < used_rects; i++) {
569 if (!implicit_rect && arith2_get_bit(&acoder)) {
572 }
573 if (!i) {
574 wmv9_mask = arith2_get_bit(&acoder) - 1;
575 if (!wmv9_mask)
577 }
579 }
580
583 if (buf_size < 1)
585 }
586
588 if (keyframe && !is_555) {
591 buf += i;
592 buf_size -= i;
593 } else if (has_mv) {
594 buf += 4;
595 buf_size -= 4;
596 if (buf_size < 1)
600 }
601
602 if (c->
mvX < 0 || c->
mvY < 0) {
604
606 return ret;
607
612 } else {
615 }
616 } else {
618 return ret;
620 return ret;
621
623 }
627
630
631 if (is_555) {
633
637
639 } else {
640 if (keyframe) {
645 }
646 if (is_rle) {
648 return ret;
653 return ret;
655
661 return ret;
662
666 } else if (!implicit_rect || wmv9_mask != -1) {
676
680 if (buf_size < 1)
689
692 }
693 } else
695 }
696
697 if (has_wmv9) {
698 for (i = 0; i < used_rects; i++) {
699 int x = wmv9rects[i].
x;
700 int y = wmv9rects[i].
y;
701 int w = wmv9rects[i].
w;
702 int h = wmv9rects[i].
h;
703 if (wmv9rects[i].coded) {
704 int WMV9codedFrameSize;
705 if (buf_size < 4 || !(WMV9codedFrameSize =
AV_RL24(buf)))
707 if (ret =
decode_wmv9(avctx, buf + 3, buf_size - 3,
708 x, y, w, h, wmv9_mask))
709 return ret;
710 buf += WMV9codedFrameSize + 3;
711 buf_size -= WMV9codedFrameSize + 3;
712 } else {
714 if (wmv9_mask != -1) {
716 wmv9_mask,
719 w, h);
720 } else {
721 do {
722 memset(dst, 0x80, w * 3);
724 } while (--h);
725 }
726 }
727 }
728 }
729
730 if (buf_size)
732
733 if (c->
mvX < 0 || c->
mvY < 0) {
736 if (ret < 0)
737 return ret;
738 }
739
740 *got_frame = 1;
741
743 }
744
746 {
748 int ret;
749
751
753 return ret;
755
757
762
765
769
771
773
776
778
780
783
786
788
790
792
795 return ret;
796
797 /* error concealment */
800
801 return 0;
802 }
803
805 {
807
809
814
815 return 0;
816 }
817
819 {
822 int ret;
825 return ret;
833 }
836 return ret;
837 }
840
843
844
845 return 0;
846 }
847
858 };
static av_cold int mss2_decode_init(AVCodecContext *avctx)
const char const char void * val
qpel_mc_func avg_qpel_pixels_tab[2][16]
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
static int arith2_get_scaled_value(int value, int n, int range)
This structure describes decoded (raw) audio or video data.
ptrdiff_t const GLvoid * data
static int decode_rle(GetBitContext *gb, uint8_t *pal_dst, int pal_stride, uint8_t *rgb_dst, int rgb_stride, uint32_t *pal, int keyframe, int kf_slipt, int slice, int w, int h)
av_cold int ff_mss12_decode_init(MSS12Context *c, int version, SliceContext *sc1, SliceContext *sc2)
static unsigned int get_bits(GetBitContext *s, int n)
Read 1-25 bits.
#define AV_LOG_WARNING
Something somehow does not look correct.
int ff_msmpeg4_decode_init(AVCodecContext *avctx)
packed RGB 8:8:8, 24bpp, RGBRGB...
int end_mb_y
end mb_y of this thread (so current thread should process start_mb_y <= row < end_mb_y) ...
static av_cold int init(AVCodecContext *avctx)
void(* mss2_gray_fill_masked)(uint8_t *dst, int dst_stride, int maskcolor, const uint8_t *mask, int mask_stride, int w, int h)
int max_b_frames
maximum number of B-frames between non-B-frames Note: The output will be delayed by max_b_frames+1 re...
int extended_mv
Ext MV in P/B (not in Simple)
void ff_er_frame_end(ERContext *s)
static void arith2_rescale_interval(ArithCoder *c, int range, int low, int high, int n)
void(* mss2_blit_wmv9)(uint8_t *dst, int dst_stride, const uint8_t *srcy, int srcy_stride, const uint8_t *srcu, const uint8_t *srcv, int srcuv_stride, int w, int h)
enum AVPixelFormat pix_fmt
Pixel format, see AV_PIX_FMT_xxx.
static av_always_inline void bytestream2_init(GetByteContext *g, const uint8_t *buf, int buf_size)
int fastuvmc
Rounding of qpel vector to hpel ? (not in Simple)
int end_mb_x
Horizontal macroblock limit (used only by mss2)
uint64_t_TMPL AV_WL64 unsigned int_TMPL AV_WL32 unsigned int_TMPL AV_WL24 unsigned int_TMPL AV_WL16 uint64_t_TMPL AV_WB64 unsigned int_TMPL AV_WB32 unsigned int_TMPL AV_WB24 unsigned int_TMPL AV_RB16
int frmrtq_postproc
3bits,
int ff_mss12_decode_rect(SliceContext *sc, ArithCoder *acoder, int x, int y, int width, int height)
#define av_assert0(cond)
assert() equivalent, that is always enabled.
int res_transtab
reserved, always 0
void void avpriv_request_sample(void *avc, const char *msg,...) av_printf_format(2
Log a generic warning message about a missing feature.
AVFrame * av_frame_alloc(void)
Allocate an AVFrame and set its fields to default values.
av_cold int ff_vc1_init_common(VC1Context *v)
Init VC-1 specific tables and VC1Context members.
qpel_mc_func(* qpel_put)[16]
int av_frame_ref(AVFrame *dst, const AVFrame *src)
Set up a new reference to the data described by the source frame.
Picture current_picture
copy of the current picture structure.
int profile
Sequence header data for all Profiles TODO: choose between ints, uint8_ts and monobit flags...
static int get_bits_count(const GetBitContext *s)
int ff_vc1_decode_end(AVCodecContext *avctx)
Close a VC1/WMV3 decoder.
int(* get_model_sym)(struct ArithCoder *c, Model *m)
static int arith2_get_number(ArithCoder *c, int n)
static int arith2_get_prob(ArithCoder *c, int16_t *probs)
static av_cold int wmv9_init(AVCodecContext *avctx)
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
av_cold void ff_mss2dsp_init(MSS2DSPContext *dsp)
int res_y411
reserved, old interlaced mode
int overlap
overlapped transforms in use
static int decode_wmv9(AVCodecContext *avctx, const uint8_t *buf, int buf_size, int x, int y, int w, int h, int wmv9_mask)
void av_frame_free(AVFrame **frame)
Free the frame and any dynamically allocated objects in it, e.g.
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
#define ARITH_GET_MODEL_SYM(prefix)
void(* upsample_plane)(uint8_t *plane, int plane_stride, int w, int h)
simple assert() macros that are a bit more flexible than ISO C assert().
void ff_vc1_decode_blocks(VC1Context *v)
const char * name
Name of the codec implementation.
uint64_t_TMPL AV_WL64 unsigned int_TMPL AV_WL32 unsigned int_TMPL AV_RL24
qpel_mc_func put_qpel_pixels_tab[2][16]
int resync_marker
could this stream contain resync markers
const uint8_t * zz_8x4
Zigzag scan table for TT_8x4 coding mode.
int res_rtm_flag
reserved, set to 1
const uint8_t ff_wmv2_scantableB[64]
static char * split(char *message, char delim)
static void arith2_init(ArithCoder *c, GetByteContext *gB)
void ff_mpeg_flush(AVCodecContext *avctx)
const uint8_t * zz_4x8
Zigzag scan table for TT_4x8 coding mode.
int ff_reget_buffer(AVCodecContext *avctx, AVFrame *frame)
Identical in function to av_frame_make_writable(), except it uses ff_get_buffer() to allocate the buf...
enum AVPictureType pict_type
Picture type of the frame.
int width
picture width / height.
GLsizei GLboolean const GLfloat * value
void ff_mpeg_er_frame_start(MpegEncContext *s)
int ff_vc1_parse_frame_header(VC1Context *v, GetBitContext *gb)
static av_always_inline int get_vlc2(GetBitContext *s, VLC_TYPE(*table)[2], int bits, int max_depth)
Parse a vlc code.
static int decode_555(GetByteContext *gB, uint16_t *dst, int stride, int keyframe, int w, int h)
static int arith2_get_consumed_bytes(ArithCoder *c)
static av_always_inline int bytestream2_tell(GetByteContext *g)
int rangered
RANGEREDFRM (range reduction) syntax element present at frame level.
int format
format of the frame, -1 if unknown or unset Values correspond to enum AVPixelFormat for video frames...
int finterpflag
INTERPFRM present.
av_cold int ff_mss12_decode_end(MSS12Context *c)
int res_sprite
Simple/Main Profile sequence header.
uint64_t_TMPL AV_WL64 unsigned int_TMPL AV_WL32 unsigned int_TMPL AV_WL24 unsigned int_TMPL AV_WL16 uint64_t_TMPL AV_WB64 unsigned int_TMPL AV_WB32 unsigned int_TMPL AV_RB24
int linesize[AV_NUM_DATA_POINTERS]
For video, size in bytes of each picture line.
static int init_get_bits8(GetBitContext *s, const uint8_t *buffer, int byte_size)
Initialize GetBitContext.
int multires
frame-level RESPIC syntax element present
main external API structure.
const uint8_t ff_wmv2_scantableA[64]
static av_cold int mss2_decode_end(AVCodecContext *avctx)
int ff_get_buffer(AVCodecContext *avctx, AVFrame *frame, int flags)
Get a buffer for a frame.
#define init_vlc(vlc, nb_bits, nb_codes,bits, bits_wrap, bits_size,codes, codes_wrap, codes_size,flags)
static unsigned int get_bits1(GetBitContext *s)
static void skip_bits(GetBitContext *s, int n)
uint8_t respic
Frame-level flag for resized images.
int ff_mpv_frame_start(MpegEncContext *s, AVCodecContext *avctx)
generic function called after decoding the header and before a frame is decoded.
int quantizer_mode
2bits, quantizer mode used for sequence, see QUANT_*
int max_b_frames
max number of b-frames for encoding
int pict_type
AV_PICTURE_TYPE_I, AV_PICTURE_TYPE_P, AV_PICTURE_TYPE_B, ...
#define MAX_WMV9_RECTANGLES
int vstransform
variable-size [48]x[48] transform type + info
void av_frame_unref(AVFrame *frame)
Unreference all the buffers referenced by frame and reset the frame fields.
uint8_t * data[AV_NUM_DATA_POINTERS]
pointer to the picture/channel planes.
qpel_mc_func(* qpel_avg)[16]
void ff_vc1_init_transposed_scantables(VC1Context *v)
struct AVCodecContext * avctx
static int decode(AVCodecContext *avctx, void *data, int *got_sub, AVPacket *avpkt)
GLint GLenum GLboolean GLsizei stride
planar YUV 4:2:0, 12bpp, (1 Cr & Cb sample per 2x2 Y samples)
common internal api header.
void ff_mss12_slicecontext_reset(SliceContext *sc)
void(* mss2_blit_wmv9_masked)(uint8_t *dst, int dst_stride, int maskcolor, const uint8_t *mask, int mask_stride, const uint8_t *srcy, int srcy_stride, const uint8_t *srcu, const uint8_t *srcv, int srcuv_stride, int w, int h)
int res_fasttx
reserved, always 1
#define AV_PIX_FMT_RGB555
enum AVDiscard skip_loop_filter
Skip loop filtering for selected frames.
Microsoft Screen 2 (aka Windows Media Video V9 Screen) decoder DSP routines.
#define AV_INPUT_BUFFER_PADDING_SIZE
Required number of additionally allocated bytes at the end of the input bitstream for decoding...
static av_always_inline int diff(const uint32_t a, const uint32_t b)
void ff_mpv_frame_end(MpegEncContext *s)
VLC_TYPE(* table)[2]
code, bits
int key_frame
1 -> keyframe, 0-> not
static const uint8_t * align_get_bits(GetBitContext *s)
int bitrtq_postproc
5bits, quantized framerate-based postprocessing strength
static int decode_pal_v2(MSS12Context *ctx, const uint8_t *buf, int buf_size)
union ArithCoder::@77 gbc
#define ARITH_GET_BIT(prefix)
int ff_vc1_decode_init_alloc_tables(VC1Context *v)
static void arith2_normalise(ArithCoder *c)
static int mss2_decode_frame(AVCodecContext *avctx, void *data, int *got_frame, AVPacket *avpkt)
int dquant
How qscale varies with MBs, 2bits (not in Simple)
av_cold void ff_qpeldsp_init(QpelDSPContext *c)
#define FFSWAP(type, a, b)
int(* get_number)(struct ArithCoder *c, int n)
This structure stores compressed data.
void ff_free_vlc(VLC *vlc)
#define AV_GET_BUFFER_FLAG_REF
The decoder will keep a reference to the frame and may reuse it later.
av_cold void ff_vc1dsp_init(VC1DSPContext *dsp)
void * av_mallocz(size_t size)
Allocate a block of size bytes with alignment suitable for all memory accesses (including vectors if ...
#define AV_CODEC_CAP_DR1
Codec uses get_buffer() for allocating buffers and supports custom allocators.
Common header for Microsoft Screen 1 and 2.