AOMedia AV1 Codec
aq_cyclicrefresh.h
1 /*
2  * Copyright (c) 2016, Alliance for Open Media. All rights reserved
3  *
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.
10  */
11 
12 #ifndef AOM_AV1_ENCODER_AQ_CYCLICREFRESH_H_
13 #define AOM_AV1_ENCODER_AQ_CYCLICREFRESH_H_
14 
15 #include "av1/common/blockd.h"
16 #include "av1/encoder/block.h"
17 #include "av1/encoder/tokenize.h"
18 
19 #ifdef __cplusplus
20 extern "C" {
21 #endif
22 
23 // The segment ids used in cyclic refresh: from base (no boost) to increasing
24 // boost (higher delta-qp).
25 #define CR_SEGMENT_ID_BASE 0
26 #define CR_SEGMENT_ID_BOOST1 1
27 #define CR_SEGMENT_ID_BOOST2 2
28 
29 // Maximum rate target ratio for setting segment delta-qp.
30 #define CR_MAX_RATE_TARGET_RATIO 4.0
31 
49  int sb_index;
73  int rdmult;
81  int8_t *map;
86  int64_t thresh_rate_sb;
91  int64_t thresh_dist_sb;
96  int16_t motion_thresh;
105 
107  int qindex_delta[3];
108  double weight_segment;
109  int apply_cyclic_refresh;
110  int skip_over4x4;
112 };
113 
114 struct AV1_COMP;
115 
116 typedef struct CYCLIC_REFRESH CYCLIC_REFRESH;
117 
118 CYCLIC_REFRESH *av1_cyclic_refresh_alloc(int mi_rows, int mi_cols);
119 
120 void av1_cyclic_refresh_free(CYCLIC_REFRESH *cr);
121 
138  double correction_factor);
139 
158 int av1_cyclic_refresh_rc_bits_per_mb(const struct AV1_COMP *cpi, int i,
159  double correction_factor);
160 
182  MACROBLOCK *const x, int mi_row, int mi_col,
183  BLOCK_SIZE bsize);
184 
210  MACROBLOCK *const x, int mi_row,
211  int mi_col, BLOCK_SIZE bsize,
212  int64_t rate, int64_t dist, int skip,
213  RUN_TYPE dry_run);
214 
230 
248  CYCLIC_REFRESH *const cyclic_refresh, const MACROBLOCK *const x);
249 
263 void av1_cyclic_refresh_postencode(struct AV1_COMP *const cpi);
264 
276 
293 
307 void av1_cyclic_refresh_setup(struct AV1_COMP *const cpi);
308 
309 int av1_cyclic_refresh_get_rdmult(const CYCLIC_REFRESH *cr);
310 
311 void av1_cyclic_refresh_reset_resize(struct AV1_COMP *const cpi);
312 
313 static INLINE int cyclic_refresh_segment_id_boosted(int segment_id) {
314  return segment_id == CR_SEGMENT_ID_BOOST1 ||
315  segment_id == CR_SEGMENT_ID_BOOST2;
316 }
317 
318 static INLINE int cyclic_refresh_segment_id(int segment_id) {
319  if (segment_id == CR_SEGMENT_ID_BOOST1)
320  return CR_SEGMENT_ID_BOOST1;
321  else if (segment_id == CR_SEGMENT_ID_BOOST2)
322  return CR_SEGMENT_ID_BOOST2;
323  else
324  return CR_SEGMENT_ID_BASE;
325 }
326 
327 #ifdef __cplusplus
328 } // extern "C"
329 #endif
330 
331 #endif // AOM_AV1_ENCODER_AQ_CYCLICREFRESH_H_
void av1_cyclic_refresh_postencode(struct AV1_COMP *const cpi)
Update stats after encoding frame.
int av1_cyclic_refresh_rc_bits_per_mb(const struct AV1_COMP *cpi, int i, double correction_factor)
Estimate the bits per mb, for given q = i and delta-q.
void av1_cyclic_refresh_update_parameters(struct AV1_COMP *const cpi)
Set the global/frame level parameters for cyclic refresh.
void av1_cyclic_refresh_update_segment(const struct AV1_COMP *cpi, MACROBLOCK *const x, int mi_row, int mi_col, BLOCK_SIZE bsize, int64_t rate, int64_t dist, int skip, RUN_TYPE dry_run)
Update segment_id for block based on mode selected.
int av1_cyclic_refresh_estimate_bits_at_q(const struct AV1_COMP *cpi, double correction_factor)
Estimate the bits, incorporating the delta-q from the segments.
void av1_accumulate_cyclic_refresh_counters(CYCLIC_REFRESH *const cyclic_refresh, const MACROBLOCK *const x)
Accumulate cyclic refresh counters.
void av1_cyclic_refresh_set_golden_update(struct AV1_COMP *const cpi)
Set golden frame update interval nased on cyclic refresh.
void av1_init_cyclic_refresh_counters(MACROBLOCK *const x)
Initialize counters used for cyclic refresh.
void av1_cyclic_reset_segment_skip(const struct AV1_COMP *cpi, MACROBLOCK *const x, int mi_row, int mi_col, BLOCK_SIZE bsize)
Update segment_id for blocks are skipped.
void av1_cyclic_refresh_setup(struct AV1_COMP *const cpi)
Setup the cyclic background refresh.
Top level encoder structure.
Definition: encoder.h:2632
The stucture of CYCLIC_REFRESH.
Definition: aq_cyclicrefresh.h:36
int16_t motion_thresh
Definition: aq_cyclicrefresh.h:96
int64_t thresh_rate_sb
Definition: aq_cyclicrefresh.h:86
int time_for_refresh
Definition: aq_cyclicrefresh.h:55
int rdmult
Definition: aq_cyclicrefresh.h:73
int max_qdelta_perc
Definition: aq_cyclicrefresh.h:45
int8_t * map
Definition: aq_cyclicrefresh.h:81
int cnt_zeromv
Definition: aq_cyclicrefresh.h:77
int64_t thresh_dist_sb
Definition: aq_cyclicrefresh.h:91
int actual_num_seg2_blocks
Definition: aq_cyclicrefresh.h:69
int target_num_seg_blocks
Definition: aq_cyclicrefresh.h:59
int sb_index
Definition: aq_cyclicrefresh.h:49
int rate_boost_fac
Definition: aq_cyclicrefresh.h:104
int percent_refresh
Definition: aq_cyclicrefresh.h:41
int actual_num_seg1_blocks
Definition: aq_cyclicrefresh.h:64
double rate_ratio_qdelta
Definition: aq_cyclicrefresh.h:100
Encoder's parameters related to the current coding block.
Definition: block.h:778