FFmpeg coverage

Directory: ../../../ffmpeg/
File: src/libavcodec/vvc/intra_utils.c
Date: 2025-11-10 11:20:04
Exec Total Coverage
Lines: 117 118 99.2%
Functions: 11 11 100.0%
Branches: 105 114 92.1%
Line Branch Exec Source
1 /*
2 * VVC intra prediction utils
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
23 #include <stdint.h>
24 #include <stdlib.h>
25 #include "libavutil/avassert.h"
26 #include "libavutil/macros.h"
27 #include "libavutil/common.h"
28 #include "ctu.h"
29 #include "intra.h"
30 #include "ps.h"
31 #include "dec.h"
32
33 113399 int ff_vvc_get_mip_size_id(const int w, const int h)
34 {
35
4/4
✓ Branch 0 taken 32953 times.
✓ Branch 1 taken 80446 times.
✓ Branch 2 taken 16464 times.
✓ Branch 3 taken 16489 times.
 113399 if (w == 4 && h == 4)
36 16464 return 0;
37
8/8
✓ Branch 0 taken 80446 times.
✓ Branch 1 taken 16489 times.
✓ Branch 2 taken 55983 times.
✓ Branch 3 taken 24463 times.
✓ Branch 4 taken 18528 times.
✓ Branch 5 taken 37455 times.
✓ Branch 6 taken 10937 times.
✓ Branch 7 taken 7591 times.
 96935 if ((w == 4 || h == 4) || (w == 8 && h == 8))
38 51889 return 1;
39 45046 return 2;
40 }
41
42 321254 int ff_vvc_nscale_derive(const int w, const int h, const int mode)
43 {
44 int side_size, nscale;
45
4/6
✓ Branch 0 taken 321254 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 161115 times.
✓ Branch 3 taken 160139 times.
✗ Branch 4 not taken.
✓ Branch 5 taken 161115 times.
 321254 av_assert0(mode < INTRA_LT_CCLM && !(mode > INTRA_HORZ && mode < INTRA_VERT));
46
3/6
✓ Branch 0 taken 321254 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 321254 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 321254 times.
✗ Branch 5 not taken.
 321254 if (mode == INTRA_PLANAR || mode == INTRA_DC ||
47
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 321254 times.
 321254 mode == INTRA_HORZ || mode == INTRA_VERT) {
48 ✗ nscale = (av_log2(w) + av_log2(h) - 2) >> 2;
49 } else {
50 321254 const int intra_pred_angle = ff_vvc_intra_pred_angle_derive(mode);
51 321254 const int inv_angle = ff_vvc_intra_inv_angle_derive(intra_pred_angle);
52
2/2
✓ Branch 0 taken 161115 times.
✓ Branch 1 taken 160139 times.
 321254 if (mode >= INTRA_VERT)
53 161115 side_size = h;
54
2/2
✓ Branch 0 taken 160139 times.
✓ Branch 1 taken 161115 times.
 321254 if (mode <= INTRA_HORZ)
55 160139 side_size = w;
56 321254 nscale = FFMIN(2, av_log2(side_size) - av_log2(3 * inv_angle - 2) + 8);
57 }
58 321254 return nscale;
59 }
60
61 1111317 int ff_vvc_need_pdpc(const int w, const int h, const uint8_t bdpcm_flag, const int mode, const int ref_idx)
62 {
63
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1111317 times.
 1111317 av_assert0(mode < INTRA_LT_CCLM);
64
7/8
✓ Branch 0 taken 1111317 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 1011331 times.
✓ Branch 3 taken 99986 times.
✓ Branch 4 taken 966517 times.
✓ Branch 5 taken 44814 times.
✓ Branch 6 taken 955534 times.
✓ Branch 7 taken 10983 times.
 1111317 if ((w >= 4 && h >= 4) && !ref_idx && !bdpcm_flag) {
65 int nscale;
66
6/6
✓ Branch 0 taken 601257 times.
✓ Branch 1 taken 354277 times.
✓ Branch 2 taken 521781 times.
✓ Branch 3 taken 79476 times.
✓ Branch 4 taken 464047 times.
✓ Branch 5 taken 57734 times.
 955534 if (mode == INTRA_PLANAR || mode == INTRA_DC ||
67
2/2
✓ Branch 0 taken 59962 times.
✓ Branch 1 taken 404085 times.
 464047 mode == INTRA_HORZ || mode == INTRA_VERT)
68 551449 return 1;
69
4/4
✓ Branch 0 taken 281408 times.
✓ Branch 1 taken 122677 times.
✓ Branch 2 taken 164679 times.
✓ Branch 3 taken 116729 times.
 404085 if (mode > INTRA_HORZ && mode < INTRA_VERT)
70 164679 return 0;
71 239406 nscale = ff_vvc_nscale_derive(w, h, mode);
72 239406 return nscale >= 0;
73
74 }
75 155783 return 0;
76 }
77
78 3870583 static const ReconstructedArea* get_reconstructed_area(const VVCLocalContext *lc, const int x, const int y, const int c_idx)
79 {
80 3870583 const int ch_type = c_idx > 0;
81
2/2
✓ Branch 0 taken 37325045 times.
✓ Branch 1 taken 64336 times.
 37389381 for (int i = lc->num_ras[ch_type] - 1; i >= 0; i--) {
82 37325045 const ReconstructedArea* a = &lc->ras[ch_type][i];
83 37325045 const int r = (a->x + a->w);
84 37325045 const int b = (a->y + a->h);
85
8/8
✓ Branch 0 taken 20234962 times.
✓ Branch 1 taken 17090083 times.
✓ Branch 2 taken 6589590 times.
✓ Branch 3 taken 13645372 times.
✓ Branch 4 taken 4307083 times.
✓ Branch 5 taken 2282507 times.
✓ Branch 6 taken 3096240 times.
✓ Branch 7 taken 1210843 times.
 37325045 if (a->x <= x && x < r && a->y <= y && y < b)
86 3096240 return a;
87
88 //it's too far away, no need check it;
89
4/4
✓ Branch 0 taken 13645372 times.
✓ Branch 1 taken 20583433 times.
✓ Branch 2 taken 710007 times.
✓ Branch 3 taken 12935365 times.
 34228805 if (x >= r && y >= b)
90 710007 break;
91 }
92 774343 return NULL;
93 }
94
95 1242734 int ff_vvc_get_top_available(const VVCLocalContext *lc, const int x, const int y, int target_size, const int c_idx)
96 {
97 1242734 const VVCFrameContext *fc = lc->fc;
98 1242734 const VVCSPS *sps = fc->ps.sps;
99 1242734 const int hs = sps->hshift[c_idx];
100 1242734 const int vs = sps->vshift[c_idx];
101 1242734 const int log2_ctb_size_v = sps->ctb_log2_size_y - vs;
102 1242734 const int end_of_ctb_x = ((lc->cu->x0 >> sps->ctb_log2_size_y) + 1) << sps->ctb_log2_size_y;
103 1242734 const int y0b = av_zero_extend(y, log2_ctb_size_v);
104 1242734 const int max_x = FFMIN(fc->ps.pps->width, end_of_ctb_x) >> hs;
105 const ReconstructedArea *a;
106 1242734 int px = x;
107
108
2/2
✓ Branch 0 taken 126317 times.
✓ Branch 1 taken 1116417 times.
 1242734 if (!y0b) {
109
2/2
✓ Branch 0 taken 30107 times.
✓ Branch 1 taken 96210 times.
 126317 if (!lc->ctb_up_flag)
110 30107 return 0;
111 96210 target_size = FFMIN(target_size, (lc->end_of_tiles_x >> hs) - x);
112
2/2
✓ Branch 0 taken 4760 times.
✓ Branch 1 taken 91450 times.
 96210 if (sps->r->sps_entropy_coding_sync_enabled_flag)
113 4760 target_size = FFMIN(target_size, (end_of_ctb_x >> hs) - x);
114 96210 return target_size;
115 }
116
117 1116417 target_size = FFMAX(0, FFMIN(target_size, max_x - x));
118
4/4
✓ Branch 0 taken 1940241 times.
✓ Branch 1 taken 730945 times.
✓ Branch 3 taken 1554769 times.
✓ Branch 4 taken 385472 times.
 2671186 while (target_size > 0 && (a = get_reconstructed_area(lc, px, y - 1, c_idx))) {
119 1554769 const int sz = FFMIN(target_size, a->x + a->w - px);
120 1554769 px += sz;
121 1554769 target_size -= sz;
122 }
123 1116417 return px - x;
124 }
125
126 1240662 int ff_vvc_get_left_available(const VVCLocalContext *lc, const int x, const int y, int target_size, const int c_idx)
127 {
128 1240662 const VVCFrameContext *fc = lc->fc;
129 1240662 const VVCSPS *sps = fc->ps.sps;
130 1240662 const int hs = sps->hshift[c_idx];
131 1240662 const int vs = sps->vshift[c_idx];
132 1240662 const int log2_ctb_size_h = sps->ctb_log2_size_y - hs;
133 1240662 const int x0b = av_zero_extend(x, log2_ctb_size_h);
134 1240662 const int end_of_ctb_y = ((lc->cu->y0 >> sps->ctb_log2_size_y) + 1) << sps->ctb_log2_size_y;
135 1240662 const int max_y = FFMIN(fc->ps.pps->height, end_of_ctb_y) >> vs;
136 const ReconstructedArea *a;
137 1240662 int py = y;
138
139
4/4
✓ Branch 0 taken 163491 times.
✓ Branch 1 taken 1077171 times.
✓ Branch 2 taken 30451 times.
✓ Branch 3 taken 133040 times.
 1240662 if (!x0b && !lc->ctb_left_flag)
140 30451 return 0;
141
142 1210211 target_size = FFMAX(0, FFMIN(target_size, max_y - y));
143
2/2
✓ Branch 0 taken 133040 times.
✓ Branch 1 taken 1077171 times.
 1210211 if (!x0b)
144 133040 return target_size;
145
146
4/4
✓ Branch 0 taken 1930342 times.
✓ Branch 1 taken 688300 times.
✓ Branch 3 taken 1541471 times.
✓ Branch 4 taken 388871 times.
 2618642 while (target_size > 0 && (a = get_reconstructed_area(lc, x - 1, py, c_idx))) {
147 1541471 const int sz = FFMIN(target_size, a->y + a->h - py);
148 1541471 py += sz;
149 1541471 target_size -= sz;
150 }
151 1077171 return py - y;
152 }
153
154 3711759 static int less(const void *a, const void *b)
155 {
156 3711759 return *(const int*)a - *(const int*)b;
157 }
158
159 997918 int ff_vvc_ref_filter_flag_derive(const int mode)
160 {
161 static const int modes[] = { -14, -12, -10, -6, INTRA_PLANAR, 2, 34, 66, 72, 76, 78, 80};
162 997918 return bsearch(&mode, modes, FF_ARRAY_ELEMS(modes), sizeof(int), less) != NULL;
163 }
164
165 1123992 int ff_vvc_intra_pred_angle_derive(const int pred_mode)
166 {
167 static const int angles[] = {
168 0, 1, 2, 3, 4, 6, 8, 10, 12, 14, 16, 18, 20, 23, 26, 29,
169 32, 35, 39, 45, 51, 57, 64, 73, 86, 102, 128, 171, 256, 341, 512
170 };
171 1123992 int sign = 1, idx, intra_pred_angle;
172
2/2
✓ Branch 0 taken 568901 times.
✓ Branch 1 taken 555091 times.
 1123992 if (pred_mode > INTRA_DIAG) {
173 568901 idx = pred_mode - INTRA_VERT;
174
2/2
✓ Branch 0 taken 504305 times.
✓ Branch 1 taken 50786 times.
 555091 } else if (pred_mode > 0) {
175 504305 idx = INTRA_HORZ - pred_mode;
176 } else {
177 50786 idx = INTRA_HORZ - 2 - pred_mode;
178 }
179
2/2
✓ Branch 0 taken 412734 times.
✓ Branch 1 taken 711258 times.
 1123992 if (idx < 0) {
180 412734 idx = -idx;
181 412734 sign = -1;
182 }
183 1123992 intra_pred_angle = sign * angles[idx];
184 1123992 return intra_pred_angle;
185 }
186
187 804471 int ff_vvc_intra_inv_angle_derive(const int intra_pred_angle)
188 {
189 av_assert2(intra_pred_angle != 0);
190
2/2
✓ Branch 0 taken 598104 times.
✓ Branch 1 taken 206367 times.
 804471 if (intra_pred_angle > 0)
191 598104 return ROUNDED_DIV(32*512, intra_pred_angle);
192 else
193 206367 return -ROUNDED_DIV(32*512, -intra_pred_angle);
194 }
195
196 //8.4.5.2.7 Wide angle intra prediction mode mapping process
197 1286153 int ff_vvc_wide_angle_mode_mapping(const CodingUnit *cu,
198 const int tb_width, const int tb_height, const int c_idx, int pred_mode_intra)
199 {
200 int nw, nh, wh_ratio, min, max;
201
202
4/4
✓ Branch 0 taken 237142 times.
✓ Branch 1 taken 1049011 times.
✓ Branch 2 taken 8338 times.
✓ Branch 3 taken 228804 times.
 1286153 if (cu->isp_split_type == ISP_NO_SPLIT || c_idx) {
203 1057349 nw = tb_width;
204 1057349 nh = tb_height;
205 } else {
206 228804 nw = cu->cb_width;
207 228804 nh = cu->cb_height;
208 }
209 1286153 wh_ratio = FFABS(ff_log2(nw) - ff_log2(nh));
210
2/2
✓ Branch 0 taken 303387 times.
✓ Branch 1 taken 982766 times.
 1286153 max = (wh_ratio > 1) ? (8 + 2 * wh_ratio) : 8;
211
2/2
✓ Branch 0 taken 303387 times.
✓ Branch 1 taken 982766 times.
 1286153 min = (wh_ratio > 1) ? (60 - 2 * wh_ratio) : 60;
212
213
6/6
✓ Branch 0 taken 511707 times.
✓ Branch 1 taken 774446 times.
✓ Branch 2 taken 282766 times.
✓ Branch 3 taken 228941 times.
✓ Branch 4 taken 45531 times.
✓ Branch 5 taken 237235 times.
 1286153 if (nw > nh && pred_mode_intra >=2 && pred_mode_intra < max)
214 45531 pred_mode_intra += 65;
215
5/6
✓ Branch 0 taken 336603 times.
✓ Branch 1 taken 904019 times.
✓ Branch 2 taken 336603 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 15892 times.
✓ Branch 5 taken 320711 times.
 1240622 else if (nh > nw && pred_mode_intra <= 66 && pred_mode_intra > min)
216 15892 pred_mode_intra -= 67;
217 1286153 return pred_mode_intra;
218 }
219