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 is_zero_level; /*!< CDEF filtering level ON/OFF */
51 int dir[CDEF_NBLOCKS]
52 [CDEF_NBLOCKS]; /*!< CDEF filter direction for all 8x8 sub-blocks*/
53 int var[CDEF_NBLOCKS][CDEF_NBLOCKS]; /*!< variance for all 8x8 sub-blocks */
54
55 int dst_stride; /*!< CDEF destination buffer stride */
56 int coffset; /*!< current superblock offset in a row */
57 int roffset; /*!< current row offset */
58 } CdefBlockInfo;
59
sign(int i)60 static INLINE int sign(int i) { return i < 0 ? -1 : 1; }
61
constrain(int diff,int threshold,int damping)62 static INLINE int constrain(int diff, int threshold, int damping) {
63 if (!threshold) return 0;
64
65 const int shift = AOMMAX(0, damping - get_msb(threshold));
66 return sign(diff) *
67 AOMMIN(abs(diff), AOMMAX(0, threshold - (abs(diff) >> shift)));
68 }
69
70 #ifdef __cplusplus
71 extern "C" {
72 #endif
73
74 int av1_cdef_compute_sb_list(const CommonModeInfoParams *const mi_params,
75 int mi_row, int mi_col, cdef_list *dlist,
76 BLOCK_SIZE bsize);
77
78 typedef void (*cdef_init_fb_row_t)(
79 const AV1_COMMON *const cm, const MACROBLOCKD *const xd,
80 CdefBlockInfo *const fb_info, uint16_t **const linebuf, uint16_t *const src,
81 struct AV1CdefSyncData *const cdef_sync, int fbr);
82
83 /*!\brief Function for applying CDEF to a frame
84 *
85 * \ingroup in_loop_cdef
86 * This function applies CDEF to a frame.
87 *
88 * \param[in, out] frame Compressed frame buffer
89 * \param[in, out] cm Pointer to top level common structure
90 * \param[in] xd Pointer to common current coding block structure
91 * \param[in] cdef_init_fb_row_fn Function Pointer
92 *
93 * \return Nothing is returned. Instead, the filtered frame is output in
94 * \c frame.
95 */
96 void av1_cdef_frame(YV12_BUFFER_CONFIG *frame, AV1_COMMON *const cm,
97 MACROBLOCKD *xd, cdef_init_fb_row_t cdef_init_fb_row_fn);
98 void av1_cdef_fb_row(const AV1_COMMON *const cm, MACROBLOCKD *xd,
99 uint16_t **const linebuf, uint16_t **const colbuf,
100 uint16_t *const src, int fbr,
101 cdef_init_fb_row_t cdef_init_fb_row_fn,
102 struct AV1CdefSyncData *const cdef_sync);
103 void 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_
113