Newer
Older
Martin Christoph Hierholzer
committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
#!/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 != None :
print("ERROR: Found two feeders!")
print(node.attrib['name'])
sys.exit(1)
feeder = peerName
if feeder == None :
print("ERROR: No feeders found!")
print(node.attrib['name'])
sys.exit(1)
for m in moduleList :
if not m in moduleConnections.keys():
moduleConnections[m] = set()
if not feeder 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 not label 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 == None :
continue
print(nodeIds[source] + ' -> { ', end="")
for target in moduleConnections[source] :
print(nodeIds[target]+' ', end="")
print('}')
print('}')