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