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 
26 enum { TOP, LEFT, BOTTOM, RIGHT, BOUNDARIES } UENUM1BYTE(BOUNDARY);
27 
28 struct 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];
39  int xdec;
40  int ydec;
41  int mi_wide_l2;
42  int mi_high_l2;
43  int frame_boundary[BOUNDARIES];
45  int damping;
47  int level;
49  int cdef_count;
51  int dir[CDEF_NBLOCKS]
52  [CDEF_NBLOCKS];
53  int var[CDEF_NBLOCKS][CDEF_NBLOCKS];
55  int dst_stride;
56  int coffset;
57  int roffset;
59 
60 static INLINE int sign(int i) { return i < 0 ? -1 : 1; }
61 
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 
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_
int sec_strength
Definition: cdef.h:48
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.
Parameters related to CDEF Block.
Definition: cdef.h:31
int damping
Definition: cdef.h:45
int coffset
Definition: cdef.h:56
int cdef_count
Definition: cdef.h:49
int dst_stride
Definition: cdef.h:55
int mi_wide_l2
Definition: cdef.h:41
int coeff_shift
Definition: cdef.h:46
Params related to MB_MODE_INFO arrays and related info.
Definition: av1_common_int.h:505
int ydec
Definition: cdef.h:40
int xdec
Definition: cdef.h:39
int mi_high_l2
Definition: cdef.h:42
YV12 frame buffer data structure.
Definition: yv12config.h:38
uint16_t * src
Definition: cdef.h:32
Variables related to current coding block.
Definition: blockd.h:577
Top level common structure used by both encoder and decoder.
Definition: av1_common_int.h:751
uint8_t * dst
Definition: cdef.h:35
int level
Definition: cdef.h:47
int roffset
Definition: cdef.h:57
int is_zero_level
Definition: cdef.h:50