Go to the documentation of this file.
38 #if !NVDECAPI_CHECK_VERSION(9, 0)
39 #define cudaVideoSurfaceFormat_YUV444 2
40 #define cudaVideoSurfaceFormat_YUV444_16Bit 3
59 #define CHECK_CU(x) FF_CUDA_CHECK_DL(logctx, decoder->cudl, x)
80 int shift_h = 0, shift_v = 0;
84 if (shift_h == 1 && shift_v == 1)
85 return cudaVideoChromaFormat_420;
86 else if (shift_h == 1 && shift_v == 0)
87 return cudaVideoChromaFormat_422;
88 else if (shift_h == 0 && shift_v == 0)
89 return cudaVideoChromaFormat_444;
95 CUVIDDECODECREATEINFO *
params,
void *logctx)
98 CUVIDDECODECAPS caps = { 0 };
100 caps.eCodecType =
params->CodecType;
101 caps.eChromaFormat =
params->ChromaFormat;
102 caps.nBitDepthMinus8 =
params->bitDepthMinus8;
104 if (!
decoder->cvdl->cuvidGetDecoderCaps) {
105 av_log(logctx,
AV_LOG_WARNING,
"Used Nvidia driver is too old to perform a capability check.\n");
107 #
if defined(_WIN32) || defined(__CYGWIN__)
112 ". Continuing blind.\n");
122 caps.bIsSupported ?
"yes" :
"no", caps.nMaxMBCount);
124 caps.nMinWidth, caps.nMaxWidth);
126 caps.nMinHeight, caps.nMaxHeight);
128 if (!caps.bIsSupported) {
133 if (
params->ulWidth > caps.nMaxWidth ||
params->ulWidth < caps.nMinWidth) {
135 (
int)
params->ulWidth, caps.nMinWidth, caps.nMaxWidth);
139 if (
params->ulHeight > caps.nMaxHeight ||
params->ulHeight < caps.nMinHeight) {
141 (
int)
params->ulHeight, caps.nMinHeight, caps.nMaxHeight);
145 if ((
params->ulWidth *
params->ulHeight) / 256 > caps.nMaxMBCount) {
147 (
int)(
params->ulWidth *
params->ulHeight) / 256, caps.nMaxMBCount);
159 void *logctx =
decoder->hw_device_ref->data;
168 cuvid_free_functions(&
decoder->cvdl);
174 CUVIDDECODECREATEINFO *
params,
void *logctx)
201 decoder->cuda_ctx = device_hwctx->cuda_ctx;
202 decoder->cudl = device_hwctx->internal->cuda_dl;
203 decoder->stream = device_hwctx->stream;
205 ret = cuvid_load_functions(&
decoder->cvdl, logctx);
259 ctx->bitstream_len = 0;
260 ctx->bitstream_allocated = 0;
264 ctx->slice_offsets_allocated = 0;
280 CUVIDDECODECREATEINFO
params = { 0 };
282 cudaVideoSurfaceFormat output_format;
283 int cuvid_codec_type, cuvid_chroma_format, chroma_444;
291 if (cuvid_codec_type < 0) {
297 if (cuvid_chroma_format < 0) {
301 chroma_444 =
ctx->supports_444 && cuvid_chroma_format == cudaVideoChromaFormat_444;
312 cudaVideoSurfaceFormat_NV12;
317 cudaVideoSurfaceFormat_P016;
331 params.OutputFormat = output_format;
332 params.CodecType = cuvid_codec_type;
333 params.ChromaFormat = cuvid_chroma_format;
339 if (
params.ulNumDecodeSurfaces > 32) {
340 av_log(avctx,
AV_LOG_WARNING,
"Using more than 32 (%d) decode surfaces might cause nvdec to fail.\n",
341 (
int)
params.ulNumDecodeSurfaces);
357 if (!
ctx->decoder_pool) {
385 void *logctx =
decoder->hw_device_ref->data;
386 CUdeviceptr devptr = (CUdeviceptr)opaque;
412 CUVIDPROCPARAMS vpp = { 0 };
418 unsigned int pitch,
i;
420 int shift_h = 0, shift_v = 0;
423 vpp.progressive_frame = 1;
424 vpp.output_stream =
decoder->stream;
445 if (!
frame->buf[1]) {
450 unmap_data->idx = cf->idx;
455 for (
i = 0;
frame->linesize[
i];
i++) {
457 frame->linesize[
i] = pitch;
464 if (!
frame->buf[1]) {
483 ctx->bitstream_len = 0;
494 if (!cf->decoder_ref) {
505 cf->idx = *(
unsigned int*)cf->idx_ref->data;
522 void *logctx = avctx;
523 CUVIDPICPARAMS *pp = &
ctx->pic_params;
529 pp->nBitstreamDataLen =
ctx->bitstream_len;
530 pp->pBitstreamData =
ctx->bitstream;
531 pp->nNumSlices =
ctx->nb_slices;
532 pp->pSliceDataOffsets =
ctx->slice_offsets;
563 (
ctx->nb_slices + 1) *
sizeof(*
ctx->slice_offsets));
595 int cuvid_codec_type, cuvid_chroma_format, chroma_444;
602 if (cuvid_codec_type < 0) {
608 if (cuvid_chroma_format < 0) {
612 chroma_444 = supports_444 && cuvid_chroma_format == cudaVideoChromaFormat_444;
626 if (!frames_ctx->
pool)
629 switch (sw_desc->comp[0].depth) {
static AVBufferRef * nvdec_decoder_frame_alloc(void *opaque, int size)
static int nvdec_decoder_create(AVBufferRef **out, AVBufferRef *hw_device_ref, CUVIDDECODECREATEINFO *params, void *logctx)
#define AV_LOG_WARNING
Something somehow does not look correct.
@ AV_PIX_FMT_CUDA
HW acceleration through CUDA.
AVPixelFormat
Pixel format.
AVBufferRef * av_buffer_alloc(int size)
Allocate an AVBuffer of the given size using av_malloc().
Filter the word “frame” indicates either a video frame or a group of audio as stored in an AVFrame structure Format for each input and each output the list of supported formats For video that means pixel format For audio that means channel sample they are references to shared objects When the negotiation mechanism computes the intersection of the formats supported at each end of a all references to both lists are replaced with a reference to the intersection And when a single format is eventually chosen for a link amongst the remaining all references to the list are updated That means that if a filter requires that its input and output have the same format amongst a supported all it has to do is use a reference to the same list of formats query_formats can leave some formats unset and return AVERROR(EAGAIN) to cause the negotiation mechanism toagain later. That can be used by filters with complex requirements to use the format negotiated on one link to set the formats supported on another. Frame references ownership and permissions
static int map_avcodec_id(enum AVCodecID id)
const AVPixFmtDescriptor * av_pix_fmt_desc_get(enum AVPixelFormat pix_fmt)
uint8_t * data
The data buffer.
static int map_chroma_format(enum AVPixelFormat pix_fmt)
enum AVPixelFormat format
The pixel format identifying the underlying HW surface type.
int ff_nvdec_get_ref_idx(AVFrame *frame)
This struct stores per-frame lavc-internal data and is attached to it via private_ref.
This structure describes decoded (raw) audio or video data.
void(* free)(struct AVHWFramesContext *ctx)
This field may be set by the caller before calling av_hwframe_ctx_init().
int depth
Number of bits in the component.
#define AV_LOG_VERBOSE
Detailed information.
AVBufferRef * av_buffer_create(uint8_t *data, int size, void(*free)(void *opaque, uint8_t *data), void *opaque, int flags)
Create an AVBuffer from an existing array.
void(* hwaccel_priv_free)(void *priv)
int width
The allocated dimensions of the frames in this pool.
AVBufferPool * av_buffer_pool_init(int size, AVBufferRef *(*alloc)(int size))
Allocate and initialize a buffer pool.
static const chunk_decoder decoder[8]
int ff_nvdec_start_frame(AVCodecContext *avctx, AVFrame *frame)
int thread_count
thread count is used to decide how many independent tasks should be passed to execute()
int av_pix_fmt_get_chroma_sub_sample(enum AVPixelFormat pix_fmt, int *h_shift, int *v_shift)
Utility function to access log2_chroma_w log2_chroma_h from the pixel format AVPixFmtDescriptor.
This struct aggregates all the (hardware/vendor-specific) "high-level" state, i.e.
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
#define cudaVideoSurfaceFormat_YUV444
AVBufferRef * av_buffer_pool_get(AVBufferPool *pool)
Allocate a new AVBuffer, reusing an old buffer from the pool when available.
AVBufferPool * pool
A pool from which the frames are allocated by av_hwframe_get_buffer().
void * av_fast_realloc(void *ptr, unsigned int *size, size_t min_size)
Reallocate the given buffer if it is not large enough, otherwise do nothing.
#define AV_PIX_FMT_YUV444P16
#define AV_BUFFER_FLAG_READONLY
Always treat the buffer as read-only, even when it has only one reference.
static void nvdec_decoder_free(void *opaque, uint8_t *data)
static enum AVPixelFormat pix_fmt
int ff_decode_get_hw_frames_ctx(AVCodecContext *avctx, enum AVHWDeviceType dev_type)
Make sure avctx.hw_frames_ctx is set.
int ff_nvdec_simple_end_frame(AVCodecContext *avctx)
enum AVPixelFormat sw_format
The pixel format identifying the actual data layout of the hardware frames.
void av_buffer_unref(AVBufferRef **buf)
Free a given reference and automatically free the buffer if there are no more references to it.
int ff_nvdec_decode_init(AVCodecContext *avctx)
AVBufferRef * device_ref
A reference to the parent AVHWDeviceContext.
AVBufferRef * hw_device_ref
struct AVCodecInternal * internal
Private context used for internal data.
void av_buffer_pool_uninit(AVBufferPool **ppool)
Mark the pool as being available for freeing.
AVCodecID
Identify the syntax and semantics of the bitstream.
#define cudaVideoSurfaceFormat_YUV444_16Bit
int(* post_process)(void *logctx, AVFrame *frame)
The callback to perform some delayed processing on the frame right before it is returned to the calle...
void * hwaccel_priv_data
hwaccel-specific private data
static int nvdec_test_capabilities(NVDECDecoder *decoder, CUVIDDECODECREATEINFO *params, void *logctx)
it s the only field you need to keep assuming you have a context There is some magic you don t need to care about around this just let it vf offset
static void nvdec_free_dummy(struct AVHWFramesContext *ctx)
int ff_nvdec_decode_uninit(AVCodecContext *avctx)
#define i(width, name, range_min, range_max)
static AVBufferRef * hw_device_ctx
static AVBufferRef * nvdec_alloc_dummy(int size)
static void nvdec_fdd_priv_free(void *priv)
int ff_nvdec_end_frame(AVCodecContext *avctx)
void * av_mallocz(size_t size)
Allocate a memory block with alignment suitable for all memory accesses (including vectors if availab...
AVBufferPool * av_buffer_pool_init2(int size, void *opaque, AVBufferRef *(*alloc)(void *opaque, int size), void(*pool_free)(void *opaque))
Allocate and initialize a buffer pool with a more complex allocator.
AVBufferRef * hw_frames_ctx
A reference to the AVHWFramesContext describing the input (for encoding) or output (decoding) frames.
static int nvdec_retrieve_data(void *logctx, AVFrame *frame)
This struct describes a set or pool of "hardware" frames (i.e.
This struct is allocated as AVHWDeviceContext.hwctx.
@ AV_PIX_FMT_NV12
planar YUV 4:2:0, 12bpp, 1 plane for Y and 1 plane for the UV components, which are interleaved (firs...
these buffered frames must be flushed immediately if a new input produces new the filter must not call request_frame to get more It must just process the frame or queue it The task of requesting more frames is left to the filter s request_frame method or the application If a filter has several the filter must be ready for frames arriving randomly on any input any filter with several inputs will most likely require some kind of queuing mechanism It is perfectly acceptable to have a limited queue and to drop frames when the inputs are too unbalanced request_frame For filters that do not use the this method is called when a frame is wanted on an output For a it should directly call filter_frame on the corresponding output For a if there are queued frames already one of these frames should be pushed If the filter should request a frame on one of its repeatedly until at least one frame has been pushed Return or at least make progress towards producing a frame
const char const char * params
unsigned int nb_allocated
main external API structure.
int ff_nvdec_simple_decode_slice(AVCodecContext *avctx, const uint8_t *buffer, uint32_t size)
static void nvdec_unmap_mapped_frame(void *opaque, uint8_t *data)
the frame and frame reference mechanism is intended to as much as expensive copies of that data while still allowing the filters to produce correct results The data is stored in buffers represented by AVFrame structures Several references can point to the same frame buffer
int ff_nvdec_frame_params(AVCodecContext *avctx, AVBufferRef *hw_frames_ctx, int dpb_size, int supports_444)
AVComponentDescriptor comp[4]
Parameters that describe how pixels are packed.
AVBufferRef * av_buffer_ref(AVBufferRef *buf)
Create a new reference to an AVBuffer.
@ AV_PIX_FMT_YUV444P
planar YUV 4:4:4, 24bpp, (1 Cr & Cb sample per 1x1 Y samples)
int coded_width
Bitstream width / height, may be different from width/height e.g.
int initial_pool_size
Initial size of the frame pool.
A reference to a data buffer.
Descriptor that unambiguously describes how the bits of a pixel are stored in the up to 4 data planes...
void * hwaccel_priv
Per-frame private data for hwaccels.
#define AVERROR_BUG
Internal bug, also see AVERROR_BUG2.
enum AVPixelFormat sw_pix_fmt
Nominal unaccelerated pixel format, see AV_PIX_FMT_xxx.
AVBufferRef * decoder_ref
@ AV_CODEC_ID_MPEG2VIDEO
preferred ID for MPEG-1/2 video decoding