sources/gst-plugins/gst-nvsegvisual/gstnvsegvisual.h

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 2023-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 * SPDX-License-Identifier: Apache-2.0
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
 
#ifndef __GST_NVSEG_VISUAL_H__
#define __GST_NVSEG_VISUAL_H__
 
#include <gst/base/gstbasetransform.h>
#include <gst/video/video.h>
 
#include <cuda.h>
#include <cuda_runtime.h>
 
/* Package and library details required for plugin_init */
#define PACKAGE "nvsegvisual"
#define VERSION "1.0"
#define LICENSE "Proprietary"
#define DESCRIPTION "NVIDIA segmentation visualization plugin for integration with DeepStream on DGPU"
#define BINARY_PACKAGE "NVIDIA DeepStream Segmantation Visualization Plugin"
#define URL "http://nvidia.com/"
 
 
G_BEGIN_DECLS
/* Standard boilerplate stuff */
typedef struct _GstNvSegVisual GstNvSegVisual;
typedef struct _GstNvSegVisualClass GstNvSegVisualClass;
 
/* Standard boilerplate stuff */
#define GST_TYPE_NV_SEG_VISUAL (gst_nvseg_visual_get_type())
#define GST_NV_SEG_VISUAL(obj) (G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_NV_SEG_VISUAL,GstNvSegVisual))
#define GST_NV_SEG_VISUAL_CLASS(klass) (G_TYPE_CHECK_CLASS_CAST((klass),GST_TYPE_NV_SEG_VISUAL,GstNvSegVisualClass))
#define GST_NV_SEG_VISUAL_GET_CLASS(obj) (G_TYPE_INSTANCE_GET_CLASS((obj), GST_TYPE_NV_SEG_VISUAL, GstNvSegVisualClass))
#define GST_IS_NV_SEG_VISUAL(obj) (G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_NV_SEG_VISUAL))
#define GST_IS_NV_SEG_VISUAL_CLASS(klass) (G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_NV_SEG_VISUAL))
#define GST_NV_SEG_VISUAL_CAST(obj)  ((GstNvSegVisual *)(obj))
 
struct _GstNvSegVisual
{
  GstBaseTransform base_trans;
 
  // Unique ID of the element. The labels generated by the element will be
  // updated at index `unique_id` of attr_info array in NvDsObjectParams.
  guint unique_id;
 
  // Frame number of the current input buffer
  guint64 frame_num;
 
  // Input video info (resolution, color format, framerate, etc)
  GstVideoInfo video_info;
 
  // Resolution at which frames/objects should be processed
  gint output_width;
  gint output_height;
 
  // GPU ID on which we expect to execute the task
  guint gpu_id;
 
  // Buffer pool size used for optical flow output
  guint pool_size;
 
  // Segmentation-Meta id for which segmentation is visualize
  gint operate_on_seg_meta_id;
 
  gboolean original_background;
  guint class_id;
  float alpha;
  gboolean flag_alloc_dev_memory;
  unsigned char* class2BGR_device;
  cudaStream_t stream;
  gboolean gpu_on;
 
  int *class_map_device;
 
  gint batch_size;
  guint num_batch_buffers;
 
  gint input_feature;
  gint output_feature;
  gint cuda_mem_type;
  GstVideoFormat input_fmt;
  GstVideoFormat output_fmt;
 
  GstVideoInfo in_info;
  GstVideoInfo out_info;
 
  GstCaps *sinkcaps;
  GstCaps *srccaps;
 
  GstBufferPool *pool;
};
 
// Boiler plate stuff
struct _GstNvSegVisualClass
{
  GstBaseTransformClass parent_class;
};
 
GType gst_nvseg_visual_get_type (void);
 
G_END_DECLS
#endif /* __GST_NVSEG_VISUAL_H__ */