FFmpeg coverage

Directory: ../../../ffmpeg/
File: src/libavfilter/colorspace.c
Date: 2026-10-29 17:07:12
Exec Total Coverage
Lines: 76 105 72.4%
Functions: 5 7 71.4%
Branches: 12 40 30.0%
Line Branch Exec Source
1 /*
2 * Copyright (c) 2016 Ronald S. Bultje <rsbultje@gmail.com>
3 * This file is part of FFmpeg.
4 *
5 * FFmpeg is free software; you can redistribute it and/or
6 * modify it under the terms of the GNU Lesser General Public
7 * License as published by the Free Software Foundation; either
8 * version 2.1 of the License, or (at your option) any later version.
9 *
10 * FFmpeg is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 * Lesser General Public License for more details.
14 *
15 * You should have received a copy of the GNU Lesser General Public
16 * License along with FFmpeg; if not, write to the Free Software
17 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
18 */
19
20 #include "libavutil/frame.h"
21 #include "libavutil/mastering_display_metadata.h"
22 #include "libavutil/pixdesc.h"
23
24 #include "colorspace.h"
25
26
27 20 void ff_matrix_invert_3x3(const double in[3][3], double out[3][3])
28 {
29 20 double m00 = in[0][0], m01 = in[0][1], m02 = in[0][2],
30 20 m10 = in[1][0], m11 = in[1][1], m12 = in[1][2],
31 20 m20 = in[2][0], m21 = in[2][1], m22 = in[2][2];
32 int i, j;
33 double det;
34
35 20 out[0][0] = (m11 * m22 - m21 * m12);
36 20 out[0][1] = -(m01 * m22 - m21 * m02);
37 20 out[0][2] = (m01 * m12 - m11 * m02);
38 20 out[1][0] = -(m10 * m22 - m20 * m12);
39 20 out[1][1] = (m00 * m22 - m20 * m02);
40 20 out[1][2] = -(m00 * m12 - m10 * m02);
41 20 out[2][0] = (m10 * m21 - m20 * m11);
42 20 out[2][1] = -(m00 * m21 - m20 * m01);
43 20 out[2][2] = (m00 * m11 - m10 * m01);
44
45 20 det = m00 * out[0][0] + m10 * out[0][1] + m20 * out[0][2];
46 20 det = 1.0 / det;
47
48
2/2
✓ Branch 0 taken 60 times.
✓ Branch 1 taken 20 times.
 80 for (i = 0; i < 3; i++) {
49
2/2
✓ Branch 0 taken 180 times.
✓ Branch 1 taken 60 times.
 240 for (j = 0; j < 3; j++)
50 180 out[i][j] *= det;
51 }
52 20 }
53
54 5 void ff_matrix_mul_3x3(double dst[3][3],
55 const double src1[3][3], const double src2[3][3])
56 {
57 int m, n;
58
59
2/2
✓ Branch 0 taken 15 times.
✓ Branch 1 taken 5 times.
 20 for (m = 0; m < 3; m++)
60
2/2
✓ Branch 0 taken 45 times.
✓ Branch 1 taken 15 times.
 60 for (n = 0; n < 3; n++)
61 45 dst[m][n] = src2[m][0] * src1[0][n] +
62 45 src2[m][1] * src1[1][n] +
63 45 src2[m][2] * src1[2][n];
64 5 }
65
66 180471 void ff_matrix_mul_3x3_vec(double dst[3], const double vec[3], const double mat[3][3])
67 {
68 int m;
69
70
2/2
✓ Branch 0 taken 541413 times.
✓ Branch 1 taken 180471 times.
 721884 for (m = 0; m < 3; m++)
71 541413 dst[m] = vec[0] * mat[m][0] +
72 541413 vec[1] * mat[m][1] +
73 541413 vec[2] * mat[m][2];
74 180471 }
75
76 /*
77 * see e.g. http://www.brucelindbloom.com/index.html?Eqn_RGB_XYZ_Matrix.html
78 */
79 15 void ff_fill_rgb2xyz_table(const AVPrimaryCoefficients *coeffs,
80 const AVWhitepointCoefficients *wp,
81 double rgb2xyz[3][3])
82 {
83 double i[3][3], sr, sg, sb, zw;
84 15 double xr = av_q2d(coeffs->r.x), yr = av_q2d(coeffs->r.y);
85 15 double xg = av_q2d(coeffs->g.x), yg = av_q2d(coeffs->g.y);
86 15 double xb = av_q2d(coeffs->b.x), yb = av_q2d(coeffs->b.y);
87 15 double xw = av_q2d(wp->x), yw = av_q2d(wp->y);
88
89 15 rgb2xyz[0][0] = xr;
90 15 rgb2xyz[0][1] = xg;
91 15 rgb2xyz[0][2] = xb;
92 15 rgb2xyz[1][0] = yr;
93 15 rgb2xyz[1][1] = yg;
94 15 rgb2xyz[1][2] = yb;
95 15 rgb2xyz[2][0] = 1.0 - xr - yr;
96 15 rgb2xyz[2][1] = 1.0 - xg - yg;
97 15 rgb2xyz[2][2] = 1.0 - xb - yb;
98 15 ff_matrix_invert_3x3(rgb2xyz, i);
99 15 zw = 1.0 - xw - yw;
100 15 sr = i[0][0] * xw + i[0][1] * yw + i[0][2] * zw;
101 15 sg = i[1][0] * xw + i[1][1] * yw + i[1][2] * zw;
102 15 sb = i[2][0] * xw + i[2][1] * yw + i[2][2] * zw;
103 15 rgb2xyz[0][0] *= sr;
104 15 rgb2xyz[0][1] *= sg;
105 15 rgb2xyz[0][2] *= sb;
106 15 rgb2xyz[1][0] *= sr;
107 15 rgb2xyz[1][1] *= sg;
108 15 rgb2xyz[1][2] *= sb;
109 15 rgb2xyz[2][0] *= sr;
110 15 rgb2xyz[2][1] *= sg;
111 15 rgb2xyz[2][2] *= sb;
112 15 }
113 static const double ycgco_matrix[3][3] =
114 {
115 { 0.25, 0.5, 0.25 },
116 { -0.25, 0.5, -0.25 },
117 { 0.5, 0, -0.5 },
118 };
119
120 static const double gbr_matrix[3][3] =
121 {
122 { 0, 1, 0 },
123 { 0, -0.5, 0.5 },
124 { 0.5, -0.5, 0 },
125 };
126
127 18690 void ff_fill_rgb2yuv_table(const AVLumaCoefficients *coeffs,
128 double rgb2yuv[3][3])
129 {
130 double bscale, rscale;
131 18690 double cr = av_q2d(coeffs->cr), cg = av_q2d(coeffs->cg), cb = av_q2d(coeffs->cb);
132
133 // special ycgco matrix
134
1/6
✗ Branch 0 not taken.
✓ Branch 1 taken 18690 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
 18690 if (cr == 0.25 && cg == 0.5 && cb == 0.25) {
135 ✗ memcpy(rgb2yuv, ycgco_matrix, sizeof(double) * 9);
136 ✗ return;
137
1/6
✗ Branch 0 not taken.
✓ Branch 1 taken 18690 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
 18690 } else if (cr == 1 && cg == 1 && cb == 1) {
138 ✗ memcpy(rgb2yuv, gbr_matrix, sizeof(double) * 9);
139 ✗ return;
140 }
141
142 18690 rgb2yuv[0][0] = cr;
143 18690 rgb2yuv[0][1] = cg;
144 18690 rgb2yuv[0][2] = cb;
145 18690 bscale = 0.5 / (cb - 1.0);
146 18690 rscale = 0.5 / (cr - 1.0);
147 18690 rgb2yuv[1][0] = bscale * cr;
148 18690 rgb2yuv[1][1] = bscale * cg;
149 18690 rgb2yuv[1][2] = 0.5;
150 18690 rgb2yuv[2][0] = 0.5;
151 18690 rgb2yuv[2][1] = rscale * cg;
152 18690 rgb2yuv[2][2] = rscale * cb;
153 }
154
155 ✗ double ff_determine_signal_peak(AVFrame *in)
156 {
157 ✗ AVFrameSideData *sd = av_frame_get_side_data(in, AV_FRAME_DATA_CONTENT_LIGHT_LEVEL);
158 ✗ double peak = 0;
159
160 ✗ if (sd) {
161 ✗ AVContentLightMetadata *clm = (AVContentLightMetadata *)sd->data;
162 ✗ peak = clm->MaxCLL / REFERENCE_WHITE;
163 }
164
165 ✗ sd = av_frame_get_side_data(in, AV_FRAME_DATA_MASTERING_DISPLAY_METADATA);
166 ✗ if (!peak && sd) {
167 ✗ AVMasteringDisplayMetadata *metadata = (AVMasteringDisplayMetadata *)sd->data;
168 ✗ if (metadata->has_luminance)
169 ✗ peak = av_q2d(metadata->max_luminance) / REFERENCE_WHITE;
170 }
171
172 // For untagged source, use peak of 10000 if SMPTE ST.2084
173 // otherwise assume HLG with reference display peak 1000.
174 ✗ if (!peak)
175 ✗ peak = in->color_trc == AVCOL_TRC_SMPTE2084 ? 100.0f : 10.0f;
176
177 ✗ return peak;
178 }
179
180 ✗ void ff_update_hdr_metadata(AVFrame *in, double peak)
181 {
182 ✗ AVFrameSideData *sd = av_frame_get_side_data(in, AV_FRAME_DATA_CONTENT_LIGHT_LEVEL);
183
184 ✗ if (sd) {
185 ✗ AVContentLightMetadata *clm = (AVContentLightMetadata *)sd->data;
186 ✗ clm->MaxCLL = (unsigned)(peak * REFERENCE_WHITE);
187 }
188
189 ✗ sd = av_frame_get_side_data(in, AV_FRAME_DATA_MASTERING_DISPLAY_METADATA);
190 ✗ if (sd) {
191 ✗ AVMasteringDisplayMetadata *metadata = (AVMasteringDisplayMetadata *)sd->data;
192 ✗ if (metadata->has_luminance)
193 ✗ metadata->max_luminance = av_d2q(peak * REFERENCE_WHITE, 10000);
194 }
195 ✗ }
196