From d3261de520c231ac0de864dfb3d8aa221c4a5a1f Mon Sep 17 00:00:00 2001
From: Bob Mottram <bob@freedombone.net>
Date: Wed, 23 Dec 2020 16:19:18 +0000
Subject: [PATCH] Plot modules call graph

---
 tests.py | 30 ++++++++++++++++++++++++++++--
 1 file changed, 28 insertions(+), 2 deletions(-)

diff --git a/tests.py b/tests.py
index ec4a2b830..9519c743e 100644
--- a/tests.py
+++ b/tests.py
@@ -2846,7 +2846,7 @@ def testFunctions():
                     assert False
         print('Function: ' + name + ' ✓')
 
-    print('Constructing call graph')
+    print('Constructing function call graph')
     for modName, modProperties in modules.items():
         lineCtr = 0
         for line in modules[modName]['lines']:
@@ -2856,8 +2856,34 @@ def testFunctions():
                     getFunctionCalls(name, modules[modName]['lines'],
                                      lineCtr, functionProperties)
                 functionProperties[name]['calls'] = callsList.copy()
+                # keep track of which module calls which other module
+                for fn in callsList:
+                    modCall = functionProperties[fn]['module']
+                    if modCall != modName:
+                        if modules[modName].get('calls'):
+                            if modCall not in modules[modName]['calls']:
+                                modules[modName]['calls'].append(modCall)
+                        else:
+                            modules[modName]['calls'] = [modCall]
             lineCtr += 1
+    callGraphStr = 'digraph EpicyonModules {\n\n'
+    callGraphStr += '  graph [fontsize=10 fontname="Verdana" compound=true];\n'
+    callGraphStr += '  node [shape=record fontsize=10 fontname="Verdana"];\n\n'
+    for modName, modProperties in modules.items():
+        if not modProperties.get('calls'):
+            continue
+        for modCall in modProperties['calls']:
+            callGraphStr += '  "' + modName + '" -> "' + modCall + '";\n'
+    callGraphStr += '\n}\n'
+    with open('epicyon_modules.dot', 'w+') as fp:
+        fp.write(callGraphStr)
+        print('Modules call graph saved to epicyon_modules.dot')
+        print('Plot using: ' +
+              'sfdp -x -Goverlap=false -Goverlap_scaling=2 ' +
+              '-Gsep=+100 -Tx11 epicyon_modules.dot')
+
     callGraphStr = 'digraph Epicyon {\n\n'
+    callGraphStr += '  size="8,6"; ratio=fill;\n'
     callGraphStr += '  graph [fontsize=10 fontname="Verdana" compound=true];\n'
     callGraphStr += '  node [shape=record fontsize=10 fontname="Verdana"];\n\n'
 
@@ -2889,7 +2915,7 @@ def testFunctions():
     with open('epicyon.dot', 'w+') as fp:
         fp.write(callGraphStr)
         print('Call graph saved to epicyon.dot')
-        print('Convert to image with: ' +
+        print('Plot using: ' +
               'sfdp -x -Goverlap=prism -Goverlap_scaling=8 ' +
               '-Gsep=+120 -Tx11 epicyon.dot')