deepstream_secondary_preprocess.h 2.13 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
/*
 * SPDX-FileCopyrightText: Copyright (c) 2022 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_SECONDARY_PREPROCESS_H__
#define __NVGSTDS_SECONDARY_PREPROCESS_H__

#include <gst/gst.h>
#include "deepstream_preprocess.h"

#ifdef __cplusplus
extern "C"
{
#endif

typedef struct
{
  GstElement *queue;
  GstElement *secondary_preprocess;
  GstElement *tee;
  GstElement *sink;
  gboolean create;
  guint num_children;
  gint parent_index;
} NvDsSecondaryPreProcessBinSubBin;

typedef struct
{
  GstElement *bin;
  GstElement *tee;
  GstElement *queue;
  gulong wait_for_secondary_preprocess_process_buf_probe_id;
  gboolean stop;
  gboolean flush;
  NvDsSecondaryPreProcessBinSubBin sub_bins[MAX_SECONDARY_GIE_BINS];
  GMutex wait_lock;
  GCond wait_cond;
} NvDsSecondaryPreProcessBin;

/**
 * Initialize @ref NvDsSecondaryPreProcessBin. It creates and adds secondary preprocess and
 * other elements needed for processing to the bin.
 * It also sets properties mentioned in the configuration file under
 * group @ref CONFIG_GROUP_SECONDARY_PREPROCESS
 *
 * @param[in] num_secondary_gie number of secondary preprocess.
 * @param[in] config_array array of pointers of type @ref NvDsPreProcessConfig
 *            parsed from configuration file.
 * @param[in] bin pointer to @ref NvDsSecondaryPreProcessBin to be filled.
 *
 * @return true if bin created successfully.
 */
gboolean create_secondary_preprocess_bin (guint num_secondary_preprocess,
    guint primary_gie_unique_id,
    NvDsPreProcessConfig *config_array,
    NvDsSecondaryPreProcessBin *bin);

/**
 * Release the resources.
 */
void destroy_secondary_preprocess_bin (NvDsSecondaryPreProcessBin *bin);

#ifdef __cplusplus
}
#endif

#endif