Dump出app的pipeline的全部结构有助于快速了解一个陌生app的组成,也有助于分析和查找问题或者进行展示说明。在Deepstream开发的app里按要求增加一些代码将pipeline结构图导出成DOT语言脚本文件 *.dot,并且安装AT&T实验室的用于绘制DOT语言脚本描述的图形的开源工具包graphviz,使用这个工具将*.dot文件转换输出成图片文件即可。
1) 首先,安装graphviz
sudo apt-get install graphviz
2) 在app代码里需要增加相关代码,我可以们参考deepstream-app的代码在自己的app里增加对应的代码:
static gboolean
bus_callback (GstBus * bus, GstMessage * message, gpointer data)
{
...
case GST_MESSAGE_STATE_CHANGED:{
GstState oldstate, newstate;
gst_message_parse_state_changed (message, &oldstate, &newstate, NULL);
if (GST_ELEMENT (GST_MESSAGE_SRC (message)) == appCtx->pipeline.pipeline) {
switch (newstate) {
case GST_STATE_PLAYING:
NVGSTDS_INFO_MSG_V ("Pipeline running\n");
GST_DEBUG_BIN_TO_DOT_FILE_WITH_TS (GST_BIN (appCtx->
pipeline.pipeline), GST_DEBUG_GRAPH_SHOW_ALL,
"ds-app-playing");
break;
case GST_STATE_PAUSED:
if (oldstate == GST_STATE_PLAYING) {
NVGSTDS_INFO_MSG_V ("Pipeline paused\n");
}
break;
case GST_STATE_READY:
GST_DEBUG_BIN_TO_DOT_FILE_WITH_TS (GST_BIN (appCtx->pipeline.
pipeline), GST_DEBUG_GRAPH_SHOW_ALL, "ds-app-ready");
if (oldstate == GST_STATE_NULL) {
NVGSTDS_INFO_MSG_V ("Pipeline ready\n");
} else {
NVGSTDS_INFO_MSG_V ("Pipeline stopped\n");
}
break;
case GST_STATE_NULL:
g_mutex_lock (&appCtx->app_lock);
g_cond_broadcast (&appCtx->app_cond);
g_mutex_unlock (&appCtx->app_lock);
break;
default:
break;
}
}
break;
}
...
也就是pipeline的状态监听回调函数bus_callback()函数里,当pipeline的状态转为READY或者PLAYING时或者是进入这两种状态时都调用调用类似下面这句:
ST_DEBUG_BIN_TO_DOT_FILE_WITH_TS (GST_BIN (pipeline),
GST_DEBUG_GRAPH_SHOW_ALL, "state-name");
即可。
3) 设置一个环境变量GST_DEBUG_DUMP_DOT_DIR,可以在执行程序前设置环境变量,指定将*.dot文件保存在哪里,例如输出到执行命令的当前目录下:
export GST_DEBUG_DUMP_DOT_DIR=.
也可以写在程序里,在 "gst_init(&argc, &argv);" 之前调用:
g_setenv("GST_DEBUG_DUMP_DOT_DIR", ".", TRUE);
4) 执行app,例如执行deepstream-app:
cd /opt/nvidia/deepstream/deepstream/samples/configs/deepstream-app
../../../sources/apps/sample_apps/deepstream-app/deepstream-app -c source4_1080p_dec_infer-resnet_tracker_sgie_tiled_display_int8.txt
可以看到有类似于0.01.7.724900223-ds-app-ready.dot和0.01.7.232944784-ds-app-playing.dot这样的文件生成了。
5) 使用graphviz基于*.dot文件生成图片文件:
dot -Tpng 0.01.7.724900223-ds-app-ready.dot > pipeline-ready.png
dot -Tpng 0.01.7.232944784-ds-app-playing.dot > pipeline-playing.png
例如Deepstream的示例deepstream-app的pipeline的READY和PLAYING状态下的结构图分别如下: