macroblock.h
资源名称:chapter15.rar [点击查看]
上传用户:hjq518
上传日期:2021-12-09
资源大小:5084k
文件大小:10k
源码类别:
Audio
开发平台:
Visual C++
- /*****************************************************************************
- * macroblock.h: h264 encoder library
- *****************************************************************************
- * Copyright (C) 2003 Laurent Aimar
- * $Id: macroblock.h,v 1.1 2004/06/03 19:27:07 fenrir Exp $
- *
- * Authors: Laurent Aimar <fenrir@via.ecp.fr>
- *
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- * GNU General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111, USA.
- *****************************************************************************/
- #ifndef _MACROBLOCK_H
- #define _MACROBLOCK_H 1
- enum macroblock_position_e
- {
- MB_LEFT = 0x01,
- MB_TOP = 0x02,
- MB_TOPRIGHT = 0x04,
- MB_TOPLEFT = 0x08,
- MB_PRIVATE = 0x10,
- };
- /* XXX mb_type isn't the one written in the bitstream -> only internal usage */
- #define IS_INTRA(type) ( (type) == I_4x4 || (type) == I_8x8 || (type) == I_16x16 )
- #define IS_SKIP(type) ( (type) == P_SKIP || (type) == B_SKIP )
- #define IS_DIRECT(type) ( (type) == B_DIRECT )
- enum mb_class_e
- {
- I_4x4 = 0,
- I_8x8 = 1,
- I_16x16 = 2,
- I_PCM = 3,
- P_L0 = 4,
- P_8x8 = 5,
- P_SKIP = 6,
- B_DIRECT = 7,
- B_L0_L0 = 8,
- B_L0_L1 = 9,
- B_L0_BI = 10,
- B_L1_L0 = 11,
- B_L1_L1 = 12,
- B_L1_BI = 13,
- B_BI_L0 = 14,
- B_BI_L1 = 15,
- B_BI_BI = 16,
- B_8x8 = 17,
- B_SKIP = 18,
- };
- static const int x264_mb_type_fix[19] =
- {
- I_4x4, I_4x4, I_16x16, I_PCM,
- P_L0, P_8x8, P_SKIP,
- B_DIRECT, B_L0_L0, B_L0_L1, B_L0_BI, B_L1_L0, B_L1_L1,
- B_L1_BI, B_BI_L0, B_BI_L1, B_BI_BI, B_8x8, B_SKIP
- };
- static const int x264_mb_type_list0_table[19][2] =
- {
- {0,0}, {0,0}, {0,0}, {0,0}, /* INTRA */
- {1,1}, /* P_L0 */
- {0,0}, /* P_8x8 */
- {1,1}, /* P_SKIP */
- {0,0}, /* B_DIRECT */
- {1,1}, {1,0}, {1,1}, /* B_L0_* */
- {0,1}, {0,0}, {0,1}, /* B_L1_* */
- {1,1}, {1,0}, {1,1}, /* B_BI_* */
- {0,0}, /* B_8x8 */
- {0,0} /* B_SKIP */
- };
- static const int x264_mb_type_list1_table[19][2] =
- {
- {0,0}, {0,0}, {0,0}, {0,0}, /* INTRA */
- {0,0}, /* P_L0 */
- {0,0}, /* P_8x8 */
- {0,0}, /* P_SKIP */
- {0,0}, /* B_DIRECT */
- {0,0}, {0,1}, {0,1}, /* B_L0_* */
- {1,0}, {1,1}, {1,1}, /* B_L1_* */
- {1,0}, {1,1}, {1,1}, /* B_BI_* */
- {0,0}, /* B_8x8 */
- {0,0} /* B_SKIP */
- };
- #define IS_SUB4x4(type) ( (type ==D_L0_4x4)||(type ==D_L1_4x4)||(type ==D_BI_4x4))
- #define IS_SUB4x8(type) ( (type ==D_L0_4x8)||(type ==D_L1_4x8)||(type ==D_BI_4x8))
- #define IS_SUB8x4(type) ( (type ==D_L0_8x4)||(type ==D_L1_8x4)||(type ==D_BI_8x4))
- #define IS_SUB8x8(type) ( (type ==D_L0_8x8)||(type ==D_L1_8x8)||(type ==D_BI_8x8)||(type ==D_DIRECT_8x8))
- enum mb_partition_e
- {
- /* sub partition type for P_8x8 and B_8x8 */
- D_L0_4x4 = 0,
- D_L0_8x4 = 1,
- D_L0_4x8 = 2,
- D_L0_8x8 = 3,
- /* sub partition type for B_8x8 only */
- D_L1_4x4 = 4,
- D_L1_8x4 = 5,
- D_L1_4x8 = 6,
- D_L1_8x8 = 7,
- D_BI_4x4 = 8,
- D_BI_8x4 = 9,
- D_BI_4x8 = 10,
- D_BI_8x8 = 11,
- D_DIRECT_8x8 = 12,
- /* partition */
- D_8x8 = 13,
- D_16x8 = 14,
- D_8x16 = 15,
- D_16x16 = 16,
- };
- static const int x264_mb_partition_listX_table[2][17] =
- {{
- 1, 1, 1, 1, /* D_L0_* */
- 0, 0, 0, 0, /* D_L1_* */
- 1, 1, 1, 1, /* D_BI_* */
- 0, /* D_DIRECT_8x8 */
- 0, 0, 0, 0 /* 8x8 .. 16x16 */
- },
- {
- 0, 0, 0, 0, /* D_L0_* */
- 1, 1, 1, 1, /* D_L1_* */
- 1, 1, 1, 1, /* D_BI_* */
- 0, /* D_DIRECT_8x8 */
- 0, 0, 0, 0 /* 8x8 .. 16x16 */
- }};
- static const int x264_mb_partition_count_table[17] =
- {
- /* sub L0 */
- 4, 2, 2, 1,
- /* sub L1 */
- 4, 2, 2, 1,
- /* sub BI */
- 4, 2, 2, 1,
- /* Direct */
- 1,
- /* Partition */
- 4, 2, 2, 1
- };
- static const int x264_mb_partition_pixel_table[17] =
- {
- 6, 4, 5, 3, 6, 4, 5, 3, 6, 4, 5, 3, 3, 3, 1, 2, 0
- };
- static const int x264_zigzag_scan4[16] =
- {
- 0, 1, 4, 8, 5, 2, 3, 6, 9, 12, 13, 10, 7, 11, 14, 15
- };
- static const int x264_zigzag_scan8[64] =
- {
- 0, 1, 8, 16, 9, 2, 3, 10, 17, 24, 32, 25, 18, 11, 4, 5,
- 12, 19, 26, 33, 40, 48, 41, 34, 27, 20, 13, 6, 7, 14, 21, 28,
- 35, 42, 49, 56, 57, 50, 43, 36, 29, 22, 15, 23, 30, 37, 44, 51,
- 58, 59, 52, 45, 38, 31, 39, 46, 53, 60, 61, 54, 47, 55, 62, 63
- };
- static const uint8_t block_idx_x[16] =
- {
- 0, 1, 0, 1, 2, 3, 2, 3, 0, 1, 0, 1, 2, 3, 2, 3
- };
- static const uint8_t block_idx_y[16] =
- {
- 0, 0, 1, 1, 0, 0, 1, 1, 2, 2, 3, 3, 2, 2, 3, 3
- };
- static const uint8_t block_idx_xy[4][4] =
- {
- { 0, 2, 8, 10 },
- { 1, 3, 9, 11 },
- { 4, 6, 12, 14 },
- { 5, 7, 13, 15 }
- };
- static const int i_chroma_qp_table[52] =
- {
- 0, 1, 2, 3, 4, 5, 6, 7, 8, 9,
- 10, 11, 12, 13, 14, 15, 16, 17, 18, 19,
- 20, 21, 22, 23, 24, 25, 26, 27, 28, 29,
- 29, 30, 31, 32, 32, 33, 34, 34, 35, 35,
- 36, 36, 37, 37, 37, 38, 38, 38, 39, 39,
- 39, 39
- };
- enum cabac_ctx_block_cat_e
- {
- DCT_LUMA_DC = 0,
- DCT_LUMA_AC = 1,
- DCT_LUMA_4x4 = 2,
- DCT_CHROMA_DC = 3,
- DCT_CHROMA_AC = 4,
- DCT_LUMA_8x8 = 5,
- };
- void x264_macroblock_cache_init( x264_t *h );
- void x264_macroblock_slice_init( x264_t *h );
- void x264_macroblock_cache_load( x264_t *h, int i_mb_x, int i_mb_y );
- void x264_macroblock_cache_save( x264_t *h );
- void x264_macroblock_cache_end( x264_t *h );
- void x264_macroblock_bipred_init( x264_t *h );
- /* x264_mb_predict_mv_16x16:
- * set mvp with predicted mv for D_16x16 block
- * h->mb. need only valid values from other blocks */
- void x264_mb_predict_mv_16x16( x264_t *h, int i_list, int i_ref, int mvp[2] );
- /* x264_mb_predict_mv_pskip:
- * set mvp with predicted mv for P_SKIP
- * h->mb. need only valid values from other blocks */
- void x264_mb_predict_mv_pskip( x264_t *h, int mv[2] );
- /* x264_mb_predict_mv:
- * set mvp with predicted mv for all blocks except SKIP and DIRECT
- * h->mb. need valid ref/partition/sub of current block to be valid
- * and valid mv/ref from other blocks. */
- void x264_mb_predict_mv( x264_t *h, int i_list, int idx, int i_width, int mvp[2] );
- /* x264_mb_predict_mv_direct16x16:
- * set h->mb.cache.mv and h->mb.cache.ref for B_SKIP or B_DIRECT
- * h->mb. need only valid values from other blocks.
- * return 1 on success, 0 on failure.
- * if b_changed != NULL, set it to whether refs or mvs differ from
- * before this functioncall. */
- int x264_mb_predict_mv_direct16x16( x264_t *h, int *b_changed );
- /* x264_mb_load_mv_direct8x8:
- * set h->mb.cache.mv and h->mb.cache.ref for B_DIRECT
- * must be called only after x264_mb_predict_mv_direct16x16 */
- void x264_mb_load_mv_direct8x8( x264_t *h, int idx );
- /* x264_mb_predict_mv_ref16x16:
- * set mvc with D_16x16 prediction.
- * uses all neighbors, even those that didn't end up using this ref.
- * h->mb. need only valid values from other blocks */
- void x264_mb_predict_mv_ref16x16( x264_t *h, int i_list, int i_ref, int mvc[8][2], int *i_mvc );
- int x264_mb_predict_intra4x4_mode( x264_t *h, int idx );
- int x264_mb_predict_non_zero_code( x264_t *h, int idx );
- /* x264_mb_transform_8x8_allowed:
- * check whether any partition is smaller than 8x8 (or at least
- * might be, according to just partition type.)
- * doesn't check for intra or cbp */
- int x264_mb_transform_8x8_allowed( x264_t *h );
- void x264_mb_encode_i4x4( x264_t *h, int idx, int i_qscale );
- void x264_mb_encode_i8x8( x264_t *h, int idx, int i_qscale );
- void x264_mb_mc( x264_t *h );
- static inline void x264_macroblock_cache_ref( x264_t *h, int x, int y, int width, int height, int i_list, int ref )
- {
- int dy, dx;
- for( dy = 0; dy < height; dy++ )
- {
- for( dx = 0; dx < width; dx++ )
- {
- h->mb.cache.ref[i_list][X264_SCAN8_0+x+dx+8*(y+dy)] = ref;
- }
- }
- }
- static inline void x264_macroblock_cache_mv( x264_t *h, int x, int y, int width, int height, int i_list, int mvx, int mvy )
- {
- int dy, dx;
- for( dy = 0; dy < height; dy++ )
- {
- for( dx = 0; dx < width; dx++ )
- {
- h->mb.cache.mv[i_list][X264_SCAN8_0+x+dx+8*(y+dy)][0] = mvx;
- h->mb.cache.mv[i_list][X264_SCAN8_0+x+dx+8*(y+dy)][1] = mvy;
- }
- }
- }
- static inline void x264_macroblock_cache_mvd( x264_t *h, int x, int y, int width, int height, int i_list, int mdx, int mdy )
- {
- int dy, dx;
- for( dy = 0; dy < height; dy++ )
- {
- for( dx = 0; dx < width; dx++ )
- {
- h->mb.cache.mvd[i_list][X264_SCAN8_0+x+dx+8*(y+dy)][0] = mdx;
- h->mb.cache.mvd[i_list][X264_SCAN8_0+x+dx+8*(y+dy)][1] = mdy;
- }
- }
- }
- static inline void x264_macroblock_cache_skip( x264_t *h, int x, int y, int width, int height, int b_skip )
- {
- int dy, dx;
- for( dy = 0; dy < height; dy++ )
- {
- for( dx = 0; dx < width; dx++ )
- {
- h->mb.cache.skip[X264_SCAN8_0+x+dx+8*(y+dy)] = b_skip;
- }
- }
- }
- static inline void x264_macroblock_cache_intra8x8_pred( x264_t *h, int x, int y, int i_mode )
- {
- int *cache = &h->mb.cache.intra4x4_pred_mode[X264_SCAN8_0+x+8*y];
- cache[0] = cache[1] = cache[8] = cache[9] = i_mode;
- }
- #endif