| /* |
| * Dolby Vision RPU decoder |
| * |
| * Copyright (C) 2021 Jan Ekström |
| * Copyright (C) 2021 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 |
| */ |
| |
| #ifndef AVCODEC_DOVI_RPU_H |
| #define AVCODEC_DOVI_RPU_H |
| |
| #include "libavutil/dovi_meta.h" |
| #include "libavutil/frame.h" |
| #include "avcodec.h" |
| |
| #define DOVI_MAX_DM_ID 15 |
| typedef struct DOVIContext { |
| void *logctx; |
| |
| /** |
| * Enable tri-state. For encoding only. FF_DOVI_AUTOMATIC enables Dolby |
| * Vision only if avctx->decoded_side_data contains an AVDOVIMetadata. |
| */ |
| #define FF_DOVI_AUTOMATIC -1 |
| int enable; |
| |
| /** |
| * Currently active dolby vision configuration, or {0} for none. |
| * Set by the user when decoding. Generated by ff_dovi_configure() |
| * when encoding. |
| * |
| * Note: sizeof(cfg) is not part of the libavutil ABI, so users should |
| * never pass &cfg to any other library calls. This is included merely as |
| * a way to look up the values of fields known at compile time. |
| */ |
| AVDOVIDecoderConfigurationRecord cfg; |
| |
| /** |
| * Currently active RPU data header, updates on every ff_dovi_rpu_parse() |
| * or ff_dovi_rpu_generate(). |
| */ |
| AVDOVIRpuDataHeader header; |
| |
| /** |
| * Currently active data mappings, or NULL. Points into memory owned by the |
| * corresponding rpu/vdr_ref, which becomes invalid on the next call to |
| * ff_dovi_rpu_parse() or ff_dovi_rpu_generate(). |
| */ |
| const AVDOVIDataMapping *mapping; |
| const AVDOVIColorMetadata *color; |
| |
| /** |
| * Currently active extension blocks, updates on every ff_dovi_rpu_parse() |
| * or ff_dovi_rpu_generate(). |
| */ |
| AVDOVIDmData *ext_blocks; |
| int num_ext_blocks; |
| |
| /** |
| * Private fields internal to dovi_rpu.c |
| */ |
| struct DOVIVdr *vdr[DOVI_MAX_DM_ID+1]; ///< RefStruct references |
| uint8_t *rpu_buf; ///< temporary buffer |
| unsigned rpu_buf_sz; |
| |
| } DOVIContext; |
| |
| void ff_dovi_ctx_replace(DOVIContext *s, const DOVIContext *s0); |
| |
| /** |
| * Completely reset a DOVIContext, preserving only logctx. |
| */ |
| void ff_dovi_ctx_unref(DOVIContext *s); |
| |
| /** |
| * Partially reset the internal state. Resets per-frame state, but preserves |
| * the stream-wide configuration record. |
| */ |
| void ff_dovi_ctx_flush(DOVIContext *s); |
| |
| /** |
| * Parse the contents of a Dovi RPU NAL and update the parsed values in the |
| * DOVIContext struct. |
| * |
| * Returns 0 or an error code. |
| * |
| * Note: `DOVIContext.cfg` should be initialized before calling into this |
| * function. If not done, the profile will be guessed according to HEVC |
| * semantics. |
| */ |
| int ff_dovi_rpu_parse(DOVIContext *s, const uint8_t *rpu, size_t rpu_size, |
| int err_recognition); |
| |
| /** |
| * Attach the decoded AVDOVIMetadata as side data to an AVFrame. |
| */ |
| int ff_dovi_attach_side_data(DOVIContext *s, AVFrame *frame); |
| |
| /** |
| * Configure the encoder for Dolby Vision encoding. Generates a configuration |
| * record in s->cfg, and attaches it to avctx->coded_side_data. Sets the correct |
| * profile and compatibility ID based on the tagged AVCodecContext colorspace |
| * metadata, and the correct level based on the resolution and tagged framerate. |
| * |
| * Returns 0 or a negative error code. |
| */ |
| int ff_dovi_configure(DOVIContext *s, AVCodecContext *avctx); |
| |
| |
| /*************************************************** |
| * The following section is for internal use only. * |
| ***************************************************/ |
| |
| typedef struct DOVIVdr { |
| AVDOVIDataMapping mapping; |
| AVDOVIColorMetadata color; |
| } DOVIVdr; |
| |
| enum { |
| RPU_COEFF_FIXED = 0, |
| RPU_COEFF_FLOAT = 1, |
| }; |
| |
| /** |
| * Synthesize a Dolby Vision RPU reflecting the current state. Note that this |
| * assumes all previous calls to `ff_dovi_rpu_generate` have been appropriately |
| * signalled, i.e. it will not re-send already transmitted redundant data. |
| * |
| * Mutates the internal state of DOVIContext to reflect the change. |
| * Returns 0 or a negative error code. |
| * |
| * This generates a fully formed RPU ready for inclusion in the bitstream, |
| * including the EMDF header (profile 10) or NAL encapsulation (otherwise). |
| */ |
| int ff_dovi_rpu_generate(DOVIContext *s, const AVDOVIMetadata *metadata, |
| uint8_t **out_rpu, int *out_size); |
| |
| /** |
| * Internal helper function to guess the correct DV profile for HEVC. |
| * |
| * Returns the profile number or 0 if unknown. |
| */ |
| int ff_dovi_guess_profile_hevc(const AVDOVIRpuDataHeader *hdr); |
| |
| #endif /* AVCODEC_DOVI_RPU_H */ |