gstreamer/subprojects/gst-plugins-bad/sys/nvcodec/gstcudaipcclient.h
2024-01-30 14:30:41 +00:00

145 lines
5 KiB
C++

/* 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/cuda/gstcuda.h>
#include "gstcudaipc.h"
#include <memory>
#include <vector>
G_BEGIN_DECLS
#define GST_TYPE_CUDA_IPC_CLIENT (gst_cuda_ipc_client_get_type())
#define GST_CUDA_IPC_CLIENT(obj) (G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_CUDA_IPC_CLIENT,GstCudaIpcClient))
#define GST_CUDA_IPC_CLIENT_CLASS(klass) (G_TYPE_CHECK_CLASS_CAST((klass), GST_TYPE_CUDA_IPC_CLIENT,GstCudaIpcClientClass))
#define GST_CUDA_IPC_CLIENT_GET_CLASS(obj) (G_TYPE_INSTANCE_GET_CLASS((obj), GST_TYPE_CUDA_IPC_CLIENT,GstCudaIpcClientClass))
#define GST_IS_CUDA_IPC_CLIENT(obj) (G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_CUDA_IPC_CLIENT))
#define GST_IS_CUDA_IPC_CLIENT_CLASS(klass) (G_TYPE_CHECK_CLASS_TYPE((klass), GST_TYPE_CUDA_IPC_CLIENT))
struct GstCudaIpcClientConn;
struct GstCudaIpcClientPrivate;
enum GstCudaIpcIOMode
{
GST_CUDA_IPC_IO_COPY,
GST_CUDA_IPC_IO_IMPORT,
};
#define GST_TYPE_CUDA_IPC_IO_MODE (gst_cuda_ipc_io_mode_get_type ())
GType gst_cuda_ipc_io_mode_get_type (void);
struct GstCudaIpcClient
{
GstObject parent;
GstCudaContext *context;
GstCudaStream *stream;
GstCudaIpcIOMode io_mode;
guint buffer_size;
GstCudaIpcClientPrivate *priv;
};
struct GstCudaIpcClientClass
{
GstObjectClass parent_class;
bool (*send_msg) (GstCudaIpcClient * client,
GstCudaIpcClientConn * conn);
bool (*wait_msg) (GstCudaIpcClient * client,
GstCudaIpcClientConn * conn);
void (*terminate) (GstCudaIpcClient * client);
void (*invoke) (GstCudaIpcClient * client);
void (*set_flushing) (GstCudaIpcClient * client,
bool flushing);
void (*loop) (GstCudaIpcClient * client);
bool (*config) (GstCudaIpcClient * client,
GstCudaPid pid,
gboolean use_mmap);
};
GType gst_cuda_ipc_client_get_type (void);
GstFlowReturn gst_cuda_ipc_client_get_sample (GstCudaIpcClient * client,
GstSample ** sample);
void gst_cuda_ipc_client_set_flushing (GstCudaIpcClient * client,
bool flushing);
GstCaps * gst_cuda_ipc_client_get_caps (GstCudaIpcClient * client);
GstFlowReturn gst_cuda_ipc_client_run (GstCudaIpcClient * client);
void gst_cuda_ipc_client_stop (GstCudaIpcClient * client);
/* subclass methods */
void gst_cuda_ipc_client_new_connection (GstCudaIpcClient * client,
std::shared_ptr<GstCudaIpcClientConn> conn);
void gst_cuda_ipc_client_send_msg_finish (GstCudaIpcClient * client,
bool result);
void gst_cuda_ipc_client_wait_msg_finish (GstCudaIpcClient * client,
bool result);
void gst_cuda_ipc_client_have_mmap_data (GstCudaIpcClient * client,
GstClockTime pts,
const GstCudaIpcMemLayout & layout,
GstCaps * caps,
GstCudaSharableHandle server_handle,
GstCudaSharableHandle client_handle,
std::vector <guint8> & meta);
void gst_cuda_ipc_client_on_idle (GstCudaIpcClient * client);
void gst_cuda_ipc_client_abort (GstCudaIpcClient * client);
G_DEFINE_AUTOPTR_CLEANUP_FUNC(GstCudaIpcClient, gst_object_unref)
G_END_DECLS
struct GstCudaIpcClientConn : public OVERLAPPED
{
GstCudaIpcClientConn ()
{
client_msg.resize (GST_CUDA_IPC_PKT_HEADER_SIZE);
server_msg.resize (GST_CUDA_IPC_PKT_HEADER_SIZE);
}
virtual ~GstCudaIpcClientConn()
{
gst_clear_object (&context);
}
GstCudaIpcClient *client;
GstCudaContext *context = nullptr;
GstCudaIpcPktType type;
std::vector<guint8> client_msg;
std::vector<guint8> server_msg;
};