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;
50  int dir[CDEF_NBLOCKS]
51  [CDEF_NBLOCKS];
52  int var[CDEF_NBLOCKS][CDEF_NBLOCKS];
54  int dst_stride;
55  int coffset;
56  int roffset;
58 
59 static INLINE int sign(int i) { return i < 0 ? -1 : 1; }
60 
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 
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  struct aom_internal_error_info *error_info);
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_
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.
Top level common structure used by both encoder and decoder.
Definition: av1_common_int.h:752
Parameters related to CDEF Block.
Definition: cdef.h:31
int cdef_count
Definition: cdef.h:49
int sec_strength
Definition: cdef.h:48
int roffset
Definition: cdef.h:56
int ydec
Definition: cdef.h:40
int damping
Definition: cdef.h:45
int coeff_shift
Definition: cdef.h:46
uint16_t * src
Definition: cdef.h:32
int xdec
Definition: cdef.h:39
int dst_stride
Definition: cdef.h:54
int mi_wide_l2
Definition: cdef.h:41
int mi_high_l2
Definition: cdef.h:42
int coffset
Definition: cdef.h:55
int level
Definition: cdef.h:47
uint8_t * dst
Definition: cdef.h:35
Params related to MB_MODE_INFO arrays and related info.
Definition: av1_common_int.h:503
Variables related to current coding block.
Definition: blockd.h:570
YV12 frame buffer data structure.
Definition: yv12config.h:44