blob: 4250afab21a5510d5350ed3a61919b1b9331251e [file] [log] [blame]
// Copyright 2019 Joe Drago. All rights reserved.
// SPDX-License-Identifier: BSD-2-Clause
#include "avif/internal.h"
#include <math.h>
#include <string.h>
struct avifColorPrimariesTable
{
avifColorPrimaries colorPrimariesEnum;
const char * name;
float primaries[8]; // rX, rY, gX, gY, bX, bY, wX, wY
};
static const struct avifColorPrimariesTable avifColorPrimariesTables[] = {
{ AVIF_COLOR_PRIMARIES_BT709, "BT.709", { 0.64f, 0.33f, 0.3f, 0.6f, 0.15f, 0.06f, 0.3127f, 0.329f } },
{ AVIF_COLOR_PRIMARIES_BT470M, "BT.470-6 System M", { 0.67f, 0.33f, 0.21f, 0.71f, 0.14f, 0.08f, 0.310f, 0.316f } },
{ AVIF_COLOR_PRIMARIES_BT470BG, "BT.470-6 System BG", { 0.64f, 0.33f, 0.29f, 0.60f, 0.15f, 0.06f, 0.3127f, 0.3290f } },
{ AVIF_COLOR_PRIMARIES_BT601, "BT.601", { 0.630f, 0.340f, 0.310f, 0.595f, 0.155f, 0.070f, 0.3127f, 0.3290f } },
{ AVIF_COLOR_PRIMARIES_SMPTE240, "SMPTE 240M", { 0.630f, 0.340f, 0.310f, 0.595f, 0.155f, 0.070f, 0.3127f, 0.3290f } },
{ AVIF_COLOR_PRIMARIES_GENERIC_FILM, "Generic film", { 0.681f, 0.319f, 0.243f, 0.692f, 0.145f, 0.049f, 0.310f, 0.316f } },
{ AVIF_COLOR_PRIMARIES_BT2020, "BT.2020", { 0.708f, 0.292f, 0.170f, 0.797f, 0.131f, 0.046f, 0.3127f, 0.3290f } },
{ AVIF_COLOR_PRIMARIES_XYZ, "XYZ", { 1.0f, 0.0f, 0.0f, 1.0f, 0.0f, 0.0f, 0.3333f, 0.3333f } },
{ AVIF_COLOR_PRIMARIES_SMPTE431, "SMPTE RP 431-2", { 0.680f, 0.320f, 0.265f, 0.690f, 0.150f, 0.060f, 0.314f, 0.351f } },
{ AVIF_COLOR_PRIMARIES_SMPTE432, "SMPTE EG 432-1 (DCI P3)", { 0.680f, 0.320f, 0.265f, 0.690f, 0.150f, 0.060f, 0.3127f, 0.3290f } },
{ AVIF_COLOR_PRIMARIES_EBU3213, "EBU Tech. 3213-E", { 0.630f, 0.340f, 0.295f, 0.605f, 0.155f, 0.077f, 0.3127f, 0.3290f } }
};
static const int avifColorPrimariesTableSize = sizeof(avifColorPrimariesTables) / sizeof(avifColorPrimariesTables[0]);
void avifColorPrimariesGetValues(avifColorPrimaries acp, float outPrimaries[8])
{
for (int i = 0; i < avifColorPrimariesTableSize; ++i) {
if (avifColorPrimariesTables[i].colorPrimariesEnum == acp) {
memcpy(outPrimaries, avifColorPrimariesTables[i].primaries, sizeof(avifColorPrimariesTables[i].primaries));
return;
}
}
// if we get here, the color primaries are unknown. Just return a reasonable default.
memcpy(outPrimaries, avifColorPrimariesTables[0].primaries, sizeof(avifColorPrimariesTables[0].primaries));
}
static avifBool matchesTo3RoundedPlaces(float a, float b)
{
return (fabsf(a - b) < 0.001f);
}
static avifBool primariesMatch(const float p1[8], const float p2[8])
{
return matchesTo3RoundedPlaces(p1[0], p2[0]) && matchesTo3RoundedPlaces(p1[1], p2[1]) &&
matchesTo3RoundedPlaces(p1[2], p2[2]) && matchesTo3RoundedPlaces(p1[3], p2[3]) && matchesTo3RoundedPlaces(p1[4], p2[4]) &&
matchesTo3RoundedPlaces(p1[5], p2[5]) && matchesTo3RoundedPlaces(p1[6], p2[6]) && matchesTo3RoundedPlaces(p1[7], p2[7]);
}
avifColorPrimaries avifColorPrimariesFind(const float inPrimaries[8], const char ** outName)
{
if (outName) {
*outName = NULL;
}
for (int i = 0; i < avifColorPrimariesTableSize; ++i) {
if (primariesMatch(inPrimaries, avifColorPrimariesTables[i].primaries)) {
if (outName) {
*outName = avifColorPrimariesTables[i].name;
}
return avifColorPrimariesTables[i].colorPrimariesEnum;
}
}
return AVIF_COLOR_PRIMARIES_UNKNOWN;
}
avifResult avifTransferCharacteristicsGetGamma(avifTransferCharacteristics atc, float * gamma)
{
switch (atc) {
case AVIF_TRANSFER_CHARACTERISTICS_BT470M:
*gamma = 2.2f;
return AVIF_RESULT_OK;
case AVIF_TRANSFER_CHARACTERISTICS_BT470BG:
*gamma = 2.8f;
return AVIF_RESULT_OK;
case AVIF_TRANSFER_CHARACTERISTICS_LINEAR:
*gamma = 1.0f;
return AVIF_RESULT_OK;
default:
return AVIF_RESULT_INVALID_ARGUMENT;
}
}
avifTransferCharacteristics avifTransferCharacteristicsFindByGamma(float gamma)
{
if (matchesTo3RoundedPlaces(gamma, 2.2f)) {
return AVIF_TRANSFER_CHARACTERISTICS_BT470M;
} else if (matchesTo3RoundedPlaces(gamma, 1.0f)) {
return AVIF_TRANSFER_CHARACTERISTICS_LINEAR;
} else if (matchesTo3RoundedPlaces(gamma, 2.8f)) {
return AVIF_TRANSFER_CHARACTERISTICS_BT470BG;
}
return AVIF_TRANSFER_CHARACTERISTICS_UNKNOWN;
}
struct avifMatrixCoefficientsTable
{
avifMatrixCoefficients matrixCoefficientsEnum;
const char * name;
const float kr;
const float kb;
};
// https://www.itu.int/rec/T-REC-H.273-201612-I/en
static const struct avifMatrixCoefficientsTable matrixCoefficientsTables[] = {
//{ AVIF_MATRIX_COEFFICIENTS_IDENTITY, "Identity", 0.0f, 0.0f, }, // Handled elsewhere
{ AVIF_MATRIX_COEFFICIENTS_BT709, "BT.709", 0.2126f, 0.0722f },
{ AVIF_MATRIX_COEFFICIENTS_FCC, "FCC USFC 73.682", 0.30f, 0.11f },
{ AVIF_MATRIX_COEFFICIENTS_BT470BG, "BT.470-6 System BG", 0.299f, 0.114f },
{ AVIF_MATRIX_COEFFICIENTS_BT601, "BT.601", 0.299f, 0.114f },
{ AVIF_MATRIX_COEFFICIENTS_SMPTE240, "SMPTE ST 240", 0.212f, 0.087f },
//{ AVIF_MATRIX_COEFFICIENTS_YCGCO, "YCgCo", 0.0f, 0.0f, }, // Handled elsewhere
{ AVIF_MATRIX_COEFFICIENTS_BT2020_NCL, "BT.2020 (non-constant luminance)", 0.2627f, 0.0593f },
//{ AVIF_MATRIX_COEFFICIENTS_BT2020_CL, "BT.2020 (constant luminance)", 0.2627f, 0.0593f }, // FIXME: It is not an linear transformation.
//{ AVIF_MATRIX_COEFFICIENTS_SMPTE2085, "ST 2085", 0.0f, 0.0f }, // FIXME: ST2085 can't represent using Kr and Kb.
//{ AVIF_MATRIX_COEFFICIENTS_CHROMA_DERIVED_CL, "Chromaticity-derived constant luminance system", 0.0f, 0.0f } // FIXME: It is not an linear transformation.
//{ AVIF_MATRIX_COEFFICIENTS_ICTCP, "BT.2100-0 ICtCp", 0.0f, 0.0f }, // FIXME: This can't represent using Kr and Kb.
};
static const int avifMatrixCoefficientsTableSize = sizeof(matrixCoefficientsTables) / sizeof(matrixCoefficientsTables[0]);
static avifBool calcYUVInfoFromCICP(const avifImage * image, float coeffs[3])
{
if (image->matrixCoefficients == AVIF_MATRIX_COEFFICIENTS_CHROMA_DERIVED_NCL) {
float primaries[8];
avifColorPrimariesGetValues(image->colorPrimaries, primaries);
float const rX = primaries[0];
float const rY = primaries[1];
float const gX = primaries[2];
float const gY = primaries[3];
float const bX = primaries[4];
float const bY = primaries[5];
float const wX = primaries[6];
float const wY = primaries[7];
float const rZ = 1.0f - (rX + rY); // (Eq. 34)
float const gZ = 1.0f - (gX + gY); // (Eq. 35)
float const bZ = 1.0f - (bX + bY); // (Eq. 36)
float const wZ = 1.0f - (wX + wY); // (Eq. 37)
float const kr = (rY * (wX * (gY * bZ - bY * gZ) + wY * (bX * gZ - gX * bZ) + wZ * (gX * bY - bX * gY))) /
(wY * (rX * (gY * bZ - bY * gZ) + gX * (bY * rZ - rY * bZ) + bX * (rY * gZ - gY * rZ)));
// (Eq. 32)
float const kb = (bY * (wX * (rY * gZ - gY * rZ) + wY * (gX * rZ - rX * gZ) + wZ * (rX * gY - gX * rY))) /
(wY * (rX * (gY * bZ - bY * gZ) + gX * (bY * rZ - rY * bZ) + bX * (rY * gZ - gY * rZ)));
// (Eq. 33)
coeffs[0] = kr;
coeffs[2] = kb;
coeffs[1] = 1.0f - coeffs[0] - coeffs[2];
return AVIF_TRUE;
} else {
for (int i = 0; i < avifMatrixCoefficientsTableSize; ++i) {
const struct avifMatrixCoefficientsTable * const table = &matrixCoefficientsTables[i];
if (table->matrixCoefficientsEnum == image->matrixCoefficients) {
coeffs[0] = table->kr;
coeffs[2] = table->kb;
coeffs[1] = 1.0f - coeffs[0] - coeffs[2];
return AVIF_TRUE;
}
}
}
return AVIF_FALSE;
}
void avifCalcYUVCoefficients(const avifImage * image, float * outR, float * outG, float * outB)
{
// (As of ISO/IEC 23000-22:2019 Amendment 2)
// MIAF Section 7.3.6.4 "Colour information property":
//
// If a coded image has no associated colour property, the default property is defined as having
// colour_type equal to 'nclx' with properties as follows:
// - colour_primaries equal to 1,
// - transfer_characteristics equal to 13,
// - matrix_coefficients equal to 5 or 6 (which are functionally identical), and
// - full_range_flag equal to 1.
// Only if the colour information property of the image matches these default values, the colour
// property may be omitted; all other images shall have an explicitly declared colour space via
// association with a property of this type.
//
// See here for the discussion: https://github.com/AOMediaCodec/av1-avif/issues/77#issuecomment-676526097
// matrix_coefficients of [5,6] == BT.601:
float kr = 0.299f;
float kb = 0.114f;
float kg = 1.0f - kr - kb;
float coeffs[3];
if (calcYUVInfoFromCICP(image, coeffs)) {
kr = coeffs[0];
kg = coeffs[1];
kb = coeffs[2];
}
*outR = kr;
*outG = kg;
*outB = kb;
}