/* * VVC inter prediction DSP * * Copyright (C) 2022 Nuo Mi * * This file is part of FFmpeg. * * FFmpeg is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * FFmpeg is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with FFmpeg; if not, write to the Free Software * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA */ #include "libavcodec/h26x/h2656_inter_template.c" #include "libavutil/imgutils.h" #define TMP_STRIDE EDGE_EMU_BUFFER_STRIDE static void av_always_inline FUNC(put_scaled)(uint8_t *_dst, const ptrdiff_t _dst_stride, const uint8_t *const _src, ptrdiff_t _src_stride, const int src_height, const int _x, const int _y, const int dx, const int dy, const int height, const int8_t *hf, const int8_t *vf, const int width, const int is_uni, const int is_chroma) { int16_t tmp_array[TMP_STRIDE * MAX_PB_SIZE]; int16_t *tmp = tmp_array; pixel *dst = (pixel*)_dst; int16_t *dst16 = (int16_t*)_dst; const ptrdiff_t dst_stride = _dst_stride / sizeof(pixel); const ptrdiff_t src_stride = _src_stride / sizeof(pixel); const int shift = FFMAX(2, 14 - BIT_DEPTH); const int offset = 1 << (shift - 1); const int taps = is_chroma ? VVC_INTER_CHROMA_TAPS : VVC_INTER_LUMA_TAPS; const int extra = is_chroma ? CHROMA_EXTRA : LUMA_EXTRA; const int extra_before = is_chroma ? CHROMA_EXTRA_BEFORE : LUMA_EXTRA_BEFORE; const int shift1 = 6 - is_chroma; const int shift2 = 4 + is_chroma; const int x0 = SCALED_INT(_x); const int y0 = SCALED_INT(_y); for (int i = 0; i < width; i++) { const int tx = _x + dx * i; const int x = SCALED_INT(tx) - x0; const int mx = av_zero_extend(tx >> shift1, shift2); const int8_t *filter = hf + mx * taps; const pixel *src = (pixel*)_src - extra_before * src_stride; for (int j = 0; j < src_height + extra; j++) { tmp[j] = (is_chroma ? CHROMA_FILTER(src, 1) : LUMA_FILTER(src, 1)) >> (BIT_DEPTH - 8); src += src_stride; } tmp += TMP_STRIDE; } for (int i = 0; i < height; i++) { const int ty = _y + dy * i; const int x = SCALED_INT(ty) - y0; const int mx = av_zero_extend(ty >> shift1, shift2); const int8_t *filter = vf + mx * taps; tmp = tmp_array + extra_before; for (int j = 0; j < width; j++) { const int val = (is_chroma ? CHROMA_FILTER(tmp, 1) : LUMA_FILTER(tmp, 1)) >> 6; if (is_uni) dst[j] = av_clip_pixel((val + offset) >> shift); else dst16[j] = val; tmp += TMP_STRIDE; } if (is_uni) dst += dst_stride; else dst16 += dst_stride; } } static void FUNC(put_luma_scaled)(int16_t *_dst, const uint8_t *_src, ptrdiff_t _src_stride, const int src_height, const int x, const int y, const int dx, const int dy, const int height, const int8_t *hf, const int8_t *vf, const int width) { FUNC(put_scaled)((uint8_t *)_dst, MAX_PB_SIZE * sizeof(pixel), _src, _src_stride, src_height, x, y, dx, dy, height, hf, vf, width, 0, 0); } static void FUNC(put_chroma_scaled)(int16_t *_dst, const uint8_t *_src, ptrdiff_t _src_stride, const int src_height, const int x, const int y, const int dx, const int dy, const int height, const int8_t *hf, const int8_t *vf, const int width) { FUNC(put_scaled)((uint8_t *)_dst, MAX_PB_SIZE * sizeof(pixel), _src, _src_stride, src_height, x, y, dx, dy, height, hf, vf, width, 0, 1); } static void FUNC(put_uni_luma_scaled)(uint8_t *_dst, const ptrdiff_t _dst_stride, const uint8_t *_src, ptrdiff_t _src_stride, const int src_height, const int x, const int y, const int dx, const int dy, const int height, const int8_t *hf, const int8_t *vf, const int width) { FUNC(put_scaled)(_dst, _dst_stride, _src, _src_stride, src_height, x, y, dx, dy, height, hf, vf, width, 1, 0); } static void FUNC(put_uni_chroma_scaled)(uint8_t *_dst, const ptrdiff_t _dst_stride, const uint8_t *_src, ptrdiff_t _src_stride, const int src_height, const int x, const int y, const int dx, const int dy, const int height, const int8_t *hf, const int8_t *vf, const int width) { FUNC(put_scaled)(_dst, _dst_stride, _src, _src_stride, src_height, x, y, dx, dy, height, hf, vf, width, 1, 1); } static void av_always_inline FUNC(put_uni_w_scaled)(uint8_t *_dst, const ptrdiff_t _dst_stride, const uint8_t *const _src, ptrdiff_t _src_stride, const int src_height, const int _x, const int _y, const int dx, const int dy, const int denom, const int wx, const int _ox, const int height, const int8_t *hf, const int8_t *vf, const int width, const int is_chroma) { int16_t tmp_array[TMP_STRIDE * MAX_PB_SIZE]; int16_t *tmp = tmp_array; pixel *dst = (pixel*)_dst; const ptrdiff_t dst_stride = _dst_stride / sizeof(pixel); const ptrdiff_t src_stride = _src_stride / sizeof(pixel); const int shift = FFMAX(2, 14 - BIT_DEPTH); const int offset = 1 << (shift - 1); const int ox = _ox * (1 << (BIT_DEPTH - 8)); const int taps = is_chroma ? VVC_INTER_CHROMA_TAPS : VVC_INTER_LUMA_TAPS; const int extra = is_chroma ? CHROMA_EXTRA : LUMA_EXTRA; const int extra_before = is_chroma ? CHROMA_EXTRA_BEFORE : LUMA_EXTRA_BEFORE; const int shift1 = 6 - is_chroma; const int shift2 = 4 + is_chroma; const int x0 = SCALED_INT(_x); const int y0 = SCALED_INT(_y); for (int i = 0; i < width; i++) { const int tx = _x + dx * i; const int x = SCALED_INT(tx) - x0; const int mx = av_zero_extend(tx >> shift1, shift2); const int8_t *filter = hf + mx * taps; const pixel *src = (pixel*)_src - extra_before * src_stride; for (int j = 0; j < src_height + extra; j++) { tmp[j] = (is_chroma ? CHROMA_FILTER(src, 1) : LUMA_FILTER(src, 1)) >> (BIT_DEPTH - 8); src += src_stride; } tmp += TMP_STRIDE; } for (int i = 0; i < height; i++) { const int ty = _y + dy * i; const int x = SCALED_INT(ty) - y0; const int mx = av_zero_extend(ty >> shift1, shift2); const int8_t *filter = vf + mx * taps; tmp = tmp_array + extra_before; for (int j = 0; j < width; j++) { const int val = (is_chroma ? CHROMA_FILTER(tmp, 1) : LUMA_FILTER(tmp, 1)) >> 6; dst[j] = av_clip_pixel(((wx * val + offset) >> shift) + ox); tmp += TMP_STRIDE; } dst += dst_stride; } } static void FUNC(put_uni_luma_w_scaled)(uint8_t *_dst, const ptrdiff_t _dst_stride, const uint8_t *_src, ptrdiff_t _src_stride, const int src_height, const int x, const int y, const int dx, const int dy, const int denom, const int wx, const int ox, const int height, const int8_t *hf, const int8_t *vf, const int width) { FUNC(put_uni_w_scaled)(_dst, _dst_stride, _src, _src_stride, src_height, x, y, dx, dy, denom, wx, ox, height, hf, vf, width, 0); } static void FUNC(put_uni_chroma_w_scaled)(uint8_t *_dst, const ptrdiff_t _dst_stride, const uint8_t *_src, ptrdiff_t _src_stride, const int src_height, const int x, const int y, const int dx, const int dy, const int denom, const int wx, const int ox, const int height, const int8_t *hf, const int8_t *vf, const int width) { FUNC(put_uni_w_scaled)(_dst, _dst_stride, _src, _src_stride, src_height, x, y, dx, dy, denom, wx, ox, height, hf, vf, width, 1); } #undef TMP_STRIDE static void FUNC(avg)(uint8_t *_dst, const ptrdiff_t _dst_stride, const int16_t *src0, const int16_t *src1, const int width, const int height) { pixel *dst = (pixel*)_dst; const ptrdiff_t dst_stride = _dst_stride / sizeof(pixel); const int shift = FFMAX(3, 15 - BIT_DEPTH); const int offset = 1 << (shift - 1); for (int y = 0; y < height; y++) { for (int x = 0; x < width; x++) dst[x] = av_clip_pixel((src0[x] + src1[x] + offset) >> shift); src0 += MAX_PB_SIZE; src1 += MAX_PB_SIZE; dst += dst_stride; } } static void FUNC(w_avg)(uint8_t *_dst, const ptrdiff_t _dst_stride, const int16_t *src0, const int16_t *src1, const int width, const int height, const int denom, const int w0, const int w1, const int o0, const int o1) { pixel *dst = (pixel*)_dst; const ptrdiff_t dst_stride = _dst_stride / sizeof(pixel); const int shift = denom + FFMAX(3, 15 - BIT_DEPTH); const int offset = ((o0 + o1) * (1 << (BIT_DEPTH - 8)) + 1) * (1 << (shift - 1)); for (int y = 0; y < height; y++) { for (int x = 0; x < width; x++) dst[x] = av_clip_pixel((src0[x] * w0 + src1[x] * w1 + offset) >> shift); src0 += MAX_PB_SIZE; src1 += MAX_PB_SIZE; dst += dst_stride; } } static void FUNC(put_ciip)(uint8_t *_dst, const ptrdiff_t _dst_stride, const int width, const int height, const uint8_t *_inter, const ptrdiff_t _inter_stride, const int intra_weight) { pixel *dst = (pixel *)_dst; pixel *inter = (pixel *)_inter; const size_t dst_stride = _dst_stride / sizeof(pixel); const size_t inter_stride = _inter_stride / sizeof(pixel); const int inter_weight = 4 - intra_weight; for (int y = 0; y < height; y++) { for (int x = 0; x < width; x++) dst[x] = (dst[x] * intra_weight + inter[x] * inter_weight + 2) >> 2; dst += dst_stride; inter += inter_stride; } } static void FUNC(put_gpm)(uint8_t *_dst, ptrdiff_t dst_stride, const int width, const int height, const int16_t *src0, const int16_t *src1, const uint8_t *weights, const int step_x, const int step_y) { const int shift = FFMAX(5, 17 - BIT_DEPTH); const int offset = 1 << (shift - 1); pixel *dst = (pixel *)_dst; dst_stride /= sizeof(pixel); for (int y = 0; y < height; y++) { for (int x = 0; x < width; x++) { const uint8_t w = weights[x * step_x]; dst[x] = av_clip_pixel((src0[x] * w + src1[x] * (8 - w) + offset) >> shift); } dst += dst_stride; src0 += MAX_PB_SIZE; src1 += MAX_PB_SIZE; weights += step_y; } } //8.5.6.3.3 Luma integer sample fetching process, add one extra pad line static void FUNC(bdof_fetch_samples)(int16_t *_dst, const uint8_t *_src, const ptrdiff_t _src_stride, const int x_frac, const int y_frac, const int width, const int height) { const int x_off = (x_frac >> 3) - 1; const int y_off = (y_frac >> 3) - 1; const ptrdiff_t src_stride = _src_stride / sizeof(pixel); const pixel *src = (pixel*)_src + (x_off) + y_off * src_stride; int16_t *dst = _dst - 1 - MAX_PB_SIZE; const int shift = 14 - BIT_DEPTH; const int bdof_width = width + 2 * BDOF_BORDER_EXT; // top for (int i = 0; i < bdof_width; i++) dst[i] = src[i] << shift; dst += MAX_PB_SIZE; src += src_stride; for (int i = 0; i < height; i++) { dst[0] = src[0] << shift; dst[1 + width] = src[1 + width] << shift; dst += MAX_PB_SIZE; src += src_stride; } for (int i = 0; i < bdof_width; i++) dst[i] = src[i] << shift; } //8.5.6.3.3 Luma integer sample fetching process static void FUNC(fetch_samples)(int16_t *_dst, const uint8_t *_src, const ptrdiff_t _src_stride, const int x_frac, const int y_frac) { FUNC(bdof_fetch_samples)(_dst, _src, _src_stride, x_frac, y_frac, AFFINE_MIN_BLOCK_SIZE, AFFINE_MIN_BLOCK_SIZE); } static void FUNC(prof_grad_filter)(int16_t *gradient_h, int16_t *gradient_v, const ptrdiff_t gradient_stride, const int16_t *_src, const ptrdiff_t src_stride, const int width, const int height) { const int shift = 6; const int16_t *src = _src; for (int y = 0; y < height; y++) { const int16_t *p = src; for (int x = 0; x < width; x++) { gradient_h[x] = (p[1] >> shift) - (p[-1] >> shift); gradient_v[x] = (p[src_stride] >> shift) - (p[-src_stride] >> shift); p++; } gradient_h += gradient_stride; gradient_v += gradient_stride; src += src_stride; } } static void FUNC(apply_prof)(int16_t *dst, const int16_t *src, const int16_t *diff_mv_x, const int16_t *diff_mv_y) { const int limit = (1 << FFMAX(13, BIT_DEPTH + 1)); ///< dILimit int16_t gradient_h[AFFINE_MIN_BLOCK_SIZE * AFFINE_MIN_BLOCK_SIZE]; int16_t gradient_v[AFFINE_MIN_BLOCK_SIZE * AFFINE_MIN_BLOCK_SIZE]; FUNC(prof_grad_filter)(gradient_h, gradient_v, AFFINE_MIN_BLOCK_SIZE, src, MAX_PB_SIZE, AFFINE_MIN_BLOCK_SIZE, AFFINE_MIN_BLOCK_SIZE); for (int y = 0; y < AFFINE_MIN_BLOCK_SIZE; y++) { for (int x = 0; x < AFFINE_MIN_BLOCK_SIZE; x++) { const int o = y * AFFINE_MIN_BLOCK_SIZE + x; const int di = gradient_h[o] * diff_mv_x[o] + gradient_v[o] * diff_mv_y[o]; const int val = src[x] + av_clip(di, -limit, limit - 1); dst[x] = val; } src += MAX_PB_SIZE; dst += MAX_PB_SIZE; } } static void FUNC(apply_prof_uni)(uint8_t *_dst, const ptrdiff_t _dst_stride, const int16_t *src, const int16_t *diff_mv_x, const int16_t *diff_mv_y) { const int limit = (1 << FFMAX(13, BIT_DEPTH + 1)); ///< dILimit pixel *dst = (pixel*)_dst; const ptrdiff_t dst_stride = _dst_stride / sizeof(pixel); const int shift = 14 - BIT_DEPTH; #if BIT_DEPTH < 14 const int offset = 1 << (shift - 1); #else const int offset = 0; #endif int16_t gradient_h[AFFINE_MIN_BLOCK_SIZE * AFFINE_MIN_BLOCK_SIZE]; int16_t gradient_v[AFFINE_MIN_BLOCK_SIZE * AFFINE_MIN_BLOCK_SIZE]; FUNC(prof_grad_filter)(gradient_h, gradient_v, AFFINE_MIN_BLOCK_SIZE, src, MAX_PB_SIZE, AFFINE_MIN_BLOCK_SIZE, AFFINE_MIN_BLOCK_SIZE); for (int y = 0; y < AFFINE_MIN_BLOCK_SIZE; y++) { for (int x = 0; x < AFFINE_MIN_BLOCK_SIZE; x++) { const int o = y * AFFINE_MIN_BLOCK_SIZE + x; const int di = gradient_h[o] * diff_mv_x[o] + gradient_v[o] * diff_mv_y[o]; const int val = src[x] + av_clip(di, -limit, limit - 1); dst[x] = av_clip_pixel((val + offset) >> shift); } src += MAX_PB_SIZE; dst += dst_stride; } } static void FUNC(apply_prof_uni_w)(uint8_t *_dst, const ptrdiff_t _dst_stride, const int16_t *src, const int16_t *diff_mv_x, const int16_t *diff_mv_y, const int denom, const int wx, const int _ox) { const int limit = (1 << FFMAX(13, BIT_DEPTH + 1)); ///< dILimit pixel *dst = (pixel*)_dst; const ptrdiff_t dst_stride = _dst_stride / sizeof(pixel); const int shift = denom + FFMAX(2, 14 - BIT_DEPTH); const int offset = 1 << (shift - 1); const int ox = _ox * (1 << (BIT_DEPTH - 8)); int16_t gradient_h[AFFINE_MIN_BLOCK_SIZE * AFFINE_MIN_BLOCK_SIZE]; int16_t gradient_v[AFFINE_MIN_BLOCK_SIZE * AFFINE_MIN_BLOCK_SIZE]; FUNC(prof_grad_filter)(gradient_h, gradient_v, AFFINE_MIN_BLOCK_SIZE, src, MAX_PB_SIZE, AFFINE_MIN_BLOCK_SIZE, AFFINE_MIN_BLOCK_SIZE); for (int y = 0; y < AFFINE_MIN_BLOCK_SIZE; y++) { for (int x = 0; x < AFFINE_MIN_BLOCK_SIZE; x++) { const int o = y * AFFINE_MIN_BLOCK_SIZE + x; const int di = gradient_h[o] * diff_mv_x[o] + gradient_v[o] * diff_mv_y[o]; const int val = src[x] + av_clip(di, -limit, limit - 1); dst[x] = av_clip_pixel(((val * wx + offset) >> shift) + ox); } src += MAX_PB_SIZE; dst += dst_stride; } } static void FUNC(derive_bdof_vx_vy)(const int16_t *_src0, const int16_t *_src1, const int pad_left, const int pad_top, const int pad_right, const int pad_bottom, const int16_t **gradient_h, const int16_t **gradient_v, int* vx, int* vy) { const int shift2 = 4; const int shift3 = 1; const int thres = 1 << 4; int sgx2 = 0, sgy2 = 0, sgxgy = 0, sgxdi = 0, sgydi = 0; for (int y = -1; y < BDOF_MIN_BLOCK_SIZE + 1; y++) { const int dy = y + (pad_top && y < 0) - (pad_bottom && y == BDOF_MIN_BLOCK_SIZE); // we pad for the first and last row const int16_t *src0 = _src0 + dy * MAX_PB_SIZE; const int16_t *src1 = _src1 + dy * MAX_PB_SIZE; for (int x = -1; x < BDOF_MIN_BLOCK_SIZE + 1; x++) { const int dx = x + (pad_left && x < 0) - (pad_right && x == BDOF_MIN_BLOCK_SIZE); // we pad for the first and last col const int diff = (src0[dx] >> shift2) - (src1[dx] >> shift2); const int idx = BDOF_BLOCK_SIZE * dy + dx; const int temph = (gradient_h[0][idx] + gradient_h[1][idx]) >> shift3; const int tempv = (gradient_v[0][idx] + gradient_v[1][idx]) >> shift3; sgx2 += FFABS(temph); sgy2 += FFABS(tempv); sgxgy += VVC_SIGN(tempv) * temph; sgxdi += -VVC_SIGN(temph) * diff; sgydi += -VVC_SIGN(tempv) * diff; } } *vx = sgx2 > 0 ? av_clip((sgxdi * (1 << 2)) >> av_log2(sgx2) , -thres + 1, thres - 1) : 0; *vy = sgy2 > 0 ? av_clip(((sgydi * (1 << 2)) - ((*vx * sgxgy) >> 1)) >> av_log2(sgy2), -thres + 1, thres - 1) : 0; } static void FUNC(apply_bdof_min_block)(pixel* dst, const ptrdiff_t dst_stride, const int16_t *src0, const int16_t *src1, const int16_t **gh, const int16_t **gv, const int vx, const int vy) { const int shift4 = 15 - BIT_DEPTH; const int offset4 = 1 << (shift4 - 1); for (int y = 0; y < BDOF_MIN_BLOCK_SIZE; y++) { for (int x = 0; x < BDOF_MIN_BLOCK_SIZE; x++) { const int idx = y * BDOF_BLOCK_SIZE + x; const int bdof_offset = vx * (gh[0][idx] - gh[1][idx]) + vy * (gv[0][idx] - gv[1][idx]); dst[x] = av_clip_pixel((src0[x] + offset4 + src1[x] + bdof_offset) >> shift4); } dst += dst_stride; src0 += MAX_PB_SIZE; src1 += MAX_PB_SIZE; } } static void FUNC(apply_bdof)(uint8_t *_dst, const ptrdiff_t _dst_stride, const int16_t *_src0, const int16_t *_src1, const int block_w, const int block_h) { int16_t gradient_h[2][BDOF_BLOCK_SIZE * BDOF_BLOCK_SIZE]; int16_t gradient_v[2][BDOF_BLOCK_SIZE * BDOF_BLOCK_SIZE]; int vx, vy; const ptrdiff_t dst_stride = _dst_stride / sizeof(pixel); pixel* dst = (pixel*)_dst; FUNC(prof_grad_filter)(gradient_h[0], gradient_v[0], BDOF_BLOCK_SIZE, _src0, MAX_PB_SIZE, block_w, block_h); FUNC(prof_grad_filter)(gradient_h[1], gradient_v[1], BDOF_BLOCK_SIZE, _src1, MAX_PB_SIZE, block_w, block_h); for (int y = 0; y < block_h; y += BDOF_MIN_BLOCK_SIZE) { for (int x = 0; x < block_w; x += BDOF_MIN_BLOCK_SIZE) { const int16_t* src0 = _src0 + y * MAX_PB_SIZE + x; const int16_t* src1 = _src1 + y * MAX_PB_SIZE + x; pixel *d = dst + x; const int idx = BDOF_BLOCK_SIZE * y + x; const int16_t* gh[] = { gradient_h[0] + idx, gradient_h[1] + idx }; const int16_t* gv[] = { gradient_v[0] + idx, gradient_v[1] + idx }; FUNC(derive_bdof_vx_vy)(src0, src1, !x, !y, x + BDOF_MIN_BLOCK_SIZE == block_w, y + BDOF_MIN_BLOCK_SIZE == block_h, gh, gv, &vx, &vy); FUNC(apply_bdof_min_block)(d, dst_stride, src0, src1, gh, gv, vx, vy); } dst += BDOF_MIN_BLOCK_SIZE * dst_stride; } } #define DMVR_FILTER(src, stride) \ (filter[0] * src[x] + \ filter[1] * src[x + stride]) #define DMVR_FILTER2(filter, src0, src1) \ (filter[0] * src0 + filter[1] * src1) //8.5.3.2.2 Luma sample bilinear interpolation process static void FUNC(dmvr)(int16_t *dst, const uint8_t *_src, const ptrdiff_t _src_stride, const int height, const intptr_t mx, const intptr_t my, const int width) { #if BIT_DEPTH != 10 const pixel *src = (const pixel *)_src; const ptrdiff_t src_stride = _src_stride / sizeof(pixel); #if BIT_DEPTH > 10 const int shift4 = BIT_DEPTH - 10; const int offset4 = 1 << (shift4 - 1); #define DMVR_SHIFT(s) (((s) + offset4) >> shift4) #else #define DMVR_SHIFT(s) ((s) << (10 - BIT_DEPTH)) #endif // BIT_DEPTH > 10 for (int y = 0; y < height; y++) { for (int x = 0; x < width; x++) dst[x] = DMVR_SHIFT(src[x]); src += src_stride; dst += MAX_PB_SIZE; } #undef DMVR_SHIFT #else av_image_copy_plane((uint8_t*)dst, sizeof(int16_t) * MAX_PB_SIZE, _src, _src_stride, width * sizeof(pixel), height); #endif // BIT_DEPTH != 10 } //8.5.3.2.2 Luma sample bilinear interpolation process static void FUNC(dmvr_h)(int16_t *dst, const uint8_t *_src, const ptrdiff_t _src_stride, const int height, const intptr_t mx, const intptr_t my, const int width) { const pixel *src = (const pixel*)_src; const ptrdiff_t src_stride = _src_stride / sizeof(pixel); const int8_t *filter = ff_vvc_inter_luma_dmvr_filters[mx]; const int shift1 = BIT_DEPTH - 6; const int offset1 = 1 << (shift1 - 1); for (int y = 0; y < height; y++) { for (int x = 0; x < width; x++) dst[x] = (DMVR_FILTER(src, 1) + offset1) >> shift1; src += src_stride; dst += MAX_PB_SIZE; } } //8.5.3.2.2 Luma sample bilinear interpolation process static void FUNC(dmvr_v)(int16_t *dst, const uint8_t *_src, const ptrdiff_t _src_stride, const int height, const intptr_t mx, const intptr_t my, const int width) { const pixel *src = (pixel*)_src; const ptrdiff_t src_stride = _src_stride / sizeof(pixel); const int8_t *filter = ff_vvc_inter_luma_dmvr_filters[my]; const int shift1 = BIT_DEPTH - 6; const int offset1 = 1 << (shift1 - 1); for (int y = 0; y < height; y++) { for (int x = 0; x < width; x++) dst[x] = (DMVR_FILTER(src, src_stride) + offset1) >> shift1; src += src_stride; dst += MAX_PB_SIZE; } } //8.5.3.2.2 Luma sample bilinear interpolation process static void FUNC(dmvr_hv)(int16_t *dst, const uint8_t *_src, const ptrdiff_t _src_stride, const int height, const intptr_t mx, const intptr_t my, const int width) { int16_t tmp_array[MAX_PB_SIZE * 2]; int16_t *tmp0 = tmp_array; int16_t *tmp1 = tmp_array + MAX_PB_SIZE; const pixel *src = (const pixel*)_src; const ptrdiff_t src_stride = _src_stride / sizeof(pixel); const int8_t *filter_x = ff_vvc_inter_luma_dmvr_filters[mx]; const int8_t *filter_y = ff_vvc_inter_luma_dmvr_filters[my]; const int shift1 = BIT_DEPTH - 6; const int offset1 = 1 << (shift1 - 1); const int shift2 = 4; const int offset2 = 1 << (shift2 - 1); src -= BILINEAR_EXTRA_BEFORE * src_stride; for (int x = 0; x < width; x++) tmp0[x] = (DMVR_FILTER2(filter_x, src[x], src[x + 1]) + offset1) >> shift1; src += src_stride; for (int y = 1; y < height + BILINEAR_EXTRA; y++) { for (int x = 0; x < width; x++) { tmp1[x] = (DMVR_FILTER2(filter_x, src[x], src[x + 1]) + offset1) >> shift1; dst[x] = (DMVR_FILTER2(filter_y, tmp0[x], tmp1[x]) + offset2) >> shift2; } src += src_stride; dst += MAX_PB_SIZE; FFSWAP(int16_t *, tmp0, tmp1); } } #define PEL_FUNC(dst, C, idx1, idx2, a) \ do { \ for (int w = 0; w < 7; w++) \ inter->dst[C][w][idx1][idx2] = FUNC(a); \ } while (0) \ #define DIR_FUNCS(d, C, c) \ PEL_FUNC(put_##d, C, 0, 0, put_##d##_pixels); \ PEL_FUNC(put_##d, C, 0, 1, put_##d##_##c##_h); \ PEL_FUNC(put_##d, C, 1, 0, put_##d##_##c##_v); \ PEL_FUNC(put_##d, C, 1, 1, put_##d##_##c##_hv); \ PEL_FUNC(put_##d##_w, C, 0, 0, put_##d##_w_pixels); \ PEL_FUNC(put_##d##_w, C, 0, 1, put_##d##_##c##_w_h); \ PEL_FUNC(put_##d##_w, C, 1, 0, put_##d##_##c##_w_v); \ PEL_FUNC(put_##d##_w, C, 1, 1, put_##d##_##c##_w_hv); #define FUNCS(C, c) \ PEL_FUNC(put, C, 0, 0, put_pixels); \ PEL_FUNC(put, C, 0, 1, put_##c##_h); \ PEL_FUNC(put, C, 1, 0, put_##c##_v); \ PEL_FUNC(put, C, 1, 1, put_##c##_hv); \ DIR_FUNCS(uni, C, c); \ static void FUNC(ff_vvc_inter_dsp_init)(VVCInterDSPContext *const inter) { FUNCS(LUMA, luma); FUNCS(CHROMA, chroma); for (int i = 0; i < FF_ARRAY_ELEMS(inter->put_scaled[LUMA]); i++) { inter->put_scaled[LUMA][i] = FUNC(put_luma_scaled); inter->put_scaled[CHROMA][i] = FUNC(put_chroma_scaled); inter->put_uni_scaled[LUMA][i] = FUNC(put_uni_luma_scaled); inter->put_uni_scaled[CHROMA][i] = FUNC(put_uni_chroma_scaled); inter->put_uni_w_scaled[LUMA][i] = FUNC(put_uni_luma_w_scaled); inter->put_uni_w_scaled[CHROMA][i] = FUNC(put_uni_chroma_w_scaled); } inter->avg = FUNC(avg); inter->w_avg = FUNC(w_avg); inter->dmvr[0][0] = FUNC(dmvr); inter->dmvr[0][1] = FUNC(dmvr_h); inter->dmvr[1][0] = FUNC(dmvr_v); inter->dmvr[1][1] = FUNC(dmvr_hv); inter->put_ciip = FUNC(put_ciip); inter->put_gpm = FUNC(put_gpm); inter->fetch_samples = FUNC(fetch_samples); inter->bdof_fetch_samples = FUNC(bdof_fetch_samples); inter->apply_prof = FUNC(apply_prof); inter->apply_prof_uni = FUNC(apply_prof_uni); inter->apply_prof_uni_w = FUNC(apply_prof_uni_w); inter->apply_bdof = FUNC(apply_bdof); inter->sad = vvc_sad; } #undef FUNCS #undef PEL_FUNC #undef DMVR_FUNCS