Added visualization of in and out edges for a Connection to dot repre…

…sentation.
brics · Jan 18, 2016 · d7979db · d7979db
1 parent 57cd7d4
commit d7979db
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/examples/wm_function_blocks.cpp b/examples/wm_function_blocks.cpp
@@ -96,7 +96,7 @@ int main(int argc, char **argv) {
 	vector<brics_3d::rsg::Id> output;
 	vector<brics_3d::rsg::Id> output2;
 //	wm->executeFunctionBlock("pointcloudloader", input, output);
-	wmStructureVisualizer->setIsActive(false);
+	wmStructureVisualizer->setIsActive(true);
 	wm->executeFunctionBlock("osmloader", input, output);
 	wmStructureVisualizer->setIsActive(true);
 	wm->executeFunctionBlock("octreefilter", output, output2); // "stack them together"

diff --git a/src/brics_3d/worldModel/sceneGraph/DotGraphGenerator.cpp b/src/brics_3d/worldModel/sceneGraph/DotGraphGenerator.cpp
@@ -156,6 +156,24 @@ void DotGraphGenerator::doHandleConnection(Connection* connection) {
     nodes << "fillcolor=lightblue ";
     nodes <<"]";
 	nodes << ";" << std::endl;
+
+	/* Add further edges for in going arcs */
+	for (unsigned int sourceIndex = 0; sourceIndex < connection->getNumberOfSourceNodes(); ++sourceIndex) {
+		if(config.abbreviateIds) {
+			edges << "\"" << uuidToUnsignedInt(connection->getSourceNode(sourceIndex)->getId()) << "\"" << " -> " << "\"" << uuidToUnsignedInt(connection->getId()) << "\"" << " [ style=dashed, label=\"in\" ] " << ";" << std::endl;
+		} else {
+			edges << "\"" << connection->getSourceNode(sourceIndex)->getId() << "\"" << " -> " << "\"" << connection->getId() << "\"" << " [ style=dashed, label=\"in\" ] " << ";" << std::endl;
+		}
+	}
+	/* Add further edges for out going arcs */
+	for (unsigned int targetIndex = 0; targetIndex < connection->getNumberOfTargetNodes(); ++targetIndex) {
+		if(config.abbreviateIds) {
+			edges << "\"" << uuidToUnsignedInt(connection->getId())  << "\"" << " -> " << "\"" << uuidToUnsignedInt(connection->getTargetNode(targetIndex)->getId()) << "\"" << " [ style=dashed, label=\"out\" ] " << ";" << std::endl;
+		} else {
+			edges << "\"" << connection->getId() << "\"" << " -> " << "\"" << connection->getTargetNode(targetIndex)->getId() << "\"" << " [ style=dashed, label=\"out\" ] " << ";" << std::endl;
+		}
+	}
+
 }
 
 void DotGraphGenerator::doHandleTransform(Transform* node) {