AOMedia AV1 Codec: av1/common/cdef.h Source File

| AOMedia AV1 Codec

cdef.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#ifndef AOM_AV1_COMMON_CDEF_H_
12#define AOM_AV1_COMMON_CDEF_H_
13
14#define CDEF_STRENGTH_BITS 6
15
16#define CDEF_PRI_STRENGTHS 16
17#define CDEF_SEC_STRENGTHS 4
18
19#include "config/aom_config.h"
20
21#include "aom/aom_integer.h"
22#include "aom_ports/mem.h"
23#include "av1/common/av1_common_int.h"
24#include "av1/common/cdef_block.h"
25
26enum { TOP, LEFT, BOTTOM, RIGHT, BOUNDARIES } UENUM1BYTE(BOUNDARY);
27
28struct AV1CdefSyncData;
29
31 typedef struct {
32 uint16_t *src;
33 uint16_t *top_linebuf[MAX_MB_PLANE];
34 uint16_t *bot_linebuf[MAX_MB_PLANE];
35 uint8_t *dst;
36 cdef_list
37 dlist[MI_SIZE_64X64 * MI_SIZE_64X64];
38
39 int xdec;
40 int ydec;
41 int mi_wide_l2;
42 int mi_high_l2;
43 int frame_boundary[BOUNDARIES];
44
45 int damping;
46 int coeff_shift;
47 int level;
48 int sec_strength;
49 int cdef_count;
50 int dir[CDEF_NBLOCKS]
51 [CDEF_NBLOCKS];
52 int var[CDEF_NBLOCKS][CDEF_NBLOCKS];
53
54 int dst_stride;
55 int coffset;
56 int roffset;
57} CdefBlockInfo;
58
59static inline int sign(int i) { return i < 0 ? -1 : 1; }
60
61static inline int constrain(int diff, int threshold, int damping) {
62 if (!threshold) return 0;
63
64 int shift = damping - get_msb(threshold);
65 shift = AOMMAX(0, shift);
66 return sign(diff) * clamp(threshold - (abs(diff) >> shift), 0, abs(diff));
67}
68
69#ifdef __cplusplus
70extern "C" {
71#endif
72
73int av1_cdef_compute_sb_list(const CommonModeInfoParams *const mi_params,
74 int mi_row, int mi_col, cdef_list *dlist,
75 BLOCK_SIZE bsize);
76
77typedef void (*cdef_init_fb_row_t)(
78 const AV1_COMMON *const cm, const MACROBLOCKD *const xd,
79 CdefBlockInfo *const fb_info, uint16_t **const linebuf, uint16_t *const src,
80 struct AV1CdefSyncData *const cdef_sync, int fbr);
81
95 void av1_cdef_frame(YV12_BUFFER_CONFIG *frame, AV1_COMMON *const cm,
96 MACROBLOCKD *xd, cdef_init_fb_row_t cdef_init_fb_row_fn);
97void av1_cdef_fb_row(const AV1_COMMON *const cm, MACROBLOCKD *xd,
98 uint16_t **const linebuf, uint16_t **const colbuf,
99 uint16_t *const src, int fbr,
100 cdef_init_fb_row_t cdef_init_fb_row_fn,
101 struct AV1CdefSyncData *const cdef_sync,
102 struct aom_internal_error_info *error_info);
103void av1_cdef_init_fb_row(const AV1_COMMON *const cm,
104 const MACROBLOCKD *const xd,
105 CdefBlockInfo *const fb_info,
106 uint16_t **const linebuf, uint16_t *const src,
107 struct AV1CdefSyncData *const cdef_sync, int fbr);
108
109#ifdef __cplusplus
110} // extern "C"
111#endif
112#endif // AOM_AV1_COMMON_CDEF_H_
av1_cdef_frame
void av1_cdef_frame(YV12_BUFFER_CONFIG *frame, AV1_COMMON *const cm, MACROBLOCKD *xd, cdef_init_fb_row_t cdef_init_fb_row_fn)
Function for applying CDEF to a frame.
CdefBlockInfo
Parameters related to CDEF Block.
Definition cdef.h:31
CdefBlockInfo::frame_boundary
int frame_boundary[BOUNDARIES]
Definition cdef.h:43
CdefBlockInfo::cdef_count
int cdef_count
Definition cdef.h:49
CdefBlockInfo::sec_strength
int sec_strength
Definition cdef.h:48
CdefBlockInfo::roffset
int roffset
Definition cdef.h:56
CdefBlockInfo::ydec
int ydec
Definition cdef.h:40
CdefBlockInfo::damping
int damping
Definition cdef.h:45
CdefBlockInfo::dlist
cdef_list dlist[MI_SIZE_64X64 *MI_SIZE_64X64]
Definition cdef.h:37
CdefBlockInfo::coeff_shift
int coeff_shift
Definition cdef.h:46
CdefBlockInfo::dir
int dir[((1<< MAX_SB_SIZE_LOG2)/8)][((1<< MAX_SB_SIZE_LOG2)/8)]
Definition cdef.h:51
CdefBlockInfo::src
uint16_t * src
Definition cdef.h:32
CdefBlockInfo::xdec
int xdec
Definition cdef.h:39
CdefBlockInfo::dst_stride
int dst_stride
Definition cdef.h:54
CdefBlockInfo::mi_wide_l2
int mi_wide_l2
Definition cdef.h:41
CdefBlockInfo::bot_linebuf
uint16_t * bot_linebuf[3]
Definition cdef.h:34
CdefBlockInfo::mi_high_l2
int mi_high_l2
Definition cdef.h:42
CdefBlockInfo::coffset
int coffset
Definition cdef.h:55
CdefBlockInfo::level
int level
Definition cdef.h:47
CdefBlockInfo::var
int var[((1<< MAX_SB_SIZE_LOG2)/8)][((1<< MAX_SB_SIZE_LOG2)/8)]
Definition cdef.h:52
CdefBlockInfo::top_linebuf
uint16_t * top_linebuf[3]
Definition cdef.h:33
CdefBlockInfo::dst
uint8_t * dst
Definition cdef.h:35
CommonModeInfoParams
Params related to MB_MODE_INFO arrays and related info.
Definition av1_common_int.h:511