mirror of
https://github.com/FFmpeg/FFmpeg.git
synced 2024-11-26 19:01:44 +02:00
e01e30ede1
It involves less allocations, in particular no allocations after the entry has been created. Therefore creating a new reference from an existing one can't fail and therefore need not be checked. It also avoids indirections and casts. Also note that nvdec_decoder_frame_init() (the callback to initialize new entries from the pool) does not use atomics to read and replace the number of entries currently used by the pool. This relies on nvdec (like most other hwaccels) not being run in a truely frame-threaded way. Tested-by: Timo Rothenpieler <timo@rothenpieler.org> Signed-off-by: Andreas Rheinhardt <andreas.rheinhardt@outlook.com>
86 lines
2.8 KiB
C
86 lines
2.8 KiB
C
/*
|
|
* HW decode acceleration through NVDEC
|
|
*
|
|
* Copyright (c) 2016 Anton Khirnov
|
|
*
|
|
* 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_NVDEC_H
|
|
#define AVCODEC_NVDEC_H
|
|
|
|
#include "compat/cuda/dynlink_loader.h"
|
|
|
|
#include <stdint.h>
|
|
|
|
#include "libavutil/buffer.h"
|
|
#include "libavutil/frame.h"
|
|
|
|
#include "avcodec.h"
|
|
|
|
#if defined(NVDECAPI_MAJOR_VERSION) && defined(NVDECAPI_MINOR_VERSION)
|
|
# define NVDECAPI_CHECK_VERSION(major, minor) \
|
|
((major) < NVDECAPI_MAJOR_VERSION || ((major) == NVDECAPI_MAJOR_VERSION && (minor) <= NVDECAPI_MINOR_VERSION))
|
|
#else
|
|
/* version macros were added in SDK 8.1 ffnvcodec */
|
|
# define NVDECAPI_CHECK_VERSION(major, minor) \
|
|
((major) < 8 || ((major) == 8 && (minor) <= 0))
|
|
#endif
|
|
|
|
typedef struct NVDECFrame {
|
|
unsigned int idx;
|
|
unsigned int ref_idx;
|
|
unsigned int *idx_ref; ///< RefStruct reference
|
|
unsigned int *ref_idx_ref; ///< RefStruct reference
|
|
struct NVDECDecoder *decoder; ///< RefStruct reference
|
|
} NVDECFrame;
|
|
|
|
typedef struct NVDECContext {
|
|
CUVIDPICPARAMS pic_params;
|
|
|
|
struct FFRefStructPool *decoder_pool;
|
|
|
|
struct NVDECDecoder *decoder; ///< RefStruct reference
|
|
|
|
uint8_t *bitstream;
|
|
int bitstream_len;
|
|
unsigned int bitstream_allocated;
|
|
uint8_t *bitstream_internal;
|
|
|
|
unsigned *slice_offsets;
|
|
int nb_slices;
|
|
unsigned int slice_offsets_allocated;
|
|
|
|
int supports_444;
|
|
} NVDECContext;
|
|
|
|
int ff_nvdec_decode_init(AVCodecContext *avctx);
|
|
int ff_nvdec_decode_uninit(AVCodecContext *avctx);
|
|
int ff_nvdec_start_frame(AVCodecContext *avctx, AVFrame *frame);
|
|
int ff_nvdec_start_frame_sep_ref(AVCodecContext *avctx, AVFrame *frame, int has_sep_ref);
|
|
int ff_nvdec_end_frame(AVCodecContext *avctx);
|
|
int ff_nvdec_simple_end_frame(AVCodecContext *avctx);
|
|
int ff_nvdec_simple_decode_slice(AVCodecContext *avctx, const uint8_t *buffer,
|
|
uint32_t size);
|
|
int ff_nvdec_frame_params(AVCodecContext *avctx,
|
|
AVBufferRef *hw_frames_ctx,
|
|
int dpb_size,
|
|
int supports_444);
|
|
int ff_nvdec_get_ref_idx(AVFrame *frame);
|
|
|
|
#endif /* AVCODEC_NVDEC_H */
|