cdef_block.h 2.66 KB
Newer Older
Steinar Midtskogen's avatar
Steinar Midtskogen committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
/*
 * Copyright (c) 2016, Alliance for Open Media. All rights reserved
 *
 * This source code is subject to the terms of the BSD 2 Clause License and
 * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
 * was not distributed with this source code in the LICENSE file, you can
 * obtain it at www.aomedia.org/license/software. If the Alliance for Open
 * Media Patent License 1.0 was not distributed with this source code in the
 * PATENTS file, you can obtain it at www.aomedia.org/license/patent.
 */

#if !defined(_CDEF_BLOCK_H)
#define _CDEF_BLOCK_H (1)

#include "./odintrin.h"

#define CDEF_BLOCKSIZE 64
#define CDEF_BLOCKSIZE_LOG2 6
19
#define CDEF_NBLOCKS ((1 << MAX_SB_SIZE_LOG2) / 8)
20
#define CDEF_SB_SHIFT (MAX_SB_SIZE_LOG2 - CDEF_BLOCKSIZE_LOG2)
Steinar Midtskogen's avatar
Steinar Midtskogen committed
21
22
23
24
25
26

/* We need to buffer three vertical lines. */
#define CDEF_VBORDER (3)
/* We only need to buffer three horizontal pixels too, but let's align to
   16 bytes (8 x 16 bits) to make vectorization easier. */
#define CDEF_HBORDER (8)
27
28
#define CDEF_BSTRIDE \
  ALIGN_POWER_OF_TWO((1 << MAX_SB_SIZE_LOG2) + 2 * CDEF_HBORDER, 3)
Steinar Midtskogen's avatar
Steinar Midtskogen committed
29
30

#define CDEF_VERY_LARGE (30000)
31
32
#define CDEF_INBUF_SIZE \
  (CDEF_BSTRIDE * ((1 << MAX_SB_SIZE_LOG2) + 2 * CDEF_VBORDER))
Steinar Midtskogen's avatar
Steinar Midtskogen committed
33

34
35
36
37
38
39
40
// Filter configuration
#define CDEF_CAP 1   // 1 = Cap change to largest diff
#define CDEF_FULL 0  // 1 = 7x7 filter, 0 = 5x5 filter

#if CDEF_FULL
extern const int cdef_pri_taps[2][3];
extern const int cdef_sec_taps[2][2];
41
DECLARE_ALIGNED(16, extern const int, cdef_directions[8][3]);
42
43
44
#else
extern const int cdef_pri_taps[2][2];
extern const int cdef_sec_taps[2][2];
45
DECLARE_ALIGNED(16, extern const int, cdef_directions[8][2]);
46
47
#endif

Steinar Midtskogen's avatar
Steinar Midtskogen committed
48
49
50
51
52
53
typedef struct {
  uint8_t by;
  uint8_t bx;
  uint8_t skip;
} cdef_list;

54
55
56
57
typedef void (*cdef_filter_block_func)(uint8_t *dst8, uint16_t *dst16,
                                       int dstride, const uint16_t *in,
                                       int pri_strength, int sec_strength,
                                       int dir, int pri_damping,
58
59
                                       int sec_damping, int bsize, int max,
                                       int coeff_shift);
60
61
void copy_cdef_16bit_to_16bit(uint16_t *dst, int dstride, uint16_t *src,
                              cdef_list *dlist, int cdef_count, int bsize);
Steinar Midtskogen's avatar
Steinar Midtskogen committed
62

63
64
65
66
67
68
69
void cdef_filter_fb(uint8_t *dst8, uint16_t *dst16, int dstride, uint16_t *in,
                    int xdec, int ydec, int dir[CDEF_NBLOCKS][CDEF_NBLOCKS],
                    int *dirinit, int var[CDEF_NBLOCKS][CDEF_NBLOCKS], int pli,
                    cdef_list *dlist, int cdef_count, int level,
                    int sec_strength, int pri_damping, int sec_damping,
                    int coeff_shift);
#endif