yading@10: /* yading@10: * MPEG4 encoder/decoder internal header. yading@10: * Copyright (c) 2000,2001 Fabrice Bellard yading@10: * Copyright (c) 2002-2010 Michael Niedermayer yading@10: * yading@10: * This file is part of FFmpeg. yading@10: * yading@10: * FFmpeg is free software; you can redistribute it and/or yading@10: * modify it under the terms of the GNU Lesser General Public yading@10: * License as published by the Free Software Foundation; either yading@10: * version 2.1 of the License, or (at your option) any later version. yading@10: * yading@10: * FFmpeg is distributed in the hope that it will be useful, yading@10: * but WITHOUT ANY WARRANTY; without even the implied warranty of yading@10: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU yading@10: * Lesser General Public License for more details. yading@10: * yading@10: * You should have received a copy of the GNU Lesser General Public yading@10: * License along with FFmpeg; if not, write to the Free Software yading@10: * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA yading@10: */ yading@10: yading@10: #ifndef AVCODEC_MPEG4VIDEO_H yading@10: #define AVCODEC_MPEG4VIDEO_H yading@10: yading@10: #include yading@10: #include "get_bits.h" yading@10: #include "mpegvideo.h" yading@10: #include "rl.h" yading@10: yading@10: // shapes yading@10: #define RECT_SHAPE 0 yading@10: #define BIN_SHAPE 1 yading@10: #define BIN_ONLY_SHAPE 2 yading@10: #define GRAY_SHAPE 3 yading@10: yading@10: #define SIMPLE_VO_TYPE 1 yading@10: #define CORE_VO_TYPE 3 yading@10: #define MAIN_VO_TYPE 4 yading@10: #define NBIT_VO_TYPE 5 yading@10: #define ARTS_VO_TYPE 10 yading@10: #define ACE_VO_TYPE 12 yading@10: #define ADV_SIMPLE_VO_TYPE 17 yading@10: yading@10: // aspect_ratio_info yading@10: #define EXTENDED_PAR 15 yading@10: yading@10: //vol_sprite_usage / sprite_enable yading@10: #define STATIC_SPRITE 1 yading@10: #define GMC_SPRITE 2 yading@10: yading@10: #define MOTION_MARKER 0x1F001 yading@10: #define DC_MARKER 0x6B001 yading@10: yading@10: #define VOS_STARTCODE 0x1B0 yading@10: #define USER_DATA_STARTCODE 0x1B2 yading@10: #define GOP_STARTCODE 0x1B3 yading@10: #define VISUAL_OBJ_STARTCODE 0x1B5 yading@10: #define VOP_STARTCODE 0x1B6 yading@10: yading@10: /* dc encoding for mpeg4 */ yading@10: extern const uint8_t ff_mpeg4_DCtab_lum[13][2]; yading@10: extern const uint8_t ff_mpeg4_DCtab_chrom[13][2]; yading@10: yading@10: extern const uint16_t ff_mpeg4_intra_vlc[103][2]; yading@10: extern RLTable ff_mpeg4_rl_intra; yading@10: yading@10: /* Note this is identical to the intra rvlc except that it is reordered. */ yading@10: extern RLTable ff_rvlc_rl_inter; yading@10: extern RLTable ff_rvlc_rl_intra; yading@10: yading@10: extern const uint16_t ff_sprite_trajectory_tab[15][2]; yading@10: extern const uint8_t ff_mb_type_b_tab[4][2]; yading@10: yading@10: /* these matrixes will be permuted for the idct */ yading@10: extern const int16_t ff_mpeg4_default_intra_matrix[64]; yading@10: extern const int16_t ff_mpeg4_default_non_intra_matrix[64]; yading@10: yading@10: extern const uint8_t ff_mpeg4_y_dc_scale_table[32]; yading@10: extern const uint8_t ff_mpeg4_c_dc_scale_table[32]; yading@10: extern const uint16_t ff_mpeg4_resync_prefix[8]; yading@10: yading@10: extern const uint8_t ff_mpeg4_dc_threshold[8]; yading@10: yading@10: void ff_mpeg4_encode_mb(MpegEncContext *s, yading@10: int16_t block[6][64], yading@10: int motion_x, int motion_y); yading@10: void ff_mpeg4_pred_ac(MpegEncContext * s, int16_t *block, int n, yading@10: int dir); yading@10: void ff_set_mpeg4_time(MpegEncContext * s); yading@10: void ff_mpeg4_encode_picture_header(MpegEncContext *s, int picture_number); yading@10: yading@10: int ff_mpeg4_decode_picture_header(MpegEncContext * s, GetBitContext *gb); yading@10: void ff_mpeg4_encode_video_packet_header(MpegEncContext *s); yading@10: void ff_mpeg4_clean_buffers(MpegEncContext *s); yading@10: void ff_mpeg4_stuffing(PutBitContext * pbc); yading@10: void ff_mpeg4_init_partitions(MpegEncContext *s); yading@10: void ff_mpeg4_merge_partitions(MpegEncContext *s); yading@10: void ff_clean_mpeg4_qscales(MpegEncContext *s); yading@10: int ff_mpeg4_decode_partitions(MpegEncContext *s); yading@10: int ff_mpeg4_get_video_packet_prefix_length(MpegEncContext *s); yading@10: int ff_mpeg4_decode_video_packet_header(MpegEncContext *s); yading@10: void ff_mpeg4_init_direct_mv(MpegEncContext *s); yading@10: void ff_mpeg4videodec_static_init(void); yading@10: yading@10: /** yading@10: * yading@10: * @return the mb_type yading@10: */ yading@10: int ff_mpeg4_set_direct_mv(MpegEncContext *s, int mx, int my); yading@10: yading@10: extern uint8_t ff_mpeg4_static_rl_table_store[3][2][2*MAX_RUN + MAX_LEVEL + 3]; yading@10: yading@10: yading@10: #if 0 //3IV1 is quite rare and it slows things down a tiny bit yading@10: #define IS_3IV1 s->codec_tag == AV_RL32("3IV1") yading@10: #else yading@10: #define IS_3IV1 0 yading@10: #endif yading@10: yading@10: yading@10: /** yading@10: * Predict the dc. yading@10: * encoding quantized level -> quantized diff yading@10: * decoding quantized diff -> quantized level yading@10: * @param n block index (0-3 are luma, 4-5 are chroma) yading@10: * @param dir_ptr pointer to an integer where the prediction direction will be stored yading@10: */ yading@10: static inline int ff_mpeg4_pred_dc(MpegEncContext * s, int n, int level, int *dir_ptr, int encoding) yading@10: { yading@10: int a, b, c, wrap, pred, scale, ret; yading@10: int16_t *dc_val; yading@10: yading@10: /* find prediction */ yading@10: if (n < 4) { yading@10: scale = s->y_dc_scale; yading@10: } else { yading@10: scale = s->c_dc_scale; yading@10: } yading@10: if(IS_3IV1) yading@10: scale= 8; yading@10: yading@10: wrap= s->block_wrap[n]; yading@10: dc_val = s->dc_val[0] + s->block_index[n]; yading@10: yading@10: /* B C yading@10: * A X yading@10: */ yading@10: a = dc_val[ - 1]; yading@10: b = dc_val[ - 1 - wrap]; yading@10: c = dc_val[ - wrap]; yading@10: yading@10: /* outside slice handling (we can't do that by memset as we need the dc for error resilience) */ yading@10: if(s->first_slice_line && n!=3){ yading@10: if(n!=2) b=c= 1024; yading@10: if(n!=1 && s->mb_x == s->resync_mb_x) b=a= 1024; yading@10: } yading@10: if(s->mb_x == s->resync_mb_x && s->mb_y == s->resync_mb_y+1){ yading@10: if(n==0 || n==4 || n==5) yading@10: b=1024; yading@10: } yading@10: yading@10: if (abs(a - b) < abs(b - c)) { yading@10: pred = c; yading@10: *dir_ptr = 1; /* top */ yading@10: } else { yading@10: pred = a; yading@10: *dir_ptr = 0; /* left */ yading@10: } yading@10: /* we assume pred is positive */ yading@10: pred = FASTDIV((pred + (scale >> 1)), scale); yading@10: yading@10: if(encoding){ yading@10: ret = level - pred; yading@10: }else{ yading@10: level += pred; yading@10: ret= level; yading@10: if(s->err_recognition&(AV_EF_BITSTREAM|AV_EF_AGGRESSIVE)){ yading@10: if(level<0){ yading@10: av_log(s->avctx, AV_LOG_ERROR, "dc<0 at %dx%d\n", s->mb_x, s->mb_y); yading@10: return -1; yading@10: } yading@10: if(level*scale > 2048 + scale){ yading@10: av_log(s->avctx, AV_LOG_ERROR, "dc overflow at %dx%d\n", s->mb_x, s->mb_y); yading@10: return -1; yading@10: } yading@10: } yading@10: } yading@10: level *=scale; yading@10: if(level&(~2047)){ yading@10: if(level<0) yading@10: level=0; yading@10: else if(!(s->workaround_bugs&FF_BUG_DC_CLIP)) yading@10: level=2047; yading@10: } yading@10: dc_val[0]= level; yading@10: yading@10: return ret; yading@10: } yading@10: #endif /* AVCODEC_MPEG4VIDEO_H */