looprestoration.h (3115B)
1 /* 2 * Copyright © 2018-2021, 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_LOOPRESTORATION_H 29 #define DAV1D_SRC_LOOPRESTORATION_H 30 31 #include <stdint.h> 32 #include <stddef.h> 33 34 #include "common/bitdepth.h" 35 36 enum LrEdgeFlags { 37 LR_HAVE_LEFT = 1 << 0, 38 LR_HAVE_RIGHT = 1 << 1, 39 LR_HAVE_TOP = 1 << 2, 40 LR_HAVE_BOTTOM = 1 << 3, 41 }; 42 43 #ifdef BITDEPTH 44 typedef const pixel (*const_left_pixel_row)[4]; 45 #else 46 typedef const void *const_left_pixel_row; 47 #endif 48 49 typedef union LooprestorationParams { 50 ALIGN(int16_t filter[2][8], 16); 51 struct { 52 uint32_t s0, s1; 53 int16_t w0, w1; 54 } sgr; 55 } LooprestorationParams; 56 57 // Although the spec applies restoration filters over 4x4 blocks, 58 // they can be applied to a bigger surface. 59 // * w is constrained by the restoration unit size (w <= 256) 60 // * h is constrained by the stripe height (h <= 64) 61 // The filter functions are allowed to do aligned writes past the right 62 // edge of the buffer, aligned up to the minimum loop restoration unit size 63 // (which is 32 pixels for subsampled chroma and 64 pixels for luma). 64 #define decl_lr_filter_fn(name) \ 65 void (name)(pixel *dst, ptrdiff_t dst_stride, \ 66 const_left_pixel_row left, \ 67 const pixel *lpf, int w, int h, \ 68 const LooprestorationParams *params, \ 69 enum LrEdgeFlags edges HIGHBD_DECL_SUFFIX) 70 typedef decl_lr_filter_fn(*looprestorationfilter_fn); 71 72 typedef struct Dav1dLoopRestorationDSPContext { 73 looprestorationfilter_fn wiener[2]; /* 7-tap, 5-tap */ 74 looprestorationfilter_fn sgr[3]; /* 5x5, 3x3, mix */ 75 } Dav1dLoopRestorationDSPContext; 76 77 bitfn_decls(void dav1d_loop_restoration_dsp_init, Dav1dLoopRestorationDSPContext *c, int bpc); 78 79 #endif /* DAV1D_SRC_LOOPRESTORATION_H */