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
114
115
116
117
118
119
120
121
122
123
|
/*
* 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_UTILS_H
#define SWSCALE_UTILS_H
#include "libavutil/csp.h"
#include "libavutil/pixdesc.h"
#include "swscale.h"
static inline int ff_q_isnan(const AVRational a)
{
return !a.num && !a.den;
}
/* Like av_cmp_q but considers NaN == NaN */
static inline int ff_q_equal(const AVRational a, const AVRational b)
{
return (ff_q_isnan(a) && ff_q_isnan(b)) || !av_cmp_q(a, b);
}
static inline int ff_cie_xy_equal(const AVCIExy a, const AVCIExy b)
{
return ff_q_equal(a.x, b.x) && ff_q_equal(a.y, b.y);
}
static inline int ff_prim_equal(const AVPrimaryCoefficients *a,
const AVPrimaryCoefficients *b)
{
return ff_cie_xy_equal(a->r, b->r) &&
ff_cie_xy_equal(a->g, b->g) &&
ff_cie_xy_equal(a->b, b->b);
}
enum {
FIELD_TOP, /* top/even rows, or progressive */
FIELD_BOTTOM, /* bottom/odd rows */
};
typedef struct SwsColor {
enum AVColorPrimaries prim;
enum AVColorTransferCharacteristic trc;
AVPrimaryCoefficients gamut; /* mastering display gamut */
AVRational min_luma; /* minimum luminance in nits */
AVRational max_luma; /* maximum luminance in nits */
AVRational frame_peak; /* per-frame/scene peak luminance, or 0 */
AVRational frame_avg; /* per-frame/scene average luminance, or 0 */
} SwsColor;
/* Subset of AVFrame parameters that uniquely determine pixel representation */
typedef struct SwsFormat {
int width, height;
int interlaced;
enum AVPixelFormat format;
enum AVColorRange range;
enum AVColorSpace csp;
enum AVChromaLocation loc;
const AVPixFmtDescriptor *desc; /* convenience */
SwsColor color;
} SwsFormat;
/**
* This function also sanitizes and strips the input data, removing irrelevant
* fields for certain formats.
*/
SwsFormat ff_fmt_from_frame(const AVFrame *frame, int field);
static inline int ff_color_equal(const SwsColor *c1, const SwsColor *c2)
{
return c1->prim == c2->prim &&
c1->trc == c2->trc &&
ff_q_equal(c1->min_luma, c2->min_luma) &&
ff_q_equal(c1->max_luma, c2->max_luma) &&
ff_prim_equal(&c1->gamut, &c2->gamut);
}
/* Tests only the static components of a colorspace, ignoring per-frame data */
static inline int ff_fmt_equal(const SwsFormat *fmt1, const SwsFormat *fmt2)
{
return fmt1->width == fmt2->width &&
fmt1->height == fmt2->height &&
fmt1->interlaced == fmt2->interlaced &&
fmt1->format == fmt2->format &&
fmt1->range == fmt2->range &&
fmt1->csp == fmt2->csp &&
fmt1->loc == fmt2->loc &&
ff_color_equal(&fmt1->color, &fmt2->color);
}
static inline int ff_fmt_align(enum AVPixelFormat fmt)
{
const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(fmt);
if (desc->flags & AV_PIX_FMT_FLAG_BAYER) {
return 2;
} else {
return 1 << desc->log2_chroma_h;
}
}
int ff_test_fmt(const SwsFormat *fmt, int output);
/* Returns 1 if the formats are incomplete, 0 otherwise */
int ff_infer_colors(SwsColor *src, SwsColor *dst);
#endif /* SWSCALE_UTILS_H */
|