Power-Flow-Analyzer-with-Network-Modification/parser.py at main · isha240724/Power-Flow-Analyzer-with-Network-Modification · 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
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
class ParsedData:

    def __init__(self):
        self.buses = []
        self.loads = []
        self.generators = []
        self.branches = []
        self.bus_names = {}

        self.bus_voltage = {}
        self.bus_angle = {}

        self.transformers = []

def parse_raw(lines):

    data = ParsedData()

    # ---------------- BUS ----------------
    for line in lines[2:]:

        if "END OF BUS DATA" in line.upper():
            break

        parts = line.split(",")

        try:

            bus_no = int(parts[0].strip())
            ide = int(parts[3].strip())

            bus_name = parts[1].replace("'", "").strip()

            vm = float(parts[7].strip())
            va = float(parts[8].strip())

            data.buses.append(bus_no)

            data.bus_names[bus_no] = bus_name
            data.bus_voltage[bus_no] = vm
            data.bus_angle[bus_no] = va

            if not hasattr(data, "bus_types"):
                data.bus_types = {}

            data.bus_types[bus_no] = ide

        except:
            pass
    # ---------------- LOAD ----------------
    load_start = None
    load_end = None

    for i, line in enumerate(lines):

        if "END OF BUS DATA" in line.upper():
            load_start = i + 1

        if "END OF LOAD DATA" in line.upper():
            load_end = i
            break

    for line in lines[load_start:load_end]:

        parts = line.split(",")

        try:
            bus_no = int(parts[0].strip())
            data.loads.append(bus_no)

        except:
            pass

    # ---------------- GENERATOR ----------------
    gen_start = None
    gen_end = None

    for i, line in enumerate(lines):

        if "END OF LOAD DATA" in line.upper():
            gen_start = i + 1

        if "END OF GENERATOR DATA" in line.upper():
            gen_end = i
            break

    for line in lines[gen_start:gen_end]:

        parts = line.split(",")

        try:
            bus_no = int(parts[0].strip())
            data.generators.append(bus_no)

        except:
            pass

    # ---------------- BRANCH ----------------
    branch_start = None
    branch_end = None

    for i, line in enumerate(lines):

        if "END OF GENERATOR DATA" in line.upper():
            branch_start = i + 1

        if "END OF BRANCH DATA" in line.upper():
            branch_end = i
            break

    for line in lines[branch_start:branch_end]:

        parts = line.split(",")

        try:
            from_bus = int(parts[0].strip())
            to_bus = int(parts[1].strip())

            r = float(parts[3].strip())
            x = float(parts[4].strip())

            data.branches.append(
                (from_bus, to_bus, r, x)
            )
        except:
            pass
    # ---------------- TRANSFORMER ----------------
    trafo_start = None
    trafo_end = None

    for i, line in enumerate(lines):

        if "END OF BRANCH DATA" in line.upper():
            trafo_start = i + 1

        if "END OF TRANSFORMER DATA" in line.upper():
            trafo_end = i
            break

    if trafo_start and trafo_end:

        for line in lines[trafo_start:trafo_end]:

            parts = line.split(",")

            try:
                from_bus = int(parts[0].strip())
                to_bus = int(parts[1].strip())

                data.transformers.append((from_bus, to_bus))

            except:
                pass
    return data