dct.h
上传用户:lctgjx
上传日期:2022-06-04
资源大小:8887k
文件大小:5k
源码类别:

流媒体/Mpeg4/MP4

开发平台:

Visual C++

  1. /*****************************************************************************
  2.  * dct.h: h264 encoder library
  3.  *****************************************************************************
  4.  * Copyright (C) 2004-2008 Loren Merritt <lorenm@u.washington.edu>
  5.  *
  6.  * This program is free software; you can redistribute it and/or modify
  7.  * it under the terms of the GNU General Public License as published by
  8.  * the Free Software Foundation; either version 2 of the License, or
  9.  * (at your option) any later version.
  10.  *
  11.  * This program 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
  14.  * GNU General Public License for more details.
  15.  *
  16.  * You should have received a copy of the GNU General Public License
  17.  * along with this program; if not, write to the Free Software
  18.  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02111, USA.
  19.  *****************************************************************************/
  20. #ifndef X264_DCT_H
  21. #define X264_DCT_H
  22. /* the inverse of the scaling factors introduced by 8x8 fdct */
  23. #define W(i) (i==0 ? FIX8(1.0000) :
  24.               i==1 ? FIX8(0.8859) :
  25.               i==2 ? FIX8(1.6000) :
  26.               i==3 ? FIX8(0.9415) :
  27.               i==4 ? FIX8(1.2651) :
  28.               i==5 ? FIX8(1.1910) :0)
  29. static const uint16_t x264_dct8_weight_tab[64] = {
  30.     W(0), W(3), W(4), W(3),  W(0), W(3), W(4), W(3),
  31.     W(3), W(1), W(5), W(1),  W(3), W(1), W(5), W(1),
  32.     W(4), W(5), W(2), W(5),  W(4), W(5), W(2), W(5),
  33.     W(3), W(1), W(5), W(1),  W(3), W(1), W(5), W(1),
  34.     W(0), W(3), W(4), W(3),  W(0), W(3), W(4), W(3),
  35.     W(3), W(1), W(5), W(1),  W(3), W(1), W(5), W(1),
  36.     W(4), W(5), W(2), W(5),  W(4), W(5), W(2), W(5),
  37.     W(3), W(1), W(5), W(1),  W(3), W(1), W(5), W(1)
  38. };
  39. #undef W
  40. #define W(i) (i==0 ? FIX8(1.76777) :
  41.               i==1 ? FIX8(1.11803) :
  42.               i==2 ? FIX8(0.70711) :0)
  43. static const uint16_t x264_dct4_weight_tab[16] = {
  44.     W(0), W(1), W(0), W(1),
  45.     W(1), W(2), W(1), W(2),
  46.     W(0), W(1), W(0), W(1),
  47.     W(1), W(2), W(1), W(2)
  48. };
  49. #undef W
  50. /* inverse squared */
  51. #define W(i) (i==0 ? FIX8(3.125) :
  52.               i==1 ? FIX8(1.25) :
  53.               i==2 ? FIX8(0.5) :0)
  54. static const uint16_t x264_dct4_weight2_tab[16] = {
  55.     W(0), W(1), W(0), W(1),
  56.     W(1), W(2), W(1), W(2),
  57.     W(0), W(1), W(0), W(1),
  58.     W(1), W(2), W(1), W(2)
  59. };
  60. #undef W
  61. #define W(i) (i==0 ? FIX8(1.00000) :
  62.               i==1 ? FIX8(0.78487) :
  63.               i==2 ? FIX8(2.56132) :
  64.               i==3 ? FIX8(0.88637) :
  65.               i==4 ? FIX8(1.60040) :
  66.               i==5 ? FIX8(1.41850) :0)
  67. static const uint16_t x264_dct8_weight2_tab[64] = {
  68.     W(0), W(3), W(4), W(3),  W(0), W(3), W(4), W(3),
  69.     W(3), W(1), W(5), W(1),  W(3), W(1), W(5), W(1),
  70.     W(4), W(5), W(2), W(5),  W(4), W(5), W(2), W(5),
  71.     W(3), W(1), W(5), W(1),  W(3), W(1), W(5), W(1),
  72.     W(0), W(3), W(4), W(3),  W(0), W(3), W(4), W(3),
  73.     W(3), W(1), W(5), W(1),  W(3), W(1), W(5), W(1),
  74.     W(4), W(5), W(2), W(5),  W(4), W(5), W(2), W(5),
  75.     W(3), W(1), W(5), W(1),  W(3), W(1), W(5), W(1)
  76. };
  77. #undef W
  78. extern int x264_dct4_weight2_zigzag[2][16]; // [2] = {frame, field}
  79. extern int x264_dct8_weight2_zigzag[2][64];
  80. typedef struct
  81. {
  82.     // pix1  stride = FENC_STRIDE
  83.     // pix2  stride = FDEC_STRIDE
  84.     // p_dst stride = FDEC_STRIDE
  85.     void (*sub4x4_dct)   ( int16_t dct[4][4], uint8_t *pix1, uint8_t *pix2 );
  86.     void (*add4x4_idct)  ( uint8_t *p_dst, int16_t dct[4][4] );
  87.     void (*sub8x8_dct)   ( int16_t dct[4][4][4], uint8_t *pix1, uint8_t *pix2 );
  88.     void (*sub8x8_dct_dc)( int16_t dct[2][2], uint8_t *pix1, uint8_t *pix2 );
  89.     void (*add8x8_idct)  ( uint8_t *p_dst, int16_t dct[4][4][4] );
  90.     void (*add8x8_idct_dc) ( uint8_t *p_dst, int16_t dct[2][2] );
  91.     void (*sub16x16_dct) ( int16_t dct[16][4][4], uint8_t *pix1, uint8_t *pix2 );
  92.     void (*add16x16_idct)( uint8_t *p_dst, int16_t dct[16][4][4] );
  93.     void (*add16x16_idct_dc) ( uint8_t *p_dst, int16_t dct[4][4] );
  94.     void (*sub8x8_dct8)  ( int16_t dct[8][8], uint8_t *pix1, uint8_t *pix2 );
  95.     void (*add8x8_idct8) ( uint8_t *p_dst, int16_t dct[8][8] );
  96.     void (*sub16x16_dct8) ( int16_t dct[4][8][8], uint8_t *pix1, uint8_t *pix2 );
  97.     void (*add16x16_idct8)( uint8_t *p_dst, int16_t dct[4][8][8] );
  98.     void (*dct4x4dc) ( int16_t d[4][4] );
  99.     void (*idct4x4dc)( int16_t d[4][4] );
  100. } x264_dct_function_t;
  101. typedef struct
  102. {
  103.     void (*scan_8x8)( int16_t level[64], int16_t dct[8][8] );
  104.     void (*scan_4x4)( int16_t level[16], int16_t dct[4][4] );
  105.     int  (*sub_8x8)  ( int16_t level[64], const uint8_t *p_src, uint8_t *p_dst );
  106.     int  (*sub_4x4)  ( int16_t level[16], const uint8_t *p_src, uint8_t *p_dst );
  107.     int  (*sub_4x4ac)( int16_t level[16], const uint8_t *p_src, uint8_t *p_dst, int16_t *dc );
  108.     void (*interleave_8x8_cavlc)( int16_t *dst, int16_t *src, uint8_t *nnz );
  109. } x264_zigzag_function_t;
  110. void x264_dct_init( int cpu, x264_dct_function_t *dctf );
  111. void x264_dct_init_weights( void );
  112. void x264_zigzag_init( int cpu, x264_zigzag_function_t *pf, int b_interlaced );
  113. #endif