| AOMedia AV1 Codec

av1_loopfilter.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_COMMON_AV1_LOOPFILTER_H_
13#define AOM_AV1_COMMON_AV1_LOOPFILTER_H_
14
15#include "config/aom_config.h"
16
17#include "aom/internal/aom_codec_internal.h"
18
19#include "aom_ports/mem.h"
20#include "av1/common/blockd.h"
21#include "av1/common/seg_common.h"
22
23#ifdef __cplusplus
24 extern "C" {
25#endif
26
27#define MAX_LOOP_FILTER 63
28#define MAX_SHARPNESS 7
29
30#define SIMD_WIDTH 16
31
32 enum lf_path {
33 LF_PATH_420,
34 LF_PATH_444,
35 LF_PATH_SLOW,
36};
37
39 enum { VERT_EDGE = 0, HORZ_EDGE = 1, NUM_EDGE_DIRS } UENUM1BYTE(EDGE_DIR);
40 typedef struct {
41 uint64_t bits[4];
42} FilterMask;
43
44 struct loopfilter {
45 int filter_level[2];
46 int filter_level_u;
47 int filter_level_v;
48
49 int backup_filter_level[2];
50 int backup_filter_level_u;
51 int backup_filter_level_v;
52
53 int sharpness_level;
54
55 uint8_t mode_ref_delta_enabled;
56 uint8_t mode_ref_delta_update;
57
58 // 0 = Intra, Last, Last2+Last3,
59 // GF, BRF, ARF2, ARF
60 int8_t ref_deltas[REF_FRAMES];
61
62 // 0 = ZERO_MV, MV
63 int8_t mode_deltas[MAX_MODE_LF_DELTAS];
64};
65
66// Need to align this structure so when it is declared and
67// passed it can be loaded into vector registers.
68 typedef struct {
69 DECLARE_ALIGNED(SIMD_WIDTH, uint8_t, mblim[SIMD_WIDTH]);
70 DECLARE_ALIGNED(SIMD_WIDTH, uint8_t, lim[SIMD_WIDTH]);
71 DECLARE_ALIGNED(SIMD_WIDTH, uint8_t, hev_thr[SIMD_WIDTH]);
72} loop_filter_thresh;
73
74 typedef struct {
75 loop_filter_thresh lfthr[MAX_LOOP_FILTER + 1];
76 uint8_t lvl[MAX_MB_PLANE][MAX_SEGMENTS][2][REF_FRAMES][MAX_MODE_LF_DELTAS];
77} loop_filter_info_n;
78
79 typedef struct AV1_DEBLOCKING_PARAMETERS {
80 // length of the filter applied to the outer edge
81 uint8_t filter_length;
82 // deblocking limits
83 const loop_filter_thresh *lfthr;
84} AV1_DEBLOCKING_PARAMETERS;
85
86 typedef struct LoopFilterWorkerData {
87 YV12_BUFFER_CONFIG *frame_buffer;
88 struct AV1Common *cm;
89 struct macroblockd_plane planes[MAX_MB_PLANE];
90 // TODO(Ranjit): When the filter functions are modified to use xd->lossless
91 // add lossless as a member here.
92 MACROBLOCKD *xd;
93
94 AV1_DEBLOCKING_PARAMETERS params_buf[MAX_MIB_SIZE];
95 TX_SIZE tx_buf[MAX_MIB_SIZE];
96 struct aom_internal_error_info error_info;
97} LFWorkerData;
99
100/* assorted loopfilter functions which get used elsewhere */
101 struct AV1Common;
102 struct macroblockd;
103 struct AV1LfSyncData;
104
105 void av1_loop_filter_init(struct AV1Common *cm);
106
107 void av1_loop_filter_frame_init(struct AV1Common *cm, int plane_start,
108 int plane_end);
109
110 void av1_filter_block_plane_vert(const struct AV1Common *const cm,
111 const MACROBLOCKD *const xd, const int plane,
112 const MACROBLOCKD_PLANE *const plane_ptr,
113 const uint32_t mi_row, const uint32_t mi_col);
114
115 void av1_filter_block_plane_horz(const struct AV1Common *const cm,
116 const MACROBLOCKD *const xd, const int plane,
117 const MACROBLOCKD_PLANE *const plane_ptr,
118 const uint32_t mi_row, const uint32_t mi_col);
119
120 void av1_filter_block_plane_vert_opt(
121 const struct AV1Common *const cm, const MACROBLOCKD *const xd,
122 const MACROBLOCKD_PLANE *const plane_ptr, const uint32_t mi_row,
123 const uint32_t mi_col, AV1_DEBLOCKING_PARAMETERS *params_buf,
124 TX_SIZE *tx_buf, int num_mis_in_lpf_unit_height_log2);
125
126 void av1_filter_block_plane_vert_opt_chroma(
127 const struct AV1Common *const cm, const MACROBLOCKD *const xd,
128 const MACROBLOCKD_PLANE *const plane_ptr, const uint32_t mi_row,
129 const uint32_t mi_col, AV1_DEBLOCKING_PARAMETERS *params_buf,
130 TX_SIZE *tx_buf, int plane, bool joint_filter_chroma,
131 int num_mis_in_lpf_unit_height_log2);
132
133 void av1_filter_block_plane_horz_opt(
134 const struct AV1Common *const cm, const MACROBLOCKD *const xd,
135 const MACROBLOCKD_PLANE *const plane_ptr, const uint32_t mi_row,
136 const uint32_t mi_col, AV1_DEBLOCKING_PARAMETERS *params_buf,
137 TX_SIZE *tx_buf, int num_mis_in_lpf_unit_height_log2);
138
139 void av1_filter_block_plane_horz_opt_chroma(
140 const struct AV1Common *const cm, const MACROBLOCKD *const xd,
141 const MACROBLOCKD_PLANE *const plane_ptr, const uint32_t mi_row,
142 const uint32_t mi_col, AV1_DEBLOCKING_PARAMETERS *params_buf,
143 TX_SIZE *tx_buf, int plane, bool joint_filter_chroma,
144 int num_mis_in_lpf_unit_height_log2);
145
146#ifdef __cplusplus
147} // extern "C"
148#endif
149
150#endif // AOM_AV1_COMMON_AV1_LOOPFILTER_H_
AV1Common
Top level common structure used by both encoder and decoder.
Definition av1_common_int.h:764
macroblockd
Variables related to current coding block.
Definition blockd.h:570