mirror of
https://github.com/FFmpeg/FFmpeg.git
synced 2025-02-14 22:22:59 +02:00
1f0c500784
Logic is loosely on equivalent decisions in libplacebo. The basic idea is to try and be a bit conservative by treating AVCOL_*_UNSPECIFIED as a no-op, unless the other primaries set are non-standard / wide-gamut or HDR. This helps avoid unintended or unexpected colorspace conversions, while forcing it in cases where we are almost certain it is needed. The major departure from libplacebo semantics is that we no default to a 1000:1 contrast ration for SDR displays, instead modelling them as idealized devices with an infinite contrast ratio. In either case, setting SWS_STRICT overrides this behavior in favor of always requiring explicit colorspace metadata.
124 lines
3.9 KiB
C
124 lines
3.9 KiB
C
/*
|
|
* 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 */
|