1 /* GStreamer 2 * Copyright (C) <2019> Seungha Yang <seungha.yang@navercorp.com> 3 * 4 * This library is free software; you can redistribute it and/or 5 * modify it under the terms of the GNU Library General Public 6 * License as published by the Free Software Foundation; either 7 * version 2 of the License, or (at your option) any later version. 8 * 9 * This library is distributed in the hope that it will be useful, 10 * but WITHOUT ANY WARRANTY; without even the implied warranty of 11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 12 * Library General Public License for more details. 13 * 14 * You should have received a copy of the GNU Library General Public 15 * License along with this library; if not, write to the 16 * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor, 17 * Boston, MA 02110-1301, USA. 18 */ 19 20 #ifndef __GST_CUDA_BASE_TRANSFORM_H__ 21 #define __GST_CUDA_BASE_TRANSFORM_H__ 22 23 #include <gst/gst.h> 24 #include <gst/base/gstbasetransform.h> 25 #include <gst/video/video.h> 26 #include "gstcudacontext.h" 27 #include "gstcudabufferpool.h" 28 29 G_BEGIN_DECLS 30 31 #define GST_TYPE_CUDA_BASE_TRANSFORM (gst_cuda_base_transform_get_type()) 32 #define GST_CUDA_BASE_TRANSFORM(obj) (G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_CUDA_BASE_TRANSFORM,GstCudaBaseTransform)) 33 #define GST_CUDA_BASE_TRANSFORM_CLASS(klass) (G_TYPE_CHECK_CLASS_CAST((klass), GST_TYPE_CUDA_BASE_TRANSFORM,GstCudaBaseTransformClass)) 34 #define GST_CUDA_BASE_TRANSFORM_GET_CLASS(obj) (G_TYPE_INSTANCE_GET_CLASS((obj), GST_TYPE_CUDA_BASE_TRANSFORM,GstCudaBaseTransformClass)) 35 #define GST_IS_CUDA_BASE_TRANSFORM(obj) (G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_CUDA_BASE_TRANSFORM)) 36 #define GST_IS_CUDA_BASE_TRANSFORM_CLASS(klass) (G_TYPE_CHECK_CLASS_TYPE((klass), GST_TYPE_CUDA_BASE_TRANSFORM)) 37 38 typedef struct _GstCudaBaseTransform GstCudaBaseTransform; 39 typedef struct _GstCudaBaseTransformClass GstCudaBaseTransformClass; 40 41 struct _GstCudaBaseTransform 42 { 43 GstBaseTransform parent; 44 45 gboolean negotiated; 46 47 GstVideoInfo in_info; 48 GstVideoInfo out_info; 49 50 GstCudaContext *context; 51 CUstream cuda_stream; 52 53 gint device_id; 54 }; 55 56 struct _GstCudaBaseTransformClass 57 { 58 GstBaseTransformClass parent_class; 59 60 gboolean (*set_info) (GstCudaBaseTransform *filter, 61 GstCaps *incaps, GstVideoInfo *in_info, 62 GstCaps *outcaps, GstVideoInfo *out_info); 63 64 GstFlowReturn (*transform_frame) (GstCudaBaseTransform *filter, 65 GstVideoFrame *in_frame, 66 GstCudaMemory *in_cuda_mem, 67 GstVideoFrame *out_frame, 68 GstCudaMemory *out_cuda_mem); 69 }; 70 71 GType gst_cuda_base_transform_get_type (void); 72 73 G_END_DECLS 74 75 #endif /* __GST_CUDA_BASE_TRANSFORM_H__ */ 76