FFmpeg coverage


Directory: ../../../ffmpeg/
File: src/libavutil/film_grain_params.c
Date: 2024-12-12 01:08:13
Exec Total Coverage
Lines: 0 47 0.0%
Functions: 0 3 0.0%
Branches: 0 72 0.0%

Line Branch Exec Source
1 /**
2 * This file is part of FFmpeg.
3 *
4 * FFmpeg is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Lesser General Public
6 * License as published by the Free Software Foundation; either
7 * version 2.1 of the License, or (at your option) any later version.
8 *
9 * FFmpeg is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * Lesser General Public License for more details.
13 *
14 * You should have received a copy of the GNU Lesser General Public
15 * License along with FFmpeg; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17 */
18
19 #include "film_grain_params.h"
20 #include "mem.h"
21 #include "pixdesc.h"
22
23 AVFilmGrainParams *av_film_grain_params_alloc(size_t *size)
24 {
25 AVFilmGrainParams *params = av_mallocz(sizeof(AVFilmGrainParams));
26
27 if (size)
28 *size = sizeof(*params);
29
30 return params;
31 }
32
33 AVFilmGrainParams *av_film_grain_params_create_side_data(AVFrame *frame)
34 {
35 AVFilmGrainParams *fgp;
36 AVFrameSideData *side_data = av_frame_new_side_data(frame,
37 AV_FRAME_DATA_FILM_GRAIN_PARAMS,
38 sizeof(AVFilmGrainParams));
39 if (!side_data)
40 return NULL;
41
42 fgp = (AVFilmGrainParams *) side_data->data;
43 *fgp = (AVFilmGrainParams) {
44 .color_range = AVCOL_RANGE_UNSPECIFIED,
45 .color_primaries = AVCOL_PRI_UNSPECIFIED,
46 .color_trc = AVCOL_TRC_UNSPECIFIED,
47 .color_space = AVCOL_SPC_UNSPECIFIED,
48 };
49
50 return fgp;
51 }
52
53 const AVFilmGrainParams *av_film_grain_params_select(const AVFrame *frame)
54 {
55 const AVFilmGrainParams *fgp, *best = NULL;
56 const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(frame->format);
57 int bit_depth_luma, bit_depth_chroma;
58 if (!desc)
59 return NULL;
60
61 /* There are no YUV formats with different bit depth per component,
62 * so just check both against the first component for simplicity */
63 bit_depth_luma = bit_depth_chroma = desc->comp[0].depth;
64
65 for (int i = 0; i < frame->nb_side_data; i++) {
66 if (frame->side_data[i]->type != AV_FRAME_DATA_FILM_GRAIN_PARAMS)
67 continue;
68 fgp = (const AVFilmGrainParams*)frame->side_data[i]->data;
69 if (fgp->width && fgp->width > frame->width ||
70 fgp->height && fgp->height > frame->height)
71 continue;
72
73 #define CHECK(a, b, unspec) \
74 if ((a) != (unspec) && (b) != (unspec) && (a) != (b)) \
75 continue
76
77 CHECK(fgp->bit_depth_luma, bit_depth_luma, 0);
78 CHECK(fgp->bit_depth_chroma, bit_depth_chroma, 0);
79 CHECK(fgp->color_range, frame->color_range, AVCOL_RANGE_UNSPECIFIED);
80 CHECK(fgp->color_primaries, frame->color_primaries, AVCOL_PRI_UNSPECIFIED);
81 CHECK(fgp->color_trc, frame->color_trc, AVCOL_TRC_UNSPECIFIED);
82 CHECK(fgp->color_space, frame->colorspace, AVCOL_SPC_UNSPECIFIED);
83
84 switch (fgp->type) {
85 case AV_FILM_GRAIN_PARAMS_NONE:
86 continue;
87 case AV_FILM_GRAIN_PARAMS_AV1:
88 /* AOM FGS needs an exact match for the chroma resolution */
89 if (fgp->subsampling_x != desc->log2_chroma_w ||
90 fgp->subsampling_y != desc->log2_chroma_h)
91 continue;
92 break;
93 case AV_FILM_GRAIN_PARAMS_H274:
94 /* H.274 FGS can be adapted to any lower chroma resolution */
95 if (fgp->subsampling_x > desc->log2_chroma_w ||
96 fgp->subsampling_y > desc->log2_chroma_h)
97 continue;
98 break;
99 }
100
101 if (!best || best->width < fgp->width || best->height < fgp->height)
102 best = fgp;
103 }
104
105 return best;
106 }
107