AOMedia AV1 Codec: av1/encoder/aq_cyclicrefresh.h Source File

| 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
20extern "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
36 struct CYCLIC_REFRESH {
41 int percent_refresh;
45 int max_qdelta_perc;
49 int sb_index;
55 int time_for_refresh;
59 int target_num_seg_blocks;
64 int actual_num_seg1_blocks;
69 int actual_num_seg2_blocks;
73 int rdmult;
77 int cnt_zeromv;
81 int8_t *map;
86 int64_t thresh_rate_sb;
91 int64_t thresh_dist_sb;
96 int16_t motion_thresh;
100 double rate_ratio_qdelta;
104 int rate_boost_fac;
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
115struct AV1_COMP;
116
117typedef struct CYCLIC_REFRESH CYCLIC_REFRESH;
118
119 CYCLIC_REFRESH *av1_cyclic_refresh_alloc(int mi_rows, int mi_cols);
120
121void av1_cyclic_refresh_free(CYCLIC_REFRESH *cr);
122
138 int av1_cyclic_refresh_estimate_bits_at_q(const struct AV1_COMP *cpi,
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
182 void av1_cyclic_reset_segment_skip(const struct AV1_COMP *cpi,
183 MACROBLOCK *const x, int mi_row, int mi_col,
184 BLOCK_SIZE bsize);
185
210 void av1_cyclic_refresh_update_segment(const struct AV1_COMP *cpi,
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
230 void av1_init_cyclic_refresh_counters(MACROBLOCK *const x);
231
248 void av1_accumulate_cyclic_refresh_counters(
249 CYCLIC_REFRESH *const cyclic_refresh, const MACROBLOCK *const x);
250
264 void av1_cyclic_refresh_postencode(struct AV1_COMP *const cpi);
265
276 void av1_cyclic_refresh_set_golden_update(struct AV1_COMP *const cpi);
277
293 void av1_cyclic_refresh_update_parameters(struct AV1_COMP *const cpi);
294
308 void av1_cyclic_refresh_setup(struct AV1_COMP *const cpi);
309
310int av1_cyclic_refresh_get_rdmult(const CYCLIC_REFRESH *cr);
311
312void av1_cyclic_refresh_reset_resize(struct AV1_COMP *const cpi);
313
314int av1_cyclic_refresh_disable_lf_cdef(struct AV1_COMP *const cpi);
315
316static 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
321static 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_
block.h
av1_cyclic_refresh_postencode
void av1_cyclic_refresh_postencode(struct AV1_COMP *const cpi)
Update stats after encoding frame.
av1_cyclic_refresh_rc_bits_per_mb
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.
av1_cyclic_refresh_update_parameters
void av1_cyclic_refresh_update_parameters(struct AV1_COMP *const cpi)
Set the global/frame level parameters for cyclic refresh.
av1_cyclic_refresh_update_segment
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.
av1_cyclic_refresh_estimate_bits_at_q
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.
av1_accumulate_cyclic_refresh_counters
void av1_accumulate_cyclic_refresh_counters(CYCLIC_REFRESH *const cyclic_refresh, const MACROBLOCK *const x)
Accumulate cyclic refresh counters.
av1_cyclic_refresh_set_golden_update
void av1_cyclic_refresh_set_golden_update(struct AV1_COMP *const cpi)
Set golden frame update interval nased on cyclic refresh.
av1_init_cyclic_refresh_counters
void av1_init_cyclic_refresh_counters(MACROBLOCK *const x)
Initialize counters used for cyclic refresh.
av1_cyclic_reset_segment_skip
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.
av1_cyclic_refresh_setup
void av1_cyclic_refresh_setup(struct AV1_COMP *const cpi)
Setup the cyclic background refresh.
CYCLIC_REFRESH
The stucture of CYCLIC_REFRESH.
Definition: aq_cyclicrefresh.h:36
CYCLIC_REFRESH::motion_thresh
int16_t motion_thresh
Definition: aq_cyclicrefresh.h:96
CYCLIC_REFRESH::thresh_rate_sb
int64_t thresh_rate_sb
Definition: aq_cyclicrefresh.h:86
CYCLIC_REFRESH::time_for_refresh
int time_for_refresh
Definition: aq_cyclicrefresh.h:55
CYCLIC_REFRESH::rdmult
int rdmult
Definition: aq_cyclicrefresh.h:73
CYCLIC_REFRESH::max_qdelta_perc
int max_qdelta_perc
Definition: aq_cyclicrefresh.h:45
CYCLIC_REFRESH::map
int8_t * map
Definition: aq_cyclicrefresh.h:81
CYCLIC_REFRESH::cnt_zeromv
int cnt_zeromv
Definition: aq_cyclicrefresh.h:77
CYCLIC_REFRESH::thresh_dist_sb
int64_t thresh_dist_sb
Definition: aq_cyclicrefresh.h:91
CYCLIC_REFRESH::actual_num_seg2_blocks
int actual_num_seg2_blocks
Definition: aq_cyclicrefresh.h:69
CYCLIC_REFRESH::target_num_seg_blocks
int target_num_seg_blocks
Definition: aq_cyclicrefresh.h:59
CYCLIC_REFRESH::sb_index
int sb_index
Definition: aq_cyclicrefresh.h:49
CYCLIC_REFRESH::rate_boost_fac
int rate_boost_fac
Definition: aq_cyclicrefresh.h:104
CYCLIC_REFRESH::percent_refresh
int percent_refresh
Definition: aq_cyclicrefresh.h:41
CYCLIC_REFRESH::actual_num_seg1_blocks
int actual_num_seg1_blocks
Definition: aq_cyclicrefresh.h:64
CYCLIC_REFRESH::rate_ratio_qdelta
double rate_ratio_qdelta
Definition: aq_cyclicrefresh.h:100
macroblock
Encoder's parameters related to the current coding block.
Definition: block.h:813