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

流媒体/Mpeg4/MP4

开发平台:

Visual C++

  1. /*****************************************************************************
  2.  * pixel.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_PIXEL_H
  21. #define X264_PIXEL_H
  22. // SSD assumes all args aligned
  23. // other cmp functions assume first arg aligned
  24. typedef int  (*x264_pixel_cmp_t) ( uint8_t *, int, uint8_t *, int );
  25. typedef void (*x264_pixel_cmp_x3_t) ( uint8_t *, uint8_t *, uint8_t *, uint8_t *, int, int[3] );
  26. typedef void (*x264_pixel_cmp_x4_t) ( uint8_t *, uint8_t *, uint8_t *, uint8_t *, uint8_t *, int, int[4] );
  27. enum
  28. {
  29.     PIXEL_16x16 = 0,
  30.     PIXEL_16x8  = 1,
  31.     PIXEL_8x16  = 2,
  32.     PIXEL_8x8   = 3,
  33.     PIXEL_8x4   = 4,
  34.     PIXEL_4x8   = 5,
  35.     PIXEL_4x4   = 6,
  36.     PIXEL_4x2   = 7,
  37.     PIXEL_2x4   = 8,
  38.     PIXEL_2x2   = 9,
  39. };
  40. static const struct {
  41.     int w;
  42.     int h;
  43. } x264_pixel_size[7] = {
  44.     { 16, 16 },
  45.     { 16,  8 }, {  8, 16 },
  46.     {  8,  8 },
  47.     {  8,  4 }, {  4,  8 },
  48.     {  4,  4 }
  49. };
  50. static const uint8_t x264_size2pixel[5][5] = {
  51.     { 0, },
  52.     { 0, PIXEL_4x4, PIXEL_8x4, 0, 0 },
  53.     { 0, PIXEL_4x8, PIXEL_8x8, 0, PIXEL_16x8 },
  54.     { 0, },
  55.     { 0, 0,        PIXEL_8x16, 0, PIXEL_16x16 }
  56. };
  57. typedef struct
  58. {
  59.     x264_pixel_cmp_t  sad[7];
  60.     x264_pixel_cmp_t  ssd[7];
  61.     x264_pixel_cmp_t satd[7];
  62.     x264_pixel_cmp_t ssim[7];
  63.     x264_pixel_cmp_t sa8d[4];
  64.     x264_pixel_cmp_t mbcmp[7]; /* either satd or sad for subpel refine and mode decision */
  65.     x264_pixel_cmp_t mbcmp_unaligned[7]; /* unaligned mbcmp for subpel */
  66.     x264_pixel_cmp_t fpelcmp[7]; /* either satd or sad for fullpel motion search */
  67.     x264_pixel_cmp_x3_t fpelcmp_x3[7];
  68.     x264_pixel_cmp_x4_t fpelcmp_x4[7];
  69.     x264_pixel_cmp_t sad_aligned[7]; /* Aligned SAD for mbcmp */
  70.     int (*var2_8x8)( uint8_t *, int, uint8_t *, int, int * );
  71.     int (*var[4])( uint8_t *pix, int stride );
  72.     uint64_t (*hadamard_ac[4])( uint8_t *pix, int stride );
  73.     void (*ssim_4x4x2_core)( const uint8_t *pix1, int stride1,
  74.                              const uint8_t *pix2, int stride2, int sums[2][4] );
  75.     float (*ssim_end4)( int sum0[5][4], int sum1[5][4], int width );
  76.     /* multiple parallel calls to cmp. */
  77.     x264_pixel_cmp_x3_t sad_x3[7];
  78.     x264_pixel_cmp_x4_t sad_x4[7];
  79.     x264_pixel_cmp_x3_t satd_x3[7];
  80.     x264_pixel_cmp_x4_t satd_x4[7];
  81.     /* abs-diff-sum for successive elimination.
  82.      * may round width up to a multiple of 16. */
  83.     int (*ads[7])( int enc_dc[4], uint16_t *sums, int delta,
  84.                    uint16_t *cost_mvx, int16_t *mvs, int width, int thresh );
  85.     /* calculate satd or sad of V, H, and DC modes.
  86.      * may be NULL, in which case just use pred+satd instead. */
  87.     void (*intra_mbcmp_x3_16x16)( uint8_t *fenc, uint8_t *fdec  , int res[3] );
  88.     void (*intra_satd_x3_16x16) ( uint8_t *fenc, uint8_t *fdec  , int res[3] );
  89.     void (*intra_sad_x3_16x16)  ( uint8_t *fenc, uint8_t *fdec  , int res[3] );
  90.     void (*intra_mbcmp_x3_8x8c) ( uint8_t *fenc, uint8_t *fdec  , int res[3] );
  91.     void (*intra_satd_x3_8x8c)  ( uint8_t *fenc, uint8_t *fdec  , int res[3] );
  92.     void (*intra_sad_x3_8x8c)   ( uint8_t *fenc, uint8_t *fdec  , int res[3] );
  93.     void (*intra_mbcmp_x3_4x4)  ( uint8_t *fenc, uint8_t *fdec  , int res[3] );
  94.     void (*intra_satd_x3_4x4)   ( uint8_t *fenc, uint8_t *fdec  , int res[3] );
  95.     void (*intra_sad_x3_4x4)    ( uint8_t *fenc, uint8_t *fdec  , int res[3] );
  96.     void (*intra_mbcmp_x3_8x8)  ( uint8_t *fenc, uint8_t edge[33], int res[3] );
  97.     void (*intra_sa8d_x3_8x8)   ( uint8_t *fenc, uint8_t edge[33], int res[3] );
  98.     void (*intra_sad_x3_8x8)    ( uint8_t *fenc, uint8_t edge[33], int res[3] );
  99. } x264_pixel_function_t;
  100. void x264_pixel_init( int cpu, x264_pixel_function_t *pixf );
  101. int64_t x264_pixel_ssd_wxh( x264_pixel_function_t *pf, uint8_t *pix1, int i_pix1, uint8_t *pix2, int i_pix2, int i_width, int i_height );
  102. float x264_pixel_ssim_wxh( x264_pixel_function_t *pf, uint8_t *pix1, int i_pix1, uint8_t *pix2, int i_pix2, int i_width, int i_height, void *buf );
  103. #endif