1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
/*
* Copyright (C) 2024 Niklas Haas
*
* This file is part of FFmpeg.
*
* FFmpeg is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* FFmpeg is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with FFmpeg; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
*/
#ifndef SWSCALE_CSPUTILS_H
#define SWSCALE_CSPUTILS_H
#include <stdint.h>
#include <stdbool.h>
#include <math.h>
#include "libavutil/attributes.h"
#include "libavutil/common.h"
#include "libavutil/csp.h"
#include "libavutil/pixfmt.h"
/* Shared constants and helpers for colorspace mapping */
#define fmixf(a, b, x) ((b) * (x) + (a) * (1 - (x)))
static inline float smoothstepf(float edge0, float edge1, float x)
{
if (edge0 == edge1)
return x >= edge0;
x = (x - edge0) / (edge1 - edge0);
x = av_clipf(x, 0.0f, 1.0f);
return x * x * (3.0f - 2.0f * x);
}
/* 3x3 matrix math */
typedef struct SwsMatrix3x3 {
float m[3][3];
} SwsMatrix3x3;
void ff_sws_matrix3x3_mul(SwsMatrix3x3 *a, const SwsMatrix3x3 *b);
void ff_sws_matrix3x3_invert(SwsMatrix3x3 *mat);
void ff_sws_matrix3x3_apply(const SwsMatrix3x3 *mat, float vec[3]);
SwsMatrix3x3 ff_sws_ipt_rgb2lms(const AVColorPrimariesDesc *prim);
SwsMatrix3x3 ff_sws_ipt_lms2rgb(const AVColorPrimariesDesc *prim);
/* Converts to/from XYZ (relative to the given white point, no adaptation) */
SwsMatrix3x3 ff_sws_rgb2xyz(const AVColorPrimariesDesc *prim);
SwsMatrix3x3 ff_sws_xyz2rgb(const AVColorPrimariesDesc *prim);
/* Returns an RGB -> RGB adaptation matrix */
SwsMatrix3x3 ff_sws_get_adaptation(const AVPrimaryCoefficients *prim,
AVWhitepointCoefficients from,
AVWhitepointCoefficients to);
/* Integer math definitions / helpers */
typedef struct v3u8_t {
uint8_t x, y, z;
} v3u8_t;
typedef struct v2u16_t {
uint16_t x, y;
} v2u16_t;
typedef struct v3u16_t {
uint16_t x, y, z;
} v3u16_t;
/* Fast perceptual quantizer */
static const float PQ_M1 = 2610./4096 * 1./4,
PQ_M2 = 2523./4096 * 128,
PQ_C1 = 3424./4096,
PQ_C2 = 2413./4096 * 32,
PQ_C3 = 2392./4096 * 32;
enum { PQ_LUT_SIZE = 1024 };
extern const float ff_pq_eotf_lut[PQ_LUT_SIZE+1];
static inline float pq_eotf(float x)
{
float idxf = av_clipf(x, 0.0f, 1.0f) * (PQ_LUT_SIZE - 1);
int ipart = floorf(idxf);
float fpart = idxf - ipart;
return fmixf(ff_pq_eotf_lut[ipart], ff_pq_eotf_lut[ipart + 1], fpart);
}
static inline float pq_oetf(float x)
{
x = powf(fmaxf(x * 1e-4f, 0.0f), PQ_M1);
x = (PQ_C1 + PQ_C2 * x) / (1.0f + PQ_C3 * x);
return powf(x, PQ_M2);
}
/* Misc colorspace math / helpers */
/**
* Returns true if 'b' is entirely contained in 'a'. Useful for figuring out if
* colorimetric clipping will occur or not.
*/
bool ff_prim_superset(const AVPrimaryCoefficients *a, const AVPrimaryCoefficients *b);
#endif /* SWSCALE_CSPUTILS_H */
|
