drmhdr.h

Go to the documentation of this file.

// SPDX-License-Identifier: AGPL-3.0-or-later
 
#ifndef __DRMHDR_H
#define __DRMHDR_H
 
extern "C" {
#include <libavutil/frame.h>
#include <libavutil/mastering_display_metadata.h>
}
 
#include <drm_mode.h>
 
class cVideoRender;
 
struct vector {
    float f[4];
};
 
struct colorspace {
    struct vector r, g, b;
    struct vector whitepoint;
    const char *name;
    const char *whitepoint_name;
};
 
static struct colorspace bt709 = {
    .r =          {{ 0.640f,  0.330f,  }},
    .g =          {{ 0.300f,  0.600f,  }},
    .b =          {{ 0.150f,  0.060f,  }},
    .whitepoint = {{ 0.3127f, 0.3290f, }},
    .name = "BT.709",
    .whitepoint_name = "D65",
};
 
static struct colorspace bt2020 = {
    .r =          {{ 0.708f,  0.292f,  }},
    .g =          {{ 0.170f,  0.797f,  }},
    .b =          {{ 0.131f,  0.046f,  }},
    .whitepoint = {{ 0.3127f, 0.3290f, }},
    .name = "BT.2020",
    .whitepoint_name = "D65",
};
 
 
static struct colorspace bt470bg = {
    .r =          {{ 0.640f,  0.330f,  }},
    .g =          {{ 0.290f,  0.600f,  }},
    .b =          {{ 0.150f,  0.060f,  }},
    .whitepoint = {{ 0.3127f, 0.3290f, }},
    .name = "BT.470 B/G",
    .whitepoint_name = "D65",
};
 
static struct colorspace *const colorspaces[] = {
    &bt709, &bt2020, &bt470bg,
};
 
#define ARRAY_LENGTH(a) (sizeof(a) / sizeof(a)[0])
static inline struct colorspace *colorspace_lookup(const char *name)
{
    unsigned i;
 
    if (!name)
        return NULL;
 
    for (i = 0; i < ARRAY_LENGTH(colorspaces); i++) {
        struct colorspace *c = colorspaces[i];
        if (!strcmp(c->name, name))
            return c;
    }
 
    return NULL;
}
 
class cHdrMetadata {
public:
    cHdrMetadata(cVideoRender *render) : m_pRender(render) { };
    int Build(struct hdr_output_metadata *,int, int, AVFrameSideData *, AVFrameSideData *);
    int GetColorPrimaries(void) { return m_colorPrimaries; };
    int GetColorTrc(void) { return m_colorTrc; };
 
private:
    cVideoRender *m_pRender;                       
    int m_colorPrimaries = -1;                     
    int m_colorTrc = -1;                           
    AVMasteringDisplayMetadata m_mdMetadata = { }; 
    AVContentLightMetadata m_clMetadata = { };     
    struct colorspace m_hdr10;                     
};
 
#endif