Last active
August 28, 2021 20:39
-
-
Save ganindu7/48c1062e8d0877641fd6b08748aa1b61 to your computer and use it in GitHub Desktop.
deepstream_nvdsanalytics_test.cpp
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
/* | |
* Copyright (c) 2020, NVIDIA CORPORATION. All rights reserved. | |
* | |
* Permission is hereby granted, free of charge, to any person obtaining a | |
* copy of this software and associated documentation files (the "Software"), | |
* to deal in the Software without restriction, including without limitation | |
* the rights to use, copy, modify, merge, publish, distribute, sublicense, | |
* and/or sell copies of the Software, and to permit persons to whom the | |
* Software is furnished to do so, subject to the following conditions: | |
* | |
* The above copyright notice and this permission notice shall be included in | |
* all copies or substantial portions of the Software. | |
* | |
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | |
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | |
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL | |
* THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER | |
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING | |
* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER | |
* DEALINGS IN THE SOFTWARE. | |
*/ | |
#include <gst/gst.h> | |
#include <glib.h> | |
#include <stdio.h> | |
#include <math.h> | |
#include <string.h> | |
#include <sys/time.h> | |
#include <iostream> | |
#include <vector> | |
#include <unordered_map> | |
#include <sstream> | |
#include "gstnvdsmeta.h" | |
#include "nvds_analytics_meta.h" | |
#ifndef PLATFORM_TEGRA | |
#include "gst-nvmessage.h" | |
#endif | |
#define MAX_DISPLAY_LEN 64 | |
#define PGIE_CLASS_ID_VEHICLE 0 | |
#define PGIE_CLASS_ID_PERSON 2 | |
/* The muxer output resolution must be set if the input streams will be of | |
* different resolution. The muxer will scale all the input frames to this | |
* resolution. */ | |
#define MUXER_OUTPUT_WIDTH 1920 | |
#define MUXER_OUTPUT_HEIGHT 1080 | |
/* Muxer batch formation timeout, for e.g. 40 millisec. Should ideally be set | |
* based on the fastest source's framerate. */ | |
#define MUXER_BATCH_TIMEOUT_USEC 40000 | |
#define TILED_OUTPUT_WIDTH 1920 | |
#define TILED_OUTPUT_HEIGHT 1080 | |
/* NVIDIA Decoder source pad memory feature. This feature signifies that source | |
* pads having this capability will push GstBuffers containing cuda buffers. */ | |
#define GST_CAPS_FEATURES_NVMM "memory:NVMM" | |
gchar pgie_classes_str[4][32] = { "Vehicle", "TwoWheeler", "Person", | |
"RoadSign" | |
}; | |
typedef struct _CustomData { | |
GstElement *mpipeline; | |
GstElement *msrc; | |
GMainLoop *loop; | |
gboolean keep_playing; | |
}CustomData; | |
/** process keyboard input */ | |
static gboolean handle_keyboard(GIOChannel *source, GIOCondition cond, CustomData *data) | |
{ | |
gchar *str = NULL; | |
if(g_io_channel_read_line(source, &str, NULL, NULL, NULL) != G_IO_STATUS_NORMAL) | |
{ | |
return TRUE; | |
} | |
switch(g_ascii_tolower(str[0])) | |
{ | |
case 's': | |
g_print("exit signal recived..\n"); | |
data->keep_playing = FALSE; | |
// gst_element_set_state(data->mpipeline, GST_STATE_CHANGE_PLAYING_TO_PAUSED); | |
gst_element_set_state(data->mpipeline, GST_STATE_NULL); | |
g_main_loop_quit(data->loop); | |
break; | |
default: | |
break; | |
} | |
g_free(str); | |
return TRUE; | |
} | |
/* nvdsanalytics_src_pad_buffer_probe will extract metadata received on tiler sink pad | |
* and extract nvanalytics metadata etc. */ | |
static GstPadProbeReturn | |
nvdsanalytics_src_pad_buffer_probe (GstPad * pad, GstPadProbeInfo * info, | |
gpointer u_data) | |
{ | |
GstBuffer *buf = (GstBuffer *) info->data; | |
guint num_rects = 0; | |
NvDsObjectMeta *obj_meta = NULL; | |
guint vehicle_count = 0; | |
guint person_count = 0; | |
NvDsMetaList * l_frame = NULL; | |
NvDsMetaList * l_obj = NULL; | |
NvDsBatchMeta *batch_meta = gst_buffer_get_nvds_batch_meta (buf); | |
for (l_frame = batch_meta->frame_meta_list; l_frame != NULL; | |
l_frame = l_frame->next) { | |
NvDsFrameMeta *frame_meta = (NvDsFrameMeta *) (l_frame->data); | |
std::stringstream out_string; | |
vehicle_count = 0; | |
num_rects = 0; | |
person_count = 0; | |
for (l_obj = frame_meta->obj_meta_list; l_obj != NULL; | |
l_obj = l_obj->next) { | |
obj_meta = (NvDsObjectMeta *) (l_obj->data); | |
if (obj_meta->class_id == PGIE_CLASS_ID_VEHICLE) { | |
vehicle_count++; | |
num_rects++; | |
} | |
if (obj_meta->class_id == PGIE_CLASS_ID_PERSON) { | |
person_count++; | |
num_rects++; | |
} | |
// Access attached user meta for each object | |
for (NvDsMetaList *l_user_meta = obj_meta->obj_user_meta_list; l_user_meta != NULL; | |
l_user_meta = l_user_meta->next) { | |
NvDsUserMeta *user_meta = (NvDsUserMeta *) (l_user_meta->data); | |
if(user_meta->base_meta.meta_type == NVDS_USER_OBJ_META_NVDSANALYTICS) | |
{ | |
NvDsAnalyticsObjInfo * user_meta_data = (NvDsAnalyticsObjInfo *)user_meta->user_meta_data; | |
if (user_meta_data->dirStatus.length()){ | |
g_print ("object %lu moving in %s\n", obj_meta->object_id, user_meta_data->dirStatus.c_str()); | |
} | |
} | |
} | |
} | |
/* Iterate user metadata in frames to search analytics metadata */ | |
for (NvDsMetaList * l_user = frame_meta->frame_user_meta_list; | |
l_user != NULL; l_user = l_user->next) { | |
NvDsUserMeta *user_meta = (NvDsUserMeta *) l_user->data; | |
if (user_meta->base_meta.meta_type != NVDS_USER_FRAME_META_NVDSANALYTICS) | |
continue; | |
/* convert to metadata */ | |
NvDsAnalyticsFrameMeta *meta = | |
(NvDsAnalyticsFrameMeta *) user_meta->user_meta_data; | |
/* Get the labels from nvdsanalytics config file */ | |
for (std::pair<std::string, uint32_t> status : meta->objInROIcnt){ | |
out_string << "Objs in ROI "; | |
out_string << status.first; | |
out_string << " = "; | |
out_string << status.second; | |
} | |
for (std::pair<std::string, uint32_t> status : meta->objLCCumCnt){ | |
out_string << " LineCrossing Cumulative "; | |
out_string << status.first; | |
out_string << " = "; | |
out_string << status.second; | |
} | |
for (std::pair<std::string, uint32_t> status : meta->objLCCurrCnt){ | |
out_string << " LineCrossing Current Frame "; | |
out_string << status.first; | |
out_string << " = "; | |
out_string << status.second; | |
} | |
for (std::pair<std::string, bool> status : meta->ocStatus){ | |
out_string << " Overcrowding status "; | |
out_string << status.first; | |
out_string << " = "; | |
out_string << status.second; | |
} | |
} | |
g_print ("Frame Number = %d of Stream = %d, Number of objects = %d " | |
"Vehicle Count = %d Person Count = %d %s\n", | |
frame_meta->frame_num, frame_meta->pad_index, | |
num_rects, vehicle_count, person_count, out_string.str().c_str()); | |
} | |
return GST_PAD_PROBE_OK; | |
} | |
static gboolean | |
bus_call (GstBus * bus, GstMessage * msg, gpointer data) | |
{ | |
GMainLoop *loop = (GMainLoop *) data; | |
switch (GST_MESSAGE_TYPE (msg)) { | |
case GST_MESSAGE_EOS: | |
g_print ("End of stream\n"); | |
g_main_loop_quit (loop); | |
break; | |
case GST_MESSAGE_WARNING: | |
{ | |
gchar *debug; | |
GError *error; | |
gst_message_parse_warning (msg, &error, &debug); | |
g_printerr ("WARNING from element %s: %s\n", | |
GST_OBJECT_NAME (msg->src), error->message); | |
g_free (debug); | |
g_printerr ("Warning: %s\n", error->message); | |
g_error_free (error); | |
break; | |
} | |
case GST_MESSAGE_ERROR: | |
{ | |
gchar *debug; | |
GError *error; | |
gst_message_parse_error (msg, &error, &debug); | |
g_printerr ("ERROR from element %s: %s\n", | |
GST_OBJECT_NAME (msg->src), error->message); | |
if (debug) | |
g_printerr ("Error details: %s\n", debug); | |
g_free (debug); | |
g_error_free (error); | |
g_main_loop_quit (loop); | |
break; | |
} | |
#ifndef PLATFORM_TEGRA | |
case GST_MESSAGE_ELEMENT: | |
{ | |
if (gst_nvmessage_is_stream_eos (msg)) { | |
guint stream_id; | |
if (gst_nvmessage_parse_stream_eos (msg, &stream_id)) { | |
g_print ("Got EOS from stream %d\n", stream_id); | |
} | |
} | |
break; | |
} | |
#endif | |
default: | |
break; | |
} | |
return TRUE; | |
} | |
static void | |
cb_newpad (GstElement * decodebin, GstPad * decoder_src_pad, gpointer data) | |
{ | |
g_print ("In cb_newpad\n"); | |
GstCaps *caps = gst_pad_get_current_caps (decoder_src_pad); | |
const GstStructure *str = gst_caps_get_structure (caps, 0); | |
const gchar *name = gst_structure_get_name (str); | |
GstElement *source_bin = (GstElement *) data; | |
GstCapsFeatures *features = gst_caps_get_features (caps, 0); | |
/* Need to check if the pad created by the decodebin is for video and not | |
* audio. */ | |
if (!strncmp (name, "video", 5)) { | |
/* Link the decodebin pad only if decodebin has picked nvidia | |
* decoder plugin nvdec_*. We do this by checking if the pad caps contain | |
* NVMM memory features. */ | |
if (gst_caps_features_contains (features, GST_CAPS_FEATURES_NVMM)) { | |
/* Get the source bin ghost pad */ | |
GstPad *bin_ghost_pad = gst_element_get_static_pad (source_bin, "src"); | |
if (!gst_ghost_pad_set_target (GST_GHOST_PAD (bin_ghost_pad), | |
decoder_src_pad)) { | |
g_printerr ("Failed to link decoder src pad to source bin ghost pad\n"); | |
} | |
gst_object_unref (bin_ghost_pad); | |
} else { | |
g_printerr ("Error: Decodebin did not pick nvidia decoder plugin.\n"); | |
} | |
} | |
} | |
static void | |
decodebin_child_added (GstChildProxy * child_proxy, GObject * object, | |
gchar * name, gpointer user_data) | |
{ | |
g_print ("Decodebin child added: %s\n", name); | |
if (g_strrstr (name, "decodebin") == name) { | |
g_signal_connect (G_OBJECT (object), "child-added", | |
G_CALLBACK (decodebin_child_added), user_data); | |
} | |
} | |
static GstElement * | |
create_source_bin (guint index, gchar * uri) | |
{ | |
GstElement *bin = NULL, *uri_decode_bin = NULL; | |
gchar bin_name[16] = { }; | |
g_snprintf (bin_name, 15, "source-bin-%02d", index); | |
/* Create a source GstBin to abstract this bin's content from the rest of the | |
* pipeline */ | |
bin = gst_bin_new (bin_name); | |
/* Source element for reading from the uri. | |
* We will use decodebin and let it figure out the container format of the | |
* stream and the codec and plug the appropriate demux and decode plugins. */ | |
uri_decode_bin = gst_element_factory_make ("uridecodebin", "uri-decode-bin"); | |
if (!bin || !uri_decode_bin) { | |
g_printerr ("One element in source bin could not be created.\n"); | |
return NULL; | |
} | |
/* We set the input uri to the source element */ | |
g_object_set (G_OBJECT (uri_decode_bin), "uri", uri, NULL); | |
/* Connect to the "pad-added" signal of the decodebin which generates a | |
* callback once a new pad for raw data has beed created by the decodebin */ | |
g_signal_connect (G_OBJECT (uri_decode_bin), "pad-added", | |
G_CALLBACK (cb_newpad), bin); | |
g_signal_connect (G_OBJECT (uri_decode_bin), "child-added", | |
G_CALLBACK (decodebin_child_added), bin); | |
gst_bin_add (GST_BIN (bin), uri_decode_bin); | |
/* We need to create a ghost pad for the source bin which will act as a proxy | |
* for the video decoder src pad. The ghost pad will not have a target right | |
* now. Once the decode bin creates the video decoder and generates the | |
* cb_newpad callback, we will set the ghost pad target to the video decoder | |
* src pad. */ | |
if (!gst_element_add_pad (bin, gst_ghost_pad_new_no_target ("src", | |
GST_PAD_SRC))) { | |
g_printerr ("Failed to add ghost pad in source bin\n"); | |
return NULL; | |
} | |
return bin; | |
} | |
int | |
main (int argc, char *argv[]) | |
{ | |
CustomData data; | |
GstElementFactory *source_factory, *sink_factory; | |
GMainLoop *loop = NULL; | |
GstElement *pipeline = NULL, *streammux = NULL, *sink = NULL, *pgie = NULL, | |
*nvtracker = NULL, *nvdsanalytics = NULL, | |
*nvvidconv = NULL, *nvosd = NULL, *tiler = NULL, | |
*queue1, *queue2, *queue3, *queue4, *queue5, *queue6, *queue7; | |
#ifdef PLATFORM_TEGRA | |
GstElement *transform = NULL; | |
#endif | |
GstBus *bus = NULL; | |
guint bus_watch_id; | |
GstPad *nvdsanalytics_src_pad = NULL; | |
guint i, num_sources; | |
guint tiler_rows, tiler_columns; | |
guint pgie_batch_size; | |
/* Check input arguments */ | |
// if (argc < 2) { | |
// g_printerr ("Usage: %s <uri1> [uri2] ... [uriN] \n", argv[0]); | |
// return -1; | |
// } | |
source_factory = gst_element_factory_find("nvarguscamerasrc"); | |
// sink_factory = gst_element_factory_find("nveglglessink"); | |
data.msrc = gst_element_factory_create(source_factory, "source-camera"); | |
num_sources = 1; | |
/* Standard GStreamer initialization */ | |
gst_init (&argc, &argv); | |
loop = g_main_loop_new (NULL, FALSE); | |
/* Create gstreamer elements */ | |
/* Create Pipeline element that will form a connection of other elements */ | |
pipeline = gst_pipeline_new ("nvdsanalytics-test-pipeline"); | |
/* Create nvstreammux instance to form batches from one or more sources. */ | |
streammux = gst_element_factory_make ("nvstreammux", "stream-muxer"); | |
if (!pipeline || !streammux) { | |
g_printerr ("One element could not be created. Exiting.\n"); | |
return -1; | |
} | |
gst_bin_add_many(GST_BIN (pipeline), streammux, data.msrc, NULL); | |
GstPad *srcpad, *sinkpad; | |
srcpad = gst_element_get_static_pad(data.msrc, "src"); | |
sinkpad = gst_element_get_request_pad (streammux, "sink_0"); | |
if(!srcpad || !sinkpad) | |
{ | |
g_printerr("pad creation failed\n"); | |
} | |
if(gst_pad_link(srcpad, sinkpad) != GST_PAD_LINK_OK) | |
{ | |
g_printerr("Pad link failed \n"); | |
} | |
gst_object_unref (srcpad); | |
gst_object_unref (sinkpad); | |
// for (i = 0; i < num_sources; i++) { | |
// GstPad *sinkpad, *srcpad; | |
// gchar pad_name[16] = { }; | |
// GstElement *source_bin = create_source_bin (i, argv[i + 1]); | |
// if (!source_bin) { | |
// g_printerr ("Failed to create source bin. Exiting.\n"); | |
// return -1; | |
// } | |
// gst_bin_add (GST_BIN (pipeline), source_bin); | |
// g_snprintf (pad_name, 15, "sink_%u", i); | |
// sinkpad = gst_element_get_request_pad (streammux, pad_name); | |
// if (!sinkpad) { | |
// g_printerr ("Streammux request sink pad failed. Exiting.\n"); | |
// return -1; | |
// } | |
// srcpad = gst_element_get_static_pad (source_bin, "src"); | |
// if (!srcpad) { | |
// g_printerr ("Failed to get src pad of source bin. Exiting.\n"); | |
// return -1; | |
// } | |
// if (gst_pad_link (srcpad, sinkpad) != GST_PAD_LINK_OK) { | |
// g_printerr ("Failed to link source bin to stream muxer. Exiting.\n"); | |
// return -1; | |
// } | |
// gst_object_unref (srcpad); | |
// gst_object_unref (sinkpad); | |
// } | |
/* Use nvinfer to infer on batched frame. */ | |
pgie = gst_element_factory_make ("nvinfer", "primary-nvinference-engine"); | |
/* Use nvtracker to track detections on batched frame. */ | |
nvtracker = gst_element_factory_make ("nvtracker", "nvtracker"); | |
/* Use nvdsanalytics to perform analytics on object */ | |
nvdsanalytics = gst_element_factory_make ("nvdsanalytics", "nvdsanalytics"); | |
/* Use nvtiler to composite the batched frames into a 2D tiled array based | |
* on the source of the frames. */ | |
tiler = gst_element_factory_make ("nvmultistreamtiler", "nvtiler"); | |
/* Use convertor to convert from NV12 to RGBA as required by nvosd */ | |
nvvidconv = gst_element_factory_make ("nvvideoconvert", "nvvideo-converter"); | |
/* Create OSD to draw on the converted RGBA buffer */ | |
nvosd = gst_element_factory_make ("nvdsosd", "nv-onscreendisplay"); | |
/* Add queue elements between every two elements */ | |
queue1 = gst_element_factory_make ("queue", "queue1"); | |
queue2 = gst_element_factory_make ("queue", "queue2"); | |
queue3 = gst_element_factory_make ("queue", "queue3"); | |
queue4 = gst_element_factory_make ("queue", "queue4"); | |
queue5 = gst_element_factory_make ("queue", "queue5"); | |
queue6 = gst_element_factory_make ("queue", "queue6"); | |
queue7 = gst_element_factory_make ("queue", "queue7"); | |
/* Finally render the osd output */ | |
#ifdef PLATFORM_TEGRA | |
transform = gst_element_factory_make ("nvegltransform", "nvegl-transform"); | |
#endif | |
sink = gst_element_factory_make ("nveglglessink", "nvvideo-renderer"); | |
if (!pgie || !nvtracker || !nvdsanalytics || !tiler || !nvvidconv || | |
!nvosd || !sink || !queue1 || !queue2 || !queue3 || !queue4 || !queue5 || | |
!queue6 || !queue7 | |
) { | |
g_printerr ("One element could not be created. Exiting.\n"); | |
return -1; | |
} | |
#ifdef PLATFORM_TEGRA | |
g_print("\n\n\n****************** Platform tegra selected **********************\n\n\n\n"); | |
if(!transform) { | |
g_printerr ("One tegra element could not be created. Exiting.\n"); | |
return -1; | |
} | |
#endif | |
g_object_set (G_OBJECT (streammux), "width", MUXER_OUTPUT_WIDTH, "height", | |
MUXER_OUTPUT_HEIGHT, "batch-size", num_sources, | |
"batched-push-timeout", MUXER_BATCH_TIMEOUT_USEC, NULL); | |
/* Configure the nvinfer element using the nvinfer config file. */ | |
g_object_set (G_OBJECT (pgie), | |
"config-file-path", "nvdsanalytics_pgie_config.txt", NULL); | |
/* Configure the nvtracker element for using the particular tracker algorithm. */ | |
g_object_set (G_OBJECT (nvtracker), | |
"ll-lib-file", "/opt/nvidia/deepstream/deepstream-5.0/lib/libnvds_nvdcf.so", | |
"ll-config-file", "tracker_config.yml", "tracker-width", 640, "tracker-height", 480, | |
NULL); | |
/* Configure the nvdsanalytics element for using the particular analytics config file*/ | |
g_object_set (G_OBJECT (nvdsanalytics), | |
"config-file", "config_nvdsanalytics.txt", | |
NULL); | |
/* Override the batch-size set in the config file with the number of sources. */ | |
g_object_get (G_OBJECT (pgie), "batch-size", &pgie_batch_size, NULL); | |
if (pgie_batch_size != num_sources) { | |
g_printerr | |
("WARNING: Overriding infer-config batch-size (%d) with number of sources (%d)\n", | |
pgie_batch_size, num_sources); | |
g_object_set (G_OBJECT (pgie), "batch-size", num_sources, NULL); | |
} | |
tiler_rows = (guint) sqrt (num_sources); | |
tiler_columns = (guint) ceil (1.0 * num_sources / tiler_rows); | |
/* we set the tiler properties here */ | |
g_object_set (G_OBJECT (tiler), "rows", tiler_rows, "columns", tiler_columns, | |
"width", TILED_OUTPUT_WIDTH, "height", TILED_OUTPUT_HEIGHT, NULL); | |
g_object_set (G_OBJECT (sink), "qos", 0, NULL); | |
/* we add a message handler */ | |
bus = gst_pipeline_get_bus (GST_PIPELINE (pipeline)); | |
bus_watch_id = gst_bus_add_watch (bus, bus_call, loop); | |
gst_object_unref (bus); | |
data.loop = loop; | |
GIOChannel *io_stdin; | |
io_stdin = g_io_channel_unix_new(fileno(stdin)); | |
g_io_add_watch(io_stdin, G_IO_IN, (GIOFunc)handle_keyboard, &data); | |
/* Set up the pipeline */ | |
/* we add all elements into the pipeline */ | |
#ifdef PLATFORM_TEGRA | |
gst_bin_add_many (GST_BIN (pipeline), queue1, pgie, queue2, nvtracker, queue3, | |
nvdsanalytics , queue4, tiler, queue5, | |
nvvidconv, queue6, nvosd, queue7, transform, sink, | |
NULL); | |
/* we link the elements together, with queues in between | |
* nvstreammux -> nvinfer -> nvtracker -> nvdsanalytics -> nvtiler -> | |
* nvvideoconvert -> nvosd -> transform -> sink | |
*/ | |
if (!gst_element_link_many (streammux,queue1, pgie , queue2, nvtracker, | |
queue3, nvdsanalytics, queue4, tiler, queue5, | |
nvvidconv, queue6, nvosd, queue7, transform, sink, NULL)) { | |
g_printerr ("Elements could not be linked. Exiting.\n"); | |
return -1; | |
} | |
#else | |
gst_bin_add_many (GST_BIN (pipeline), queue1, pgie, queue2, | |
nvtracker, queue3, nvdsanalytics, queue4, tiler, queue5, | |
nvvidconv, queue6, nvosd, queue7, sink, NULL); | |
/* we link the elements together | |
* nvstreammux -> nvinfer -> nvtracker -> nvdsanalytics -> nvtiler -> | |
* nvvideoconvert -> nvosd -> sink | |
*/ | |
if (!gst_element_link_many (streammux, queue1, pgie, queue2, nvtracker, | |
queue3, nvdsanalytics, queue4, tiler, queue5, nvvidconv, queue6, | |
nvosd, queue7, sink, NULL)) { | |
g_printerr ("Elements could not be linked. Exiting.\n"); | |
return -1; | |
} | |
#endif | |
/* Lets add probe to get informed of the meta data generated, we add probe to | |
* the sink pad of the nvdsanalytics element, since by that time, the buffer | |
* would have had got all the metadata. | |
*/ | |
nvdsanalytics_src_pad = gst_element_get_static_pad (nvdsanalytics, "src"); | |
if (!nvdsanalytics_src_pad) | |
g_print ("Unable to get src pad\n"); | |
else | |
gst_pad_add_probe (nvdsanalytics_src_pad, GST_PAD_PROBE_TYPE_BUFFER, | |
nvdsanalytics_src_pad_buffer_probe, NULL, NULL); | |
gst_object_unref (nvdsanalytics_src_pad); | |
/* Set the pipeline to "playing" state */ | |
g_print ("Now playing: stuff\n"); | |
// for (i = 0; i < num_sources; i++) { | |
// g_print (" %s,", argv[i + 1]); | |
// } | |
// g_print ("\n"); | |
gst_element_set_state (pipeline, GST_STATE_PLAYING); | |
/* Wait till pipeline encounters an error or EOS */ | |
g_print ("Running...\n"); | |
g_main_loop_run (loop); | |
/* Out of the main loop, clean up nicely */ | |
g_print ("Returned, stopping playback\n"); | |
gst_element_set_state (pipeline, GST_STATE_NULL); | |
g_print ("Deleting pipeline\n"); | |
gst_object_unref (GST_OBJECT (pipeline)); | |
g_source_remove (bus_watch_id); | |
g_main_loop_unref (loop); | |
return 0; | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment