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

流媒体/Mpeg4/MP4

开发平台:

Visual C++

  1. /*****************************************************************************
  2.  * frame.h: h264 encoder library
  3.  *****************************************************************************
  4.  * Copyright (C) 2003-2008 x264 project
  5.  *
  6.  * Authors: Laurent Aimar <fenrir@via.ecp.fr>
  7.  *          Loren Merritt <lorenm@u.washington.edu>
  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., 51 Franklin Street, Fifth Floor, Boston, MA  02111, USA.
  22.  *****************************************************************************/
  23. #ifndef X264_FRAME_H
  24. #define X264_FRAME_H
  25. /* number of pixels past the edge of the frame, for motion estimation/compensation */
  26. #define PADH 32
  27. #define PADV 32
  28. typedef struct
  29. {
  30.     /* */
  31.     int     i_poc;
  32.     int     i_type;
  33.     int     i_qpplus1;
  34.     int64_t i_pts;
  35.     x264_param_t *param;
  36.     int     i_frame;    /* Presentation frame number */
  37.     int     i_frame_num; /* Coded frame number */
  38.     int     b_kept_as_ref;
  39.     uint8_t b_fdec;
  40.     uint8_t b_last_minigop_bframe; /* this frame is the last b in a sequence of bframes */
  41.     uint8_t i_bframes;   /* number of bframes following this nonb in coded order */
  42.     float   f_qp_avg_rc; /* QPs as decided by ratecontrol */
  43.     float   f_qp_avg_aq; /* QPs as decided by AQ in addition to ratecontrol */
  44.     /* YUV buffer */
  45.     int     i_plane;
  46.     int     i_stride[3];
  47.     int     i_width[3];
  48.     int     i_lines[3];
  49.     int     i_stride_lowres;
  50.     int     i_width_lowres;
  51.     int     i_lines_lowres;
  52.     uint8_t *plane[3];
  53.     uint8_t *filtered[4]; /* plane[0], H, V, HV */
  54.     uint8_t *lowres[4]; /* half-size copy of input frame: Orig, H, V, HV */
  55.     uint16_t *integral;
  56.     /* for unrestricted mv we allocate more data than needed
  57.      * allocated data are stored in buffer */
  58.     uint8_t *buffer[4];
  59.     uint8_t *buffer_lowres[4];
  60.     /* motion data */
  61.     int8_t  *mb_type;
  62.     int16_t (*mv[2])[2];
  63.     int16_t (*lowres_mvs[2][X264_BFRAME_MAX+1])[2];
  64.     uint16_t (*lowres_costs[X264_BFRAME_MAX+2][X264_BFRAME_MAX+2]);
  65.     /* Actually a width-2 bitfield with 4 values per uint8_t. */
  66.     uint8_t  (*lowres_inter_types[X264_BFRAME_MAX+2][X264_BFRAME_MAX+2]);
  67.     int     *lowres_mv_costs[2][X264_BFRAME_MAX+1];
  68.     int8_t  *ref[2];
  69.     int     i_ref[2];
  70.     int     ref_poc[2][16];
  71.     int     inv_ref_poc[16]; // inverse values (list0 only) to avoid divisions in MB encoding
  72.     /* for adaptive B-frame decision.
  73.      * contains the SATD cost of the lowres frame encoded in various modes
  74.      * FIXME: how big an array do we need? */
  75.     int     i_cost_est[X264_BFRAME_MAX+2][X264_BFRAME_MAX+2];
  76.     int     i_cost_est_aq[X264_BFRAME_MAX+2][X264_BFRAME_MAX+2];
  77.     int     i_satd; // the i_cost_est of the selected frametype
  78.     int     i_intra_mbs[X264_BFRAME_MAX+2];
  79.     int     *i_row_satds[X264_BFRAME_MAX+2][X264_BFRAME_MAX+2];
  80.     int     *i_row_satd;
  81.     int     *i_row_bits;
  82.     int     *i_row_qp;
  83.     float   *f_qp_offset;
  84.     float   *f_qp_offset_aq;
  85.     int     b_intra_calculated;
  86.     uint16_t *i_intra_cost;
  87.     uint16_t *i_propagate_cost;
  88.     uint16_t *i_inv_qscale_factor;
  89.     /* vbv */
  90.     uint8_t i_planned_type[X264_LOOKAHEAD_MAX+1];
  91.     int i_planned_satd[X264_LOOKAHEAD_MAX+1];
  92.     /* threading */
  93.     int     i_lines_completed; /* in pixels */
  94.     int     i_reference_count; /* number of threads using this frame (not necessarily the number of pointers) */
  95.     x264_pthread_mutex_t mutex;
  96.     x264_pthread_cond_t  cv;
  97. } x264_frame_t;
  98. /* synchronized frame list */
  99. typedef struct
  100. {
  101.    x264_frame_t **list;
  102.    int i_max_size;
  103.    int i_size;
  104.    x264_pthread_mutex_t     mutex;
  105.    x264_pthread_cond_t      cv_fill;  /* event signaling that the list became fuller */
  106.    x264_pthread_cond_t      cv_empty; /* event signaling that the list became emptier */
  107. } x264_synch_frame_list_t;
  108. typedef void (*x264_deblock_inter_t)( uint8_t *pix, int stride, int alpha, int beta, int8_t *tc0 );
  109. typedef void (*x264_deblock_intra_t)( uint8_t *pix, int stride, int alpha, int beta );
  110. typedef struct
  111. {
  112.     x264_deblock_inter_t deblock_v_luma;
  113.     x264_deblock_inter_t deblock_h_luma;
  114.     x264_deblock_inter_t deblock_v_chroma;
  115.     x264_deblock_inter_t deblock_h_chroma;
  116.     x264_deblock_intra_t deblock_v_luma_intra;
  117.     x264_deblock_intra_t deblock_h_luma_intra;
  118.     x264_deblock_intra_t deblock_v_chroma_intra;
  119.     x264_deblock_intra_t deblock_h_chroma_intra;
  120. } x264_deblock_function_t;
  121. x264_frame_t *x264_frame_new( x264_t *h, int b_fdec );
  122. void          x264_frame_delete( x264_frame_t *frame );
  123. int           x264_frame_copy_picture( x264_t *h, x264_frame_t *dst, x264_picture_t *src );
  124. void          x264_frame_expand_border( x264_t *h, x264_frame_t *frame, int mb_y, int b_end );
  125. void          x264_frame_expand_border_filtered( x264_t *h, x264_frame_t *frame, int mb_y, int b_end );
  126. void          x264_frame_expand_border_lowres( x264_frame_t *frame );
  127. void          x264_frame_expand_border_mod16( x264_t *h, x264_frame_t *frame );
  128. void          x264_frame_deblock( x264_t *h );
  129. void          x264_frame_deblock_row( x264_t *h, int mb_y );
  130. void          x264_frame_filter( x264_t *h, x264_frame_t *frame, int mb_y, int b_end );
  131. void          x264_frame_init_lowres( x264_t *h, x264_frame_t *frame );
  132. void          x264_deblock_init( int cpu, x264_deblock_function_t *pf );
  133. void          x264_frame_cond_broadcast( x264_frame_t *frame, int i_lines_completed );
  134. void          x264_frame_cond_wait( x264_frame_t *frame, int i_lines_completed );
  135. void          x264_frame_push( x264_frame_t **list, x264_frame_t *frame );
  136. x264_frame_t *x264_frame_pop( x264_frame_t **list );
  137. void          x264_frame_unshift( x264_frame_t **list, x264_frame_t *frame );
  138. x264_frame_t *x264_frame_shift( x264_frame_t **list );
  139. void          x264_frame_push_unused( x264_t *h, x264_frame_t *frame );
  140. x264_frame_t *x264_frame_pop_unused( x264_t *h, int b_fdec );
  141. void          x264_frame_sort( x264_frame_t **list, int b_dts );
  142. void          x264_frame_delete_list( x264_frame_t **list );
  143. int           x264_synch_frame_list_init( x264_synch_frame_list_t *slist, int nelem );
  144. void          x264_synch_frame_list_delete( x264_synch_frame_list_t *slist );
  145. void          x264_synch_frame_list_push( x264_synch_frame_list_t *slist, x264_frame_t *frame );
  146. #define x264_frame_sort_dts(list) x264_frame_sort(list, 1)
  147. #define x264_frame_sort_pts(list) x264_frame_sort(list, 0)
  148. #endif