ffmpeg/libavutil/dovi_meta.c
Niklas Haas 4f55e16f2b avutil/dovi_meta: add dolby vision extension blocks
As well as accessors plus a function for allocating this struct with
extension blocks,

Definitions generously taken from quietvoid/dovi_tool, which is
assembled as a collection of various patent fragments, as well as output
by the official Dolby Vision bitstream verifier tool.
2024-04-03 16:16:25 +02:00

77 lines
2.3 KiB
C

/*
* Copyright (c) 2020 Jun Zhao<barryjzhao@tencent.com>
*
* 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
*/
#include <string.h>
#include "dovi_meta.h"
#include "mem.h"
AVDOVIDecoderConfigurationRecord *av_dovi_alloc(size_t *size)
{
AVDOVIDecoderConfigurationRecord *dovi =
av_mallocz(sizeof(AVDOVIDecoderConfigurationRecord));
if (!dovi)
return NULL;
if (size)
*size = sizeof(*dovi);
return dovi;
}
typedef struct AVDOVIMetadataInternal {
AVDOVIMetadata metadata;
AVDOVIRpuDataHeader header;
AVDOVIDataMapping mapping;
AVDOVIColorMetadata color;
AVDOVIDmData ext_blocks[AV_DOVI_MAX_EXT_BLOCKS];
} AVDOVIMetadataInternal;
AVDOVIMetadata *av_dovi_metadata_alloc(size_t *size)
{
AVDOVIMetadataInternal *dovi = av_mallocz(sizeof(AVDOVIMetadataInternal));
if (!dovi)
return NULL;
if (size)
*size = sizeof(*dovi);
dovi->metadata = (struct AVDOVIMetadata) {
.header_offset = offsetof(AVDOVIMetadataInternal, header),
.mapping_offset = offsetof(AVDOVIMetadataInternal, mapping),
.color_offset = offsetof(AVDOVIMetadataInternal, color),
.ext_block_offset = offsetof(AVDOVIMetadataInternal, ext_blocks),
.ext_block_size = sizeof(AVDOVIDmData),
};
return &dovi->metadata;
}
AVDOVIDmData *av_dovi_find_level(const AVDOVIMetadata *data, uint8_t level)
{
for (int i = 0; i < data->num_ext_blocks; i++) {
AVDOVIDmData *ext = av_dovi_get_ext(data, i);
if (ext->level == level)
return ext;
}
return NULL;
}