diff options
author | Anton Khirnov <anton@khirnov.net> | 2017-02-11 16:49:34 +0100 |
---|---|---|
committer | Anton Khirnov <anton@khirnov.net> | 2017-07-26 23:24:34 +0200 |
commit | b9129ec4668c511e0a79e25c6f25d748cee172c9 (patch) | |
tree | f3ace45abdc775d65ed2361958e26966d33c47ca /libavcodec/cuvid.h | |
parent | 704311b2946d74a80f65906961cd9baaa18683a3 (diff) | |
download | ffmpeg-streaming-b9129ec4668c511e0a79e25c6f25d748cee172c9.zip ffmpeg-streaming-b9129ec4668c511e0a79e25c6f25d748cee172c9.tar.gz |
h264dec: add a CUVID hwaccel
Some parts of the code are based on a patch by
Timo Rothenpieler <timo@rothenpieler.org>
Diffstat (limited to 'libavcodec/cuvid.h')
-rw-r--r-- | libavcodec/cuvid.h | 61 |
1 files changed, 61 insertions, 0 deletions
diff --git a/libavcodec/cuvid.h b/libavcodec/cuvid.h new file mode 100644 index 0000000..62e376b --- /dev/null +++ b/libavcodec/cuvid.h @@ -0,0 +1,61 @@ +/* + * HW decode acceleration through CUVID + * + * Copyright (c) 2016 Anton Khirnov + * + * This file is part of Libav. + * + * Libav 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. + * + * Libav 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 Libav; if not, write to the Free Software Foundation, + * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA + */ + +#ifndef AVCODEC_CUVID_H +#define AVCODEC_CUVID_H + +#include <cuviddec.h> +#include <stdint.h> + +#include "libavutil/buffer.h" +#include "libavutil/frame.h" + +#include "avcodec.h" + +typedef struct CUVIDFrame { + unsigned int idx; + AVBufferRef *idx_ref; + AVBufferRef *decoder_ref; +} CUVIDFrame; + +typedef struct CUVIDContext { + CUVIDPICPARAMS pic_params; + + AVBufferPool *decoder_pool; + + AVBufferRef *decoder_ref; + + uint8_t *bitstream; + int bitstream_len; + unsigned int bitstream_allocated; + + unsigned *slice_offsets; + int nb_slices; + unsigned int slice_offsets_allocated; +} CUVIDContext; + +int ff_cuvid_decode_init(AVCodecContext *avctx, unsigned int dpb_size); +int ff_cuvid_decode_uninit(AVCodecContext *avctx); +int ff_cuvid_start_frame(AVCodecContext *avctx, AVFrame *frame); +int ff_cuvid_end_frame(AVCodecContext *avctx); + +#endif /* AVCODEC_CUVID_H */ |