Split the mpeg4 encoder and decoder off h263.c
[libav.git] / libavcodec / mpeg4video.h
CommitLineData
ca334dd1
MN
1/*
2 * MPEG4 encoder/decoder internal header.
3 * Copyright (c) 2000,2001 Fabrice Bellard
4 * Copyright (c) 2002-2010 Michael Niedermayer <michaelni@gmx.at>
5 *
6 * This file is part of FFmpeg.
7 *
8 * FFmpeg is free software; you can redistribute it and/or
9 * modify it under the terms of the GNU Lesser General Public
10 * License as published by the Free Software Foundation; either
11 * version 2.1 of the License, or (at your option) any later version.
12 *
13 * FFmpeg is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 * Lesser General Public License for more details.
17 *
18 * You should have received a copy of the GNU Lesser General Public
19 * License along with FFmpeg; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21 */
22
23#ifndef AVCODEC_MPEG4VIDEO_H
24#define AVCODEC_MPEG4VIDEO_H
25
26// shapes
27#define RECT_SHAPE 0
28#define BIN_SHAPE 1
29#define BIN_ONLY_SHAPE 2
30#define GRAY_SHAPE 3
31
32#define SIMPLE_VO_TYPE 1
33#define CORE_VO_TYPE 3
34#define MAIN_VO_TYPE 4
35#define NBIT_VO_TYPE 5
36#define ARTS_VO_TYPE 10
37#define ACE_VO_TYPE 12
38#define ADV_SIMPLE_VO_TYPE 17
39
40// aspect_ratio_info
41#define EXTENDED_PAR 15
42
43//vol_sprite_usage / sprite_enable
44#define STATIC_SPRITE 1
45#define GMC_SPRITE 2
46
47#define MOTION_MARKER 0x1F001
48#define DC_MARKER 0x6B001
49
50#define VOS_STARTCODE 0x1B0
51#define USER_DATA_STARTCODE 0x1B2
52#define GOP_STARTCODE 0x1B3
53#define VISUAL_OBJ_STARTCODE 0x1B5
54#define VOP_STARTCODE 0x1B6
55
56/* dc encoding for mpeg4 */
57extern const uint8_t DCtab_lum[13][2];
58extern const uint8_t DCtab_chrom[13][2];
59
60extern const uint16_t intra_vlc[103][2];
61extern const int8_t intra_level[102];
62extern const int8_t intra_run[102];
63extern RLTable rl_intra;
64
65/* Note this is identical to the intra rvlc except that it is reordered. */
66extern const uint16_t inter_rvlc[170][2];
67extern const int8_t inter_rvlc_run[169];
68extern const int8_t inter_rvlc_level[169];
69extern RLTable rvlc_rl_inter;
70
71extern const uint16_t intra_rvlc[170][2];
72extern const int8_t intra_rvlc_run[169];
73extern const int8_t intra_rvlc_level[169];
74extern RLTable rvlc_rl_intra;
75
76extern const uint16_t sprite_trajectory_tab[15][2];
77extern const uint8_t mb_type_b_tab[4][2];
78extern const AVRational pixel_aspect[16];
79
80/* these matrixes will be permuted for the idct */
81extern const int16_t ff_mpeg4_default_intra_matrix[64];
82extern const int16_t ff_mpeg4_default_non_intra_matrix[64];
83
84extern const uint8_t ff_mpeg4_y_dc_scale_table[32];
85extern const uint8_t ff_mpeg4_c_dc_scale_table[32];
86extern const uint16_t ff_mpeg4_resync_prefix[8];
87
88extern const uint8_t mpeg4_dc_threshold[8];
89
90void mpeg4_encode_mb(MpegEncContext *s,
91 DCTELEM block[6][64],
92 int motion_x, int motion_y);
93void mpeg4_pred_ac(MpegEncContext * s, DCTELEM *block, int n,
94 int dir);
95void ff_set_mpeg4_time(MpegEncContext * s);
96void mpeg4_encode_picture_header(MpegEncContext *s, int picture_number);
97
98int ff_mpeg4_decode_picture_header(MpegEncContext * s, GetBitContext *gb);
99void ff_mpeg4_encode_video_packet_header(MpegEncContext *s);
100void ff_mpeg4_clean_buffers(MpegEncContext *s);
101void ff_mpeg4_stuffing(PutBitContext * pbc);
102void ff_mpeg4_init_partitions(MpegEncContext *s);
103void ff_mpeg4_merge_partitions(MpegEncContext *s);
104void ff_clean_mpeg4_qscales(MpegEncContext *s);
105int ff_mpeg4_decode_partitions(MpegEncContext *s);
106int ff_mpeg4_get_video_packet_prefix_length(MpegEncContext *s);
107int mpeg4_decode_video_packet_header(MpegEncContext *s);
108
109extern uint8_t ff_mpeg4_static_rl_table_store[3][2][2*MAX_RUN + MAX_LEVEL + 3];
110
111
112#if 0 //3IV1 is quite rare and it slows things down a tiny bit
113#define IS_3IV1 s->codec_tag == AV_RL32("3IV1")
114#else
115#define IS_3IV1 0
116#endif
117
118
119/**
120 * predicts the dc.
121 * encoding quantized level -> quantized diff
122 * decoding quantized diff -> quantized level
123 * @param n block index (0-3 are luma, 4-5 are chroma)
124 * @param dir_ptr pointer to an integer where the prediction direction will be stored
125 */
126static inline int ff_mpeg4_pred_dc(MpegEncContext * s, int n, int level, int *dir_ptr, int encoding)
127{
128 int a, b, c, wrap, pred, scale, ret;
129 int16_t *dc_val;
130
131 /* find prediction */
132 if (n < 4) {
133 scale = s->y_dc_scale;
134 } else {
135 scale = s->c_dc_scale;
136 }
137 if(IS_3IV1)
138 scale= 8;
139
140 wrap= s->block_wrap[n];
141 dc_val = s->dc_val[0] + s->block_index[n];
142
143 /* B C
144 * A X
145 */
146 a = dc_val[ - 1];
147 b = dc_val[ - 1 - wrap];
148 c = dc_val[ - wrap];
149
150 /* outside slice handling (we can't do that by memset as we need the dc for error resilience) */
151 if(s->first_slice_line && n!=3){
152 if(n!=2) b=c= 1024;
153 if(n!=1 && s->mb_x == s->resync_mb_x) b=a= 1024;
154 }
155 if(s->mb_x == s->resync_mb_x && s->mb_y == s->resync_mb_y+1){
156 if(n==0 || n==4 || n==5)
157 b=1024;
158 }
159
160 if (abs(a - b) < abs(b - c)) {
161 pred = c;
162 *dir_ptr = 1; /* top */
163 } else {
164 pred = a;
165 *dir_ptr = 0; /* left */
166 }
167 /* we assume pred is positive */
168 pred = FASTDIV((pred + (scale >> 1)), scale);
169
170 if(encoding){
171 ret = level - pred;
172 }else{
173 level += pred;
174 ret= level;
175 if(s->error_recognition>=3){
176 if(level<0){
177 av_log(s->avctx, AV_LOG_ERROR, "dc<0 at %dx%d\n", s->mb_x, s->mb_y);
178 return -1;
179 }
180 if(level*scale > 2048 + scale){
181 av_log(s->avctx, AV_LOG_ERROR, "dc overflow at %dx%d\n", s->mb_x, s->mb_y);
182 return -1;
183 }
184 }
185 }
186 level *=scale;
187 if(level&(~2047)){
188 if(level<0)
189 level=0;
190 else if(!(s->workaround_bugs&FF_BUG_DC_CLIP))
191 level=2047;
192 }
193 dc_val[0]= level;
194
195 return ret;
196}
197#endif