FFmpeg  4.4.6
nvdec.h
Go to the documentation of this file.
1 /*
2  * HW decode acceleration through NVDEC
3  *
4  * Copyright (c) 2016 Anton Khirnov
5  *
6  * This file is part of FFmpeg.
7  *
8  * FFmpeg is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU Lesser General Public
10  * License as published by the Free Software Foundation; either
11  * version 2.1 of the License, or (at your option) any later version.
12  *
13  * FFmpeg is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16  * Lesser General Public License for more details.
17  *
18  * You should have received a copy of the GNU Lesser General Public
19  * License along with FFmpeg; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21  */
22 
23 #ifndef AVCODEC_NVDEC_H
24 #define AVCODEC_NVDEC_H
25 
27 
28 #include <stdint.h>
29 
30 #include "libavutil/buffer.h"
31 #include "libavutil/frame.h"
32 
33 #include "avcodec.h"
34 
35 #if defined(NVDECAPI_MAJOR_VERSION) && defined(NVDECAPI_MINOR_VERSION)
36 # define NVDECAPI_CHECK_VERSION(major, minor) \
37  ((major) < NVDECAPI_MAJOR_VERSION || ((major) == NVDECAPI_MAJOR_VERSION && (minor) <= NVDECAPI_MINOR_VERSION))
38 #else
39 /* version macros were added in SDK 8.1 ffnvcodec */
40 # define NVDECAPI_CHECK_VERSION(major, minor) \
41  ((major) < 8 || ((major) == 8 && (minor) <= 0))
42 #endif
43 
44 typedef struct NVDECFrame {
45  unsigned int idx;
46  unsigned int ref_idx;
50 } NVDECFrame;
51 
52 typedef struct NVDECContext {
53  CUVIDPICPARAMS pic_params;
54 
56 
58 
61  unsigned int bitstream_allocated;
63 
64  unsigned *slice_offsets;
65  int nb_slices;
67 
69 } NVDECContext;
70 
74 int ff_nvdec_start_frame_sep_ref(AVCodecContext *avctx, AVFrame *frame, int has_sep_ref);
78  uint32_t size);
80  AVBufferRef *hw_frames_ctx,
81  int dpb_size,
82  int supports_444);
84 
85 #endif /* AVCODEC_NVDEC_H */
uint8_t
Libavcodec external API header.
refcounted data buffer API
static AVFrame * frame
reference-counted frame API
int ff_nvdec_start_frame_sep_ref(AVCodecContext *avctx, AVFrame *frame, int has_sep_ref)
Definition: nvdec.c:601
int ff_nvdec_frame_params(AVCodecContext *avctx, AVBufferRef *hw_frames_ctx, int dpb_size, int supports_444)
Definition: nvdec.c:697
int ff_nvdec_simple_end_frame(AVCodecContext *avctx)
Definition: nvdec.c:665
int ff_nvdec_simple_decode_slice(AVCodecContext *avctx, const uint8_t *buffer, uint32_t size)
Definition: nvdec.c:675
int ff_nvdec_decode_init(AVCodecContext *avctx)
Definition: nvdec.c:330
int ff_nvdec_get_ref_idx(AVFrame *frame)
Definition: nvdec.c:749
int ff_nvdec_end_frame(AVCodecContext *avctx)
Definition: nvdec.c:635
int ff_nvdec_start_frame(AVCodecContext *avctx, AVFrame *frame)
Definition: nvdec.c:559
int ff_nvdec_decode_uninit(AVCodecContext *avctx)
Definition: nvdec.c:262
static char buffer[20]
Definition: seek.c:32
The buffer pool.
A reference to a data buffer.
Definition: buffer.h:84
main external API structure.
Definition: avcodec.h:536
This structure describes decoded (raw) audio or video data.
Definition: frame.h:318
unsigned int slice_offsets_allocated
Definition: nvdec.h:66
unsigned * slice_offsets
Definition: nvdec.h:64
uint8_t * bitstream
Definition: nvdec.h:59
int bitstream_len
Definition: nvdec.h:60
AVBufferPool * decoder_pool
Definition: nvdec.h:55
unsigned int bitstream_allocated
Definition: nvdec.h:61
int nb_slices
Definition: nvdec.h:65
uint8_t * bitstream_internal
Definition: nvdec.h:62
CUVIDPICPARAMS pic_params
Definition: nvdec.h:53
AVBufferRef * decoder_ref
Definition: nvdec.h:57
int supports_444
Definition: nvdec.h:68
AVBufferRef * ref_idx_ref
Definition: nvdec.h:48
AVBufferRef * decoder_ref
Definition: nvdec.h:49
AVBufferRef * idx_ref
Definition: nvdec.h:47
unsigned int ref_idx
Definition: nvdec.h:46
unsigned int idx
Definition: nvdec.h:45
int dpb_size
Definition: h264_levels.c:107
int size