| /* |
| * 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) |
| { |
| 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) |
| { |
| 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; |
| 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); |
| } |
| |
| 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 */ |
| } |