[1206] | 1 | # Usage:
|
---|
| 2 | #
|
---|
| 3 | # $ find . -name wleiden.conf > l
|
---|
| 4 | #
|
---|
| 5 | # prune demo, test and uninteresting stuff from l
|
---|
| 6 | #
|
---|
| 7 | # $ python channelplot.py
|
---|
| 8 | # $ gv channelplot.ps
|
---|
| 9 | #
|
---|
| 10 | # needs graphviz (http://www.graphviz.org/) to run
|
---|
| 11 | import os
|
---|
| 12 | import re
|
---|
| 13 | import string
|
---|
| 14 |
|
---|
| 15 | ipdict = {}
|
---|
| 16 | ipnr_node = {}
|
---|
| 17 | node_ipnr = {}
|
---|
| 18 | node_wi_channel = {}
|
---|
| 19 | node_links = {}
|
---|
| 20 | node_links_channel = {}
|
---|
| 21 | nodelabel = {}
|
---|
| 22 |
|
---|
| 23 | def parse_wleiden_conf(lines):
|
---|
| 24 | ipnr = None
|
---|
| 25 | wi = None
|
---|
| 26 | peer = None
|
---|
| 27 | omni = 0
|
---|
| 28 | for l in lines:
|
---|
| 29 | if wi != None:
|
---|
| 30 | match = re.match("^\EW.*", l)
|
---|
| 31 | if match != None:
|
---|
| 32 | wi = None
|
---|
| 33 | ipnr = None
|
---|
| 34 | peer = None
|
---|
| 35 | continue
|
---|
| 36 | match = re.match("^SDESC=omni.*", l)
|
---|
| 37 | if match != None:
|
---|
| 38 | omni = 1
|
---|
| 39 | match = re.match("^IP=([0-9]*\.[0-9]*\.[0-9]*\.[0-9]*).*", l)
|
---|
| 40 | if match != None:
|
---|
| 41 | ipnr = match.group(1)
|
---|
| 42 | if not node_ipnr.has_key(nodename):
|
---|
| 43 | node_ipnr[nodename] = []
|
---|
| 44 | node_ipnr[nodename].append(ipnr)
|
---|
| 45 | ipnr_node[ipnr] = nodename
|
---|
| 46 | match = re.match("^POINT_TO_POINT=([0-9]*\.[0-9]*\.[0-9]*\.[0-9]*).*", l)
|
---|
| 47 | if match != None:
|
---|
| 48 | peer = match.group(1)
|
---|
| 49 | if not node_links.has_key(nodename):
|
---|
| 50 | node_links[nodename] = []
|
---|
| 51 | node_links[nodename].append( (peer, wi) )
|
---|
| 52 | match = re.match("^CHANNEL=([0-9]*).*", l)
|
---|
| 53 | if match != None:
|
---|
| 54 | if omni == 1:
|
---|
| 55 | nodelabel[nodename] = "%s\\n(%s)" % (nodename, match.group(1) )
|
---|
| 56 | else:
|
---|
| 57 | node_wi_channel[(nodename, wi)] = match.group(1)
|
---|
| 58 | else:
|
---|
| 59 | match = re.match("^\$nodename='([^']*)';.*", l)
|
---|
| 60 | if match != None:
|
---|
| 61 | nodename = match.group(1).lower()
|
---|
| 62 | match = re.match("^\$config{'(wi[0-9]*).*", l)
|
---|
| 63 | if match != None:
|
---|
| 64 | wi = match.group(1)
|
---|
| 65 | omni = 0
|
---|
| 66 |
|
---|
| 67 | for fname in open("l").readlines():
|
---|
| 68 | parse_wleiden_conf(open(fname[:-1]).readlines())
|
---|
| 69 |
|
---|
| 70 | out = open("foo.dot", "w")
|
---|
| 71 | out.write("digraph channelplot {\n")
|
---|
| 72 | for node in node_ipnr.keys():
|
---|
| 73 | if nodelabel.has_key(node):
|
---|
| 74 | label = "[label=\"%s\"]" % nodelabel[node]
|
---|
| 75 | else:
|
---|
| 76 | label = ''
|
---|
| 77 | out.write("\t%s %s\n" % (node, label))
|
---|
| 78 | for node in node_ipnr.keys():
|
---|
| 79 | if node_links.has_key(node):
|
---|
| 80 | for peer, wi in node_links[node]:
|
---|
| 81 | if ipnr_node.has_key(peer):
|
---|
| 82 | if node_wi_channel.has_key( (node,wi) ):
|
---|
| 83 | label = "[label=\"%s\"]" % node_wi_channel[ (node, wi) ]
|
---|
| 84 | else:
|
---|
| 85 | label = ''
|
---|
| 86 | out.write("%s -> %s%s\n" % (node, ipnr_node[peer], label))
|
---|
| 87 | else:
|
---|
| 88 | # link naar onbekend iemand, zal wel niet
|
---|
| 89 | # belangrijk zijn.
|
---|
| 90 | pass
|
---|
| 91 | out.write("}\n")
|
---|
| 92 | out.close()
|
---|
[1938] | 93 | os.system('neato -Tjpg -Gstart=foo -Goverlap=false/scale -Gsplines=true -Gsep=2 -Gratio=fill -Gnslimit=50.0 -Grotate=90 -o channelplot.jpg foo.dot')
|
---|
[1206] | 94 |
|
---|