aaabbd67b2
Add a conditional compile flag for this feature. Also add a switch to enable the encoder to use these statistics in the second pass. Currently, the switch is turned off. Change-Id: Ia1c858c35ec90e36f19f5cffe156b97ddaa04922
127 lines
3.1 KiB
C
127 lines
3.1 KiB
C
/*
|
|
* 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 __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
#if CONFIG_FP_MB_STATS
|
|
typedef struct {
|
|
PREDICTION_MODE mode;
|
|
int err;
|
|
int_mv mv;
|
|
} FIRSTPASS_MB_STATS;
|
|
|
|
typedef struct {
|
|
FIRSTPASS_MB_STATS *mb_stats;
|
|
} FIRSTPASS_FRAME_MB_STATS;
|
|
#endif
|
|
|
|
typedef 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 = 5
|
|
} 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;
|
|
unsigned int next_iiratio;
|
|
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
|
|
FIRSTPASS_FRAME_MB_STATS this_frame_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_
|