drawModuleConnections.py

Go to the documentation of this file.

#!/bin/python3
 
import xml.etree.ElementTree as ET
import sys
 
ns = "{https://github.com/ChimeraTK/ApplicationCore}"
 
moduleConnections = {}
 
 
def parseVariable(node):
    varDir = node.find(ns + 'direction').text
 
    feeder = None
    moduleList = set()
 
    if varDir == 'control_system_to_application' or varDir == 'control_system_to_application_with_return':
        return
        feeder = 'ControlSystem'
#  else :
#    moduleList.add("ControlSystem")
 
    for peer in node.find(ns + 'connections'):
        peerType = peer.attrib['type']
        peerDir = peer.attrib['direction']
        if peerType == "ApplicationModule" or peerType == "Device":
            if peerType == "ApplicationModule":
                peerName = peer.attrib['class']
            else:
                peerName = 'Device:' + peer.attrib['name']
            if peerDir == 'consuming':
                moduleList.add(peerName)
            else:
                if feeder is not None:
                    print("ERROR: Found two feeders!")
                    print(node.attrib['name'])
                    sys.exit(1)
                feeder = peerName
 
    if feeder is None:
        print("ERROR: No feeders found!")
        print(node.attrib['name'])
        sys.exit(1)
 
    for m in moduleList:
        if m not in moduleConnections.keys():
            moduleConnections[m] = set()
    if feeder not in moduleConnections.keys():
        moduleConnections[feeder] = set()
 
    for m in moduleList:
        moduleConnections[feeder].add(m)
 
 
def parseDirectory(subtree):
 
    for node in subtree:
 
        if node.tag == ns + 'directory':
            parseDirectory(node)
        elif node.tag == ns + 'variable':
            parseVariable(node)
 
 
tree = ET.parse('llrfctrl.xml')
root = tree.getroot()
 
parseDirectory(root)
 
print("digraph {")
 
 
def makeNodeName(label):
    name = label.replace(':', '_')
    name = name.replace('/', '_')
    name = name.replace('<', '_')
    name = name.replace('>', '_')
    name = name.replace(',', '_')
    name = name.replace(' ', '_')
    return name
 
 
def defineNode(label):
    if label not in nodeIds.keys():
        nodeIds[label] = makeNodeName(label)
        print(nodeIds[label] + ' [label="' + label + '", style=filled', end="")
        if label.startswith("Device:"):
            print(', fillcolor=silver', end="")
        else:
            print(', fillcolor=lightgreen', end="")
        print(']')
 
 
nodeIds = {}
for source in moduleConnections.keys():
    defineNode(source)
    for target in moduleConnections[source]:
        defineNode(target)
 
 
for source in moduleConnections.keys():
    if source is None:
        continue
    print(nodeIds[source] + ' -> { ', end="")
    for target in moduleConnections[source]:
        print(nodeIds[target] + ' ', end="")
    print('}')
 
print('}')