Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 1 | /* |
Urvang Joshi | 8a02d76 | 2016-07-28 15:51:12 -0700 | [diff] [blame] | 2 | * Copyright (c) 2016, Alliance for Open Media. All rights reserved |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 3 | * |
Urvang Joshi | 8a02d76 | 2016-07-28 15:51:12 -0700 | [diff] [blame] | 4 | * This source code is subject to the terms of the BSD 2 Clause License and |
| 5 | * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License |
| 6 | * was not distributed with this source code in the LICENSE file, you can |
| 7 | * obtain it at www.aomedia.org/license/software. If the Alliance for Open |
| 8 | * Media Patent License 1.0 was not distributed with this source code in the |
| 9 | * PATENTS file, you can obtain it at www.aomedia.org/license/patent. |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 10 | */ |
| 11 | |
James Zern | e1cbb13 | 2018-08-22 14:10:36 -0700 | [diff] [blame] | 12 | #ifndef AOM_AV1_ENCODER_ENCODER_H_ |
| 13 | #define AOM_AV1_ENCODER_ENCODER_H_ |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 14 | |
elliottk | b8becb2 | 2019-03-04 01:01:30 -0800 | [diff] [blame] | 15 | #include <stdbool.h> |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 16 | #include <stdio.h> |
| 17 | |
Tom Finegan | 60e653d | 2018-05-22 11:34:58 -0700 | [diff] [blame] | 18 | #include "config/aom_config.h" |
| 19 | |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 20 | #include "aom/aomcx.h" |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 21 | |
| 22 | #include "av1/common/alloccommon.h" |
| 23 | #include "av1/common/entropymode.h" |
| 24 | #include "av1/common/thread_common.h" |
| 25 | #include "av1/common/onyxc_int.h" |
Fergus Simpson | d2bcbb5 | 2017-05-22 23:15:05 -0700 | [diff] [blame] | 26 | #include "av1/common/resize.h" |
Andrey Norkin | 795ba87 | 2018-03-06 13:24:14 -0800 | [diff] [blame] | 27 | #include "av1/common/timing.h" |
kyslov | 7b9d0d6 | 2018-12-21 11:12:26 -0800 | [diff] [blame] | 28 | #include "av1/common/blockd.h" |
| 29 | #include "av1/common/enums.h" |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 30 | #include "av1/encoder/aq_cyclicrefresh.h" |
Tom Finegan | 17ce8b1 | 2017-02-08 12:46:31 -0800 | [diff] [blame] | 31 | #include "av1/encoder/av1_quantize.h" |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 32 | #include "av1/encoder/context_tree.h" |
| 33 | #include "av1/encoder/encodemb.h" |
| 34 | #include "av1/encoder/firstpass.h" |
chiyotsai | d16a257 | 2019-03-05 17:48:10 -0800 | [diff] [blame] | 35 | #include "av1/encoder/level.h" |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 36 | #include "av1/encoder/lookahead.h" |
| 37 | #include "av1/encoder/mbgraph.h" |
| 38 | #include "av1/encoder/mcomp.h" |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 39 | #include "av1/encoder/ratectrl.h" |
| 40 | #include "av1/encoder/rd.h" |
| 41 | #include "av1/encoder/speed_features.h" |
| 42 | #include "av1/encoder/tokenize.h" |
kyslov | 7b9d0d6 | 2018-12-21 11:12:26 -0800 | [diff] [blame] | 43 | #include "av1/encoder/block.h" |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 44 | |
| 45 | #if CONFIG_INTERNAL_STATS |
| 46 | #include "aom_dsp/ssim.h" |
| 47 | #endif |
| 48 | #include "aom_dsp/variance.h" |
Neil Birkbeck | a2893ab | 2018-06-08 14:45:13 -0700 | [diff] [blame] | 49 | #if CONFIG_DENOISE |
| 50 | #include "aom_dsp/noise_model.h" |
| 51 | #endif |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 52 | #include "aom/internal/aom_codec_internal.h" |
| 53 | #include "aom_util/aom_thread.h" |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 54 | |
| 55 | #ifdef __cplusplus |
| 56 | extern "C" { |
| 57 | #endif |
| 58 | |
Yue Chen | 1bc5be6 | 2018-08-24 13:57:32 -0700 | [diff] [blame^] | 59 | /*!\brief Rational Number with an int64 numerator |
| 60 | * |
| 61 | * This structure holds a fractional value. |
| 62 | */ |
| 63 | typedef struct aom_rational64 { |
| 64 | int64_t num; /**< fraction numerator */ |
| 65 | int den; /**< fraction denominator */ |
| 66 | } aom_rational64_t; /**< alias for struct aom_rational */ |
| 67 | |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 68 | typedef struct { |
Jingning Han | f050fc1 | 2018-03-09 14:53:33 -0800 | [diff] [blame] | 69 | int nmv_vec_cost[MV_JOINTS]; |
| 70 | int nmv_costs[2][MV_VALS]; |
| 71 | int nmv_costs_hp[2][MV_VALS]; |
James Zern | 01a9d70 | 2017-08-25 19:09:33 +0000 | [diff] [blame] | 72 | |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 73 | FRAME_CONTEXT fc; |
| 74 | } CODING_CONTEXT; |
| 75 | |
Satish Kumar Suman | 4667aa1 | 2018-12-14 18:28:19 +0530 | [diff] [blame] | 76 | enum { |
Urvang Joshi | f70375a | 2019-03-22 23:30:19 -0700 | [diff] [blame] | 77 | REGULAR_FRAME, // regular inter frame |
| 78 | ARF_FRAME, // alternate reference frame |
| 79 | OVERLAY_FRAME, // overlay frame |
| 80 | GLD_FRAME, // golden frame |
| 81 | BRF_FRAME, // backward reference frame |
| 82 | INTERNAL_ARF_FRAME, // internal alternate reference frame |
Thomas Daede | 51020e1 | 2017-12-14 20:12:44 -0800 | [diff] [blame] | 83 | FRAME_CONTEXT_INDEXES |
Satish Kumar Suman | 4667aa1 | 2018-12-14 18:28:19 +0530 | [diff] [blame] | 84 | } UENUM1BYTE(FRAME_CONTEXT_INDEX); |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 85 | |
Satish Kumar Suman | 4667aa1 | 2018-12-14 18:28:19 +0530 | [diff] [blame] | 86 | enum { |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 87 | NORMAL = 0, |
| 88 | FOURFIVE = 1, |
| 89 | THREEFIVE = 2, |
| 90 | ONETWO = 3 |
Satish Kumar Suman | 4667aa1 | 2018-12-14 18:28:19 +0530 | [diff] [blame] | 91 | } UENUM1BYTE(AOM_SCALING); |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 92 | |
Satish Kumar Suman | 4667aa1 | 2018-12-14 18:28:19 +0530 | [diff] [blame] | 93 | enum { |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 94 | // Good Quality Fast Encoding. The encoder balances quality with the amount of |
| 95 | // time it takes to encode the output. Speed setting controls how fast. |
kyslov | 7b9d0d6 | 2018-12-21 11:12:26 -0800 | [diff] [blame] | 96 | GOOD, |
| 97 | // Realtime Fast Encoding. Will force some restrictions on bitrate |
| 98 | // constraints. |
| 99 | REALTIME |
Satish Kumar Suman | 4667aa1 | 2018-12-14 18:28:19 +0530 | [diff] [blame] | 100 | } UENUM1BYTE(MODE); |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 101 | |
Satish Kumar Suman | 4667aa1 | 2018-12-14 18:28:19 +0530 | [diff] [blame] | 102 | enum { |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 103 | FRAMEFLAGS_KEY = 1 << 0, |
| 104 | FRAMEFLAGS_GOLDEN = 1 << 1, |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 105 | FRAMEFLAGS_BWDREF = 1 << 2, |
Zoe Liu | 3ac2093 | 2017-08-30 16:35:55 -0700 | [diff] [blame] | 106 | // TODO(zoeliu): To determine whether a frame flag is needed for ALTREF2_FRAME |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 107 | FRAMEFLAGS_ALTREF = 1 << 3, |
Debargha Mukherjee | 5737825 | 2018-09-21 18:29:37 -0700 | [diff] [blame] | 108 | FRAMEFLAGS_INTRAONLY = 1 << 4, |
| 109 | FRAMEFLAGS_SWITCH = 1 << 5, |
| 110 | FRAMEFLAGS_ERROR_RESILIENT = 1 << 6, |
Satish Kumar Suman | 4667aa1 | 2018-12-14 18:28:19 +0530 | [diff] [blame] | 111 | } UENUM1BYTE(FRAMETYPE_FLAGS); |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 112 | |
Satish Kumar Suman | 4667aa1 | 2018-12-14 18:28:19 +0530 | [diff] [blame] | 113 | enum { |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 114 | NO_AQ = 0, |
| 115 | VARIANCE_AQ = 1, |
| 116 | COMPLEXITY_AQ = 2, |
| 117 | CYCLIC_REFRESH_AQ = 3, |
| 118 | AQ_MODE_COUNT // This should always be the last member of the enum |
Satish Kumar Suman | 4667aa1 | 2018-12-14 18:28:19 +0530 | [diff] [blame] | 119 | } UENUM1BYTE(AQ_MODE); |
| 120 | enum { |
Fangwen Fu | 6160df2 | 2017-04-24 09:45:51 -0700 | [diff] [blame] | 121 | NO_DELTA_Q = 0, |
Debargha Mukherjee | 4fef7a5 | 2019-04-04 01:02:06 -0700 | [diff] [blame] | 122 | DELTA_Q_OBJECTIVE = 1, // Modulation to improve objective quality |
| 123 | DELTA_Q_PERCEPTUAL = 2, // Modulation to improve perceptual quality |
| 124 | DELTA_Q_MODE_COUNT // This should always be the last member of the enum |
Satish Kumar Suman | 4667aa1 | 2018-12-14 18:28:19 +0530 | [diff] [blame] | 125 | } UENUM1BYTE(DELTAQ_MODE); |
Debargha Mukherjee | 3a4959f | 2018-02-26 15:34:03 -0800 | [diff] [blame] | 126 | |
Satish Kumar Suman | 4667aa1 | 2018-12-14 18:28:19 +0530 | [diff] [blame] | 127 | enum { |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 128 | RESIZE_NONE = 0, // No frame resizing allowed. |
Debargha Mukherjee | 7166f22 | 2017-09-05 21:32:42 -0700 | [diff] [blame] | 129 | RESIZE_FIXED = 1, // All frames are coded at the specified scale. |
| 130 | RESIZE_RANDOM = 2, // All frames are coded at a random scale. |
| 131 | RESIZE_MODES |
Satish Kumar Suman | 4667aa1 | 2018-12-14 18:28:19 +0530 | [diff] [blame] | 132 | } UENUM1BYTE(RESIZE_MODE); |
Debargha Mukherjee | 3a4959f | 2018-02-26 15:34:03 -0800 | [diff] [blame] | 133 | |
Satish Kumar Suman | 4667aa1 | 2018-12-14 18:28:19 +0530 | [diff] [blame] | 134 | enum { |
Urvang Joshi | 36a8373 | 2019-01-31 15:31:57 -0800 | [diff] [blame] | 135 | SUPERRES_NONE, // No frame superres allowed. |
| 136 | SUPERRES_FIXED, // All frames are coded at the specified scale, |
| 137 | // and super-resolved. |
| 138 | SUPERRES_RANDOM, // All frames are coded at a random scale, |
| 139 | // and super-resolved. |
| 140 | SUPERRES_QTHRESH, // Superres scale for a frame is determined based on |
| 141 | // q_index. |
| 142 | SUPERRES_AUTO, // Automatically select superres for appropriate frames. |
Debargha Mukherjee | 7166f22 | 2017-09-05 21:32:42 -0700 | [diff] [blame] | 143 | SUPERRES_MODES |
Satish Kumar Suman | 4667aa1 | 2018-12-14 18:28:19 +0530 | [diff] [blame] | 144 | } UENUM1BYTE(SUPERRES_MODE); |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 145 | |
kyslov | 7b9d0d6 | 2018-12-21 11:12:26 -0800 | [diff] [blame] | 146 | typedef enum { |
| 147 | kInvalid = 0, |
| 148 | kLowSadLowSumdiff = 1, |
| 149 | kLowSadHighSumdiff = 2, |
| 150 | kHighSadLowSumdiff = 3, |
| 151 | kHighSadHighSumdiff = 4, |
| 152 | kLowVarHighSumdiff = 5, |
| 153 | kVeryHighSad = 6, |
| 154 | } CONTENT_STATE_SB; |
| 155 | |
Satish Kumar Suman | e6d0be5 | 2019-02-14 14:33:28 +0530 | [diff] [blame] | 156 | enum { |
| 157 | SS_CFG_SRC = 0, |
| 158 | SS_CFG_LOOKAHEAD = 1, |
| 159 | SS_CFG_TOTAL = 2 |
| 160 | } UENUM1BYTE(SS_CFG_OFFSET); |
| 161 | |
Yue Chen | 7cae98f | 2018-08-24 10:43:16 -0700 | [diff] [blame] | 162 | typedef struct TplDepStats { |
| 163 | int64_t intra_cost; |
| 164 | int64_t inter_cost; |
| 165 | int64_t mc_flow; |
| 166 | int64_t mc_dep_cost; |
Yue Chen | 7cae98f | 2018-08-24 10:43:16 -0700 | [diff] [blame] | 167 | |
| 168 | int ref_frame_index; |
| 169 | int_mv mv; |
| 170 | } TplDepStats; |
| 171 | |
| 172 | typedef struct TplDepFrame { |
| 173 | uint8_t is_valid; |
| 174 | TplDepStats *tpl_stats_ptr; |
| 175 | int stride; |
| 176 | int width; |
| 177 | int height; |
| 178 | int mi_rows; |
| 179 | int mi_cols; |
| 180 | int base_qindex; |
| 181 | } TplDepFrame; |
| 182 | |
Debargha Mukherjee | 03ad12d | 2019-02-07 11:17:59 -0800 | [diff] [blame] | 183 | typedef enum { |
| 184 | COST_UPD_SB, |
| 185 | COST_UPD_SBROW, |
| 186 | COST_UPD_TILE, |
| 187 | } COST_UPDATE_TYPE; |
| 188 | |
Yue Chen | 7cae98f | 2018-08-24 10:43:16 -0700 | [diff] [blame] | 189 | #define TPL_DEP_COST_SCALE_LOG2 4 |
| 190 | |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 191 | typedef struct AV1EncoderConfig { |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 192 | BITSTREAM_PROFILE profile; |
Tom Finegan | 8ab2bba | 2018-02-28 07:36:28 -0800 | [diff] [blame] | 193 | aom_bit_depth_t bit_depth; // Codec bit-depth. |
| 194 | int width; // width of data passed to the compressor |
| 195 | int height; // height of data passed to the compressor |
| 196 | int forced_max_frame_width; // forced maximum width of frame (if != 0) |
| 197 | int forced_max_frame_height; // forced maximum height of frame (if != 0) |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 198 | unsigned int input_bit_depth; // Input bit depth. |
| 199 | double init_framerate; // set to passed in framerate |
| 200 | int64_t target_bandwidth; // bandwidth to be used in bits per second |
| 201 | |
| 202 | int noise_sensitivity; // pre processing blur: recommendation 0 |
| 203 | int sharpness; // sharpening output: recommendation 0: |
| 204 | int speed; |
| 205 | // maximum allowed bitrate for any intra frame in % of bitrate target. |
| 206 | unsigned int rc_max_intra_bitrate_pct; |
| 207 | // maximum allowed bitrate for any inter frame in % of bitrate target. |
| 208 | unsigned int rc_max_inter_bitrate_pct; |
| 209 | // percent of rate boost for golden frame in CBR mode. |
| 210 | unsigned int gf_cbr_boost_pct; |
| 211 | |
| 212 | MODE mode; |
| 213 | int pass; |
| 214 | |
| 215 | // Key Framing Operations |
| 216 | int auto_key; // autodetect cut scenes and set the keyframes |
| 217 | int key_freq; // maximum distance to key frame. |
Tarek AMARA | c981385 | 2018-03-05 18:40:18 -0500 | [diff] [blame] | 218 | int sframe_dist; |
| 219 | int sframe_mode; |
| 220 | int sframe_enabled; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 221 | int lag_in_frames; // how many frames lag before we start encoding |
Sarah Parker | 93c0314 | 2018-05-22 13:35:45 -0700 | [diff] [blame] | 222 | int fwd_kf_enabled; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 223 | |
| 224 | // ---------------------------------------------------------------- |
| 225 | // DATARATE CONTROL OPTIONS |
| 226 | |
| 227 | // vbr, cbr, constrained quality or constant quality |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 228 | enum aom_rc_mode rc_mode; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 229 | |
| 230 | // buffer targeting aggressiveness |
| 231 | int under_shoot_pct; |
| 232 | int over_shoot_pct; |
| 233 | |
| 234 | // buffering parameters |
| 235 | int64_t starting_buffer_level_ms; |
| 236 | int64_t optimal_buffer_level_ms; |
| 237 | int64_t maximum_buffer_size_ms; |
| 238 | |
| 239 | // Frame drop threshold. |
| 240 | int drop_frames_water_mark; |
| 241 | |
| 242 | // controlling quality |
| 243 | int fixed_q; |
| 244 | int worst_allowed_q; |
| 245 | int best_allowed_q; |
| 246 | int cq_level; |
| 247 | AQ_MODE aq_mode; // Adaptive Quantization mode |
Fangwen Fu | 6160df2 | 2017-04-24 09:45:51 -0700 | [diff] [blame] | 248 | DELTAQ_MODE deltaq_mode; |
Debargha Mukherjee | 4fef7a5 | 2019-04-04 01:02:06 -0700 | [diff] [blame] | 249 | int deltalf_mode; |
Debargha Mukherjee | 98a311c | 2018-03-25 16:33:11 -0700 | [diff] [blame] | 250 | int enable_cdef; |
| 251 | int enable_restoration; |
Yue Chen | 5e12951 | 2018-12-10 10:48:54 -0800 | [diff] [blame] | 252 | int enable_obmc; |
Yue Chen | 4835dc0 | 2018-05-11 15:57:43 -0700 | [diff] [blame] | 253 | int disable_trellis_quant; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 254 | int using_qm; |
Yaowu Xu | f7a1242 | 2018-01-31 15:29:20 -0800 | [diff] [blame] | 255 | int qm_y; |
| 256 | int qm_u; |
| 257 | int qm_v; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 258 | int qm_minlevel; |
| 259 | int qm_maxlevel; |
Yushin Cho | d808bfc | 2017-08-10 15:54:36 -0700 | [diff] [blame] | 260 | #if CONFIG_DIST_8X8 |
| 261 | int using_dist_8x8; |
| 262 | #endif |
Thomas Davies | af6df17 | 2016-11-09 14:04:18 +0000 | [diff] [blame] | 263 | unsigned int num_tile_groups; |
| 264 | unsigned int mtu; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 265 | |
| 266 | // Internal frame size scaling. |
Debargha Mukherjee | 29e40a6 | 2017-06-14 09:37:12 -0700 | [diff] [blame] | 267 | RESIZE_MODE resize_mode; |
Urvang Joshi | de71d14 | 2017-10-05 12:12:15 -0700 | [diff] [blame] | 268 | uint8_t resize_scale_denominator; |
| 269 | uint8_t resize_kf_scale_denominator; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 270 | |
Fergus Simpson | c4e7894 | 2017-04-10 14:59:00 -0700 | [diff] [blame] | 271 | // Frame Super-Resolution size scaling. |
| 272 | SUPERRES_MODE superres_mode; |
Urvang Joshi | de71d14 | 2017-10-05 12:12:15 -0700 | [diff] [blame] | 273 | uint8_t superres_scale_denominator; |
| 274 | uint8_t superres_kf_scale_denominator; |
Debargha Mukherjee | 7166f22 | 2017-09-05 21:32:42 -0700 | [diff] [blame] | 275 | int superres_qthresh; |
| 276 | int superres_kf_qthresh; |
Fergus Simpson | 3502d08 | 2017-04-10 12:25:07 -0700 | [diff] [blame] | 277 | |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 278 | // Enable feature to reduce the frame quantization every x frames. |
| 279 | int frame_periodic_boost; |
| 280 | |
| 281 | // two pass datarate control |
| 282 | int two_pass_vbrbias; // two pass datarate control tweaks |
| 283 | int two_pass_vbrmin_section; |
| 284 | int two_pass_vbrmax_section; |
| 285 | // END DATARATE CONTROL OPTIONS |
| 286 | // ---------------------------------------------------------------- |
| 287 | |
| 288 | int enable_auto_arf; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 289 | int enable_auto_brf; // (b)ackward (r)ef (f)rame |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 290 | |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 291 | /* Bitfield defining the error resiliency features to enable. |
| 292 | * Can provide decodable frames after losses in previous |
| 293 | * frames and decodable partitions after losses in the same frame. |
| 294 | */ |
| 295 | unsigned int error_resilient_mode; |
| 296 | |
Debargha Mukherjee | 52fb047 | 2018-03-29 15:48:11 -0700 | [diff] [blame] | 297 | unsigned int s_frame_mode; |
| 298 | |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 299 | /* Bitfield defining the parallel decoding mode where the |
| 300 | * decoding in successive frames may be conducted in parallel |
| 301 | * just by decoding the frame headers. |
| 302 | */ |
| 303 | unsigned int frame_parallel_decoding_mode; |
| 304 | |
Debargha Mukherjee | c6f24c2 | 2018-04-07 08:43:08 -0700 | [diff] [blame] | 305 | unsigned int limit; |
| 306 | |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 307 | int arnr_max_frames; |
| 308 | int arnr_strength; |
| 309 | |
| 310 | int min_gf_interval; |
| 311 | int max_gf_interval; |
Urvang Joshi | f4e775c | 2018-12-14 11:33:17 -0800 | [diff] [blame] | 312 | int gf_max_pyr_height; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 313 | |
Ravi Chaudhary | 0ec03a4 | 2018-08-17 18:53:28 +0530 | [diff] [blame] | 314 | int row_mt; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 315 | int tile_columns; |
| 316 | int tile_rows; |
Dominic Symes | 26ad0b2 | 2017-10-01 16:35:13 +0200 | [diff] [blame] | 317 | int tile_width_count; |
| 318 | int tile_height_count; |
| 319 | int tile_widths[MAX_TILE_COLS]; |
| 320 | int tile_heights[MAX_TILE_ROWS]; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 321 | |
Yue Chen | 7cae98f | 2018-08-24 10:43:16 -0700 | [diff] [blame] | 322 | int enable_tpl_model; |
| 323 | |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 324 | int max_threads; |
| 325 | |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 326 | aom_fixed_buf_t two_pass_stats_in; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 327 | |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 328 | aom_tune_metric tuning; |
| 329 | aom_tune_content content; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 330 | int use_highbitdepth; |
Andrey Norkin | 9e69463 | 2017-12-21 18:50:57 -0800 | [diff] [blame] | 331 | aom_color_primaries_t color_primaries; |
| 332 | aom_transfer_characteristics_t transfer_characteristics; |
| 333 | aom_matrix_coefficients_t matrix_coefficients; |
anorkin | 76fb126 | 2017-03-22 15:12:12 -0700 | [diff] [blame] | 334 | aom_chroma_sample_position_t chroma_sample_position; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 335 | int color_range; |
| 336 | int render_width; |
| 337 | int render_height; |
Andrey Norkin | 28e9ce2 | 2018-01-08 10:11:21 -0800 | [diff] [blame] | 338 | int timing_info_present; |
Andrey Norkin | 795ba87 | 2018-03-06 13:24:14 -0800 | [diff] [blame] | 339 | aom_timing_info_t timing_info; |
| 340 | int decoder_model_info_present_flag; |
Andrey Norkin | 2649551 | 2018-06-20 17:13:11 -0700 | [diff] [blame] | 341 | int display_model_info_present_flag; |
Wan-Teh Chang | f64b3bc | 2018-07-02 09:42:39 -0700 | [diff] [blame] | 342 | int buffer_removal_time_present; |
Andrey Norkin | 795ba87 | 2018-03-06 13:24:14 -0800 | [diff] [blame] | 343 | aom_dec_model_info_t buffer_model; |
Andrey Norkin | 6f1c2f7 | 2018-01-15 20:08:52 -0800 | [diff] [blame] | 344 | int film_grain_test_vector; |
Neil Birkbeck | eb895ef | 2018-03-14 17:51:03 -0700 | [diff] [blame] | 345 | const char *film_grain_table_filename; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 346 | |
Hui Su | 1cb1c00 | 2018-02-05 18:21:20 -0800 | [diff] [blame] | 347 | uint8_t cdf_update_mode; |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 348 | aom_superblock_size_t superblock_size; |
Yunqing Wang | eeb08a9 | 2017-07-07 21:25:18 -0700 | [diff] [blame] | 349 | unsigned int large_scale_tile; |
| 350 | unsigned int single_tile_decoding; |
Yaowu Xu | 76537ca | 2019-01-03 12:41:32 -0800 | [diff] [blame] | 351 | uint8_t monochrome; |
Debargha Mukherjee | 9713ccb | 2018-04-08 19:09:17 -0700 | [diff] [blame] | 352 | unsigned int full_still_picture_hdr; |
Jingning Han | a446f55 | 2018-02-22 15:42:12 -0800 | [diff] [blame] | 353 | int enable_dual_filter; |
Yunqing Wang | ff4fa06 | 2017-04-21 10:56:08 -0700 | [diff] [blame] | 354 | unsigned int motion_vector_unit_test; |
Maxym Dmytrychenko | cc6e0e1 | 2018-02-05 16:35:37 +0100 | [diff] [blame] | 355 | const cfg_options_t *cfg; |
Debargha Mukherjee | 9d9cf19 | 2018-12-17 11:24:21 -0800 | [diff] [blame] | 356 | int enable_rect_partitions; |
Ryan Lei | 9543c70 | 2019-03-11 11:06:14 -0700 | [diff] [blame] | 357 | int enable_ab_partitions; |
| 358 | int enable_1to4_partitions; |
| 359 | int min_partition_size; |
| 360 | int max_partition_size; |
Debargha Mukherjee | 03c43ba | 2018-12-14 13:08:08 -0800 | [diff] [blame] | 361 | int enable_intra_edge_filter; |
Debargha Mukherjee | 8c917b0 | 2018-12-17 13:45:50 -0800 | [diff] [blame] | 362 | int enable_tx64; |
Ryan Lei | 82d0a04 | 2019-03-18 15:02:26 -0700 | [diff] [blame] | 363 | int tx_size_search_method; |
Ryan Lei | 9543c70 | 2019-03-11 11:06:14 -0700 | [diff] [blame] | 364 | int enable_flip_idtx; |
Debargha Mukherjee | 0d8368a | 2018-03-25 12:23:02 -0700 | [diff] [blame] | 365 | int enable_order_hint; |
Debargha Mukherjee | 7ac3eb1 | 2018-12-12 10:26:50 -0800 | [diff] [blame] | 366 | int enable_dist_wtd_comp; |
Debargha Mukherjee | 0187d7c | 2018-03-25 11:37:56 -0700 | [diff] [blame] | 367 | int enable_ref_frame_mvs; |
Urvang Joshi | 9ad9f08 | 2019-01-22 12:31:33 -0800 | [diff] [blame] | 368 | unsigned int max_reference_frames; |
Urvang Joshi | 1a9e27e | 2019-02-19 11:17:23 -0800 | [diff] [blame] | 369 | int enable_reduced_reference_set; |
Debargha Mukherjee | 0d8368a | 2018-03-25 12:23:02 -0700 | [diff] [blame] | 370 | unsigned int allow_ref_frame_mvs; |
Debargha Mukherjee | 16ea6ba | 2018-12-10 12:01:38 -0800 | [diff] [blame] | 371 | int enable_masked_comp; |
Ryan Lei | 9543c70 | 2019-03-11 11:06:14 -0700 | [diff] [blame] | 372 | int enable_onesided_comp; |
Debargha Mukherjee | 16ea6ba | 2018-12-10 12:01:38 -0800 | [diff] [blame] | 373 | int enable_interintra_comp; |
Yue Chen | 8cbee6c | 2018-12-14 11:54:39 -0800 | [diff] [blame] | 374 | int enable_smooth_interintra; |
Debargha Mukherjee | 78e6b2c | 2018-12-11 07:50:57 -0800 | [diff] [blame] | 375 | int enable_diff_wtd_comp; |
| 376 | int enable_interinter_wedge; |
| 377 | int enable_interintra_wedge; |
Debargha Mukherjee | d2e53ca | 2018-12-10 11:34:59 -0800 | [diff] [blame] | 378 | int enable_global_motion; |
Debargha Mukherjee | 37df916 | 2018-03-25 12:48:24 -0700 | [diff] [blame] | 379 | int enable_warped_motion; |
| 380 | int allow_warped_motion; |
Yue Chen | 8f9ca58 | 2018-12-12 15:11:47 -0800 | [diff] [blame] | 381 | int enable_filter_intra; |
Debargha Mukherjee | 078ae86 | 2018-12-18 11:05:43 -0800 | [diff] [blame] | 382 | int enable_smooth_intra; |
| 383 | int enable_paeth_intra; |
Debargha Mukherjee | 8e25678 | 2018-12-18 13:44:13 -0800 | [diff] [blame] | 384 | int enable_cfl_intra; |
Urvang Joshi | 2c92b07 | 2018-03-19 17:23:31 -0700 | [diff] [blame] | 385 | int enable_superres; |
Hui Su | a9e3d14 | 2018-12-14 15:21:21 -0800 | [diff] [blame] | 386 | int enable_palette; |
Hui Su | 440e6d4 | 2018-12-17 11:47:07 -0800 | [diff] [blame] | 387 | int enable_intrabc; |
Hui Su | 94b3817 | 2018-12-19 10:40:29 -0800 | [diff] [blame] | 388 | int enable_angle_delta; |
Soo-Chul Han | 29c46fb | 2018-03-23 16:02:00 -0400 | [diff] [blame] | 389 | unsigned int save_as_annexb; |
Neil Birkbeck | a2893ab | 2018-06-08 14:45:13 -0700 | [diff] [blame] | 390 | |
| 391 | #if CONFIG_DENOISE |
| 392 | float noise_level; |
| 393 | int noise_block_size; |
| 394 | #endif |
Tom Finegan | 02b2a84 | 2018-08-24 13:50:00 -0700 | [diff] [blame] | 395 | |
| 396 | unsigned int chroma_subsampling_x; |
| 397 | unsigned int chroma_subsampling_y; |
Sarah Parker | 0b7e590 | 2018-12-10 13:35:25 -0800 | [diff] [blame] | 398 | int reduced_tx_type_set; |
Debargha Mukherjee | db45815 | 2019-01-25 11:04:09 -0800 | [diff] [blame] | 399 | int use_intra_dct_only; |
| 400 | int use_inter_dct_only; |
Debargha Mukherjee | 042af92 | 2019-02-07 05:58:26 -0800 | [diff] [blame] | 401 | int use_intra_default_tx_only; |
Sarah Parker | 740e839 | 2019-01-23 15:47:53 -0800 | [diff] [blame] | 402 | int quant_b_adapt; |
Debargha Mukherjee | 03ad12d | 2019-02-07 11:17:59 -0800 | [diff] [blame] | 403 | COST_UPDATE_TYPE coeff_cost_upd_freq; |
| 404 | COST_UPDATE_TYPE mode_cost_upd_freq; |
Satish Kumar Suman | 2990996 | 2019-01-09 10:31:21 +0530 | [diff] [blame] | 405 | int border_in_pixels; |
Hui Su | 2aa492c | 2019-03-12 15:18:18 -0700 | [diff] [blame] | 406 | AV1_LEVEL target_seq_level_idx[MAX_NUM_OPERATING_POINTS]; |
Hui Su | d909c2c | 2019-03-08 11:51:14 -0800 | [diff] [blame] | 407 | // Bit mask to specify which tier each of the 32 possible operating points |
| 408 | // conforms to. |
| 409 | unsigned int tier_mask; |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 410 | } AV1EncoderConfig; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 411 | |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 412 | static INLINE int is_lossless_requested(const AV1EncoderConfig *cfg) { |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 413 | return cfg->best_allowed_q == 0 && cfg->worst_allowed_q == 0; |
| 414 | } |
| 415 | |
Yue Chen | cc6a6ef | 2018-05-21 16:21:05 -0700 | [diff] [blame] | 416 | typedef struct FRAME_COUNTS { |
| 417 | // Note: This structure should only contain 'unsigned int' fields, or |
| 418 | // aggregates built solely from 'unsigned int' fields/elements |
| 419 | #if CONFIG_ENTROPY_STATS |
| 420 | unsigned int kf_y_mode[KF_MODE_CONTEXTS][KF_MODE_CONTEXTS][INTRA_MODES]; |
| 421 | unsigned int angle_delta[DIRECTIONAL_MODES][2 * MAX_ANGLE_DELTA + 1]; |
| 422 | unsigned int y_mode[BLOCK_SIZE_GROUPS][INTRA_MODES]; |
| 423 | unsigned int uv_mode[CFL_ALLOWED_TYPES][INTRA_MODES][UV_INTRA_MODES]; |
| 424 | unsigned int cfl_sign[CFL_JOINT_SIGNS]; |
| 425 | unsigned int cfl_alpha[CFL_ALPHA_CONTEXTS][CFL_ALPHABET_SIZE]; |
| 426 | unsigned int palette_y_mode[PALATTE_BSIZE_CTXS][PALETTE_Y_MODE_CONTEXTS][2]; |
| 427 | unsigned int palette_uv_mode[PALETTE_UV_MODE_CONTEXTS][2]; |
| 428 | unsigned int palette_y_size[PALATTE_BSIZE_CTXS][PALETTE_SIZES]; |
| 429 | unsigned int palette_uv_size[PALATTE_BSIZE_CTXS][PALETTE_SIZES]; |
| 430 | unsigned int palette_y_color_index[PALETTE_SIZES] |
| 431 | [PALETTE_COLOR_INDEX_CONTEXTS] |
| 432 | [PALETTE_COLORS]; |
| 433 | unsigned int palette_uv_color_index[PALETTE_SIZES] |
| 434 | [PALETTE_COLOR_INDEX_CONTEXTS] |
| 435 | [PALETTE_COLORS]; |
| 436 | unsigned int partition[PARTITION_CONTEXTS][EXT_PARTITION_TYPES]; |
| 437 | unsigned int txb_skip[TOKEN_CDF_Q_CTXS][TX_SIZES][TXB_SKIP_CONTEXTS][2]; |
| 438 | unsigned int eob_extra[TOKEN_CDF_Q_CTXS][TX_SIZES][PLANE_TYPES] |
| 439 | [EOB_COEF_CONTEXTS][2]; |
| 440 | unsigned int dc_sign[PLANE_TYPES][DC_SIGN_CONTEXTS][2]; |
| 441 | unsigned int coeff_lps[TX_SIZES][PLANE_TYPES][BR_CDF_SIZE - 1][LEVEL_CONTEXTS] |
| 442 | [2]; |
| 443 | unsigned int eob_flag[TX_SIZES][PLANE_TYPES][EOB_COEF_CONTEXTS][2]; |
| 444 | unsigned int eob_multi16[TOKEN_CDF_Q_CTXS][PLANE_TYPES][2][5]; |
| 445 | unsigned int eob_multi32[TOKEN_CDF_Q_CTXS][PLANE_TYPES][2][6]; |
| 446 | unsigned int eob_multi64[TOKEN_CDF_Q_CTXS][PLANE_TYPES][2][7]; |
| 447 | unsigned int eob_multi128[TOKEN_CDF_Q_CTXS][PLANE_TYPES][2][8]; |
| 448 | unsigned int eob_multi256[TOKEN_CDF_Q_CTXS][PLANE_TYPES][2][9]; |
| 449 | unsigned int eob_multi512[TOKEN_CDF_Q_CTXS][PLANE_TYPES][2][10]; |
| 450 | unsigned int eob_multi1024[TOKEN_CDF_Q_CTXS][PLANE_TYPES][2][11]; |
| 451 | unsigned int coeff_lps_multi[TOKEN_CDF_Q_CTXS][TX_SIZES][PLANE_TYPES] |
| 452 | [LEVEL_CONTEXTS][BR_CDF_SIZE]; |
| 453 | unsigned int coeff_base_multi[TOKEN_CDF_Q_CTXS][TX_SIZES][PLANE_TYPES] |
| 454 | [SIG_COEF_CONTEXTS][NUM_BASE_LEVELS + 2]; |
| 455 | unsigned int coeff_base_eob_multi[TOKEN_CDF_Q_CTXS][TX_SIZES][PLANE_TYPES] |
| 456 | [SIG_COEF_CONTEXTS_EOB][NUM_BASE_LEVELS + 1]; |
| 457 | unsigned int newmv_mode[NEWMV_MODE_CONTEXTS][2]; |
| 458 | unsigned int zeromv_mode[GLOBALMV_MODE_CONTEXTS][2]; |
| 459 | unsigned int refmv_mode[REFMV_MODE_CONTEXTS][2]; |
| 460 | unsigned int drl_mode[DRL_MODE_CONTEXTS][2]; |
| 461 | unsigned int inter_compound_mode[INTER_MODE_CONTEXTS][INTER_COMPOUND_MODES]; |
| 462 | unsigned int wedge_idx[BLOCK_SIZES_ALL][16]; |
| 463 | unsigned int interintra[BLOCK_SIZE_GROUPS][2]; |
| 464 | unsigned int interintra_mode[BLOCK_SIZE_GROUPS][INTERINTRA_MODES]; |
| 465 | unsigned int wedge_interintra[BLOCK_SIZES_ALL][2]; |
Debargha Mukherjee | 54eabb5 | 2019-02-01 16:54:33 -0800 | [diff] [blame] | 466 | unsigned int compound_type[BLOCK_SIZES_ALL][MASKED_COMPOUND_TYPES]; |
Yue Chen | cc6a6ef | 2018-05-21 16:21:05 -0700 | [diff] [blame] | 467 | unsigned int motion_mode[BLOCK_SIZES_ALL][MOTION_MODES]; |
| 468 | unsigned int obmc[BLOCK_SIZES_ALL][2]; |
| 469 | unsigned int intra_inter[INTRA_INTER_CONTEXTS][2]; |
| 470 | unsigned int comp_inter[COMP_INTER_CONTEXTS][2]; |
| 471 | unsigned int comp_ref_type[COMP_REF_TYPE_CONTEXTS][2]; |
| 472 | unsigned int uni_comp_ref[UNI_COMP_REF_CONTEXTS][UNIDIR_COMP_REFS - 1][2]; |
| 473 | unsigned int single_ref[REF_CONTEXTS][SINGLE_REFS - 1][2]; |
| 474 | unsigned int comp_ref[REF_CONTEXTS][FWD_REFS - 1][2]; |
| 475 | unsigned int comp_bwdref[REF_CONTEXTS][BWD_REFS - 1][2]; |
| 476 | unsigned int intrabc[2]; |
| 477 | |
| 478 | unsigned int txfm_partition[TXFM_PARTITION_CONTEXTS][2]; |
| 479 | unsigned int intra_tx_size[MAX_TX_CATS][TX_SIZE_CONTEXTS][MAX_TX_DEPTH + 1]; |
| 480 | unsigned int skip_mode[SKIP_MODE_CONTEXTS][2]; |
| 481 | unsigned int skip[SKIP_CONTEXTS][2]; |
| 482 | unsigned int compound_index[COMP_INDEX_CONTEXTS][2]; |
| 483 | unsigned int comp_group_idx[COMP_GROUP_IDX_CONTEXTS][2]; |
| 484 | unsigned int delta_q[DELTA_Q_PROBS][2]; |
| 485 | unsigned int delta_lf_multi[FRAME_LF_COUNT][DELTA_LF_PROBS][2]; |
| 486 | unsigned int delta_lf[DELTA_LF_PROBS][2]; |
| 487 | |
| 488 | unsigned int inter_ext_tx[EXT_TX_SETS_INTER][EXT_TX_SIZES][TX_TYPES]; |
| 489 | unsigned int intra_ext_tx[EXT_TX_SETS_INTRA][EXT_TX_SIZES][INTRA_MODES] |
| 490 | [TX_TYPES]; |
| 491 | unsigned int filter_intra_mode[FILTER_INTRA_MODES]; |
| 492 | unsigned int filter_intra[BLOCK_SIZES_ALL][2]; |
| 493 | unsigned int switchable_restore[RESTORE_SWITCHABLE_TYPES]; |
| 494 | unsigned int wiener_restore[2]; |
| 495 | unsigned int sgrproj_restore[2]; |
| 496 | #endif // CONFIG_ENTROPY_STATS |
| 497 | |
| 498 | unsigned int switchable_interp[SWITCHABLE_FILTER_CONTEXTS] |
| 499 | [SWITCHABLE_FILTERS]; |
| 500 | } FRAME_COUNTS; |
| 501 | |
Angie Chiang | 7157216 | 2018-08-06 16:20:36 -0700 | [diff] [blame] | 502 | #define INTER_MODE_RD_DATA_OVERALL_SIZE 6400 |
| 503 | |
| 504 | typedef struct { |
| 505 | int ready; |
| 506 | double a; |
| 507 | double b; |
| 508 | double dist_mean; |
Angie Chiang | 199f3d4 | 2018-08-14 18:07:02 -0700 | [diff] [blame] | 509 | double ld_mean; |
| 510 | double sse_mean; |
| 511 | double sse_sse_mean; |
| 512 | double sse_ld_mean; |
Angie Chiang | 106fa48 | 2018-08-07 18:28:40 -0700 | [diff] [blame] | 513 | int num; |
| 514 | double dist_sum; |
| 515 | double ld_sum; |
| 516 | double sse_sum; |
| 517 | double sse_sse_sum; |
| 518 | double sse_ld_sum; |
Angie Chiang | 7157216 | 2018-08-06 16:20:36 -0700 | [diff] [blame] | 519 | } InterModeRdModel; |
| 520 | |
| 521 | typedef struct { |
| 522 | int idx; |
| 523 | int64_t rd; |
| 524 | } RdIdxPair; |
| 525 | // TODO(angiebird): This is an estimated size. We still need to figure what is |
| 526 | // the maximum number of modes. |
| 527 | #define MAX_INTER_MODES 1024 |
| 528 | typedef struct inter_modes_info { |
| 529 | int num; |
| 530 | MB_MODE_INFO mbmi_arr[MAX_INTER_MODES]; |
| 531 | int mode_rate_arr[MAX_INTER_MODES]; |
| 532 | int64_t sse_arr[MAX_INTER_MODES]; |
| 533 | int64_t est_rd_arr[MAX_INTER_MODES]; |
| 534 | RdIdxPair rd_idx_pair_arr[MAX_INTER_MODES]; |
elliottk | b8becb2 | 2019-03-04 01:01:30 -0800 | [diff] [blame] | 535 | bool true_rd_arr[MAX_INTER_MODES]; |
elliottk | c8da531 | 2019-03-20 23:47:32 -0700 | [diff] [blame] | 536 | uint8_t blk_skip_arr[MAX_INTER_MODES][MAX_MIB_SIZE * MAX_MIB_SIZE]; |
elliottk | b8becb2 | 2019-03-04 01:01:30 -0800 | [diff] [blame] | 537 | RD_STATS rd_cost_arr[MAX_INTER_MODES]; |
| 538 | RD_STATS rd_cost_y_arr[MAX_INTER_MODES]; |
| 539 | RD_STATS rd_cost_uv_arr[MAX_INTER_MODES]; |
Angie Chiang | 7157216 | 2018-08-06 16:20:36 -0700 | [diff] [blame] | 540 | } InterModesInfo; |
Angie Chiang | 7157216 | 2018-08-06 16:20:36 -0700 | [diff] [blame] | 541 | |
Ravi Chaudhary | 40cdf13 | 2018-10-08 11:04:16 +0530 | [diff] [blame] | 542 | // Encoder row synchronization |
| 543 | typedef struct AV1RowMTSyncData { |
| 544 | #if CONFIG_MULTITHREAD |
| 545 | pthread_mutex_t *mutex_; |
| 546 | pthread_cond_t *cond_; |
| 547 | #endif |
| 548 | // Allocate memory to store the sb/mb block index in each row. |
| 549 | int *cur_col; |
| 550 | int sync_range; |
| 551 | int rows; |
| 552 | } AV1RowMTSync; |
| 553 | |
Ravi Chaudhary | 90a15f4 | 2018-10-11 18:56:35 +0530 | [diff] [blame] | 554 | typedef struct AV1RowMTInfo { |
| 555 | int current_mi_row; |
| 556 | int num_threads_working; |
| 557 | } AV1RowMTInfo; |
| 558 | |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 559 | // TODO(jingning) All spatially adaptive variables should go to TileDataEnc. |
| 560 | typedef struct TileDataEnc { |
| 561 | TileInfo tile_info; |
Rupert Swarbrick | 93c39e9 | 2017-07-12 11:11:02 +0100 | [diff] [blame] | 562 | int thresh_freq_fact[BLOCK_SIZES_ALL][MAX_MODES]; |
Yunqing Wang | 8c1e57c | 2016-10-25 15:15:23 -0700 | [diff] [blame] | 563 | int m_search_count; |
| 564 | int ex_search_count; |
Luc Trudeau | f816415 | 2017-04-11 16:20:51 -0400 | [diff] [blame] | 565 | CFL_CTX cfl; |
Jingning Han | 9f07be1 | 2017-04-13 09:31:40 -0700 | [diff] [blame] | 566 | DECLARE_ALIGNED(16, FRAME_CONTEXT, tctx); |
Ravi Chaudhary | e521517 | 2018-12-21 18:47:25 +0530 | [diff] [blame] | 567 | FRAME_CONTEXT *row_ctx; |
Yunqing Wang | 0e141b5 | 2017-11-02 15:08:58 -0700 | [diff] [blame] | 568 | uint8_t allow_update_cdf; |
Angie Chiang | 7157216 | 2018-08-06 16:20:36 -0700 | [diff] [blame] | 569 | InterModeRdModel inter_mode_rd_models[BLOCK_SIZES_ALL]; |
Ravi Chaudhary | 40cdf13 | 2018-10-08 11:04:16 +0530 | [diff] [blame] | 570 | AV1RowMTSync row_mt_sync; |
Ravi Chaudhary | 90a15f4 | 2018-10-11 18:56:35 +0530 | [diff] [blame] | 571 | AV1RowMTInfo row_mt_info; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 572 | } TileDataEnc; |
| 573 | |
Ravi Chaudhary | 73cf15b | 2018-08-30 10:52:51 +0530 | [diff] [blame] | 574 | typedef struct { |
| 575 | TOKENEXTRA *start; |
| 576 | TOKENEXTRA *stop; |
| 577 | unsigned int count; |
| 578 | } TOKENLIST; |
| 579 | |
Ravi Chaudhary | c5e7469 | 2018-10-08 16:05:38 +0530 | [diff] [blame] | 580 | typedef struct MultiThreadHandle { |
| 581 | int allocated_tile_rows; |
| 582 | int allocated_tile_cols; |
| 583 | int allocated_sb_rows; |
Ravi Chaudhary | 90a15f4 | 2018-10-11 18:56:35 +0530 | [diff] [blame] | 584 | int thread_id_to_tile_id[MAX_NUM_THREADS]; // Mapping of threads to tiles |
Ravi Chaudhary | c5e7469 | 2018-10-08 16:05:38 +0530 | [diff] [blame] | 585 | } MultiThreadHandle; |
| 586 | |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 587 | typedef struct RD_COUNTS { |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 588 | int64_t comp_pred_diff[REFERENCE_MODES]; |
Debargha Mukherjee | a575d23 | 2017-04-28 17:46:47 -0700 | [diff] [blame] | 589 | // Stores number of 4x4 blocks using global motion per reference frame. |
Zoe Liu | 27deb38 | 2018-03-27 15:13:56 -0700 | [diff] [blame] | 590 | int global_motion_used[REF_FRAMES]; |
Arild Fuldseth (arilfuld) | 6c20c78 | 2017-06-15 09:45:02 +0200 | [diff] [blame] | 591 | int compound_ref_used_flag; |
Zoe Liu | 8a5d343 | 2017-11-30 16:33:44 -0800 | [diff] [blame] | 592 | int skip_mode_used_flag; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 593 | } RD_COUNTS; |
| 594 | |
| 595 | typedef struct ThreadData { |
| 596 | MACROBLOCK mb; |
| 597 | RD_COUNTS rd_counts; |
| 598 | FRAME_COUNTS *counts; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 599 | PC_TREE *pc_tree; |
| 600 | PC_TREE *pc_root[MAX_MIB_SIZE_LOG2 - MIN_MIB_SIZE_LOG2 + 1]; |
wenyao.liu | 0c6f5fd | 2018-10-23 18:41:16 +0800 | [diff] [blame] | 601 | tran_low_t *tree_coeff_buf[MAX_MB_PLANE]; |
| 602 | tran_low_t *tree_qcoeff_buf[MAX_MB_PLANE]; |
| 603 | tran_low_t *tree_dqcoeff_buf[MAX_MB_PLANE]; |
Ravi Chaudhary | 5d970f4 | 2018-09-25 11:25:32 +0530 | [diff] [blame] | 604 | InterModesInfo *inter_modes_info; |
Ravi Chaudhary | 783d6a3 | 2018-08-28 18:21:02 +0530 | [diff] [blame] | 605 | uint32_t *hash_value_buffer[2][2]; |
Jingning Han | d064cf0 | 2017-06-01 10:00:39 -0700 | [diff] [blame] | 606 | int32_t *wsrc_buf; |
| 607 | int32_t *mask_buf; |
| 608 | uint8_t *above_pred_buf; |
| 609 | uint8_t *left_pred_buf; |
hui su | 5d49314 | 2017-05-08 12:06:12 -0700 | [diff] [blame] | 610 | PALETTE_BUFFER *palette_buffer; |
Urvang Joshi | 0a4cfad | 2018-09-07 11:10:39 -0700 | [diff] [blame] | 611 | CONV_BUF_TYPE *tmp_conv_dst; |
| 612 | uint8_t *tmp_obmc_bufs[2]; |
Ravi Chaudhary | 00525ef | 2018-10-31 19:52:42 +0530 | [diff] [blame] | 613 | int intrabc_used; |
Ravi Chaudhary | 84a280a | 2018-09-24 16:09:48 +0530 | [diff] [blame] | 614 | FRAME_CONTEXT *tctx; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 615 | } ThreadData; |
| 616 | |
| 617 | struct EncWorkerData; |
| 618 | |
| 619 | typedef struct ActiveMap { |
| 620 | int enabled; |
| 621 | int update; |
| 622 | unsigned char *map; |
| 623 | } ActiveMap; |
| 624 | |
Wan-Teh Chang | c25c92a | 2018-04-23 15:04:14 -0700 | [diff] [blame] | 625 | #if CONFIG_INTERNAL_STATS |
| 626 | // types of stats |
Satish Kumar Suman | 4667aa1 | 2018-12-14 18:28:19 +0530 | [diff] [blame] | 627 | enum { |
Wan-Teh Chang | c25c92a | 2018-04-23 15:04:14 -0700 | [diff] [blame] | 628 | STAT_Y, |
| 629 | STAT_U, |
| 630 | STAT_V, |
| 631 | STAT_ALL, |
| 632 | NUM_STAT_TYPES // This should always be the last member of the enum |
Satish Kumar Suman | 4667aa1 | 2018-12-14 18:28:19 +0530 | [diff] [blame] | 633 | } UENUM1BYTE(StatType); |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 634 | |
| 635 | typedef struct IMAGE_STAT { |
Urvang Joshi | b5ed350 | 2016-10-17 16:38:05 -0700 | [diff] [blame] | 636 | double stat[NUM_STAT_TYPES]; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 637 | double worst; |
| 638 | } ImageStat; |
Wan-Teh Chang | c25c92a | 2018-04-23 15:04:14 -0700 | [diff] [blame] | 639 | #endif // CONFIG_INTERNAL_STATS |
Urvang Joshi | b5ed350 | 2016-10-17 16:38:05 -0700 | [diff] [blame] | 640 | |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 641 | typedef struct { |
| 642 | int ref_count; |
| 643 | YV12_BUFFER_CONFIG buf; |
| 644 | } EncRefCntBuffer; |
| 645 | |
chiyotsai | 9c484b3 | 2019-03-07 16:01:50 -0800 | [diff] [blame] | 646 | #if CONFIG_COLLECT_PARTITION_STATS == 2 |
chiyotsai | 92ed0dd | 2019-01-25 14:50:14 -0800 | [diff] [blame] | 647 | typedef struct PartitionStats { |
| 648 | int partition_decisions[6][EXT_PARTITION_TYPES]; |
| 649 | int partition_attempts[6][EXT_PARTITION_TYPES]; |
chiyotsai | 9c484b3 | 2019-03-07 16:01:50 -0800 | [diff] [blame] | 650 | int64_t partition_times[6][EXT_PARTITION_TYPES]; |
chiyotsai | 92ed0dd | 2019-01-25 14:50:14 -0800 | [diff] [blame] | 651 | |
| 652 | int partition_redo; |
| 653 | } PartitionStats; |
| 654 | #endif |
| 655 | |
Yunqing Wang | ba25558 | 2019-02-11 16:21:12 -0800 | [diff] [blame] | 656 | #if CONFIG_COLLECT_COMPONENT_TIMING |
| 657 | #include "aom_ports/aom_timer.h" |
| 658 | // Adjust the following to add new components. |
| 659 | enum { |
Yunqing Wang | d1f32e6 | 2019-02-20 10:37:51 -0800 | [diff] [blame] | 660 | encode_frame_to_data_rate_time, |
| 661 | encode_with_recode_loop_time, |
| 662 | loop_filter_time, |
| 663 | cdef_time, |
| 664 | loop_restoration_time, |
| 665 | av1_pack_bitstream_final_time, |
| 666 | av1_encode_frame_time, |
| 667 | av1_compute_global_motion_time, |
| 668 | av1_setup_motion_field_time, |
Yunqing Wang | ba25558 | 2019-02-11 16:21:12 -0800 | [diff] [blame] | 669 | encode_sb_time, |
Yunqing Wang | d1f32e6 | 2019-02-20 10:37:51 -0800 | [diff] [blame] | 670 | first_partition_search_pass_time, |
Yunqing Wang | ba25558 | 2019-02-11 16:21:12 -0800 | [diff] [blame] | 671 | rd_pick_partition_time, |
| 672 | rd_pick_sb_modes_time, |
| 673 | av1_rd_pick_intra_mode_sb_time, |
| 674 | av1_rd_pick_inter_mode_sb_time, |
Yunqing Wang | d1f32e6 | 2019-02-20 10:37:51 -0800 | [diff] [blame] | 675 | handle_intra_mode_time, |
Yunqing Wang | ba25558 | 2019-02-11 16:21:12 -0800 | [diff] [blame] | 676 | handle_inter_mode_time, |
Yunqing Wang | d1f32e6 | 2019-02-20 10:37:51 -0800 | [diff] [blame] | 677 | do_tx_search_time, |
| 678 | handle_newmv_time, |
| 679 | compound_type_rd_time, |
| 680 | interpolation_filter_search_time, |
Yunqing Wang | ba25558 | 2019-02-11 16:21:12 -0800 | [diff] [blame] | 681 | motion_mode_rd_time, |
| 682 | kTimingComponents, |
| 683 | } UENUM1BYTE(TIMING_COMPONENT); |
| 684 | |
| 685 | static INLINE char const *get_component_name(int index) { |
| 686 | switch (index) { |
Yunqing Wang | d1f32e6 | 2019-02-20 10:37:51 -0800 | [diff] [blame] | 687 | case encode_frame_to_data_rate_time: |
| 688 | return "encode_frame_to_data_rate_time"; |
| 689 | case encode_with_recode_loop_time: return "encode_with_recode_loop_time"; |
| 690 | case loop_filter_time: return "loop_filter_time"; |
| 691 | case cdef_time: return "cdef_time"; |
| 692 | case loop_restoration_time: return "loop_restoration_time"; |
| 693 | case av1_pack_bitstream_final_time: return "av1_pack_bitstream_final_time"; |
| 694 | case av1_encode_frame_time: return "av1_encode_frame_time"; |
| 695 | case av1_compute_global_motion_time: |
| 696 | return "av1_compute_global_motion_time"; |
| 697 | case av1_setup_motion_field_time: return "av1_setup_motion_field_time"; |
| 698 | case encode_sb_time: return "encode_sb_time"; |
| 699 | case first_partition_search_pass_time: |
| 700 | return "first_partition_search_pass_time"; |
| 701 | case rd_pick_partition_time: return "rd_pick_partition_time"; |
| 702 | case rd_pick_sb_modes_time: return "rd_pick_sb_modes_time"; |
| 703 | case av1_rd_pick_intra_mode_sb_time: |
| 704 | return "av1_rd_pick_intra_mode_sb_time"; |
| 705 | case av1_rd_pick_inter_mode_sb_time: |
| 706 | return "av1_rd_pick_inter_mode_sb_time"; |
| 707 | case handle_intra_mode_time: return "handle_intra_mode_time"; |
| 708 | case handle_inter_mode_time: return "handle_inter_mode_time"; |
| 709 | case do_tx_search_time: return "do_tx_search_time"; |
| 710 | case handle_newmv_time: return "handle_newmv_time"; |
| 711 | case compound_type_rd_time: return "compound_type_rd_time"; |
| 712 | case interpolation_filter_search_time: |
| 713 | return "interpolation_filter_search_time"; |
| 714 | case motion_mode_rd_time: return "motion_mode_rd_time"; |
Yunqing Wang | ba25558 | 2019-02-11 16:21:12 -0800 | [diff] [blame] | 715 | default: assert(0); |
| 716 | } |
| 717 | return "error"; |
| 718 | } |
| 719 | #endif |
| 720 | |
Urvang Joshi | f70375a | 2019-03-22 23:30:19 -0700 | [diff] [blame] | 721 | // The maximum number of internal ARFs except ALTREF_FRAME |
| 722 | #define MAX_INTERNAL_ARFS (REF_FRAMES - BWDREF_FRAME - 1) |
| 723 | |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 724 | typedef struct AV1_COMP { |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 725 | QUANTS quants; |
| 726 | ThreadData td; |
Yue Chen | cc6a6ef | 2018-05-21 16:21:05 -0700 | [diff] [blame] | 727 | FRAME_COUNTS counts; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 728 | MB_MODE_INFO_EXT *mbmi_ext_base; |
Jingning Han | f5a4d3b | 2017-08-27 23:01:19 -0700 | [diff] [blame] | 729 | CB_COEFF_BUFFER *coeff_buffer_base; |
Yi Luo | c621023 | 2017-05-25 15:09:25 -0700 | [diff] [blame] | 730 | Dequants dequants; |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 731 | AV1_COMMON common; |
| 732 | AV1EncoderConfig oxcf; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 733 | struct lookahead_ctx *lookahead; |
| 734 | struct lookahead_entry *alt_ref_source; |
Sarah Parker | af32a7b | 2018-06-29 14:59:05 -0700 | [diff] [blame] | 735 | int no_show_kf; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 736 | |
Angie Chiang | e69a682 | 2018-03-16 13:52:44 -0700 | [diff] [blame] | 737 | int optimize_seg_arr[MAX_SEGMENTS]; |
| 738 | |
Alex Converse | f77fd0b | 2017-04-20 11:00:24 -0700 | [diff] [blame] | 739 | YV12_BUFFER_CONFIG *source; |
| 740 | YV12_BUFFER_CONFIG *last_source; // NULL for first frame and alt_ref frames |
Fergus Simpson | d2bcbb5 | 2017-05-22 23:15:05 -0700 | [diff] [blame] | 741 | YV12_BUFFER_CONFIG *unscaled_source; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 742 | YV12_BUFFER_CONFIG scaled_source; |
| 743 | YV12_BUFFER_CONFIG *unscaled_last_source; |
| 744 | YV12_BUFFER_CONFIG scaled_last_source; |
| 745 | |
Yue Chen | 7cae98f | 2018-08-24 10:43:16 -0700 | [diff] [blame] | 746 | TplDepFrame tpl_stats[MAX_LAG_BUFFERS]; |
| 747 | YV12_BUFFER_CONFIG *tpl_recon_frames[INTER_REFS_PER_FRAME + 1]; |
| 748 | |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 749 | // For a still frame, this flag is set to 1 to skip partition search. |
| 750 | int partition_search_skippable_frame; |
Yunqing Wang | 8025b88 | 2019-03-01 10:53:27 -0800 | [diff] [blame] | 751 | // The following item corresponds to two_pass_partition_search speed features. |
Yunqing Wang | c3e21ec | 2019-02-28 16:29:27 -0800 | [diff] [blame] | 752 | int two_pass_partition_search; |
Yunqing Wang | c3e21ec | 2019-02-28 16:29:27 -0800 | [diff] [blame] | 753 | |
RogerZhou | 3b63524 | 2017-09-19 10:06:46 -0700 | [diff] [blame] | 754 | double csm_rate_array[32]; |
| 755 | double m_rate_array[32]; |
| 756 | int rate_size; |
| 757 | int rate_index; |
Debargha Mukherjee | e41a667 | 2018-02-27 11:56:31 -0800 | [diff] [blame] | 758 | hash_table *previous_hash_table; |
| 759 | int previous_index; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 760 | |
Ravi Chaudhary | 0ec03a4 | 2018-08-17 18:53:28 +0530 | [diff] [blame] | 761 | unsigned int row_mt; |
David Turner | e7ebf90 | 2018-12-04 14:04:55 +0000 | [diff] [blame] | 762 | RefCntBuffer *scaled_ref_buf[INTER_REFS_PER_FRAME]; |
Urvang Joshi | 4d9f15f | 2018-11-05 15:26:22 -0800 | [diff] [blame] | 763 | |
David Turner | e7ebf90 | 2018-12-04 14:04:55 +0000 | [diff] [blame] | 764 | RefCntBuffer *last_show_frame_buf; // last show frame buffer |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 765 | |
Urvang Joshi | 06b3765 | 2018-11-06 11:17:56 -0800 | [diff] [blame] | 766 | // refresh_*_frame are boolean flags. If 'refresh_xyz_frame' is true, then |
| 767 | // after the current frame is encoded, the XYZ reference frame gets refreshed |
| 768 | // (updated) to be the current frame. |
| 769 | // |
| 770 | // Special case: 'refresh_last_frame' specifies that: |
| 771 | // - LAST_FRAME reference should be updated to be the current frame (as usual) |
| 772 | // - Also, LAST2_FRAME and LAST3_FRAME references are implicitly updated to be |
| 773 | // the two past reference frames just before LAST_FRAME that are available. |
| 774 | // |
| 775 | // Note: Usually at most one of these refresh flags is true at a time. |
| 776 | // But a key-frame is special, for which all the flags are true at once. |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 777 | int refresh_last_frame; |
| 778 | int refresh_golden_frame; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 779 | int refresh_bwd_ref_frame; |
Zoe Liu | e9b15e2 | 2017-07-19 15:53:01 -0700 | [diff] [blame] | 780 | int refresh_alt2_ref_frame; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 781 | int refresh_alt_ref_frame; |
Urvang Joshi | 06b3765 | 2018-11-06 11:17:56 -0800 | [diff] [blame] | 782 | |
David Turner | 99e990e | 2018-12-10 12:54:26 +0000 | [diff] [blame] | 783 | // For each type of reference frame, this contains the index of a reference |
| 784 | // frame buffer for a reference frame of the same type. We use this to |
| 785 | // choose our primary reference frame (which is the most recent reference |
| 786 | // frame of the same type as the current frame). |
| 787 | int fb_of_context_type[REF_FRAMES]; |
| 788 | |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 789 | int ext_refresh_frame_flags_pending; |
| 790 | int ext_refresh_last_frame; |
| 791 | int ext_refresh_golden_frame; |
Yunqing Wang | 9a50fec | 2017-11-02 17:02:00 -0700 | [diff] [blame] | 792 | int ext_refresh_bwd_ref_frame; |
| 793 | int ext_refresh_alt2_ref_frame; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 794 | int ext_refresh_alt_ref_frame; |
| 795 | |
| 796 | int ext_refresh_frame_context_pending; |
| 797 | int ext_refresh_frame_context; |
sarahparker | 21dbca4 | 2018-03-30 17:43:44 -0700 | [diff] [blame] | 798 | int ext_use_ref_frame_mvs; |
sarahparker | 27d686a | 2018-03-30 17:43:44 -0700 | [diff] [blame] | 799 | int ext_use_error_resilient; |
sarahparker | 9806fed | 2018-03-30 17:43:44 -0700 | [diff] [blame] | 800 | int ext_use_s_frame; |
Sarah Parker | 50b6d6e | 2018-04-11 19:21:54 -0700 | [diff] [blame] | 801 | int ext_use_primary_ref_none; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 802 | |
| 803 | YV12_BUFFER_CONFIG last_frame_uf; |
Debargha Mukherjee | 999d2f6 | 2016-12-15 13:23:21 -0800 | [diff] [blame] | 804 | YV12_BUFFER_CONFIG trial_frame_rst; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 805 | |
| 806 | // Ambient reconstruction err target for force key frames |
| 807 | int64_t ambient_err; |
| 808 | |
| 809 | RD_OPT rd; |
| 810 | |
| 811 | CODING_CONTEXT coding_context; |
| 812 | |
Yue Chen | b23d00a | 2017-07-28 17:01:21 -0700 | [diff] [blame] | 813 | int gmtype_cost[TRANS_TYPES]; |
Zoe Liu | 27deb38 | 2018-03-27 15:13:56 -0700 | [diff] [blame] | 814 | int gmparams_cost[REF_FRAMES]; |
Yue Chen | b23d00a | 2017-07-28 17:01:21 -0700 | [diff] [blame] | 815 | |
Jingning Han | f050fc1 | 2018-03-09 14:53:33 -0800 | [diff] [blame] | 816 | int nmv_costs[2][MV_VALS]; |
| 817 | int nmv_costs_hp[2][MV_VALS]; |
James Zern | 01a9d70 | 2017-08-25 19:09:33 +0000 | [diff] [blame] | 818 | |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 819 | int64_t last_time_stamp_seen; |
| 820 | int64_t last_end_time_stamp_seen; |
| 821 | int64_t first_time_stamp_ever; |
| 822 | |
| 823 | RATE_CONTROL rc; |
| 824 | double framerate; |
| 825 | |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 826 | struct aom_codec_pkt_list *output_pkt_list; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 827 | |
| 828 | MBGRAPH_FRAME_STATS mbgraph_stats[MAX_LAG_BUFFERS]; |
| 829 | int mbgraph_n_frames; // number of frames filled in the above |
| 830 | int static_mb_pct; // % forced skip mbs by segmentation |
| 831 | int ref_frame_flags; |
Yunqing Wang | f2e7a39 | 2017-11-08 00:27:21 -0800 | [diff] [blame] | 832 | int ext_ref_frame_flags; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 833 | |
David Turner | 04b70d8 | 2019-01-24 15:39:19 +0000 | [diff] [blame] | 834 | // speed is passed as a per-frame parameter into the encoder |
| 835 | int speed; |
| 836 | // sf contains fine-grained config set internally based on speed |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 837 | SPEED_FEATURES sf; |
| 838 | |
| 839 | unsigned int max_mv_magnitude; |
| 840 | int mv_step_param; |
| 841 | |
Zoe Liu | 77fb5be | 2017-11-02 14:36:19 -0700 | [diff] [blame] | 842 | int all_one_sided_refs; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 843 | |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 844 | uint8_t *segmentation_map; |
| 845 | |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 846 | CYCLIC_REFRESH *cyclic_refresh; |
| 847 | ActiveMap active_map; |
| 848 | |
| 849 | fractional_mv_step_fp *find_fractional_mv_step; |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 850 | av1_diamond_search_fn_t diamond_search_sad; |
Rupert Swarbrick | 93c39e9 | 2017-07-12 11:11:02 +0100 | [diff] [blame] | 851 | aom_variance_fn_ptr_t fn_ptr[BLOCK_SIZES_ALL]; |
Yunqing Wang | d1f32e6 | 2019-02-20 10:37:51 -0800 | [diff] [blame] | 852 | |
| 853 | #if CONFIG_INTERNAL_STATS |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 854 | uint64_t time_receive_data; |
| 855 | uint64_t time_compress_data; |
Yunqing Wang | d1f32e6 | 2019-02-20 10:37:51 -0800 | [diff] [blame] | 856 | #endif |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 857 | |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 858 | TWO_PASS twopass; |
| 859 | |
| 860 | YV12_BUFFER_CONFIG alt_ref_buffer; |
| 861 | |
| 862 | #if CONFIG_INTERNAL_STATS |
| 863 | unsigned int mode_chosen_counts[MAX_MODES]; |
| 864 | |
| 865 | int count; |
| 866 | uint64_t total_sq_error; |
| 867 | uint64_t total_samples; |
| 868 | ImageStat psnr; |
| 869 | |
| 870 | double total_blockiness; |
| 871 | double worst_blockiness; |
| 872 | |
| 873 | int bytes; |
| 874 | double summed_quality; |
| 875 | double summed_weights; |
| 876 | unsigned int tot_recode_hits; |
| 877 | double worst_ssim; |
| 878 | |
| 879 | ImageStat fastssim; |
| 880 | ImageStat psnrhvs; |
| 881 | |
| 882 | int b_calculate_blockiness; |
| 883 | int b_calculate_consistency; |
| 884 | |
| 885 | double total_inconsistency; |
| 886 | double worst_consistency; |
| 887 | Ssimv *ssim_vars; |
| 888 | Metrics metrics; |
| 889 | #endif |
| 890 | int b_calculate_psnr; |
Debargha Mukherjee | 0857e66 | 2019-01-04 16:22:09 -0800 | [diff] [blame] | 891 | #if CONFIG_SPEED_STATS |
| 892 | unsigned int tx_search_count; |
| 893 | #endif // CONFIG_SPEED_STATS |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 894 | |
| 895 | int droppable; |
| 896 | |
| 897 | int initial_width; |
| 898 | int initial_height; |
| 899 | int initial_mbs; // Number of MBs in the full-size frame; to be used to |
| 900 | // normalize the firstpass stats. This will differ from the |
| 901 | // number of MBs in the current frame when the frame is |
| 902 | // scaled. |
| 903 | |
Debargha Mukherjee | ccb2726 | 2017-09-25 14:19:46 -0700 | [diff] [blame] | 904 | // When resize is triggered through external control, the desired width/height |
| 905 | // are stored here until use in the next frame coded. They are effective only |
| 906 | // for |
| 907 | // one frame and are reset after use. |
| 908 | int resize_pending_width; |
| 909 | int resize_pending_height; |
| 910 | |
Satish Kumar Suman | 1de4688 | 2019-02-27 17:24:18 +0530 | [diff] [blame] | 911 | // ss_cfg[SS_CFG_LOOKAHEAD] : used in following cases |
| 912 | // -> temporal filtering |
| 913 | // -> intrabc |
| 914 | // ss_cfg[SS_CFG_SRC] : used everywhere except above mentioned cases |
Satish Kumar Suman | e6d0be5 | 2019-02-14 14:33:28 +0530 | [diff] [blame] | 915 | search_site_config ss_cfg[SS_CFG_TOTAL]; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 916 | |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 917 | TileDataEnc *tile_data; |
| 918 | int allocated_tiles; // Keep track of memory allocated for tiles. |
| 919 | |
| 920 | TOKENEXTRA *tile_tok[MAX_TILE_ROWS][MAX_TILE_COLS]; |
Ravi Chaudhary | 73cf15b | 2018-08-30 10:52:51 +0530 | [diff] [blame] | 921 | TOKENLIST *tplist[MAX_TILE_ROWS][MAX_TILE_COLS]; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 922 | |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 923 | int resize_state; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 924 | int resize_avg_qp; |
| 925 | int resize_buffer_underflow; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 926 | |
Debargha Mukherjee | f2e5bb3 | 2018-03-26 14:35:24 -0700 | [diff] [blame] | 927 | // Sequence parameters have been transmitted already and locked |
| 928 | // or not. Once locked av1_change_config cannot change the seq |
| 929 | // parameters. |
| 930 | int seq_params_locked; |
| 931 | |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 932 | // VARIANCE_AQ segment map refresh |
| 933 | int vaq_refresh; |
| 934 | |
kyslov | 7b9d0d6 | 2018-12-21 11:12:26 -0800 | [diff] [blame] | 935 | // VAR_BASED_PARTITION thresholds |
| 936 | // 0 - threshold_128x128; 1 - threshold_64x64; |
| 937 | // 2 - threshold_32x32; 3 - threshold_16x16; |
| 938 | // 4 - vbp_threshold_8x8; |
| 939 | int64_t vbp_thresholds[5]; |
| 940 | int64_t vbp_threshold_minmax; |
| 941 | int64_t vbp_threshold_sad; |
| 942 | int64_t vbp_threshold_copy; |
| 943 | BLOCK_SIZE vbp_bsize_min; |
| 944 | |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 945 | // Multi-threading |
| 946 | int num_workers; |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 947 | AVxWorker *workers; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 948 | struct EncWorkerData *tile_thr_data; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 949 | int existing_fb_idx_to_show; |
Urvang Joshi | f70375a | 2019-03-22 23:30:19 -0700 | [diff] [blame] | 950 | int is_arf_filter_off[MAX_INTERNAL_ARFS + 1]; |
Debargha Mukherjee | b98a702 | 2016-11-15 16:07:12 -0800 | [diff] [blame] | 951 | int global_motion_search_done; |
Urvang Joshi | f70375a | 2019-03-22 23:30:19 -0700 | [diff] [blame] | 952 | int internal_altref_allowed; |
Hui Su | 8587878 | 2017-11-07 14:56:31 -0800 | [diff] [blame] | 953 | // A flag to indicate if intrabc is ever used in current frame. |
| 954 | int intrabc_used; |
Hui Su | dfcbfbd | 2017-11-13 12:05:30 -0800 | [diff] [blame] | 955 | int dv_cost[2][MV_VALS]; |
| 956 | // TODO(huisu@google.com): we can update dv_joint_cost per SB. |
| 957 | int dv_joint_cost[MV_JOINTS]; |
Hui Su | ad7551e | 2018-03-14 11:13:31 -0700 | [diff] [blame] | 958 | int has_lossless_segment; |
Zoe Liu | 1d90ceb | 2018-04-16 16:53:37 -0700 | [diff] [blame] | 959 | |
Remya | 6924f4a | 2018-11-26 12:27:32 +0530 | [diff] [blame] | 960 | // Factors to control gating of compound type selection based on best |
| 961 | // approximate rd so far |
| 962 | int max_comp_type_rd_threshold_mul; |
| 963 | int max_comp_type_rd_threshold_div; |
| 964 | |
Venkat | b62a64b | 2019-01-03 17:50:40 +0530 | [diff] [blame] | 965 | unsigned int tx_domain_dist_threshold; |
| 966 | |
Venkat | 4806765 | 2019-01-09 17:59:53 +0530 | [diff] [blame] | 967 | // Factor to control R-D optimization of coeffs based on block |
| 968 | // mse. |
| 969 | unsigned int coeff_opt_dist_threshold; |
| 970 | |
Deepa K G | 964e72e | 2018-05-16 16:56:01 +0530 | [diff] [blame] | 971 | AV1LfSync lf_row_sync; |
Ravi Chaudhary | e2aa401 | 2018-06-04 14:20:00 +0530 | [diff] [blame] | 972 | AV1LrSync lr_row_sync; |
Ravi Chaudhary | ce0f5fc | 2018-05-30 16:19:32 +0530 | [diff] [blame] | 973 | AV1LrStruct lr_ctxt; |
Neil Birkbeck | a2893ab | 2018-06-08 14:45:13 -0700 | [diff] [blame] | 974 | |
| 975 | aom_film_grain_table_t *film_grain_table; |
| 976 | #if CONFIG_DENOISE |
| 977 | struct aom_denoise_and_model_t *denoise_and_model; |
| 978 | #endif |
Ranjit Kumar Tulabandu | 8105bf4 | 2018-07-27 14:16:55 +0530 | [diff] [blame] | 979 | // Stores the default value of skip flag depending on chroma format |
| 980 | // Set as 1 for monochrome and 3 for other color formats |
| 981 | int default_interp_skip_flags; |
Wei-Ting Lin | a8c0245 | 2018-08-13 11:04:06 -0700 | [diff] [blame] | 982 | int preserve_arf_as_gld; |
Ravi Chaudhary | c5e7469 | 2018-10-08 16:05:38 +0530 | [diff] [blame] | 983 | MultiThreadHandle multi_thread_ctxt; |
Ravi Chaudhary | 40cdf13 | 2018-10-08 11:04:16 +0530 | [diff] [blame] | 984 | void (*row_mt_sync_read_ptr)(AV1RowMTSync *const, int, int); |
| 985 | void (*row_mt_sync_write_ptr)(AV1RowMTSync *const, int, int, const int); |
Ravi Chaudhary | 90a15f4 | 2018-10-11 18:56:35 +0530 | [diff] [blame] | 986 | #if CONFIG_MULTITHREAD |
| 987 | pthread_mutex_t *row_mt_mutex_; |
| 988 | #endif |
Aniket Dhok | f6d7ed8 | 2019-01-04 14:05:57 +0530 | [diff] [blame] | 989 | // Set if screen content is set or relevant tools are enabled |
| 990 | int is_screen_content_type; |
chiyotsai | 9c484b3 | 2019-03-07 16:01:50 -0800 | [diff] [blame] | 991 | #if CONFIG_COLLECT_PARTITION_STATS == 2 |
chiyotsai | 92ed0dd | 2019-01-25 14:50:14 -0800 | [diff] [blame] | 992 | PartitionStats partition_stats; |
| 993 | #endif |
Yunqing Wang | ba25558 | 2019-02-11 16:21:12 -0800 | [diff] [blame] | 994 | |
| 995 | #if CONFIG_COLLECT_COMPONENT_TIMING |
| 996 | // component_time[] are initialized to zero while encoder starts. |
| 997 | uint64_t component_time[kTimingComponents]; |
| 998 | struct aom_usec_timer component_timer[kTimingComponents]; |
| 999 | // frame_component_time[] are initialized to zero at beginning of each frame. |
| 1000 | uint64_t frame_component_time[kTimingComponents]; |
| 1001 | #endif |
Hui Su | 81a59f1 | 2019-03-26 16:47:52 -0700 | [diff] [blame] | 1002 | |
| 1003 | // The following data are for AV1 bitstream levels. |
Hui Su | 2aa492c | 2019-03-12 15:18:18 -0700 | [diff] [blame] | 1004 | AV1_LEVEL target_seq_level_idx[MAX_NUM_OPERATING_POINTS]; |
kyslov | abeeb7c | 2019-03-06 18:35:04 -0800 | [diff] [blame] | 1005 | int keep_level_stats; |
Hui Su | a1ff3f8 | 2019-03-12 11:59:01 -0700 | [diff] [blame] | 1006 | AV1LevelInfo level_info[MAX_NUM_OPERATING_POINTS]; |
Hui Su | 4fd1176 | 2019-03-26 16:05:07 -0700 | [diff] [blame] | 1007 | // Count the number of OBU_FRAME and OBU_FRAME_HEADER for level calculation. |
| 1008 | int frame_header_count; |
Hui Su | 81a59f1 | 2019-03-26 16:47:52 -0700 | [diff] [blame] | 1009 | FrameWindowBuffer frame_window_buffer; |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 1010 | } AV1_COMP; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 1011 | |
David Turner | cb5e36f | 2019-01-17 17:15:25 +0000 | [diff] [blame] | 1012 | typedef struct { |
| 1013 | YV12_BUFFER_CONFIG *source; |
| 1014 | YV12_BUFFER_CONFIG *last_source; |
| 1015 | int64_t ts_duration; |
| 1016 | } EncodeFrameInput; |
| 1017 | |
David Turner | 056f7cd | 2019-01-07 17:48:13 +0000 | [diff] [blame] | 1018 | // EncodeFrameParams contains per-frame encoding parameters decided upon by |
| 1019 | // av1_encode_strategy() and passed down to av1_encode() |
David Turner | 475a313 | 2019-01-18 15:17:17 +0000 | [diff] [blame] | 1020 | struct EncodeFrameParams { |
David Turner | 07dbd8e | 2019-01-08 17:16:25 +0000 | [diff] [blame] | 1021 | int error_resilient_mode; |
David Turner | 475a313 | 2019-01-18 15:17:17 +0000 | [diff] [blame] | 1022 | FRAME_TYPE frame_type; |
David Turner | a7f133c | 2019-01-22 14:47:16 +0000 | [diff] [blame] | 1023 | int primary_ref_frame; |
| 1024 | int order_offset; |
David Turner | dedd8ff | 2019-01-23 13:59:46 +0000 | [diff] [blame] | 1025 | int show_frame; |
David Turner | 6e8b4d9 | 2019-02-18 15:01:33 +0000 | [diff] [blame] | 1026 | int refresh_frame_flags; |
David Turner | 07dbd8e | 2019-01-08 17:16:25 +0000 | [diff] [blame] | 1027 | |
David Turner | e86ee0d | 2019-02-18 17:16:28 +0000 | [diff] [blame] | 1028 | int show_existing_frame; |
| 1029 | int existing_fb_idx_to_show; |
| 1030 | |
David Turner | fe3aecb | 2019-02-06 14:42:42 +0000 | [diff] [blame] | 1031 | // Bitmask of which reference buffers may be referenced by this frame |
David Turner | 07dbd8e | 2019-01-08 17:16:25 +0000 | [diff] [blame] | 1032 | int ref_frame_flags; |
| 1033 | |
David Turner | 7324576 | 2019-02-11 16:42:34 +0000 | [diff] [blame] | 1034 | // Reference buffer assignment for this frame. |
| 1035 | int remapped_ref_idx[REF_FRAMES]; |
| 1036 | |
David Turner | fe3aecb | 2019-02-06 14:42:42 +0000 | [diff] [blame] | 1037 | // Flags which determine which reference buffers are refreshed by this frame |
| 1038 | int refresh_last_frame; |
| 1039 | int refresh_golden_frame; |
| 1040 | int refresh_bwd_ref_frame; |
| 1041 | int refresh_alt2_ref_frame; |
| 1042 | int refresh_alt_ref_frame; |
| 1043 | |
David Turner | 04b70d8 | 2019-01-24 15:39:19 +0000 | [diff] [blame] | 1044 | // Speed level to use for this frame: Bigger number means faster. |
| 1045 | int speed; |
David Turner | 475a313 | 2019-01-18 15:17:17 +0000 | [diff] [blame] | 1046 | }; |
| 1047 | typedef struct EncodeFrameParams EncodeFrameParams; |
David Turner | 056f7cd | 2019-01-07 17:48:13 +0000 | [diff] [blame] | 1048 | |
| 1049 | // EncodeFrameResults contains information about the result of encoding a |
| 1050 | // single frame |
| 1051 | typedef struct { |
| 1052 | size_t size; // Size of resulting bitstream |
| 1053 | } EncodeFrameResults; |
| 1054 | |
Wan-Teh Chang | 3cac454 | 2018-06-29 10:21:39 -0700 | [diff] [blame] | 1055 | // Must not be called more than once. |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 1056 | void av1_initialize_enc(void); |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 1057 | |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 1058 | struct AV1_COMP *av1_create_compressor(AV1EncoderConfig *oxcf, |
| 1059 | BufferPool *const pool); |
| 1060 | void av1_remove_compressor(AV1_COMP *cpi); |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 1061 | |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 1062 | void av1_change_config(AV1_COMP *cpi, const AV1EncoderConfig *oxcf); |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 1063 | |
| 1064 | // receive a frames worth of data. caller can assume that a copy of this |
| 1065 | // frame is made and not just a copy of the pointer.. |
James Zern | 3e2613b | 2017-03-30 23:14:40 -0700 | [diff] [blame] | 1066 | int av1_receive_raw_frame(AV1_COMP *cpi, aom_enc_frame_flags_t frame_flags, |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 1067 | YV12_BUFFER_CONFIG *sd, int64_t time_stamp, |
| 1068 | int64_t end_time_stamp); |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 1069 | |
Andrey Norkin | 795ba87 | 2018-03-06 13:24:14 -0800 | [diff] [blame] | 1070 | int av1_get_compressed_data(AV1_COMP *cpi, unsigned int *frame_flags, |
| 1071 | size_t *size, uint8_t *dest, int64_t *time_stamp, |
| 1072 | int64_t *time_end, int flush, |
Yue Chen | 1bc5be6 | 2018-08-24 13:57:32 -0700 | [diff] [blame^] | 1073 | const aom_rational64_t *timebase); |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 1074 | |
David Turner | 056f7cd | 2019-01-07 17:48:13 +0000 | [diff] [blame] | 1075 | int av1_encode(AV1_COMP *const cpi, uint8_t *const dest, |
David Turner | cb5e36f | 2019-01-17 17:15:25 +0000 | [diff] [blame] | 1076 | const EncodeFrameInput *const frame_input, |
David Turner | 056f7cd | 2019-01-07 17:48:13 +0000 | [diff] [blame] | 1077 | const EncodeFrameParams *const frame_params, |
| 1078 | EncodeFrameResults *const frame_results); |
| 1079 | |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 1080 | int av1_get_preview_raw_frame(AV1_COMP *cpi, YV12_BUFFER_CONFIG *dest); |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 1081 | |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 1082 | int av1_get_last_show_frame(AV1_COMP *cpi, YV12_BUFFER_CONFIG *frame); |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 1083 | |
Yunqing Wang | 93b18f3 | 2018-06-08 21:08:29 -0700 | [diff] [blame] | 1084 | aom_codec_err_t av1_copy_new_frame_enc(AV1_COMMON *cm, |
| 1085 | YV12_BUFFER_CONFIG *new_frame, |
| 1086 | YV12_BUFFER_CONFIG *sd); |
Yunqing Wang | ff9bfca | 2018-06-06 11:46:08 -0700 | [diff] [blame] | 1087 | |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 1088 | int av1_use_as_reference(AV1_COMP *cpi, int ref_frame_flags); |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 1089 | |
Thomas Daede | 497d195 | 2017-08-08 17:33:06 -0700 | [diff] [blame] | 1090 | int av1_copy_reference_enc(AV1_COMP *cpi, int idx, YV12_BUFFER_CONFIG *sd); |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 1091 | |
Thomas Daede | 497d195 | 2017-08-08 17:33:06 -0700 | [diff] [blame] | 1092 | int av1_set_reference_enc(AV1_COMP *cpi, int idx, YV12_BUFFER_CONFIG *sd); |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 1093 | |
David Turner | 475a313 | 2019-01-18 15:17:17 +0000 | [diff] [blame] | 1094 | void av1_set_frame_size(AV1_COMP *cpi, int width, int height); |
| 1095 | |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 1096 | int av1_update_entropy(AV1_COMP *cpi, int update); |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 1097 | |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 1098 | int av1_set_active_map(AV1_COMP *cpi, unsigned char *map, int rows, int cols); |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 1099 | |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 1100 | int av1_get_active_map(AV1_COMP *cpi, unsigned char *map, int rows, int cols); |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 1101 | |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 1102 | int av1_set_internal_size(AV1_COMP *cpi, AOM_SCALING horiz_mode, |
| 1103 | AOM_SCALING vert_mode); |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 1104 | |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 1105 | int av1_get_quantizer(struct AV1_COMP *cpi); |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 1106 | |
Soo-Chul Han | 29c46fb | 2018-03-23 16:02:00 -0400 | [diff] [blame] | 1107 | int av1_convert_sect5obus_to_annexb(uint8_t *buffer, size_t *input_size); |
| 1108 | |
David Turner | dedd8ff | 2019-01-23 13:59:46 +0000 | [diff] [blame] | 1109 | // av1 uses 10,000,000 ticks/second as time stamp |
| 1110 | #define TICKS_PER_SEC 10000000LL |
| 1111 | |
Yue Chen | 1bc5be6 | 2018-08-24 13:57:32 -0700 | [diff] [blame^] | 1112 | static INLINE int64_t |
| 1113 | timebase_units_to_ticks(const aom_rational64_t *timestamp_ratio, int64_t n) { |
| 1114 | return n * timestamp_ratio->num / timestamp_ratio->den; |
David Turner | dedd8ff | 2019-01-23 13:59:46 +0000 | [diff] [blame] | 1115 | } |
| 1116 | |
Yue Chen | 1bc5be6 | 2018-08-24 13:57:32 -0700 | [diff] [blame^] | 1117 | static INLINE int64_t |
| 1118 | ticks_to_timebase_units(const aom_rational64_t *timestamp_ratio, int64_t n) { |
| 1119 | int64_t round = timestamp_ratio->num / 2; |
| 1120 | if (round > 0) --round; |
| 1121 | return (n * timestamp_ratio->den + round) / timestamp_ratio->num; |
David Turner | dedd8ff | 2019-01-23 13:59:46 +0000 | [diff] [blame] | 1122 | } |
Andrey Norkin | 795ba87 | 2018-03-06 13:24:14 -0800 | [diff] [blame] | 1123 | |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 1124 | static INLINE int frame_is_kf_gf_arf(const AV1_COMP *cpi) { |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 1125 | return frame_is_intra_only(&cpi->common) || cpi->refresh_alt_ref_frame || |
| 1126 | (cpi->refresh_golden_frame && !cpi->rc.is_src_frame_alt_ref); |
| 1127 | } |
| 1128 | |
Hui Su | 2d5fd74 | 2018-02-21 18:10:37 -0800 | [diff] [blame] | 1129 | // TODO(huisu@google.com, youzhou@microsoft.com): enable hash-me for HBD. |
| 1130 | static INLINE int av1_use_hash_me(const AV1_COMMON *const cm) { |
Debargha Mukherjee | 8112d2f | 2018-03-01 09:53:11 -0800 | [diff] [blame] | 1131 | return cm->allow_screen_content_tools; |
Hui Su | 2d5fd74 | 2018-02-21 18:10:37 -0800 | [diff] [blame] | 1132 | } |
| 1133 | |
| 1134 | static INLINE hash_table *av1_get_ref_frame_hash_map( |
David Turner | a21966b | 2018-12-05 14:48:49 +0000 | [diff] [blame] | 1135 | const AV1_COMMON *cm, MV_REFERENCE_FRAME ref_frame) { |
| 1136 | const int map_idx = get_ref_frame_map_idx(cm, ref_frame); |
| 1137 | RefCntBuffer *buf = |
| 1138 | (map_idx != INVALID_IDX) ? cm->ref_frame_map[map_idx] : NULL; |
| 1139 | return buf ? &buf->hash_table : NULL; |
RogerZhou | cc5d35d | 2017-08-07 22:20:15 -0700 | [diff] [blame] | 1140 | } |
RogerZhou | cc5d35d | 2017-08-07 22:20:15 -0700 | [diff] [blame] | 1141 | |
David Turner | a21966b | 2018-12-05 14:48:49 +0000 | [diff] [blame] | 1142 | static INLINE const YV12_BUFFER_CONFIG *get_ref_frame_yv12_buf( |
| 1143 | const AV1_COMMON *const cm, MV_REFERENCE_FRAME ref_frame) { |
| 1144 | const RefCntBuffer *const buf = get_ref_frame_buf(cm, ref_frame); |
David Turner | e7ebf90 | 2018-12-04 14:04:55 +0000 | [diff] [blame] | 1145 | return buf != NULL ? &buf->buf : NULL; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 1146 | } |
| 1147 | |
David Turner | a21966b | 2018-12-05 14:48:49 +0000 | [diff] [blame] | 1148 | static INLINE int enc_is_ref_frame_buf(const AV1_COMMON *const cm, |
| 1149 | const RefCntBuffer *const frame_buf) { |
Urvang Joshi | 4058217 | 2018-11-07 11:53:03 -0800 | [diff] [blame] | 1150 | MV_REFERENCE_FRAME ref_frame; |
| 1151 | for (ref_frame = LAST_FRAME; ref_frame <= ALTREF_FRAME; ++ref_frame) { |
David Turner | a21966b | 2018-12-05 14:48:49 +0000 | [diff] [blame] | 1152 | const RefCntBuffer *const buf = get_ref_frame_buf(cm, ref_frame); |
David Turner | e7ebf90 | 2018-12-04 14:04:55 +0000 | [diff] [blame] | 1153 | if (buf == NULL) continue; |
| 1154 | if (frame_buf == buf) break; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 1155 | } |
Urvang Joshi | 4058217 | 2018-11-07 11:53:03 -0800 | [diff] [blame] | 1156 | return (ref_frame <= ALTREF_FRAME); |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 1157 | } |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 1158 | |
David Turner | e3e520d | 2019-01-18 14:38:08 +0000 | [diff] [blame] | 1159 | static INLINE void alloc_frame_mvs(AV1_COMMON *const cm, RefCntBuffer *buf) { |
| 1160 | assert(buf != NULL); |
| 1161 | ensure_mv_buffer(buf, cm); |
| 1162 | buf->width = cm->width; |
| 1163 | buf->height = cm->height; |
| 1164 | } |
| 1165 | |
Hui Su | 8ac0c98 | 2018-04-03 12:17:36 -0700 | [diff] [blame] | 1166 | // Token buffer is only used for palette tokens. |
Yaowu Xu | e39b3b8 | 2017-10-31 16:11:59 -0700 | [diff] [blame] | 1167 | static INLINE unsigned int get_token_alloc(int mb_rows, int mb_cols, |
Imdad Sardharwalla | af8e264 | 2018-01-19 11:46:34 +0000 | [diff] [blame] | 1168 | int sb_size_log2, |
| 1169 | const int num_planes) { |
Rupert Swarbrick | ce63e83 | 2017-10-25 17:54:17 +0100 | [diff] [blame] | 1170 | // Calculate the maximum number of max superblocks in the image. |
Yaowu Xu | e39b3b8 | 2017-10-31 16:11:59 -0700 | [diff] [blame] | 1171 | const int shift = sb_size_log2 - 4; |
| 1172 | const int sb_size = 1 << sb_size_log2; |
| 1173 | const int sb_size_square = sb_size * sb_size; |
Rupert Swarbrick | ce63e83 | 2017-10-25 17:54:17 +0100 | [diff] [blame] | 1174 | const int sb_rows = ALIGN_POWER_OF_TWO(mb_rows, shift) >> shift; |
| 1175 | const int sb_cols = ALIGN_POWER_OF_TWO(mb_cols, shift) >> shift; |
| 1176 | |
Hui Su | 8ac0c98 | 2018-04-03 12:17:36 -0700 | [diff] [blame] | 1177 | // One palette token for each pixel. There can be palettes on two planes. |
| 1178 | const int sb_palette_toks = AOMMIN(2, num_planes) * sb_size_square; |
Rupert Swarbrick | ce63e83 | 2017-10-25 17:54:17 +0100 | [diff] [blame] | 1179 | |
Hui Su | 8ac0c98 | 2018-04-03 12:17:36 -0700 | [diff] [blame] | 1180 | return sb_rows * sb_cols * sb_palette_toks; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 1181 | } |
| 1182 | |
| 1183 | // Get the allocated token size for a tile. It does the same calculation as in |
| 1184 | // the frame token allocation. |
Rupert Swarbrick | dcb3cff | 2017-11-09 15:58:33 +0000 | [diff] [blame] | 1185 | static INLINE unsigned int allocated_tokens(TileInfo tile, int sb_size_log2, |
| 1186 | int num_planes) { |
Jingning Han | eafbd5f | 2017-03-07 11:18:17 -0800 | [diff] [blame] | 1187 | int tile_mb_rows = (tile.mi_row_end - tile.mi_row_start + 2) >> 2; |
| 1188 | int tile_mb_cols = (tile.mi_col_end - tile.mi_col_start + 2) >> 2; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 1189 | |
Rupert Swarbrick | dcb3cff | 2017-11-09 15:58:33 +0000 | [diff] [blame] | 1190 | return get_token_alloc(tile_mb_rows, tile_mb_cols, sb_size_log2, num_planes); |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 1191 | } |
| 1192 | |
Ravi Chaudhary | 73cf15b | 2018-08-30 10:52:51 +0530 | [diff] [blame] | 1193 | static INLINE void get_start_tok(AV1_COMP *cpi, int tile_row, int tile_col, |
| 1194 | int mi_row, TOKENEXTRA **tok, int sb_size_log2, |
| 1195 | int num_planes) { |
| 1196 | AV1_COMMON *const cm = &cpi->common; |
| 1197 | const int tile_cols = cm->tile_cols; |
| 1198 | TileDataEnc *this_tile = &cpi->tile_data[tile_row * tile_cols + tile_col]; |
| 1199 | const TileInfo *const tile_info = &this_tile->tile_info; |
| 1200 | |
| 1201 | const int tile_mb_cols = |
| 1202 | (tile_info->mi_col_end - tile_info->mi_col_start + 2) >> 2; |
| 1203 | const int tile_mb_row = (mi_row - tile_info->mi_row_start + 2) >> 2; |
| 1204 | |
| 1205 | *tok = cpi->tile_tok[tile_row][tile_col] + |
| 1206 | get_token_alloc(tile_mb_row, tile_mb_cols, sb_size_log2, num_planes); |
| 1207 | } |
| 1208 | |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 1209 | void av1_apply_encoding_flags(AV1_COMP *cpi, aom_enc_frame_flags_t flags); |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 1210 | |
Yaowu Xu | 6da4230 | 2017-10-10 14:45:06 -0700 | [diff] [blame] | 1211 | #define ALT_MIN_LAG 3 |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 1212 | static INLINE int is_altref_enabled(const AV1_COMP *const cpi) { |
Yaowu Xu | 6da4230 | 2017-10-10 14:45:06 -0700 | [diff] [blame] | 1213 | return cpi->oxcf.lag_in_frames >= ALT_MIN_LAG && cpi->oxcf.enable_auto_arf; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 1214 | } |
| 1215 | |
| 1216 | // TODO(zoeliu): To set up cpi->oxcf.enable_auto_brf |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 1217 | |
Urvang Joshi | 5264844 | 2016-10-13 17:27:51 -0700 | [diff] [blame] | 1218 | static INLINE void set_ref_ptrs(const AV1_COMMON *cm, MACROBLOCKD *xd, |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 1219 | MV_REFERENCE_FRAME ref0, |
| 1220 | MV_REFERENCE_FRAME ref1) { |
David Turner | a21966b | 2018-12-05 14:48:49 +0000 | [diff] [blame] | 1221 | xd->block_ref_scale_factors[0] = |
| 1222 | get_ref_scale_factors_const(cm, ref0 >= LAST_FRAME ? ref0 : 1); |
| 1223 | xd->block_ref_scale_factors[1] = |
| 1224 | get_ref_scale_factors_const(cm, ref1 >= LAST_FRAME ? ref1 : 1); |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 1225 | } |
| 1226 | |
Yaowu Xu | 4ff59b5 | 2017-04-24 12:41:56 -0700 | [diff] [blame] | 1227 | static INLINE int get_chessboard_index(int frame_index) { |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 1228 | return frame_index & 0x1; |
| 1229 | } |
| 1230 | |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 1231 | static INLINE int *cond_cost_list(const struct AV1_COMP *cpi, int *cost_list) { |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 1232 | return cpi->sf.mv.subpel_search_method != SUBPEL_TREE ? cost_list : NULL; |
| 1233 | } |
| 1234 | |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 1235 | void av1_new_framerate(AV1_COMP *cpi, double framerate); |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 1236 | |
David Turner | dedd8ff | 2019-01-23 13:59:46 +0000 | [diff] [blame] | 1237 | void av1_setup_frame_size(AV1_COMP *cpi); |
| 1238 | |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 1239 | #define LAYER_IDS_TO_IDX(sl, tl, num_tl) ((sl) * (num_tl) + (tl)) |
| 1240 | |
Cheng Chen | 09c83a5 | 2018-06-05 12:27:36 -0700 | [diff] [blame] | 1241 | // Returns 1 if a frame is scaled and 0 otherwise. |
| 1242 | static INLINE int av1_resize_scaled(const AV1_COMMON *cm) { |
| 1243 | return !(cm->superres_upscaled_width == cm->render_width && |
| 1244 | cm->superres_upscaled_height == cm->render_height); |
Fergus Simpson | fecb2ab | 2017-04-30 15:49:57 -0700 | [diff] [blame] | 1245 | } |
| 1246 | |
Cheng Chen | 09c83a5 | 2018-06-05 12:27:36 -0700 | [diff] [blame] | 1247 | static INLINE int av1_frame_scaled(const AV1_COMMON *cm) { |
| 1248 | return !av1_superres_scaled(cm) && av1_resize_scaled(cm); |
Fergus Simpson | fecb2ab | 2017-04-30 15:49:57 -0700 | [diff] [blame] | 1249 | } |
| 1250 | |
Sarah Parker | 3300552 | 2018-07-27 14:46:25 -0700 | [diff] [blame] | 1251 | // Don't allow a show_existing_frame to coincide with an error resilient |
| 1252 | // frame. An exception can be made for a forward keyframe since it has no |
| 1253 | // previous dependencies. |
| 1254 | static INLINE int encode_show_existing_frame(const AV1_COMMON *cm) { |
David Turner | d2a592e | 2018-11-16 14:59:31 +0000 | [diff] [blame] | 1255 | return cm->show_existing_frame && (!cm->error_resilient_mode || |
| 1256 | cm->current_frame.frame_type == KEY_FRAME); |
Sarah Parker | 3300552 | 2018-07-27 14:46:25 -0700 | [diff] [blame] | 1257 | } |
| 1258 | |
kyslov | 7b9d0d6 | 2018-12-21 11:12:26 -0800 | [diff] [blame] | 1259 | // Lighter version of set_offsets that only sets the mode info |
| 1260 | // pointers. |
| 1261 | static INLINE void set_mode_info_offsets(const AV1_COMP *const cpi, |
| 1262 | MACROBLOCK *const x, |
| 1263 | MACROBLOCKD *const xd, int mi_row, |
| 1264 | int mi_col) { |
| 1265 | const AV1_COMMON *const cm = &cpi->common; |
| 1266 | const int idx_str = xd->mi_stride * mi_row + mi_col; |
| 1267 | xd->mi = cm->mi_grid_visible + idx_str; |
| 1268 | xd->mi[0] = cm->mi + idx_str; |
| 1269 | x->mbmi_ext = cpi->mbmi_ext_base + (mi_row * cm->mi_cols + mi_col); |
| 1270 | } |
| 1271 | |
| 1272 | // Check to see if the given partition size is allowed for a specified number |
| 1273 | // of mi block rows and columns remaining in the image. |
| 1274 | // If not then return the largest allowed partition size |
| 1275 | static INLINE BLOCK_SIZE find_partition_size(BLOCK_SIZE bsize, int rows_left, |
| 1276 | int cols_left, int *bh, int *bw) { |
| 1277 | int int_size = (int)bsize; |
| 1278 | if (rows_left <= 0 || cols_left <= 0) { |
| 1279 | return AOMMIN(bsize, BLOCK_8X8); |
| 1280 | } else { |
| 1281 | for (; int_size > 0; int_size -= 3) { |
| 1282 | *bh = mi_size_high[int_size]; |
| 1283 | *bw = mi_size_wide[int_size]; |
| 1284 | if ((*bh <= rows_left) && (*bw <= cols_left)) { |
| 1285 | break; |
| 1286 | } |
| 1287 | } |
| 1288 | } |
| 1289 | return (BLOCK_SIZE)int_size; |
| 1290 | } |
| 1291 | |
chiyotsai | dd132d2 | 2019-03-11 14:22:49 -0700 | [diff] [blame] | 1292 | static const uint8_t av1_ref_frame_flag_list[REF_FRAMES] = { 0, |
| 1293 | AOM_LAST_FLAG, |
| 1294 | AOM_LAST2_FLAG, |
| 1295 | AOM_LAST3_FLAG, |
| 1296 | AOM_GOLD_FLAG, |
| 1297 | AOM_BWD_FLAG, |
| 1298 | AOM_ALT2_FLAG, |
| 1299 | AOM_ALT_FLAG }; |
| 1300 | |
Tom Finegan | f8d6a16 | 2018-08-21 10:47:55 -0700 | [diff] [blame] | 1301 | // Returns a Sequence Header OBU stored in an aom_fixed_buf_t, or NULL upon |
| 1302 | // failure. When a non-NULL aom_fixed_buf_t pointer is returned by this |
| 1303 | // function, the memory must be freed by the caller. Both the buf member of the |
Tom Finegan | ca1e28f | 2018-08-28 16:55:35 -0700 | [diff] [blame] | 1304 | // aom_fixed_buf_t, and the aom_fixed_buf_t pointer itself must be freed. Memory |
| 1305 | // returned must be freed via call to free(). |
| 1306 | // |
| 1307 | // Note: The OBU returned is in Low Overhead Bitstream Format. Specifically, |
| 1308 | // the obu_has_size_field bit is set, and the buffer contains the obu_size |
| 1309 | // field. |
Tom Finegan | f8d6a16 | 2018-08-21 10:47:55 -0700 | [diff] [blame] | 1310 | aom_fixed_buf_t *av1_get_global_headers(AV1_COMP *cpi); |
| 1311 | |
chiyotsai | 9c484b3 | 2019-03-07 16:01:50 -0800 | [diff] [blame] | 1312 | #if CONFIG_COLLECT_PARTITION_STATS == 2 |
chiyotsai | 92ed0dd | 2019-01-25 14:50:14 -0800 | [diff] [blame] | 1313 | static INLINE void av1_print_partition_stats(PartitionStats *part_stats) { |
| 1314 | FILE *f = fopen("partition_stats.csv", "w"); |
| 1315 | if (!f) { |
| 1316 | return; |
| 1317 | } |
| 1318 | |
| 1319 | fprintf(f, "bsize,redo,"); |
| 1320 | for (int part = 0; part < EXT_PARTITION_TYPES; part++) { |
| 1321 | fprintf(f, "decision_%d,", part); |
| 1322 | } |
| 1323 | for (int part = 0; part < EXT_PARTITION_TYPES; part++) { |
| 1324 | fprintf(f, "attempt_%d,", part); |
| 1325 | } |
chiyotsai | 9c484b3 | 2019-03-07 16:01:50 -0800 | [diff] [blame] | 1326 | for (int part = 0; part < EXT_PARTITION_TYPES; part++) { |
| 1327 | fprintf(f, "time_%d,", part); |
| 1328 | } |
chiyotsai | 92ed0dd | 2019-01-25 14:50:14 -0800 | [diff] [blame] | 1329 | fprintf(f, "\n"); |
| 1330 | |
| 1331 | const int bsizes[6] = { 128, 64, 32, 16, 8, 4 }; |
| 1332 | |
| 1333 | for (int bsize_idx = 0; bsize_idx < 6; bsize_idx++) { |
| 1334 | fprintf(f, "%d,%d,", bsizes[bsize_idx], part_stats->partition_redo); |
| 1335 | for (int part = 0; part < EXT_PARTITION_TYPES; part++) { |
| 1336 | fprintf(f, "%d,", part_stats->partition_decisions[bsize_idx][part]); |
| 1337 | } |
| 1338 | for (int part = 0; part < EXT_PARTITION_TYPES; part++) { |
| 1339 | fprintf(f, "%d,", part_stats->partition_attempts[bsize_idx][part]); |
| 1340 | } |
chiyotsai | 9c484b3 | 2019-03-07 16:01:50 -0800 | [diff] [blame] | 1341 | for (int part = 0; part < EXT_PARTITION_TYPES; part++) { |
| 1342 | fprintf(f, "%ld,", part_stats->partition_times[bsize_idx][part]); |
| 1343 | } |
chiyotsai | 92ed0dd | 2019-01-25 14:50:14 -0800 | [diff] [blame] | 1344 | fprintf(f, "\n"); |
| 1345 | } |
| 1346 | fclose(f); |
| 1347 | } |
| 1348 | |
| 1349 | static INLINE int av1_get_bsize_idx_for_part_stats(BLOCK_SIZE bsize) { |
| 1350 | assert(bsize == BLOCK_128X128 || bsize == BLOCK_64X64 || |
chiyotsai | 9c484b3 | 2019-03-07 16:01:50 -0800 | [diff] [blame] | 1351 | bsize == BLOCK_32X32 || bsize == BLOCK_16X16 || bsize == BLOCK_8X8 || |
| 1352 | bsize == BLOCK_4X4); |
chiyotsai | 92ed0dd | 2019-01-25 14:50:14 -0800 | [diff] [blame] | 1353 | switch (bsize) { |
| 1354 | case BLOCK_128X128: return 0; |
| 1355 | case BLOCK_64X64: return 1; |
| 1356 | case BLOCK_32X32: return 2; |
| 1357 | case BLOCK_16X16: return 3; |
| 1358 | case BLOCK_8X8: return 4; |
| 1359 | case BLOCK_4X4: return 5; |
| 1360 | default: assert(0 && "Invalid bsize for partition_stats."); return -1; |
| 1361 | } |
| 1362 | } |
| 1363 | #endif |
| 1364 | |
Yunqing Wang | ba25558 | 2019-02-11 16:21:12 -0800 | [diff] [blame] | 1365 | #if CONFIG_COLLECT_COMPONENT_TIMING |
| 1366 | static INLINE void start_timing(AV1_COMP *cpi, int component) { |
| 1367 | aom_usec_timer_start(&cpi->component_timer[component]); |
| 1368 | } |
| 1369 | static INLINE void end_timing(AV1_COMP *cpi, int component) { |
| 1370 | aom_usec_timer_mark(&cpi->component_timer[component]); |
| 1371 | cpi->frame_component_time[component] += |
| 1372 | aom_usec_timer_elapsed(&cpi->component_timer[component]); |
| 1373 | } |
| 1374 | static INLINE char const *get_frame_type_enum(int type) { |
| 1375 | switch (type) { |
| 1376 | case 0: return "KEY_FRAME"; |
| 1377 | case 1: return "INTER_FRAME"; |
| 1378 | case 2: return "INTRA_ONLY_FRAME"; |
| 1379 | case 3: return "S_FRAME"; |
| 1380 | default: assert(0); |
| 1381 | } |
| 1382 | return "error"; |
| 1383 | } |
| 1384 | #endif |
| 1385 | |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 1386 | #ifdef __cplusplus |
| 1387 | } // extern "C" |
| 1388 | #endif |
| 1389 | |
James Zern | e1cbb13 | 2018-08-22 14:10:36 -0700 | [diff] [blame] | 1390 | #endif // AOM_AV1_ENCODER_ENCODER_H_ |