mirror of https://git.ffmpeg.org/ffmpeg.git
136 lines
3.6 KiB
C
136 lines
3.6 KiB
C
/*
|
|
* Dolby Vision RPU decoder
|
|
*
|
|
* Copyright (C) 2021 Jan Ekström
|
|
* Copyright (C) 2021-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
|
|
*/
|
|
|
|
#include "libavutil/mem.h"
|
|
|
|
#include "dovi_rpu.h"
|
|
#include "refstruct.h"
|
|
|
|
void ff_dovi_ctx_unref(DOVIContext *s)
|
|
{
|
|
ff_refstruct_unref(&s->dm);
|
|
for (int i = 0; i < FF_ARRAY_ELEMS(s->vdr); i++)
|
|
ff_refstruct_unref(&s->vdr[i]);
|
|
ff_refstruct_unref(&s->ext_blocks);
|
|
av_free(s->rpu_buf);
|
|
|
|
*s = (DOVIContext) {
|
|
.logctx = s->logctx,
|
|
};
|
|
}
|
|
|
|
void ff_dovi_ctx_flush(DOVIContext *s)
|
|
{
|
|
ff_refstruct_unref(&s->dm);
|
|
for (int i = 0; i < FF_ARRAY_ELEMS(s->vdr); i++)
|
|
ff_refstruct_unref(&s->vdr[i]);
|
|
ff_refstruct_unref(&s->ext_blocks);
|
|
|
|
*s = (DOVIContext) {
|
|
.logctx = s->logctx,
|
|
.cfg = s->cfg,
|
|
/* preserve temporary buffer */
|
|
.rpu_buf = s->rpu_buf,
|
|
.rpu_buf_sz = s->rpu_buf_sz,
|
|
};
|
|
}
|
|
|
|
void ff_dovi_ctx_replace(DOVIContext *s, const DOVIContext *s0)
|
|
{
|
|
s->logctx = s0->logctx;
|
|
s->cfg = s0->cfg;
|
|
s->header = s0->header;
|
|
s->mapping = s0->mapping;
|
|
s->color = s0->color;
|
|
ff_refstruct_replace(&s->dm, s0->dm);
|
|
for (int i = 0; i <= DOVI_MAX_DM_ID; i++)
|
|
ff_refstruct_replace(&s->vdr[i], s0->vdr[i]);
|
|
ff_refstruct_replace(&s->ext_blocks, s0->ext_blocks);
|
|
s->num_ext_blocks = s0->num_ext_blocks;
|
|
}
|
|
|
|
int ff_dovi_guess_profile_hevc(const AVDOVIRpuDataHeader *hdr)
|
|
{
|
|
switch (hdr->vdr_rpu_profile) {
|
|
case 0:
|
|
if (hdr->bl_video_full_range_flag)
|
|
return 5;
|
|
break;
|
|
case 1:
|
|
if (hdr->el_spatial_resampling_filter_flag && !hdr->disable_residual_flag) {
|
|
if (hdr->vdr_bit_depth == 12) {
|
|
return 7;
|
|
} else {
|
|
return 4;
|
|
}
|
|
} else {
|
|
return 8;
|
|
}
|
|
}
|
|
|
|
return 0; /* unknown */
|
|
}
|
|
|
|
const AVDOVIColorMetadata ff_dovi_color_default = {
|
|
.dm_metadata_id = 0,
|
|
.scene_refresh_flag = 0,
|
|
.ycc_to_rgb_matrix = {
|
|
{ 9575, 8192 },
|
|
{ 0, 8192 },
|
|
{ 14742, 8192 },
|
|
{ 9575, 8192 },
|
|
{ 1754, 8192 },
|
|
{ 4383, 8192 },
|
|
{ 9575, 8192 },
|
|
{ 17372, 8192 },
|
|
{ 0, 8192 },
|
|
},
|
|
.ycc_to_rgb_offset = {
|
|
{ 1, 4 },
|
|
{ 2, 1 },
|
|
{ 2, 1 },
|
|
},
|
|
.rgb_to_lms_matrix = {
|
|
{ 5845, 16384 },
|
|
{ 9702, 16384 },
|
|
{ 837, 16384 },
|
|
{ 2568, 16384 },
|
|
{ 12256, 16384 },
|
|
{ 1561, 16384 },
|
|
{ 0, 16384 },
|
|
{ 679, 16384 },
|
|
{ 15705, 16384 },
|
|
},
|
|
.signal_eotf = 39322,
|
|
.signal_eotf_param0 = 15867,
|
|
.signal_eotf_param1 = 228,
|
|
.signal_eotf_param2 = 1383604,
|
|
.signal_bit_depth = 14,
|
|
.signal_color_space = 0,
|
|
.signal_chroma_format = 0,
|
|
.signal_full_range_flag = 1,
|
|
.source_min_pq = 62,
|
|
.source_max_pq = 3696,
|
|
.source_diagonal = 42,
|
|
};
|