1 /*
2 * Copyright (c) 2002 The FFmpeg Project
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 #ifndef AVCODEC_WMV2_H
22 #define AVCODEC_WMV2_H
23
24 #include "libavutil/mem_internal.h"
25
26 #include "avcodec.h"
27 #include "intrax8.h"
28 #include "mpegvideo.h"
29 #include "wmv2dsp.h"
30
31 #define SKIP_TYPE_NONE 0
32 #define SKIP_TYPE_MPEG 1
33 #define SKIP_TYPE_ROW 2
34 #define SKIP_TYPE_COL 3
35
36
37 typedef struct Wmv2Context {
38 MpegEncContext s;
39 IntraX8Context x8;
40 WMV2DSPContext wdsp;
41 int j_type_bit;
42 int j_type;
43 int abt_flag;
44 int abt_type;
45 int abt_type_table[6];
46 int per_mb_abt;
47 int per_block_abt;
48 int mspel_bit;
49 int cbp_table_index;
50 int top_left_mv_flag;
51 int per_mb_rl_bit;
52 int skip_type;
53 int hshift;
54
55 ScanTable abt_scantable[2];
56 DECLARE_ALIGNED(32, int16_t, abt_block2)[6][64];
57 } Wmv2Context;
58
59 void ff_wmv2_common_init(Wmv2Context *w);
60
61 int ff_wmv2_decode_mb(MpegEncContext *s, int16_t block[6][64]);
62 int ff_wmv2_encode_picture_header(MpegEncContext * s, int picture_number);
63 void ff_wmv2_encode_mb(MpegEncContext * s, int16_t block[6][64],
64 int motion_x, int motion_y);
65 int ff_wmv2_decode_picture_header(MpegEncContext * s);
66 int ff_wmv2_decode_secondary_picture_header(MpegEncContext * s);
67 void ff_wmv2_add_mb(MpegEncContext *s, int16_t block[6][64],
68 uint8_t *dest_y, uint8_t *dest_cb, uint8_t *dest_cr);
69
70 void ff_mspel_motion(MpegEncContext *s,
71 uint8_t *dest_y, uint8_t *dest_cb, uint8_t *dest_cr,
72 uint8_t **ref_picture, op_pixels_func (*pix_op)[4],
73 int motion_x, int motion_y, int h);
74
75
wmv2_get_cbp_table_index(MpegEncContext * s,int cbp_index)76 static av_always_inline int wmv2_get_cbp_table_index(MpegEncContext *s, int cbp_index)
77 {
78 static const uint8_t map[3][3] = {
79 { 0, 2, 1 },
80 { 1, 0, 2 },
81 { 2, 1, 0 },
82 };
83
84 return map[(s->qscale > 10) + (s->qscale > 20)][cbp_index];
85 }
86
87 #endif /* AVCODEC_WMV2_H */
88