FFmpeg coverage

Directory: ../../../ffmpeg/
File: src/libavutil/tests/color_utils.c
Date: 2025-01-20 09:27:23
Exec Total Coverage
Lines: 57 63 90.5%
Functions: 2 2 100.0%
Branches: 44 52 84.6%
Line Branch Exec Source
1 /*
2 * Copyright (c) 2015 Kevin Wheatley <kevin.j.wheatley@gmail.com>
3 *
4 * This file is part of FFmpeg.
5 *
6 * FFmpeg is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
10 *
11 * FFmpeg is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with FFmpeg; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19 */
20
21 #include <math.h>
22 #include <stdio.h>
23 #include <string.h>
24
25 #include "libavutil/csp.h"
26 #include "libavutil/macros.h"
27 #include "libavutil/pixdesc.h"
28 #include "libavutil/pixfmt.h"
29
30 10800 static inline int fuzzy_equal(double a, double b)
31 {
32 10800 const double epsilon = fmax(fmax(fabs(a), fabs(b)), 1.0) * 1e-7;
33 10800 return fabs(a - b) <= epsilon;
34 }
35
36 #define TEST_EOTF(func, input, ref) do \
37 { \
38 const double _b[3] = { (ref)[0], (ref)[1], (ref)[2] }; \
39 double _a[3] = { (input)[0], (input)[1], (input)[2] }; \
40 func(Lw, Lb, _a); \
41 for (int _i = 0; _i < 3; _i++) { \
42 if (!fuzzy_equal(_a[_i], _b[_i])) { \
43 printf("FAIL: trc=%s %s(%g, %g, %s) != %s\n" \
44 " expected {%g, %g, %g}, got {%g, %g, %g}\n", \
45 trc_name, #func, Lw, Lb, #input, #ref, \
46 _b[0], _b[1], _b[2], _a[0], _a[1], _a[2]); \
47 return 1; \
48 } \
49 } \
50 } while (0)
51
52 1 int main(int argc, char *argv[])
53 {
54 static const double test_data[] = {
55 -0.1, -0.018053968510807, -0.01, -0.00449, 0.0, 0.00316227760, 0.005,
56 0.009, 0.015, 0.1, 1.0, 52.37, 125.098765, 1999.11123, 6945.443,
57 15123.4567, 19845.88923, 98678.4231, 99999.899998
58 };
59
60
2/2
✓ Branch 0 taken 19 times.
✓ Branch 1 taken 1 times.
 20 for (enum AVColorTransferCharacteristic trc = 0; trc < AVCOL_TRC_NB; trc++) {
61 19 av_csp_trc_function func = av_csp_trc_func_from_id(trc);
62 19 av_csp_trc_function func_inv = av_csp_trc_func_inv_from_id(trc);
63 19 const char *name = av_color_transfer_name(trc);
64
2/2
✓ Branch 0 taken 3 times.
✓ Branch 1 taken 16 times.
 19 if (!func)
65 3 continue;
66
67
2/2
✓ Branch 0 taken 304 times.
✓ Branch 1 taken 16 times.
 320 for (int i = 0; i < FF_ARRAY_ELEMS(test_data); i++) {
68 304 double result = func(test_data[i]);
69 304 double roundtrip = func_inv(result);
70 304 printf("trc=%s calling func(%f) expected=%f roundtrip=%f\n",
71 304 name, test_data[i], result, roundtrip);
72
73
3/4
✓ Branch 0 taken 221 times.
✓ Branch 1 taken 83 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 221 times.
 304 if (result > 0.0 && fabs(roundtrip - test_data[i]) > 1e-7) {
74 ✗ printf(" FAIL\n");
75 ✗ return 1;
76 }
77 }
78 }
79
80
2/2
✓ Branch 0 taken 19 times.
✓ Branch 1 taken 1 times.
 20 for (enum AVColorTransferCharacteristic trc = 0; trc < AVCOL_TRC_NB; trc++) {
81 19 av_csp_eotf_function eotf = av_csp_itu_eotf(trc);
82 19 av_csp_eotf_function eotf_inv = av_csp_itu_eotf_inv(trc);
83 19 const char *trc_name = av_color_transfer_name(trc);
84
2/2
✓ Branch 0 taken 5 times.
✓ Branch 1 taken 14 times.
 19 if (!eotf)
85 5 continue;
86
87
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 13 times.
 14 if (trc == AVCOL_TRC_SMPTE2084) {
88 /* This one is equivalent to the TRC already tested above */
89 1 continue;
90
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 12 times.
 13 } else if (trc == AVCOL_TRC_SMPTE428) {
91 /* Test vectors from SMPTE RP-431-2 */
92 1 const struct { double E_xyz[3]; double luma; } tests[] = {
93 #define XYZ(X, Y, Z) { X / 4095.0, Y / 4095.0, Z / 4095.0 }
94 { XYZ( 379, 396, 389), 0.14 },
95 { XYZ( 759, 792, 778), 0.75 },
96 { XYZ(1138, 1188, 1167), 2.12 },
97 { XYZ(1518, 1584, 1556), 4.45 },
98 { XYZ(1897, 1980, 1945), 7.94 },
99 { XYZ(2276, 2376, 2334), 12.74 },
100 { XYZ(2656, 2772, 2723), 19.01 },
101 { XYZ(3035, 3168, 3112), 26.89 },
102 { XYZ(3415, 3564, 3501), 36.52 },
103 { XYZ(3794, 3960, 3890), 48.02 },
104 };
105 /* DCI reference display */
106 1 const double luminance = 48.00;
107 1 const double contrast = 2000;
108 /* Solve for Lw - Lb = luminance, Lw / Lb = contrast */
109 1 const double Lb = luminance / (contrast - 1);
110 1 const double Lw = Lb + luminance;
111
112
2/2
✓ Branch 0 taken 10 times.
✓ Branch 1 taken 1 times.
 11 for (int i = 0; i < FF_ARRAY_ELEMS(tests); i++) {
113 double L_xyz[3];
114 10 memcpy(L_xyz, tests[i].E_xyz, sizeof(L_xyz));
115 10 eotf(Lw, Lb, L_xyz);
116 10 printf("trc=%s EOTF(%g, %g, {%g, %g, %g}) = {%g, %g %g}, expected Y=%f\n",
117 trc_name, Lw, Lb,
118 10 tests[i].E_xyz[0], tests[i].E_xyz[1], tests[i].E_xyz[2],
119 10 L_xyz[0], L_xyz[1], L_xyz[2], tests[i].luma);
120
121
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 10 times.
 10 if (fabs(L_xyz[1] - tests[i].luma) > 0.01) {
122 ✗ printf(" FAIL\n");
123 ✗ return 1;
124 }
125 }
126 } else {
127 /* Normal, display-relative RGB curve */
128 static const double black_points[] = { 0.0, 1e-6, 0.1, 1.5 };
129 static const double white_points[] = { 50.0, 100.0, 203.0, 1000.0, 10000.0 };
130
131
2/2
✓ Branch 0 taken 48 times.
✓ Branch 1 taken 12 times.
 60 for (int i = 0; i < FF_ARRAY_ELEMS(black_points); i++) {
132
2/2
✓ Branch 0 taken 240 times.
✓ Branch 1 taken 48 times.
 288 for (int j = 0; j < FF_ARRAY_ELEMS(white_points); j++) {
133 240 const double Lb = black_points[i];
134 240 const double Lw = white_points[j];
135 240 const double all0[3] = { 0.0, 0.0, 0.0 };
136 const double all1[3] = { 1.0, 1.0, 1.0 };
137 240 const double black[3] = { Lb, Lb, Lb };
138 240 const double white[3] = { Lw, Lw, Lw };
139 double L_prev;
140
141
3/4
✗ Branch 2 not taken.
✓ Branch 3 taken 720 times.
✓ Branch 5 taken 720 times.
✓ Branch 6 taken 240 times.
 960 TEST_EOTF(eotf, all0, black);
142
3/4
✗ Branch 2 not taken.
✓ Branch 3 taken 720 times.
✓ Branch 5 taken 720 times.
✓ Branch 6 taken 240 times.
 960 TEST_EOTF(eotf, all1, white);
143
3/4
✗ Branch 2 not taken.
✓ Branch 3 taken 720 times.
✓ Branch 5 taken 720 times.
✓ Branch 6 taken 240 times.
 960 TEST_EOTF(eotf_inv, black, all0);
144
3/4
✗ Branch 2 not taken.
✓ Branch 3 taken 720 times.
✓ Branch 5 taken 720 times.
✓ Branch 6 taken 240 times.
 960 TEST_EOTF(eotf_inv, white, all1);
145
146 /* Test round-trip on grayscale ramp */
147
2/2
✓ Branch 0 taken 2640 times.
✓ Branch 1 taken 240 times.
 2880 for (double x = 0.0; x < 1.0; x += 0.1) {
148 2640 const double E[3] = { x, x, x };
149 2640 double L[3] = { x, x, x };
150 2640 eotf(Lw, Lb, L);
151
152 2640 printf("trc=%s EOTF(%g, %g, {%g}) = {%g}\n",
153 2640 trc_name, Lw, Lb, E[1], L[1]);
154
3/4
✗ Branch 2 not taken.
✓ Branch 3 taken 7920 times.
✓ Branch 5 taken 7920 times.
✓ Branch 6 taken 2640 times.
 10560 TEST_EOTF(eotf_inv, L, E);
155
156
3/4
✓ Branch 0 taken 2400 times.
✓ Branch 1 taken 240 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 2400 times.
 2640 if (x > 0.0 && L[1] <= L_prev) {
157 ✗ printf(" FAIL: non-monotonic!\n");
158 ✗ return 1;
159 }
160 2640 L_prev = L[1];
161 }
162 }
163 }
164 }
165 }
166 }
167