gstreamer/subprojects/gst-plugins-bad/sys/nvcodec/gstnvdecobject.h

101 lines
3.5 KiB
C
Raw Normal View History

/* GStreamer
* Copyright (C) 2023 Seungha Yang <seungha@centricular.com>
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Library General Public
* License as published by the Free Software Foundation; either
* version 2 of the License, or (at your option) any later version.
*
* This library 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
* Library General Public License for more details.
*
* You should have received a copy of the GNU Library General Public
* License along with this library; if not, write to the
* Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
* Boston, MA 02110-1301, USA.
*/
#pragma once
#include <gst/gst.h>
#include <gst/video/video.h>
#include <gst/cuda/gstcuda.h>
#include "gstcuvidloader.h"
G_BEGIN_DECLS
#define GST_TYPE_NV_DEC_OBJECT (gst_nv_dec_object_get_type())
G_DECLARE_FINAL_TYPE (GstNvDecObject,
gst_nv_dec_object, GST, NV_DEC_OBJECT, GstObject);
#define GST_TYPE_NV_DEC_SURFACE (gst_nv_dec_surface_get_type())
typedef struct _GstNvDecSurface GstNvDecSurface;
struct _GstNvDecSurface
{
GstMiniObject parent;
GstNvDecObject *object;
gint index;
gint decode_frame_index;
CUdeviceptr devptr;
guint pitch;
guint seq_num;
};
GstNvDecObject * gst_nv_dec_object_new (GstCudaContext * context,
CUVIDDECODECREATEINFO * create_info,
const GstVideoInfo * video_info,
gboolean alloc_aux_frame);
gboolean gst_nv_dec_object_reconfigure (GstNvDecObject * object,
CUVIDRECONFIGUREDECODERINFO * reconfigure_info,
const GstVideoInfo * video_info,
gboolean alloc_aux_frame);
void gst_nv_dec_object_set_flushing (GstNvDecObject * object,
gboolean flushing);
GstFlowReturn gst_nv_dec_object_acquire_surface (GstNvDecObject * object,
GstNvDecSurface ** surface);
gboolean gst_nv_dec_object_decode (GstNvDecObject * object,
CUVIDPICPARAMS * params);
GstFlowReturn gst_nv_dec_object_map_surface (GstNvDecObject * object,
GstNvDecSurface * surface,
GstCudaStream * stream);
gboolean gst_nv_dec_object_unmap_surface (GstNvDecObject * object,
GstNvDecSurface * surface);
GstFlowReturn gst_nv_dec_object_export_surface (GstNvDecObject * object,
GstNvDecSurface * surface,
GstCudaStream * stream,
GstMemory ** memory);
guint gst_nv_dec_object_get_num_free_surfaces (GstNvDecObject * object);
GType gst_nv_dec_surface_get_type (void);
static inline GstNvDecSurface *
gst_nv_dec_surface_ref (GstNvDecSurface * surface)
{
return (GstNvDecSurface *)
gst_mini_object_ref (GST_MINI_OBJECT_CAST (surface));
}
static inline void
gst_nv_dec_surface_unref (GstNvDecSurface * surface)
{
gst_mini_object_unref (GST_MINI_OBJECT_CAST (surface));
}
G_END_DECLS