From ca1d8496aeb36872dc37ef45ee84cd6f92310e15 Mon Sep 17 00:00:00 2001
From: Christophe Favergeon <Christophe.Favergeon@arm.com>
Date: Thu, 16 Dec 2021 07:53:05 +0100
Subject: [PATCH] CMSIS-DSP: Improvements to dot template in SDF

---
 SDFTools/sdf/templates/dot_template.dot | 31 ++++++++++++++++++++++++-
 1 file changed, 30 insertions(+), 1 deletion(-)

diff --git a/SDFTools/sdf/templates/dot_template.dot b/SDFTools/sdf/templates/dot_template.dot
index bd65b37f..ce22313e 100755
--- a/SDFTools/sdf/templates/dot_template.dot
+++ b/SDFTools/sdf/templates/dot_template.dot
@@ -38,6 +38,35 @@ digraph structs {
 </TABLE>>];
 {% else %}
 
+{% if not config.horizontal %}
+
+{{item.nodeID}} [label=<
+<TABLE BORDER="0" CELLBORDER="1" CELLSPACING="0" CELLPADDING="4">
+  <TR>
+    {% for id in range(0,item.maxNbIOs) -%}
+    {% if item.inputNames[id] -%}
+    {% set input %}{{item.inputNames[id]}}{% endset %}
+    <TD PORT="{{input}}"><FONT POINT-SIZE="9.0">{{input}}</FONT></TD>
+    {% else %}
+    <TD></TD>
+    {%- endif %}
+    {%- endfor %}
+  </TR>
+  <TR>
+    <TD ALIGN="CENTER" COLSPAN="{{item.maxNbIOs}}">{{ item.graphvizName }}</TD>
+  </TR>
+  <TR>
+    {% for id in range(0,item.maxNbIOs) -%}
+    {% if item.outputNames[id] -%}
+    {% set output %}{{item.outputNames[id]}}{% endset %}
+    <TD PORT="{{output}}"><FONT POINT-SIZE="9.0">{{output}}</FONT></TD>
+    {% else %}
+    <TD></TD>
+    {%- endif %}
+    {%- endfor %}
+  </TR>
+</TABLE>>];
+{% else %}
 {{item.nodeID}} [label=<
 <TABLE BORDER="0" CELLBORDER="1" CELLSPACING="0" CELLPADDING="4">
   <TR>
@@ -73,7 +102,7 @@ digraph structs {
 
 
 </TABLE>>];
-
+{% endif %}
 {% endif %}
 {% endfor %}