-
Pengchong Jin authored
Change-Id: I9956db2ba2f7d28f484daaf5022d8d1ef5db473c
bae65224
/* * Copyright (c) 2010 The WebM project authors. All Rights Reserved. * * Use of this source code is governed by a BSD-style license * that can be found in the LICENSE file in the root of the source * tree. An additional intellectual property rights grant can be found * in the file PATENTS. All contributing project authors may * be found in the AUTHORS file in the root of the source tree. */#ifndef VP9_ENCODER_VP9_FIRSTPASS_H_#define VP9_ENCODER_VP9_FIRSTPASS_H_#include "vp9/encoder/vp9_lookahead.h"#include "vp9/encoder/vp9_ratectrl.h"#ifdef __cplusplusextern "C" {#endif#if CONFIG_FP_MB_STATS#define FPMB_DCINTRA_MASK 0x01#define FPMB_NONZERO_MOTION_MASK 0x02#define FPMB_ERROR_LEVEL0_MASK 0x04#define FPMB_ERROR_LEVEL1_MASK 0x10#define FPMB_ERROR_LEVEL2_MASK 0x20#define FPMB_ERROR_LEVEL3_MASK 0x40#define FPMB_ERROR_LEVEL4_MASK 0x80#define FPMB_ERROR_LEVEL1_TH 2000#define FPMB_ERROR_LEVEL2_TH 8000#define FPMB_ERROR_LEVEL3_TH 24000#define FPMB_ERROR_LEVEL4_TH 48000typedef struct { uint8_t *mb_stats_start; uint8_t *mb_stats_end;} FIRSTPASS_MB_STATS;#endiftypedef struct { double frame; double intra_error; double coded_error; double sr_coded_error; double pcnt_inter; double pcnt_motion; double pcnt_second_ref; double pcnt_neutral; double MVr; double mvr_abs; double MVc; double mvc_abs; double MVrv; double MVcv; double mv_in_out_count; double new_mv_count; double duration; double count; int64_t spatial_layer_id;} FIRSTPASS_STATS;typedef enum { KF_UPDATE = 0, LF_UPDATE = 1, GF_UPDATE = 2, ARF_UPDATE = 3, OVERLAY_UPDATE = 4, FRAME_UPDATE_TYPES = 57172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137
} FRAME_UPDATE_TYPE;
typedef struct {
unsigned char index;
RATE_FACTOR_LEVEL rf_level[(MAX_LAG_BUFFERS * 2) + 1];
FRAME_UPDATE_TYPE update_type[(MAX_LAG_BUFFERS * 2) + 1];
unsigned char arf_src_offset[(MAX_LAG_BUFFERS * 2) + 1];
unsigned char arf_update_idx[(MAX_LAG_BUFFERS * 2) + 1];
unsigned char arf_ref_idx[(MAX_LAG_BUFFERS * 2) + 1];
int bit_allocation[(MAX_LAG_BUFFERS * 2) + 1];
} GF_GROUP;
typedef struct {
unsigned int section_intra_rating;
FIRSTPASS_STATS total_stats;
FIRSTPASS_STATS this_frame_stats;
const FIRSTPASS_STATS *stats_in;
const FIRSTPASS_STATS *stats_in_start;
const FIRSTPASS_STATS *stats_in_end;
FIRSTPASS_STATS total_left_stats;
int first_pass_done;
int64_t bits_left;
double modified_error_min;
double modified_error_max;
double modified_error_left;
double kf_intra_err_min;
double gf_intra_err_min;
#if CONFIG_FP_MB_STATS
uint8_t *frame_mb_stats_buf;
uint8_t *this_frame_mb_stats;
FIRSTPASS_MB_STATS firstpass_mb_stats;
#endif
// Projected total bits available for a key frame group of frames
int64_t kf_group_bits;
// Error score of frames still to be coded in kf group
int64_t kf_group_error_left;
int sr_update_lag;
int kf_zeromotion_pct;
int gf_zeromotion_pct;
int active_worst_quality;
GF_GROUP gf_group;
} TWO_PASS;
struct VP9_COMP;
void vp9_init_first_pass(struct VP9_COMP *cpi);
void vp9_rc_get_first_pass_params(struct VP9_COMP *cpi);
void vp9_first_pass(struct VP9_COMP *cpi);
void vp9_end_first_pass(struct VP9_COMP *cpi);
void vp9_init_second_pass(struct VP9_COMP *cpi);
void vp9_rc_get_second_pass_params(struct VP9_COMP *cpi);
// Post encode update of the rate control parameters for 2-pass
void vp9_twopass_postencode_update(struct VP9_COMP *cpi);
#ifdef __cplusplus
} // extern "C"
#endif
#endif // VP9_ENCODER_VP9_FIRSTPASS_H_