| /* |
| * Direct3D 12 HW acceleration video decoder |
| * |
| * copyright (c) 2022-2023 Wu Jianhua <toqsxw@outlook.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 |
| */ |
| |
| #ifndef AVCODEC_D3D12VA_DECODE_H |
| #define AVCODEC_D3D12VA_DECODE_H |
| |
| #include "libavutil/fifo.h" |
| #include "libavutil/hwcontext.h" |
| #include "libavutil/hwcontext_d3d12va.h" |
| #include "avcodec.h" |
| #include "internal.h" |
| #include "hwaccel_internal.h" |
| |
| /** |
| * @brief This structure is used to provide the necessary configurations and data |
| * to the FFmpeg Direct3D 12 HWAccel implementation for video decoder. |
| */ |
| typedef struct D3D12VADecodeContext { |
| AVBufferRef *decoder_ref; |
| |
| /** |
| * D3D12 video decoder |
| */ |
| ID3D12VideoDecoder *decoder; |
| |
| /** |
| * D3D12 video decoder heap |
| */ |
| ID3D12VideoDecoderHeap *decoder_heap; |
| |
| /** |
| * D3D12 configuration used to create the decoder |
| * |
| * Specified by decoders |
| */ |
| D3D12_VIDEO_DECODE_CONFIGURATION cfg; |
| |
| /** |
| * A cached queue for reusing the D3D12 command allocators and upload buffers |
| * |
| * @see https://learn.microsoft.com/en-us/windows/win32/direct3d12/recording-command-lists-and-bundles#id3d12commandallocator |
| */ |
| AVFifo *objects_queue; |
| |
| /** |
| * D3D12 command queue |
| */ |
| ID3D12CommandQueue *command_queue; |
| |
| /** |
| * D3D12 video decode command list |
| */ |
| ID3D12VideoDecodeCommandList *command_list; |
| |
| /** |
| * The array of resources used for reference frames |
| * |
| * The ref_resources.length is the same as D3D12VADecodeContext.max_num_ref |
| */ |
| ID3D12Resource **ref_resources; |
| |
| /** |
| * The array of subresources used for reference frames |
| * |
| * The ref_subresources.length is the same as D3D12VADecodeContext.max_num_ref |
| */ |
| UINT *ref_subresources; |
| |
| /** |
| * Maximum number of reference frames |
| */ |
| UINT max_num_ref; |
| |
| /** |
| * Used mask used to record reference frames indices |
| */ |
| UINT used_mask; |
| |
| /** |
| * Bitstream size for each frame |
| */ |
| UINT bitstream_size; |
| |
| /** |
| * The sync context used to sync command queue |
| */ |
| AVD3D12VASyncContext sync_ctx; |
| |
| /** |
| * A pointer to AVD3D12VADeviceContext used to create D3D12 objects |
| */ |
| AVD3D12VADeviceContext *device_ctx; |
| |
| /** |
| * Pixel format |
| */ |
| enum AVPixelFormat pix_fmt; |
| |
| /** |
| * Private to the FFmpeg AVHWAccel implementation |
| */ |
| unsigned report_id; |
| } D3D12VADecodeContext; |
| |
| /** |
| * @} |
| */ |
| #define D3D12VA_VIDEO_DEC_ASYNC_DEPTH 36 |
| #define D3D12VA_DECODE_CONTEXT(avctx) ((D3D12VADecodeContext *)((avctx)->internal->hwaccel_priv_data)) |
| #define D3D12VA_FRAMES_CONTEXT(avctx) ((AVHWFramesContext *)(avctx)->hw_frames_ctx->data) |
| |
| /** |
| * @brief Get a suitable maximum bitstream size |
| * |
| * Creating and destroying a resource on d3d12 needs sync and reallocation, so use this function |
| * to help allocate a big enough bitstream buffer to avoid recreating resources when decoding. |
| * |
| * @return the suitable size |
| */ |
| int ff_d3d12va_get_suitable_max_bitstream_size(AVCodecContext *avctx); |
| |
| /** |
| * @brief init D3D12VADecodeContext |
| * |
| * @return Error code (ret < 0 if failed) |
| */ |
| int ff_d3d12va_decode_init(AVCodecContext *avctx); |
| |
| /** |
| * @brief uninit D3D12VADecodeContext |
| * |
| * @return Error code (ret < 0 if failed) |
| */ |
| int ff_d3d12va_decode_uninit(AVCodecContext *avctx); |
| |
| /** |
| * @brief d3d12va common frame params |
| * |
| * @return Error code (ret < 0 if failed) |
| */ |
| int ff_d3d12va_common_frame_params(AVCodecContext *avctx, AVBufferRef *hw_frames_ctx); |
| |
| /** |
| * @brief d3d12va common end frame |
| * |
| * @param avctx codec context |
| * @param frame current output frame |
| * @param pp picture parameters |
| * @param pp_size the size of the picture parameters |
| * @param qm quantization matrix |
| * @param qm_size the size of the quantization matrix |
| * @param callback update decoder-specified input stream arguments |
| * @return Error code (ret < 0 if failed) |
| */ |
| int ff_d3d12va_common_end_frame(AVCodecContext *avctx, AVFrame *frame, |
| const void *pp, unsigned pp_size, |
| const void *qm, unsigned qm_size, |
| int(*)(AVCodecContext *, D3D12_VIDEO_DECODE_INPUT_STREAM_ARGUMENTS *, ID3D12Resource *)); |
| |
| #endif /* AVCODEC_D3D12VA_DEC_H */ |