ref: bc008834b9e9d08f86d18227ca05a251492a0617
dir: /src/cdef_apply_tmpl.c/
/* * Copyright © 2018, VideoLAN and dav1d authors * Copyright © 2018, Two Orioles, LLC * All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions are met: * * 1. Redistributions of source code must retain the above copyright notice, this * list of conditions and the following disclaimer. * * 2. Redistributions in binary form must reproduce the above copyright notice, * this list of conditions and the following disclaimer in the documentation * and/or other materials provided with the distribution. * * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. */ #include "config.h" #include <string.h> #include "common/intops.h" #include "src/cdef_apply.h" enum Backup2x8Flags { BACKUP_2X8_Y = 1 << 0, BACKUP_2X8_UV = 1 << 1, }; static void backup2lines(pixel *const dst[3], /*const*/ pixel *const src[3], const ptrdiff_t stride[2], const enum Dav1dPixelLayout layout) { const ptrdiff_t y_stride = PXSTRIDE(stride[0]); if (y_stride < 0) pixel_copy(dst[0] + y_stride, src[0] + 7 * y_stride, -2 * y_stride); else pixel_copy(dst[0], src[0] + 6 * y_stride, 2 * y_stride); if (layout != DAV1D_PIXEL_LAYOUT_I400) { const ptrdiff_t uv_stride = PXSTRIDE(stride[1]); if (uv_stride < 0) { const int uv_off = layout == DAV1D_PIXEL_LAYOUT_I420 ? 3 : 7; pixel_copy(dst[1] + uv_stride, src[1] + uv_off * uv_stride, -2 * uv_stride); pixel_copy(dst[2] + uv_stride, src[2] + uv_off * uv_stride, -2 * uv_stride); } else { const int uv_off = layout == DAV1D_PIXEL_LAYOUT_I420 ? 2 : 6; pixel_copy(dst[1], src[1] + uv_off * uv_stride, 2 * uv_stride); pixel_copy(dst[2], src[2] + uv_off * uv_stride, 2 * uv_stride); } } } static void backup2x8(pixel dst[3][8][2], /*const*/ pixel *const src[3], const ptrdiff_t src_stride[2], int x_off, const enum Dav1dPixelLayout layout, const enum Backup2x8Flags flag) { ptrdiff_t y_off = 0; if (flag & BACKUP_2X8_Y) { for (int y = 0; y < 8; y++, y_off += PXSTRIDE(src_stride[0])) pixel_copy(dst[0][y], &src[0][y_off + x_off - 2], 2); } if (layout == DAV1D_PIXEL_LAYOUT_I400 || !(flag & BACKUP_2X8_UV)) return; const int ss_ver = layout == DAV1D_PIXEL_LAYOUT_I420; const int ss_hor = layout != DAV1D_PIXEL_LAYOUT_I444; x_off >>= ss_hor; y_off = 0; for (int y = 0; y < (8 >> ss_ver); y++, y_off += PXSTRIDE(src_stride[1])) { pixel_copy(dst[1][y], &src[1][y_off + x_off - 2], 2); pixel_copy(dst[2][y], &src[2][y_off + x_off - 2], 2); } } static int adjust_strength(const int strength, const unsigned var) { if (!var) return 0; const int i = var >> 6 ? imin(ulog2(var >> 6), 12) : 0; return (strength * (4 + i) + 8) >> 4; } void bytefn(dav1d_cdef_brow)(Dav1dFrameContext *const f, pixel *const p[3], const Av1Filter *const lflvl, const int by_start, const int by_end) { const int bitdepth_min_8 = BITDEPTH == 8 ? 0 : f->cur.p.bpc - 8; const Dav1dDSPContext *const dsp = f->dsp; enum CdefEdgeFlags edges = CDEF_HAVE_BOTTOM | (by_start > 0 ? CDEF_HAVE_TOP : 0); pixel *ptrs[3] = { p[0], p[1], p[2] }; const int sbsz = 16; const int sb64w = f->sb128w << 1; const int damping = f->frame_hdr->cdef.damping + bitdepth_min_8; const enum Dav1dPixelLayout layout = f->cur.p.layout; const int uv_idx = DAV1D_PIXEL_LAYOUT_I444 - layout; const int ss_ver = layout == DAV1D_PIXEL_LAYOUT_I420; const int ss_hor = layout != DAV1D_PIXEL_LAYOUT_I444; static const uint8_t uv_dirs[2][8] = { { 0, 1, 2, 3, 4, 5, 6, 7 }, { 7, 0, 2, 4, 5, 6, 6, 6 } }; const uint8_t *uv_dir = uv_dirs[layout == DAV1D_PIXEL_LAYOUT_I422]; for (int bit = 0, by = by_start; by < by_end; by += 2, edges |= CDEF_HAVE_TOP) { const int tf = f->lf.top_pre_cdef_toggle; const int by_idx = by & 30; if (by + 2 >= f->bh) edges &= ~CDEF_HAVE_BOTTOM; if (edges & CDEF_HAVE_BOTTOM) // backup pre-filter data for next iteration backup2lines(f->lf.cdef_line[!tf], ptrs, f->cur.stride, layout); ALIGN_STK_16(pixel, lr_bak, 2 /* idx */, [3 /* plane */][8 /* y */][2 /* x */]); pixel *iptrs[3] = { ptrs[0], ptrs[1], ptrs[2] }; edges &= ~CDEF_HAVE_LEFT; edges |= CDEF_HAVE_RIGHT; enum Backup2x8Flags prev_flag = 0; for (int sbx = 0, last_skip = 1; sbx < sb64w; sbx++, edges |= CDEF_HAVE_LEFT) { const int sb128x = sbx >> 1; const int sb64_idx = ((by & sbsz) >> 3) + (sbx & 1); const int cdef_idx = lflvl[sb128x].cdef_idx[sb64_idx]; if (cdef_idx == -1 || (!f->frame_hdr->cdef.y_strength[cdef_idx] && !f->frame_hdr->cdef.uv_strength[cdef_idx])) { last_skip = 1; goto next_sb; } const int y_lvl = f->frame_hdr->cdef.y_strength[cdef_idx]; const int uv_lvl = f->frame_hdr->cdef.uv_strength[cdef_idx]; const enum Backup2x8Flags flag = !!y_lvl + (!!uv_lvl << 1); const int y_pri_lvl = (y_lvl >> 2) << bitdepth_min_8; int y_sec_lvl = y_lvl & 3; y_sec_lvl += y_sec_lvl == 3; y_sec_lvl <<= bitdepth_min_8; const int uv_pri_lvl = (uv_lvl >> 2) << bitdepth_min_8; int uv_sec_lvl = uv_lvl & 3; uv_sec_lvl += uv_sec_lvl == 3; uv_sec_lvl <<= bitdepth_min_8; pixel *bptrs[3] = { iptrs[0], iptrs[1], iptrs[2] }; for (int bx = sbx * sbsz; bx < imin((sbx + 1) * sbsz, f->bw); bx += 2, edges |= CDEF_HAVE_LEFT) { if (bx + 2 >= f->bw) edges &= ~CDEF_HAVE_RIGHT; // check if this 8x8 block had any coded coefficients; if not, // go to the next block const unsigned bx_mask = 3U << (bx & 14); const int bx_idx = (bx & 16) >> 4; if (!((lflvl[sb128x].noskip_mask[by_idx + 0][bx_idx] | lflvl[sb128x].noskip_mask[by_idx + 1][bx_idx]) & bx_mask)) { last_skip = 1; goto next_b; } const int do_left = last_skip ? flag : (prev_flag ^ flag) & flag; prev_flag = flag; if (do_left && edges & CDEF_HAVE_LEFT) { // we didn't backup the prefilter data because it wasn't // there, so do it here instead backup2x8(lr_bak[bit], bptrs, f->cur.stride, 0, layout, do_left); } if (edges & CDEF_HAVE_RIGHT) { // backup pre-filter data for next iteration backup2x8(lr_bak[!bit], bptrs, f->cur.stride, 8, layout, flag); } int dir; unsigned variance; if (y_pri_lvl || uv_pri_lvl) dir = dsp->cdef.dir(bptrs[0], f->cur.stride[0], &variance HIGHBD_CALL_SUFFIX); if (y_pri_lvl) { const int adj_y_pri_lvl = adjust_strength(y_pri_lvl, variance); if (adj_y_pri_lvl || y_sec_lvl) dsp->cdef.fb[0](bptrs[0], f->cur.stride[0], lr_bak[bit][0], &f->lf.cdef_line[tf][0][bx * 4], adj_y_pri_lvl, y_sec_lvl, dir, damping, edges HIGHBD_CALL_SUFFIX); } else if (y_sec_lvl) dsp->cdef.fb[0](bptrs[0], f->cur.stride[0], lr_bak[bit][0], &f->lf.cdef_line[tf][0][bx * 4], 0, y_sec_lvl, 0, damping, edges HIGHBD_CALL_SUFFIX); if (uv_lvl) { assert(layout != DAV1D_PIXEL_LAYOUT_I400); const int uvdir = uv_pri_lvl ? uv_dir[dir] : 0; for (int pl = 1; pl <= 2; pl++) { dsp->cdef.fb[uv_idx](bptrs[pl], f->cur.stride[1], lr_bak[bit][pl], &f->lf.cdef_line[tf][pl][bx * 4 >> ss_hor], uv_pri_lvl, uv_sec_lvl, uvdir, damping - 1, edges HIGHBD_CALL_SUFFIX); } } bit ^= 1; last_skip = 0; next_b: bptrs[0] += 8; bptrs[1] += 8 >> ss_hor; bptrs[2] += 8 >> ss_hor; } next_sb: iptrs[0] += sbsz * 4; iptrs[1] += sbsz * 4 >> ss_hor; iptrs[2] += sbsz * 4 >> ss_hor; } ptrs[0] += 8 * PXSTRIDE(f->cur.stride[0]); ptrs[1] += 8 * PXSTRIDE(f->cur.stride[1]) >> ss_ver; ptrs[2] += 8 * PXSTRIDE(f->cur.stride[1]) >> ss_ver; f->lf.top_pre_cdef_toggle ^= 1; } }