GridAttackSim/glmMap.py at master · crond-jaist/GridAttackSim · GitHub

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
110
111
112
113
114
#!/usr/bin/python

#Pre-requisites software required to run this script:
    #Python 2.7: Modules:: sys, re, os
    #Graphviz
        ##If you just installed Graphviz, "restart" your computer for Graphviz commandline execution to work

#Purpose of this script and How can a modeler use it:
    #Convert a GridLAB-D file (.glm) to a Graphviz file (.dot)
    #Use that .dot to generate .svg
    #Use any web browser to open the .svg file - SIMPLE

#Usage:
    #python glmMap.py inputfile.glm outputfile.dot

#Licensing: Copyright 2016; licensing falls under GridLAB-D
#Email Contact: sri at pnnl dot gov
#Devloped at: Pacific Northwest National Laboratory
#Acknowledgements: This tool is developed using a base script provided by Philip Douglass, Technical University of Denmark in 2013

#Version and Patch Notes:
    ##05/20/2016: Added    :: line spacing; object names; object color coding
	##06/02/2016: Bug Fix  :: If the node names has ':' in it, grapgviz errors out. replaced ':' with '_' as well

import sys;
import re;
import os

infile = open(sys.argv[1], 'r')
outfile = open(sys.argv[2], 'w')

lines = infile.readlines()

# .dot files begin with the 'graph' keyword.
outfile.write("digraph {\n")
outfile.write("node [shape=box]\n")
# These are state variables.  There is no error checking, so we rely on
# well formatted *.GLM files.
s =0
state = 'start'
edge_color = 'black'
edge_style = 'solid'
lengthVal = 'None'
lineLengthIncrement = 0

# Loop through each line in the input file...
while s < len(lines):
    # Discard Comments
    if re.match("//", lines[s]) == None:
        if re.search("from", lines[s])!=None:
            ts = lines[s].split()
            #Graphvis format can't handle '-' characters, so they are converted to '_'
            ns = ts[1].rstrip(';').replace('-','_').replace(':','_')
            outfile.write(ns)
            state = 'after_from'
        elif state == 'after_from' and re.search("to ", lines[s])!=None:
            ts = lines[s].split()
            ns = ts[1].rstrip(';').replace('-','_').replace(':','_')
            if edge_color == 'red':
                outfile.write(' -> ' + ns + '[style=' + edge_style + ' color='+ edge_color + ' label="'+lengthVal+'"]\n')
                outfile.write("node [shape=box]\n")
            else:
                outfile.write(' -> ' + ns + '[style=' + edge_style + ' color='+ edge_color + ' label="'+lengthVal+'"]\n')
            lengthVal = 'None'
            # After an edge is added to the graph, reset the states back to default
            state = 'start'
            edge_color = 'black'
            edge_style = 'solid'
        elif (re.search("object underground_line", lines[s]) != None) or (re.search("object overhead_line", lines[s]) != None):
            while '}' not in lines[s+lineLengthIncrement]:
                if re.search("length ", lines[s+lineLengthIncrement]) != None:
                    les = lines[s+lineLengthIncrement].split()
                    if len(les) > 2:
                        tsVal = les[1]+' '+les[2].strip(';')
                        lengthVal = tsVal
                    elif len(les)<=2:
                        tsVal = les[1].strip(';')
                        lengthVal = tsVal
                    break
                else:
                    lineLengthIncrement = lineLengthIncrement+1
                if '}' in lines[s+lineLengthIncrement]:
                    lineLengthIncrement = 0
                    lengthVal = 'None'
                    break
            if re.search("object underground_line", lines[s]) != None:
                lengthVal = "UG_line\\n"+lengthVal
            elif re.search("object overhead_line", lines[s]) != None:
                lengthVal = "OH_line\\n"+lengthVal
        elif re.search("object transformer", lines[s]) != None:
            edge_color='red'
            lengthVal = "transformer\\n"+lengthVal
            outfile.write("node [shape=oval]\n")
        elif re.search("object triplex_line", lines[s]) != None:
            edge_color='green'
            lengthVal = "triplex_line\\n"+lengthVal
        elif re.search("object fuse", lines[s]) != None:
            edge_color='blue'
            lengthVal = "Fuse\\n"+lengthVal
        elif re.search("phases ", lines[s]) != None:
            ts = lines[s].split()
            if len(ts[1].rstrip(';')) > 3:
                edge_style = 'bold'
            elif len(ts[1].rstrip(';')) > 2:
                edge_style = 'dashed'

    s+=1


outfile.write("}\n")
infile.close()
outfile.close()
os.system("dot -Tsvg -O "+sys.argv[2])