sources/apps/sample_apps/deepstream-audio/deepstream_audio.h

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 2020 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 * SPDX-License-Identifier: LicenseRef-NvidiaProprietary
 *
 * NVIDIA CORPORATION, its affiliates and licensors retain all intellectual
 * property and proprietary rights in and to this material, related
 * documentation and any modifications thereto. Any use, reproduction,
 * disclosure or distribution of this material and related documentation
 * without an express license agreement from NVIDIA CORPORATION or
 * its affiliates is strictly prohibited.
 */
 
#ifndef __NVGSTDS_AUDIO_H__
#define __NVGSTDS_AUDIO_H__
 
#ifdef __cplusplus
extern "C"
{
#endif
 
#include <gst/gst.h>
#include <stdio.h>
 
#include "deepstream_app_version.h"
#include "deepstream_common.h"
#include "deepstream_config.h"
#include "deepstream_perf.h"
#include "deepstream_audio_classifier.h"
#include "deepstream_sinks.h"
#include "deepstream_sources.h"
#include "deepstream_streammux.h"
 
 
typedef struct _AppCtx AppCtx;
 
typedef void (*bbox_generated_callback)(AppCtx *appCtx, GstBuffer *buf,
                                        NvDsBatchMeta *batch_meta, guint index);
 
typedef struct
{
  guint index;
  gulong all_bbox_buffer_probe_id;
  gulong primary_bbox_buffer_probe_id;
  GstElement *bin;
  GstElement *tee;
  NvDsAudioClassifierBin audio_classifier_bin;
  NvDsSinkBin sink_bin;
  AppCtx *appCtx;
} NvDsInstanceBin;
 
typedef struct
{
  guint bus_id;
  GstElement *pipeline;
  NvDsSrcParentBin multi_src_bin;
  //NvDsSrcBin src_bin;
  NvDsInstanceBin instance_bin;
  NvDsInstanceBin common_elements;
  AppCtx *appCtx;
} NvDsPipeline;
 
typedef struct
{
  gboolean enable_perf_measurement;
  gint file_loop;
  gboolean source_list_enabled;
  guint total_num_sources;
  guint num_source_sub_bins;
  guint num_sink_sub_bins;
  guint perf_measurement_interval_sec;
 
  gchar **uri_list;
  NvDsSourceConfig multi_source_config[MAX_SOURCE_BINS];
  NvDsStreammuxConfig streammux_config;
  NvDsGieConfig audio_classifier_config;
  NvDsSinkSubBinConfig sink_bin_sub_bin_config[MAX_SINK_BINS];
} NvDsConfig;
 
struct _AppCtx
{
  gboolean version;
  gboolean cintr;
  gboolean seeking;
  gboolean quit;
  bbox_generated_callback bbox_generated_post_analytics_cb;
  gint audio_event_id;
  gint return_value;
  guint index;
 
  GMutex app_lock;
  GCond app_cond;
 
  NvDsPipeline pipeline;
  NvDsConfig config;
  NvDsAppPerfStructInt perf_struct;
};
 
typedef struct
{
    gint anomaly_count;
    gint meta_number;
    struct timespec timespec_first_frame;
    GstClockTime gst_ts_first_frame;
    GMutex lock_stream_rtcp_sr;
    guint32 id;
    gint frameCount;
    GstClockTime last_ntp_time;
} StreamSourceInfo;
 
typedef struct
{
    StreamSourceInfo streams[MAX_SOURCE_BINS];
} TestAppCtx;
 
gboolean create_pipeline (AppCtx * appCtx, perf_callback perf_cb, bbox_generated_callback bgpa_cb);
 
gboolean pause_pipeline (AppCtx * appCtx);
gboolean resume_pipeline (AppCtx * appCtx);
gboolean seek_pipeline (AppCtx * appCtx, glong milliseconds, gboolean seek_is_relative);
 
void destroy_pipeline (AppCtx * appCtx);
void restart_pipeline (AppCtx * appCtx);
 
gboolean
parse_config_file (NvDsConfig * config, gchar * cfg_file_path);
 
#ifdef __cplusplus
}
#endif
 
#endif