1 /*
2  * Copyright © 2018, VideoLAN and dav1d authors
3  * Copyright © 2018, Two Orioles, LLC
4  * All rights reserved.
5  *
6  * Redistribution and use in source and binary forms, with or without
7  * modification, are permitted provided that the following conditions are met:
8  *
9  * 1. Redistributions of source code must retain the above copyright notice, this
10  *    list of conditions and the following disclaimer.
11  *
12  * 2. Redistributions in binary form must reproduce the above copyright notice,
13  *    this list of conditions and the following disclaimer in the documentation
14  *    and/or other materials provided with the distribution.
15  *
16  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
17  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
18  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
19  * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
20  * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
21  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
22  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
23  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
25  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26  */
27 
28 #ifndef DAV1D_SRC_FILM_GRAIN_H
29 #define DAV1D_SRC_FILM_GRAIN_H
30 
31 #include "common/bitdepth.h"
32 
33 #include "src/levels.h"
34 
35 #define GRAIN_WIDTH 82
36 #define GRAIN_HEIGHT 73
37 #define BLOCK_SIZE 32
38 #if !defined(BITDEPTH) || BITDEPTH == 8
39 #define SCALING_SIZE 256
40 typedef int8_t entry;
41 #else
42 #define SCALING_SIZE 4096
43 typedef int16_t entry;
44 #endif
45 
46 #define decl_generate_grain_y_fn(name) \
47 void (name)(entry buf[][GRAIN_WIDTH], \
48             const Dav1dFilmGrainData *const data HIGHBD_DECL_SUFFIX)
49 typedef decl_generate_grain_y_fn(*generate_grain_y_fn);
50 
51 #define decl_generate_grain_uv_fn(name) \
52 void (name)(entry buf[][GRAIN_WIDTH], \
53             const entry buf_y[][GRAIN_WIDTH], \
54             const Dav1dFilmGrainData *const data, const intptr_t uv HIGHBD_DECL_SUFFIX)
55 typedef decl_generate_grain_uv_fn(*generate_grain_uv_fn);
56 
57 #define decl_fgy_32x32xn_fn(name) \
58 void (name)(pixel *dst_row, const pixel *src_row, ptrdiff_t stride, \
59             const Dav1dFilmGrainData *data, \
60             size_t pw, const uint8_t scaling[SCALING_SIZE], \
61             const entry grain_lut[][GRAIN_WIDTH], \
62             int bh, int row_num HIGHBD_DECL_SUFFIX)
63 typedef decl_fgy_32x32xn_fn(*fgy_32x32xn_fn);
64 
65 #define decl_fguv_32x32xn_fn(name) \
66 void (name)(pixel *dst_row, const pixel *src_row, ptrdiff_t stride, \
67             const Dav1dFilmGrainData *data, int pw, \
68             const uint8_t scaling[SCALING_SIZE], \
69             const entry grain_lut[][GRAIN_WIDTH], int bh, int row_num, \
70             const pixel *luma_row, ptrdiff_t luma_stride, \
71             int uv_pl, int is_id HIGHBD_DECL_SUFFIX)
72 typedef decl_fguv_32x32xn_fn(*fguv_32x32xn_fn);
73 
74 typedef struct Dav1dFilmGrainDSPContext {
75     generate_grain_y_fn generate_grain_y;
76     generate_grain_uv_fn generate_grain_uv[3];
77 
78     fgy_32x32xn_fn fgy_32x32xn;
79     fguv_32x32xn_fn fguv_32x32xn[3];
80 } Dav1dFilmGrainDSPContext;
81 
82 bitfn_decls(void dav1d_film_grain_dsp_init, Dav1dFilmGrainDSPContext *c);
83 bitfn_decls(void dav1d_film_grain_dsp_init_x86, Dav1dFilmGrainDSPContext *c);
84 
85 #endif /* DAV1D_SRC_FILM_GRAIN_H */
86