Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
deepstream_config_yaml.h 2.94 KB
Newer Older
das-qa's avatar
Add python demo: deepstream-test1, deepstream-test2, deepstream-test3 and deepstream-test4  
das-qa committed
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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103 
/*
 * SPDX-FileCopyrightText: Copyright (c) 2022-2024 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_STREAMMUX_YAML_H_
#define _NVGSTDS_STREAMMUX_YAML_H_


#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wreturn-type"
#include <yaml-cpp/yaml.h>
#pragma GCC diagnostic pop

#include <gst/gst.h>

#ifdef __cplusplus
extern "C"
{
#endif

#include "deepstream_streammux.h"
#include "deepstream_tiled_display.h"
#include "deepstream_osd.h"
#include "deepstream_segvisual.h"
#include "deepstream_image_save.h"
#include "deepstream_c2d_msg.h"
#include "deepstream_sinks.h"
#include "deepstream_sources.h"
#include "deepstream_tracker.h"
#include "deepstream_gie.h"
#include "deepstream_preprocess.h"
#include "deepstream_dewarper.h"
#include "deepstream_dsanalytics.h"
#include "deepstream_dsexample.h"

#define _MAX_STR_LENGTH 1024

std::vector<std::string> split_string (std::string input);

gboolean
get_absolute_file_path_yaml (
    const gchar * cfg_file_path, const gchar * file_path,
    char *abs_path_str);

gboolean
parse_streammux_yaml (NvDsStreammuxConfig *config, gchar *cfg_file_path);

gboolean
parse_tiled_display_yaml (NvDsTiledDisplayConfig *config, gchar *cfg_file_path);

gboolean
parse_osd_yaml (NvDsOSDConfig *config, gchar *cfg_file_path);

gboolean
parse_segvisual_yaml (NvDsSegVisualConfig *config, gchar *cfg_file_path);

gboolean
parse_image_save_yaml (NvDsImageSave *config, gchar *cfg_file_path);

gboolean
parse_msgconsumer_yaml (NvDsMsgConsumerConfig *config, std::string group, gchar *cfg_file_path);

gboolean
parse_msgconv_yaml (NvDsSinkMsgConvBrokerConfig *config, std::string group, gchar *cfg_file_path);

gboolean
parse_sink_yaml (NvDsSinkSubBinConfig *config, std::string group, gchar * cfg_file_path);

gboolean
parse_source_yaml (NvDsSourceConfig *config, std::vector<std::string> headers,
                    std::vector<std::string> source_values,  gchar *cfg_file_path);

gboolean
parse_tracker_yaml (NvDsTrackerConfig *config, gchar *cfg_file_path);

gboolean
parse_gie_yaml (NvDsGieConfig *config, std::string group, gchar *cfg_file_path);

gboolean
parse_preprocess_yaml (NvDsPreProcessConfig *config, gchar* cfg_file_path);

gboolean
parse_dewarper_yaml (NvDsDewarperConfig * config, std::string group_str, gchar *cfg_file_path);

gboolean
parse_dsexample_yaml (NvDsDsExampleConfig *config, gchar *cfg_file_path);

gboolean
parse_dsanalytics_yaml (NvDsDsAnalyticsConfig *config, gchar* cfg_file_path);

#ifdef __cplusplus
}
#endif

#endif /* _NVGSTDS_DSEXAMPLE_H_ */