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
26 enum { TOP, LEFT, BOTTOM, RIGHT, BOUNDARIES } UENUM1BYTE(BOUNDARY);
27
28 struct AV1CdefSyncData;
29
30 /*!\brief Parameters related to CDEF Block */
31 typedef struct {
32 uint16_t *src; /*!< CDEF intermediate buffer */
33 uint16_t *top_linebuf[MAX_MB_PLANE]; /*!< CDEF top line buffer */
34 uint16_t *bot_linebuf[MAX_MB_PLANE]; /*!< CDEF bottom line buffer */
35 uint8_t *dst; /*!< CDEF destination buffer */
36 cdef_list
37 dlist[MI_SIZE_64X64 * MI_SIZE_64X64]; /*!< CDEF 8x8 block positions */
38
39 int xdec; /*!< Sub-sampling X */
40 int ydec; /*!< Sub-sampling X */
41 int mi_wide_l2; /*!< Pixels per mi unit in width */
42 int mi_high_l2; /*!< Pixels per mi unit in height */
43 int frame_boundary[BOUNDARIES]; /*!< frame boundaries */
44
45 int damping; /*!< CDEF damping factor */
46 int coeff_shift; /*!< Bit-depth based shift for calculating filter strength */
47 int level; /*!< CDEF filtering level */
48 int sec_strength; /*!< CDEF secondary strength */
49 int cdef_count; /*!< Number of CDEF sub-blocks in superblock */
50 int dir[CDEF_NBLOCKS]
51 [CDEF_NBLOCKS]; /*!< CDEF filter direction for all 8x8 sub-blocks*/
52 int var[CDEF_NBLOCKS][CDEF_NBLOCKS]; /*!< variance for all 8x8 sub-blocks */
53
54 int dst_stride; /*!< CDEF destination buffer stride */
55 int coffset; /*!< current superblock offset in a row */
56 int roffset; /*!< current row offset */
57 } CdefBlockInfo;
58
sign(int i)59 static INLINE int sign(int i) { return i < 0 ? -1 : 1; }
60
constrain(int diff,int threshold,int damping)61 static INLINE int constrain(int diff, int threshold, int damping) {
62 if (!threshold) return 0;
63
64 const int shift = AOMMAX(0, damping - get_msb(threshold));
65 return sign(diff) *
66 AOMMIN(abs(diff), AOMMAX(0, threshold - (abs(diff) >> shift)));
67 }
68
69 #ifdef __cplusplus
70 extern "C" {
71 #endif
72
73 int 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
77 typedef 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
82 /*!\brief Function for applying CDEF to a frame
83 *
84 * \ingroup in_loop_cdef
85 * This function applies CDEF to a frame.
86 *
87 * \param[in, out] frame Compressed frame buffer
88 * \param[in, out] cm Pointer to top level common structure
89 * \param[in] xd Pointer to common current coding block structure
90 * \param[in] cdef_init_fb_row_fn Function Pointer
91 *
92 * \remark Nothing is returned. Instead, the filtered frame is output in
93 * \c frame.
94 */
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);
97 void 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 void av1_cdef_init_fb_row(const AV1_COMMON *const cm,
103 const MACROBLOCKD *const xd,
104 CdefBlockInfo *const fb_info,
105 uint16_t **const linebuf, uint16_t *const src,
106 struct AV1CdefSyncData *const cdef_sync, int fbr);
107
108 #ifdef __cplusplus
109 } // extern "C"
110 #endif
111 #endif // AOM_AV1_COMMON_CDEF_H_
112