Merge tag 'v01.00.00.03' into develop
[tidl/tidl-api.git] / viewer / dot_graph.cpp
index 25b6a97550dd6194d1e3eff0500db88659f8a36e..ac2d868962fed75034123d3e1ebed68d6f6978a6 100644 (file)
@@ -1,5 +1,5 @@
 /******************************************************************************
- * Copyright (c) 2017-18, Texas Instruments Incorporated - http://www.ti.com/
+ * Copyright (c) 2017-2018, Texas Instruments Incorporated - http://www.ti.com/
  *  All rights reserved.
  *
  *  Redistribution and use in source and binary forms, with or without
@@ -102,17 +102,19 @@ void DotGraph::AddVertices()
 
         auto V = add_vertex(i, *sub);
 
-        AddVertexProperties(V, sub, layer);
+        AddVertexProperties(V, sub, layer, i);
     }
 
 }
 
 void DotGraph::AddVertexProperties(Vertex& V, Graph* g,
-                                   const sTIDL_Layer_t& layer)
+                                   const sTIDL_Layer_t& layer,
+                                   int index)
 {
     VertexPropertyMap<Graph> vpm = boost::get(vertex_attribute, *g);
     vpm[V][COLOR] = GetVertexColor(layer.layerType);
     vpm[V][STYLE] = BOLD;
+    vpm[V]["xlabel"] = std::to_string(index);
     vpm[V][LABEL] = "{";
 
     switch (layer.layerType)
@@ -121,7 +123,7 @@ void DotGraph::AddVertexProperties(Vertex& V, Graph* g,
         {
             vpm[V][LABEL] += TIDL_LayerString[layer.layerType];
             const sTIDL_ConvParams_t& p = layer.layerParams.convParams;
-            vpm[V][LABEL] += "\\n" + std::to_string(p.kernelW) + "x" +
+            vpm[V][LABEL] += " " + std::to_string(p.kernelW) + "x" +
                              std::to_string(p.kernelH);
 
             if (p.enablePooling)
@@ -245,7 +247,8 @@ std::string PoolingProperties(const sTIDL_PoolingParams_t& p)
     else if (p.poolingType == TIDL_AveragePooling)
         s += " Average";
 
-    s+= "\\n" + std::to_string(p.kernelW) + "x" + std::to_string(p.kernelH);
+    if (p.kernelW != 0 && p.kernelH != 0)
+        s+= "\\n" + std::to_string(p.kernelW) + "x" + std::to_string(p.kernelH);
 
     return s;
 }