mythhdrvideometadata.cpp

Go to the documentation of this file.

// MythTV
#include "mythhdrvideometadata.h"
#include "mythframe.h"
 
// Std
#include <cmath>
 
// FFmpeg
extern "C" {
#include "libavutil/mastering_display_metadata.h"
}
 
static inline uint16_t UINT16(double x)
{ return static_cast<uint16_t>(std::round(x)); }
 
void MythHDRVideoMetadata::Update(const AVMasteringDisplayMetadata* Display,
                                  const AVContentLightMetadata* Light)
{
    bool luminance = Display && Display->has_luminance;
    bool primaries = Display && Display->has_primaries;
    m_maxMasteringLuminance = luminance ? UINT16(av_q2d(Display->max_luminance)) : 0;
    m_minMasteringLuminance = luminance ? UINT16(av_q2d(Display->min_luminance) * 10000) : 0;
    m_whitePoint[0] = primaries ? UINT16(av_q2d(Display->white_point[0]) * 50000) : 0;
    m_whitePoint[1] = primaries ? UINT16(av_q2d(Display->white_point[1]) * 50000) : 0;
    for (size_t i = 0; i < 3; i++)
    {
        m_displayPrimaries[i][0] = primaries ? UINT16(av_q2d(Display->display_primaries[i][0]) * 50000) : 0;
        m_displayPrimaries[i][1] = primaries ? UINT16(av_q2d(Display->display_primaries[i][1]) * 50000) : 0;
    }
    m_maxContentLightLevel = Light ? static_cast<uint16_t>(Light->MaxCLL) : 0;
    m_maxFrameAverageLightLevel = Light ? static_cast<uint16_t>(Light->MaxFALL) : 0;
}
 
void MythHDRVideoMetadata::Populate(MythVideoFrame* Frame, AVFrame* AvFrame)
{
    if (!Frame)
        return;
 
    if (AvFrame)
    {
        const auto * display = av_frame_get_side_data(AvFrame, AV_FRAME_DATA_MASTERING_DISPLAY_METADATA);
        const auto * ddata   = display ? reinterpret_cast<AVMasteringDisplayMetadata*>(display->data) : nullptr;
        const auto * content = av_frame_get_side_data(AvFrame, AV_FRAME_DATA_CONTENT_LIGHT_LEVEL);
        const auto * cdata   = content ? reinterpret_cast<AVContentLightMetadata*>(content->data) : nullptr;
        if (ddata || cdata)
        {
            if (!Frame->m_hdrMetadata.get())
                Frame->m_hdrMetadata = std::make_shared<MythHDRVideoMetadata>();
            Frame->m_hdrMetadata->Update(ddata, cdata);
            return;
        }
    }
    Frame->m_hdrMetadata = nullptr;
}