FFmpeg
dsp_template.c
Go to the documentation of this file.
1 /*
2  * VVC transform and residual DSP
3  *
4  * Copyright (C) 2021 Nuo Mi
5  *
6  * This file is part of FFmpeg.
7  *
8  * FFmpeg is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU Lesser General Public
10  * License as published by the Free Software Foundation; either
11  * version 2.1 of the License, or (at your option) any later version.
12  *
13  * FFmpeg is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16  * Lesser General Public License for more details.
17  *
18  * You should have received a copy of the GNU Lesser General Public
19  * License along with FFmpeg; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21  */
22 #include "libavutil/frame.h"
24 
25 #include "dec.h"
26 #include "data.h"
27 
28 #include "inter_template.c"
29 #include "intra_template.c"
30 #include "filter_template.c"
31 
32 static void FUNC(add_residual)(uint8_t *_dst, const int *res,
33  const int w, const int h, const ptrdiff_t _stride)
34 {
35  pixel *dst = (pixel *)_dst;
36 
37  const int stride = _stride / sizeof(pixel);
38 
39  for (int y = 0; y < h; y++) {
40  for (int x = 0; x < w; x++) {
41  dst[x] = av_clip_pixel(dst[x] + *res);
42  res++;
43  }
44  dst += stride;
45  }
46 }
47 
48 static void FUNC(pred_residual_joint)(int *dst, const int *src, const int w, const int h,
49  const int c_sign, const int shift)
50 {
51  const int size = w * h;
52  for (int i = 0; i < size; i++)
53  dst[i] = (src[i] * c_sign) >> shift;
54 }
55 
56 static void FUNC(transform_bdpcm)(int *coeffs, const int width, const int height,
57  const int vertical, const int log2_transform_range)
58 {
59  int x, y;
60 
61  if (vertical) {
62  coeffs += width;
63  for (y = 0; y < height - 1; y++) {
64  for (x = 0; x < width; x++)
65  coeffs[x] = av_clip_intp2(coeffs[x] + coeffs[x - width], log2_transform_range);
66  coeffs += width;
67  }
68  } else {
69  for (y = 0; y < height; y++) {
70  for (x = 1; x < width; x++)
71  coeffs[x] = av_clip_intp2(coeffs[x] + coeffs[x - 1], log2_transform_range);
72  coeffs += width;
73  }
74  }
75 }
76 
77 // 8.7.4.6 Residual modification process for blocks using colour space conversion
78 static void FUNC(adaptive_color_transform)(int *y, int *u, int *v, const int width, const int height)
79 {
80  const int size = width * height;
81  const int bits = BIT_DEPTH + 1;
82 
83  for (int i = 0; i < size; i++) {
84  const int y0 = av_clip_intp2(y[i], bits);
85  const int cg = av_clip_intp2(u[i], bits);
86  const int co = av_clip_intp2(v[i], bits);
87  const int t = y0 - (cg >> 1);
88 
89  y[i] = cg + t;
90  u[i] = t - (co >> 1);
91  v[i] = co + u[i];
92  }
93 }
94 
95 static void FUNC(ff_vvc_itx_dsp_init)(VVCItxDSPContext *const itx)
96 {
97 #define VVC_ITX(TYPE, type, s) \
98  itx->itx[VVC_##TYPE][VVC_##TX_SIZE_##s] = ff_vvc_inv_##type##_##s; \
99 
100 #define VVC_ITX_COMMON(TYPE, type) \
101  VVC_ITX(TYPE, type, 4); \
102  VVC_ITX(TYPE, type, 8); \
103  VVC_ITX(TYPE, type, 16); \
104  VVC_ITX(TYPE, type, 32);
105 
106  itx->add_residual = FUNC(add_residual);
107  itx->pred_residual_joint = FUNC(pred_residual_joint);
108  itx->transform_bdpcm = FUNC(transform_bdpcm);
109  VVC_ITX(DCT2, dct2, 2)
110  VVC_ITX(DCT2, dct2, 64)
111  VVC_ITX_COMMON(DCT2, dct2)
112  VVC_ITX_COMMON(DCT8, dct8)
113  VVC_ITX_COMMON(DST7, dst7)
114 
115  itx->adaptive_color_transform = FUNC(adaptive_color_transform);
116 
117 #undef VVC_ITX
118 #undef VVC_ITX_COMMON
119 }
_dst
uint8_t * _dst
Definition: dsp.h:56
VVC_ITX
#define VVC_ITX(TYPE, type, s)
data.h
w
uint8_t w
Definition: llviddspenc.c:38
filter_template.c
_stride
ptrdiff_t _stride
Definition: h264pred_template.c:411
VVCItxDSPContext
Definition: dsp.h:123
u
#define u(width, name, range_min, range_max)
Definition: cbs_apv.c:83
ff_vvc_itx_dsp_init
static void FUNC() ff_vvc_itx_dsp_init(VVCItxDSPContext *const itx)
Definition: dsp_template.c:95
bits
uint8_t bits
Definition: vp3data.h:128
transform_bdpcm
static void FUNC() transform_bdpcm(int *coeffs, const int width, const int height, const int vertical, const int log2_transform_range)
Definition: dsp_template.c:56
av_clip_intp2
#define av_clip_intp2
Definition: common.h:121
pixel
uint8_t pixel
Definition: tiny_ssim.c:41
bit_depth_template.c
inter_template.c
height
#define height
Definition: dsp.h:89
shift
static int shift(int a, int b)
Definition: bonk.c:261
dst
uint8_t ptrdiff_t const uint8_t ptrdiff_t int intptr_t intptr_t int int16_t * dst
Definition: dsp.h:87
size
int size
Definition: twinvq_data.h:10344
frame.h
add_residual
static void FUNC() add_residual(uint8_t *_dst, const int *res, const int w, const int h, const ptrdiff_t _stride)
Definition: dsp_template.c:32
i
#define i(width, name, range_min, range_max)
Definition: cbs_h2645.c:256
stride
#define stride
Definition: h264pred_template.c:536
av_clip_pixel
#define av_clip_pixel(a)
Definition: bit_depth_template.c:95
pred_residual_joint
static void FUNC() pred_residual_joint(int *dst, const int *src, const int w, const int h, const int c_sign, const int shift)
Definition: dsp_template.c:48
FUNC
#define FUNC(a)
Definition: bit_depth_template.c:101
BIT_DEPTH
#define BIT_DEPTH
Definition: dsp_init.c:55
VVC_ITX_COMMON
#define VVC_ITX_COMMON(TYPE, type)
intra_template.c
h
h
Definition: vp9dsp_template.c:2070
width
#define width
Definition: dsp.h:89
src
#define src
Definition: vp8dsp.c:248
adaptive_color_transform
static void FUNC() adaptive_color_transform(int *y, int *u, int *v, const int width, const int height)
Definition: dsp_template.c:78
dec.h