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;
111  int counter_encode_maxq_scene_change;
113 };
114 
115 struct AV1_COMP;
116 
117 typedef struct CYCLIC_REFRESH CYCLIC_REFRESH;
118 
119 CYCLIC_REFRESH *av1_cyclic_refresh_alloc(int mi_rows, int mi_cols);
120 
121 void av1_cyclic_refresh_free(CYCLIC_REFRESH *cr);
122 
139  double correction_factor);
140 
159 int av1_cyclic_refresh_rc_bits_per_mb(const struct AV1_COMP *cpi, int i,
160  double correction_factor);
161 
183  MACROBLOCK *const x, int mi_row, int mi_col,
184  BLOCK_SIZE bsize);
185 
211  MACROBLOCK *const x, int mi_row,
212  int mi_col, BLOCK_SIZE bsize,
213  int64_t rate, int64_t dist, int skip,
214  RUN_TYPE dry_run);
215 
231 
249  CYCLIC_REFRESH *const cyclic_refresh, const MACROBLOCK *const x);
250 
264 void av1_cyclic_refresh_postencode(struct AV1_COMP *const cpi);
265 
277 
294 
308 void av1_cyclic_refresh_setup(struct AV1_COMP *const cpi);
309 
310 int av1_cyclic_refresh_get_rdmult(const CYCLIC_REFRESH *cr);
311 
312 void av1_cyclic_refresh_reset_resize(struct AV1_COMP *const cpi);
313 
314 int av1_cyclic_refresh_disable_lf_cdef(struct AV1_COMP *const cpi);
315 
316 static INLINE int cyclic_refresh_segment_id_boosted(int segment_id) {
317  return segment_id == CR_SEGMENT_ID_BOOST1 ||
318  segment_id == CR_SEGMENT_ID_BOOST2;
319 }
320 
321 static INLINE int cyclic_refresh_segment_id(int segment_id) {
322  if (segment_id == CR_SEGMENT_ID_BOOST1)
323  return CR_SEGMENT_ID_BOOST1;
324  else if (segment_id == CR_SEGMENT_ID_BOOST2)
325  return CR_SEGMENT_ID_BOOST2;
326  else
327  return CR_SEGMENT_ID_BASE;
328 }
329 
330 #ifdef __cplusplus
331 } // extern "C"
332 #endif
333 
334 #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:2664
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:813