AOMedia AV1 Codec
temporal_filter.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_TEMPORAL_FILTER_H_
13#define AOM_AV1_ENCODER_TEMPORAL_FILTER_H_
14
15#include <stdbool.h>
16
17#ifdef __cplusplus
18extern "C" {
19#endif
21struct AV1_COMP;
22struct AV1EncoderConfig;
23struct ThreadData;
24// TODO(any): These two variables are only used in avx2, sse2, sse4
25// implementations, where the block size is still hard coded. This should be
26// fixed to align with the c implementation.
27#define BH 32
28#define BW 32
29
30// Block size used in temporal filtering.
31#define TF_BLOCK_SIZE BLOCK_32X32
32
33// Window size for temporal filtering.
34#define TF_WINDOW_LENGTH 5
35
36// A constant number, sqrt(pi / 2), used for noise estimation.
37static const double SQRT_PI_BY_2 = 1.25331413732;
38
39// Hyper-parameters used to compute filtering weight. These hyper-parameters can
40// be tuned for a better performance.
41// 0. A scale factor used in temporal filtering to raise the filter weight from
42// `double` with range [0, 1] to `int` with range [0, 1000].
43#define TF_WEIGHT_SCALE 1000
44// 1. Weight factor used to balance the weighted-average between window error
45// and block error. The weight is for window error while the weight for block
46// error is always set as 1.
47#define TF_WINDOW_BLOCK_BALANCE_WEIGHT 5
48// 2. Threshold for using q to adjust the filtering weight. Concretely, when
49// using a small q (high bitrate), we would like to reduce the filtering
50// strength such that more detailed information can be preserved. Hence, when
51// q is smaller than this threshold, we will adjust the filtering weight
52// based on the q-value.
53#define TF_Q_DECAY_THRESHOLD 20
54// 3. Normalization factor used to normalize the motion search error. Since the
55// motion search error can be large and uncontrollable, we will simply
56// normalize it before using it to compute the filtering weight.
57#define TF_SEARCH_ERROR_NORM_WEIGHT 20
58// 4. Threshold for using `arnr_strength` to adjust the filtering strength.
59// Concretely, users can use `arnr_strength` arguments to control the
60// strength of temporal filtering. When `arnr_strength` is small enough (
61// i.e., smaller than this threshold), we will adjust the filtering weight
62// based on the strength value.
63#define TF_STRENGTH_THRESHOLD 4
64// 5. Threshold for using motion search distance to adjust the filtering weight.
65// Concretely, larger motion search vector leads to a higher probability of
66// unreliable search. Hence, we would like to reduce the filtering strength
67// when the distance is large enough. Considering that the distance actually
68// relies on the frame size, this threshold is also a resolution-based
69// threshold. Taking 720p videos as an instance, if this field equals to 0.1,
70// then the actual threshold will be 720 * 0.1 = 72. Similarly, the threshold
71// for 360p videos will be 360 * 0.1 = 36.
72#define TF_SEARCH_DISTANCE_THRESHOLD 0.1
73// 6. Threshold to identify if the q is in a relative high range.
74// Above this cutoff q, a stronger filtering is applied.
75// For a high q, the quantization throws away more information, and thus a
76// stronger filtering is less likely to distort the encoded quality, while a
77// stronger filtering could reduce bit rates.
78// Ror a low q, more details are expected to be retained. Filtering is thus
79// more conservative.
80#define TF_QINDEX_CUTOFF 128
81
82#define NOISE_ESTIMATION_EDGE_THRESHOLD 50
83
84// Sum and SSE source vs filtered frame difference returned by
85// temporal filter.
86typedef struct {
87 int64_t sum;
88 int64_t sse;
89} FRAME_DIFF;
90
96typedef struct {
100 YV12_BUFFER_CONFIG *frames[MAX_LAG_BUFFERS];
105
110
122 struct scale_factors sf;
126 double noise_levels[MAX_MB_PLANE];
148
154#define TF_INFO_BUF_COUNT 2
155
159typedef struct TEMPORAL_FILTER_INFO {
170 YV12_BUFFER_CONFIG tf_buf[TF_INFO_BUF_COUNT];
171
182 FRAME_DIFF frame_diff[TF_INFO_BUF_COUNT];
186 int tf_buf_gf_index[TF_INFO_BUF_COUNT];
190 int tf_buf_display_index_offset[TF_INFO_BUF_COUNT];
194 int tf_buf_valid[TF_INFO_BUF_COUNT];
196
202int av1_is_temporal_filter_on(const struct AV1EncoderConfig *oxcf);
203
208void av1_tf_info_alloc(TEMPORAL_FILTER_INFO *tf_info,
209 const struct AV1_COMP *cpi);
210
214void av1_tf_info_free(TEMPORAL_FILTER_INFO *tf_info);
215
219void av1_tf_info_reset(TEMPORAL_FILTER_INFO *tf_info);
220
226void av1_tf_info_filtering(TEMPORAL_FILTER_INFO *tf_info, struct AV1_COMP *cpi,
227 const GF_GROUP *gf_group);
228
235YV12_BUFFER_CONFIG *av1_tf_info_get_filtered_buf(TEMPORAL_FILTER_INFO *tf_info,
236 int gf_index,
237 FRAME_DIFF *frame_diff);
238
241// Data related to temporal filtering.
242typedef struct {
243 // Source vs filtered frame error.
244 FRAME_DIFF diff;
245 // Pointer to temporary block info used to store state in temporal filtering
246 // process.
247 MB_MODE_INFO *tmp_mbmi;
248 // Pointer to accumulator buffer used in temporal filtering process.
249 uint32_t *accum;
250 // Pointer to count buffer used in temporal filtering process.
251 uint16_t *count;
252 // Pointer to predictor used in temporal filtering process.
253 uint8_t *pred;
254} TemporalFilterData;
255
256// Data related to temporal filter multi-thread synchronization.
257typedef struct {
258#if CONFIG_MULTITHREAD
259 // Mutex lock used for dispatching jobs.
260 pthread_mutex_t *mutex_;
261#endif // CONFIG_MULTITHREAD
262 // Next temporal filter block row to be filtered.
263 int next_tf_row;
264} AV1TemporalFilterSync;
265
266// Estimates noise level from a given frame using a single plane (Y, U, or V).
267// This is an adaptation of the mehtod in the following paper:
268// Shen-Chuan Tai, Shih-Ming Yang, "A fast method for image noise
269// estimation using Laplacian operator and adaptive edge detection",
270// Proc. 3rd International Symposium on Communications, Control and
271// Signal Processing, 2008, St Julians, Malta.
272// Inputs:
273// frame: Pointer to the frame to estimate noise level from.
274// noise_level: Pointer to store the estimated noise.
275// plane_from: Index of the starting plane used for noise estimation.
276// Commonly, 0 for Y-plane, 1 for U-plane, and 2 for V-plane.
277// plane_to: Index of the end plane used for noise estimation.
278// bit_depth: Actual bit-depth instead of the encoding bit-depth of the frame.
279// edge_thresh: Edge threshold.
280void av1_estimate_noise_level(const YV12_BUFFER_CONFIG *frame,
281 double *noise_level, int plane_from, int plane_to,
282 int bit_depth, int edge_thresh);
296void av1_tf_do_filtering_row(struct AV1_COMP *cpi, struct ThreadData *td,
297 int mb_row);
298
324 const int filter_frame_lookahead_idx,
325 int gf_frame_index, FRAME_DIFF *frame_diff,
326 YV12_BUFFER_CONFIG *output_frame);
327
343 const FRAME_DIFF *frame_diff, int q_index,
344 aom_bit_depth_t bit_depth);
345
347// Helper function to get `q` used for encoding.
348int av1_get_q(const struct AV1_COMP *cpi);
349
350// Allocates memory for members of TemporalFilterData.
351// Inputs:
352// tf_data: Pointer to the structure containing temporal filter related data.
353// num_pels: Number of pixels in the block across all planes.
354// is_high_bitdepth: Whether the frame is high-bitdepth or not.
355// Returns:
356// Nothing will be returned. But the contents of tf_data will be modified.
357static AOM_INLINE bool tf_alloc_and_reset_data(TemporalFilterData *tf_data,
358 int num_pels,
359 int is_high_bitdepth) {
360 tf_data->tmp_mbmi = (MB_MODE_INFO *)malloc(sizeof(*tf_data->tmp_mbmi));
361 memset(tf_data->tmp_mbmi, 0, sizeof(*tf_data->tmp_mbmi));
362 tf_data->accum =
363 (uint32_t *)aom_memalign(16, num_pels * sizeof(*tf_data->accum));
364 tf_data->count =
365 (uint16_t *)aom_memalign(16, num_pels * sizeof(*tf_data->count));
366 memset(&tf_data->diff, 0, sizeof(tf_data->diff));
367 if (is_high_bitdepth)
368 tf_data->pred = CONVERT_TO_BYTEPTR(
369 aom_memalign(32, num_pels * 2 * sizeof(*tf_data->pred)));
370 else
371 tf_data->pred =
372 (uint8_t *)aom_memalign(32, num_pels * sizeof(*tf_data->pred));
373 if (!(tf_data->accum && tf_data->count && tf_data->pred)) {
374 aom_free(tf_data->accum);
375 aom_free(tf_data->count);
376 aom_free(tf_data->pred);
377 return false;
378 }
379 return true;
380}
381
382// Setup macroblockd params for temporal filtering process.
383// Inputs:
384// mbd: Pointer to the block for filtering.
385// tf_data: Pointer to the structure containing temporal filter related data.
386// scale: Scaling factor.
387// Returns:
388// Nothing will be returned. Contents of mbd will be modified.
389static AOM_INLINE void tf_setup_macroblockd(MACROBLOCKD *mbd,
390 TemporalFilterData *tf_data,
391 const struct scale_factors *scale) {
392 mbd->block_ref_scale_factors[0] = scale;
393 mbd->block_ref_scale_factors[1] = scale;
394 mbd->mi = &tf_data->tmp_mbmi;
395 mbd->mi[0]->motion_mode = SIMPLE_TRANSLATION;
396}
397
398// Deallocates the memory allocated for members of TemporalFilterData.
399// Inputs:
400// tf_data: Pointer to the structure containing temporal filter related data.
401// is_high_bitdepth: Whether the frame is high-bitdepth or not.
402// Returns:
403// Nothing will be returned.
404static AOM_INLINE void tf_dealloc_data(TemporalFilterData *tf_data,
405 int is_high_bitdepth) {
406 if (is_high_bitdepth)
407 tf_data->pred = (uint8_t *)CONVERT_TO_SHORTPTR(tf_data->pred);
408 free(tf_data->tmp_mbmi);
409 aom_free(tf_data->accum);
410 aom_free(tf_data->count);
411 aom_free(tf_data->pred);
412}
413
414// Saves the state prior to temporal filter process.
415// Inputs:
416// mbd: Pointer to the block for filtering.
417// input_mbmi: Backup block info to save input state.
418// input_buffer: Backup buffer pointer to save input state.
419// num_planes: Number of planes.
420// Returns:
421// Nothing will be returned. Contents of input_mbmi and input_buffer will be
422// modified.
423static INLINE void tf_save_state(MACROBLOCKD *mbd, MB_MODE_INFO ***input_mbmi,
424 uint8_t **input_buffer, int num_planes) {
425 for (int i = 0; i < num_planes; i++) {
426 input_buffer[i] = mbd->plane[i].pre[0].buf;
427 }
428 *input_mbmi = mbd->mi;
429}
430
431// Restores the initial state after temporal filter process.
432// Inputs:
433// mbd: Pointer to the block for filtering.
434// input_mbmi: Backup block info from where input state is restored.
435// input_buffer: Backup buffer pointer from where input state is restored.
436// num_planes: Number of planes.
437// Returns:
438// Nothing will be returned. Contents of mbd will be modified.
439static INLINE void tf_restore_state(MACROBLOCKD *mbd, MB_MODE_INFO **input_mbmi,
440 uint8_t **input_buffer, int num_planes) {
441 for (int i = 0; i < num_planes; i++) {
442 mbd->plane[i].pre[0].buf = input_buffer[i];
443 }
444 mbd->mi = input_mbmi;
445}
446
448#ifdef __cplusplus
449} // extern "C"
450#endif
451
452#endif // AOM_AV1_ENCODER_TEMPORAL_FILTER_H_
enum aom_bit_depth aom_bit_depth_t
Bit depth for codecThis enumeration determines the bit depth of the codec.
int av1_check_show_filtered_frame(const YV12_BUFFER_CONFIG *frame, const FRAME_DIFF *frame_diff, int q_index, aom_bit_depth_t bit_depth)
Check whether a filtered frame can be show directly.
void av1_temporal_filter(struct AV1_COMP *cpi, const int filter_frame_lookahead_idx, int gf_frame_index, FRAME_DIFF *frame_diff, YV12_BUFFER_CONFIG *output_frame)
Performs temporal filtering if needed on a source frame. For example to create a filtered alternate r...
void av1_tf_do_filtering_row(struct AV1_COMP *cpi, struct ThreadData *td, int mb_row)
Does temporal filter for a given macroblock row.
Definition: temporal_filter.c:804
Main encoder configuration data structure.
Definition: encoder.h:915
Top level encoder structure.
Definition: encoder.h:2815
Data related to the current GF/ARF group and the individual frames within the group.
Definition: firstpass.h:354
Stores the prediction/txfm mode of the current coding block.
Definition: blockd.h:222
MOTION_MODE motion_mode
The motion mode used by the inter prediction.
Definition: blockd.h:250
Temporal filter info for a gop.
Definition: temporal_filter.h:159
int tf_buf_display_index_offset[2]
Definition: temporal_filter.h:190
YV12_BUFFER_CONFIG tf_buf[2]
Definition: temporal_filter.h:170
int tf_buf_gf_index[2]
Definition: temporal_filter.h:186
FRAME_DIFF frame_diff[2]
Definition: temporal_filter.h:182
int is_temporal_filter_on
Definition: temporal_filter.h:165
int tf_buf_valid[2]
Definition: temporal_filter.h:194
YV12_BUFFER_CONFIG tf_buf_second_arf
Definition: temporal_filter.h:178
Parameters related to temporal filtering.
Definition: temporal_filter.h:96
YV12_BUFFER_CONFIG * output_frame
Definition: temporal_filter.h:109
int q_factor
Definition: temporal_filter.h:146
int num_pels
Definition: temporal_filter.h:130
int num_frames
Definition: temporal_filter.h:104
int compute_frame_diff
Definition: temporal_filter.h:118
int mb_rows
Definition: temporal_filter.h:134
int mb_cols
Definition: temporal_filter.h:138
int is_highbitdepth
Definition: temporal_filter.h:142
int filter_frame_idx
Definition: temporal_filter.h:114
Variables related to current coding block.
Definition: blockd.h:570
struct macroblockd_plane plane[3]
Definition: blockd.h:606
const struct scale_factors * block_ref_scale_factors[2]
Definition: blockd.h:687
MB_MODE_INFO ** mi
Definition: blockd.h:617
YV12 frame buffer data structure.
Definition: yv12config.h:44