/* * Copyright (C) 2017 Ericsson AB. All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions * are met: * * 1. Redistributions of source code must retain the above copyright * notice, this list of conditions and the following disclaimer. * 2. Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer * in the documentation and/or other materials provided with the * distribution. * * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. */ #ifndef __GST_NVDEC_H__ #define __GST_NVDEC_H__ #ifdef HAVE_NVCODEC_GST_GL #include <gst/gl/gl.h> #include <gst/gl/gstglfuncs.h> #endif #include <gst/video/video.h> #include "gstcuvidloader.h" #include "gstcudaloader.h" #include "gstcudacontext.h" G_BEGIN_DECLS #define GST_TYPE_NVDEC (gst_nvdec_get_type()) #define GST_NVDEC(obj) (G_TYPE_CHECK_INSTANCE_CAST((obj), GST_TYPE_NVDEC, GstNvDec)) #define GST_NVDEC_CLASS(klass) (G_TYPE_CHECK_CLASS_CAST((klass), GST_TYPE_NVDEC, GstNvDecClass)) #define GST_NVDEC_GET_CLASS(obj)(G_TYPE_INSTANCE_GET_CLASS((obj),GST_TYPE_NVDEC,GstNvDecClass)) #define GST_IS_NVDEC(obj) (G_TYPE_CHECK_INSTANCE_TYPE((obj), GST_TYPE_NVDEC)) #define GST_IS_NVDEC_CLASS(obj) (G_TYPE_CHECK_CLASS_TYPE((klass), GST_TYPE_NVDEC)) typedef struct _GstNvDec GstNvDec; typedef struct _GstNvDecClass GstNvDecClass; typedef enum { GST_NVDEC_STATE_INIT = 0, GST_NVDEC_STATE_PARSE, GST_NVDEC_STATE_DECODE, } GstNvDecState; typedef enum { GST_NVDEC_MEM_TYPE_SYSTEM = 0, GST_NVDEC_MEM_TYPE_GL, GST_NVDEC_MEM_TYPE_CUDA, /* FIXME: add support D3D11 memory */ } GstNvDecMemType; struct _GstNvDec { GstVideoDecoder parent; #ifdef HAVE_NVCODEC_GST_GL GstGLDisplay *gl_display; GstGLContext *gl_context; GstGLContext *other_gl_context; #endif guint num_decode_surface; gint max_display_delay; gboolean is_live; CUvideoparser parser; CUvideodecoder decoder; GstCudaContext *cuda_ctx; CUstream cuda_stream; GstVideoInfo out_info; GstClockTime min_latency; GstVideoCodecState *input_state; GstVideoCodecState *output_state; GstFlowReturn last_ret; GstNvDecState state; GstNvDecMemType mem_type; GstBuffer *codec_data; gboolean recv_complete_picture; }; struct _GstNvDecClass { GstVideoDecoderClass parent_class; cudaVideoCodec codec_type; guint cuda_device_id; }; GType gst_nvdec_get_type (void); void gst_nvdec_plugin_init (GstPlugin * plugin, guint device_index, cudaVideoCodec codec, const gchar * codec_name, GstCaps *sink_template, GstCaps *src_template); G_END_DECLS #endif /* __GST_NVDEC_H__ */