macroblock.h
上传用户:hjq518
上传日期:2021-12-09
资源大小:5084k
文件大小:10k
源码类别:

Audio

开发平台:

Visual C++

  1. /*****************************************************************************
  2.  * macroblock.h: h264 encoder library
  3.  *****************************************************************************
  4.  * Copyright (C) 2003 Laurent Aimar
  5.  * $Id: macroblock.h,v 1.1 2004/06/03 19:27:07 fenrir Exp $
  6.  *
  7.  * Authors: Laurent Aimar <fenrir@via.ecp.fr>
  8.  *
  9.  * This program is free software; you can redistribute it and/or modify
  10.  * it under the terms of the GNU General Public License as published by
  11.  * the Free Software Foundation; either version 2 of the License, or
  12.  * (at your option) any later version.
  13.  *
  14.  * This program is distributed in the hope that it will be useful,
  15.  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17.  * GNU General Public License for more details.
  18.  *
  19.  * You should have received a copy of the GNU General Public License
  20.  * along with this program; if not, write to the Free Software
  21.  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
  22.  *****************************************************************************/
  23. #ifndef _MACROBLOCK_H
  24. #define _MACROBLOCK_H 1
  25. enum macroblock_position_e
  26. {
  27.     MB_LEFT     = 0x01,
  28.     MB_TOP      = 0x02,
  29.     MB_TOPRIGHT = 0x04,
  30.     MB_TOPLEFT  = 0x08,
  31.     MB_PRIVATE  = 0x10,
  32. };
  33. /* XXX mb_type isn't the one written in the bitstream -> only internal usage */
  34. #define IS_INTRA(type) ( (type) == I_4x4 || (type) == I_8x8 || (type) == I_16x16 )
  35. #define IS_SKIP(type)  ( (type) == P_SKIP || (type) == B_SKIP )
  36. #define IS_DIRECT(type)  ( (type) == B_DIRECT )
  37. enum mb_class_e
  38. {
  39.     I_4x4           = 0,
  40.     I_8x8           = 1,
  41.     I_16x16         = 2,
  42.     I_PCM           = 3,
  43.     P_L0            = 4,
  44.     P_8x8           = 5,
  45.     P_SKIP          = 6,
  46.     B_DIRECT        = 7,
  47.     B_L0_L0         = 8,
  48.     B_L0_L1         = 9,
  49.     B_L0_BI         = 10,
  50.     B_L1_L0         = 11,
  51.     B_L1_L1         = 12,
  52.     B_L1_BI         = 13,
  53.     B_BI_L0         = 14,
  54.     B_BI_L1         = 15,
  55.     B_BI_BI         = 16,
  56.     B_8x8           = 17,
  57.     B_SKIP          = 18,
  58. };
  59. static const int x264_mb_type_fix[19] =
  60. {
  61.     I_4x4, I_4x4, I_16x16, I_PCM,
  62.     P_L0, P_8x8, P_SKIP,
  63.     B_DIRECT, B_L0_L0, B_L0_L1, B_L0_BI, B_L1_L0, B_L1_L1,
  64.     B_L1_BI, B_BI_L0, B_BI_L1, B_BI_BI, B_8x8, B_SKIP
  65. };
  66. static const int x264_mb_type_list0_table[19][2] =
  67. {
  68.     {0,0}, {0,0}, {0,0}, {0,0}, /* INTRA */
  69.     {1,1},                  /* P_L0 */
  70.     {0,0},                  /* P_8x8 */
  71.     {1,1},                  /* P_SKIP */
  72.     {0,0},                  /* B_DIRECT */
  73.     {1,1}, {1,0}, {1,1},    /* B_L0_* */
  74.     {0,1}, {0,0}, {0,1},    /* B_L1_* */
  75.     {1,1}, {1,0}, {1,1},    /* B_BI_* */
  76.     {0,0},                  /* B_8x8 */
  77.     {0,0}                   /* B_SKIP */
  78. };
  79. static const int x264_mb_type_list1_table[19][2] =
  80. {
  81.     {0,0}, {0,0}, {0,0}, {0,0}, /* INTRA */
  82.     {0,0},                  /* P_L0 */
  83.     {0,0},                  /* P_8x8 */
  84.     {0,0},                  /* P_SKIP */
  85.     {0,0},                  /* B_DIRECT */
  86.     {0,0}, {0,1}, {0,1},    /* B_L0_* */
  87.     {1,0}, {1,1}, {1,1},    /* B_L1_* */
  88.     {1,0}, {1,1}, {1,1},    /* B_BI_* */
  89.     {0,0},                  /* B_8x8 */
  90.     {0,0}                   /* B_SKIP */
  91. };
  92. #define IS_SUB4x4(type) ( (type ==D_L0_4x4)||(type ==D_L1_4x4)||(type ==D_BI_4x4))
  93. #define IS_SUB4x8(type) ( (type ==D_L0_4x8)||(type ==D_L1_4x8)||(type ==D_BI_4x8))
  94. #define IS_SUB8x4(type) ( (type ==D_L0_8x4)||(type ==D_L1_8x4)||(type ==D_BI_8x4))
  95. #define IS_SUB8x8(type) ( (type ==D_L0_8x8)||(type ==D_L1_8x8)||(type ==D_BI_8x8)||(type ==D_DIRECT_8x8))
  96. enum mb_partition_e
  97. {
  98.     /* sub partition type for P_8x8 and B_8x8 */
  99.     D_L0_4x4        = 0,
  100.     D_L0_8x4        = 1,
  101.     D_L0_4x8        = 2,
  102.     D_L0_8x8        = 3,
  103.     /* sub partition type for B_8x8 only */
  104.     D_L1_4x4        = 4,
  105.     D_L1_8x4        = 5,
  106.     D_L1_4x8        = 6,
  107.     D_L1_8x8        = 7,
  108.     D_BI_4x4        = 8,
  109.     D_BI_8x4        = 9,
  110.     D_BI_4x8        = 10,
  111.     D_BI_8x8        = 11,
  112.     D_DIRECT_8x8    = 12,
  113.     /* partition */
  114.     D_8x8           = 13,
  115.     D_16x8          = 14,
  116.     D_8x16          = 15,
  117.     D_16x16         = 16,
  118. };
  119. static const int x264_mb_partition_listX_table[2][17] =
  120. {{
  121.     1, 1, 1, 1, /* D_L0_* */
  122.     0, 0, 0, 0, /* D_L1_* */
  123.     1, 1, 1, 1, /* D_BI_* */
  124.     0,          /* D_DIRECT_8x8 */
  125.     0, 0, 0, 0  /* 8x8 .. 16x16 */
  126. },
  127. {
  128.     0, 0, 0, 0, /* D_L0_* */
  129.     1, 1, 1, 1, /* D_L1_* */
  130.     1, 1, 1, 1, /* D_BI_* */
  131.     0,          /* D_DIRECT_8x8 */
  132.     0, 0, 0, 0  /* 8x8 .. 16x16 */
  133. }};
  134. static const int x264_mb_partition_count_table[17] =
  135. {
  136.     /* sub L0 */
  137.     4, 2, 2, 1,
  138.     /* sub L1 */
  139.     4, 2, 2, 1,
  140.     /* sub BI */
  141.     4, 2, 2, 1,
  142.     /* Direct */
  143.     1,
  144.     /* Partition */
  145.     4, 2, 2, 1
  146. };
  147. static const int x264_mb_partition_pixel_table[17] =
  148. {
  149.     6, 4, 5, 3, 6, 4, 5, 3, 6, 4, 5, 3, 3, 3, 1, 2, 0
  150. };
  151. static const int x264_zigzag_scan4[16] =
  152. {
  153.     0,  1,  4,  8,  5,  2,  3,  6,  9, 12, 13, 10,  7, 11, 14, 15
  154. };
  155. static const int x264_zigzag_scan8[64] =
  156. {
  157.     0,  1,  8, 16,  9,  2,  3, 10, 17, 24, 32, 25, 18, 11,  4,  5,
  158.    12, 19, 26, 33, 40, 48, 41, 34, 27, 20, 13,  6,  7, 14, 21, 28,
  159.    35, 42, 49, 56, 57, 50, 43, 36, 29, 22, 15, 23, 30, 37, 44, 51,
  160.    58, 59, 52, 45, 38, 31, 39, 46, 53, 60, 61, 54, 47, 55, 62, 63
  161. };  
  162. static const uint8_t block_idx_x[16] =
  163. {
  164.     0, 1, 0, 1, 2, 3, 2, 3, 0, 1, 0, 1, 2, 3, 2, 3
  165. };
  166. static const uint8_t block_idx_y[16] =
  167. {
  168.     0, 0, 1, 1, 0, 0, 1, 1, 2, 2, 3, 3, 2, 2, 3, 3
  169. };
  170. static const uint8_t block_idx_xy[4][4] =
  171. {
  172.     { 0, 2, 8,  10 },
  173.     { 1, 3, 9,  11 },
  174.     { 4, 6, 12, 14 },
  175.     { 5, 7, 13, 15 }
  176. };
  177. static const int i_chroma_qp_table[52] =
  178. {
  179.      0,  1,  2,  3,  4,  5,  6,  7,  8,  9,
  180.     10, 11, 12, 13, 14, 15, 16, 17, 18, 19,
  181.     20, 21, 22, 23, 24, 25, 26, 27, 28, 29,
  182.     29, 30, 31, 32, 32, 33, 34, 34, 35, 35,
  183.     36, 36, 37, 37, 37, 38, 38, 38, 39, 39,
  184.     39, 39
  185. };
  186. enum cabac_ctx_block_cat_e
  187. {
  188.     DCT_LUMA_DC   = 0,
  189.     DCT_LUMA_AC   = 1,
  190.     DCT_LUMA_4x4  = 2,
  191.     DCT_CHROMA_DC = 3,
  192.     DCT_CHROMA_AC = 4,
  193.     DCT_LUMA_8x8  = 5,
  194. };
  195. void x264_macroblock_cache_init( x264_t *h );
  196. void x264_macroblock_slice_init( x264_t *h );
  197. void x264_macroblock_cache_load( x264_t *h, int i_mb_x, int i_mb_y );
  198. void x264_macroblock_cache_save( x264_t *h );
  199. void x264_macroblock_cache_end( x264_t *h );
  200. void x264_macroblock_bipred_init( x264_t *h );
  201. /* x264_mb_predict_mv_16x16:
  202.  *      set mvp with predicted mv for D_16x16 block
  203.  *      h->mb. need only valid values from other blocks */
  204. void x264_mb_predict_mv_16x16( x264_t *h, int i_list, int i_ref, int mvp[2] );
  205. /* x264_mb_predict_mv_pskip:
  206.  *      set mvp with predicted mv for P_SKIP
  207.  *      h->mb. need only valid values from other blocks */
  208. void x264_mb_predict_mv_pskip( x264_t *h, int mv[2] );
  209. /* x264_mb_predict_mv:
  210.  *      set mvp with predicted mv for all blocks except SKIP and DIRECT
  211.  *      h->mb. need valid ref/partition/sub of current block to be valid
  212.  *      and valid mv/ref from other blocks. */
  213. void x264_mb_predict_mv( x264_t *h, int i_list, int idx, int i_width, int mvp[2] );
  214. /* x264_mb_predict_mv_direct16x16:
  215.  *      set h->mb.cache.mv and h->mb.cache.ref for B_SKIP or B_DIRECT
  216.  *      h->mb. need only valid values from other blocks.
  217.  *      return 1 on success, 0 on failure.
  218.  *      if b_changed != NULL, set it to whether refs or mvs differ from
  219.  *      before this functioncall. */
  220. int x264_mb_predict_mv_direct16x16( x264_t *h, int *b_changed );
  221. /* x264_mb_load_mv_direct8x8:
  222.  *      set h->mb.cache.mv and h->mb.cache.ref for B_DIRECT
  223.  *      must be called only after x264_mb_predict_mv_direct16x16 */
  224. void x264_mb_load_mv_direct8x8( x264_t *h, int idx );
  225. /* x264_mb_predict_mv_ref16x16:
  226.  *      set mvc with D_16x16 prediction.
  227.  *      uses all neighbors, even those that didn't end up using this ref.
  228.  *      h->mb. need only valid values from other blocks */
  229. void x264_mb_predict_mv_ref16x16( x264_t *h, int i_list, int i_ref, int mvc[8][2], int *i_mvc );
  230. int  x264_mb_predict_intra4x4_mode( x264_t *h, int idx );
  231. int  x264_mb_predict_non_zero_code( x264_t *h, int idx );
  232. /* x264_mb_transform_8x8_allowed:
  233.  *      check whether any partition is smaller than 8x8 (or at least
  234.  *      might be, according to just partition type.)
  235.  *      doesn't check for intra or cbp */
  236. int  x264_mb_transform_8x8_allowed( x264_t *h );
  237. void x264_mb_encode_i4x4( x264_t *h, int idx, int i_qscale );
  238. void x264_mb_encode_i8x8( x264_t *h, int idx, int i_qscale );
  239. void x264_mb_mc( x264_t *h );
  240. static inline void x264_macroblock_cache_ref( x264_t *h, int x, int y, int width, int height, int i_list, int ref )
  241. {
  242.     int dy, dx;
  243.     for( dy = 0; dy < height; dy++ )
  244.     {
  245.         for( dx = 0; dx < width; dx++ )
  246.         {
  247.             h->mb.cache.ref[i_list][X264_SCAN8_0+x+dx+8*(y+dy)] = ref;
  248.         }
  249.     }
  250. }
  251. 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 )
  252. {
  253.     int dy, dx;
  254.     for( dy = 0; dy < height; dy++ )
  255.     {
  256.         for( dx = 0; dx < width; dx++ )
  257.         {
  258.             h->mb.cache.mv[i_list][X264_SCAN8_0+x+dx+8*(y+dy)][0] = mvx;
  259.             h->mb.cache.mv[i_list][X264_SCAN8_0+x+dx+8*(y+dy)][1] = mvy;
  260.         }
  261.     }
  262. }
  263. 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 )
  264. {
  265.     int dy, dx;
  266.     for( dy = 0; dy < height; dy++ )
  267.     {
  268.         for( dx = 0; dx < width; dx++ )
  269.         {
  270.             h->mb.cache.mvd[i_list][X264_SCAN8_0+x+dx+8*(y+dy)][0] = mdx;
  271.             h->mb.cache.mvd[i_list][X264_SCAN8_0+x+dx+8*(y+dy)][1] = mdy;
  272.         }
  273.     }
  274. }
  275. static inline void x264_macroblock_cache_skip( x264_t *h, int x, int y, int width, int height, int b_skip )
  276. {
  277.     int dy, dx;
  278.     for( dy = 0; dy < height; dy++ )
  279.     {
  280.         for( dx = 0; dx < width; dx++ )
  281.         {
  282.             h->mb.cache.skip[X264_SCAN8_0+x+dx+8*(y+dy)] = b_skip;
  283.         }
  284.     }
  285. }
  286. static inline void x264_macroblock_cache_intra8x8_pred( x264_t *h, int x, int y, int i_mode )
  287. {
  288.     int *cache = &h->mb.cache.intra4x4_pred_mode[X264_SCAN8_0+x+8*y];
  289.     cache[0] = cache[1] = cache[8] = cache[9] = i_mode;
  290. }
  291. #endif