/
cyclus2fco.py
executable file
·172 lines (145 loc) · 5.94 KB
/
cyclus2fco.py
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
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
#!/usr/bin/env python
from __future__ import print_function, unicode_literals
import os
import sys
import subprocess
import numpy as np
import cymetric as cym
global r_power_E
global r_deployed
global r_flow
global c_inv
global wr_inv
global mass_conversion
fco_sheet = ['Data-Front End', 'Data-Reactor', 'Data-Recycle', 'Data-Inventories', 'Data-Economics',
'Data-Wildcard', 'Graph_Data', 'Settings', 'Pres_Graphs', 'Graphs (C1-4)', 'Version Notes']
# Half-life > 1h
nucs_Pu_list = ["238pu", "239pu", "240pu", "241pu", "242pu"]
nucs_U = "230U,231U,232U,233U,234U,235U,236U,237U,238U,240U"
nucs_Pu = "234Pu,235Pu,236Pu,237Pu,238Pu,239Pu,240Pu,241Pu,242Pu,243Pu,244Pu,245Pu,246Pu,247Pu"
nucs_Am = "239Am,240Am,241Am,242Am,243Am,244Am,245Am"
nucs_Cm = "238Cm,240Cm,241Cm,242Cm,243Cm,244Cm,245Cm,246Cm,247Cm,248Cm,250Cm"
nucs_Np = "234Np,235Np,236Np,237Np,238Np,239Np"
nucs_FP = "H3,GE74,AS75,GE76,SE77,SE78,SE79,SE80,BR81,SE82,KR82,KR83,KR84,KR85,RB85,KR86,SR86,RB87,SR88,SR89,Y89,SR90,ZR90,SR91,Y91,ZR91,ZR92,Y93,ZR93,ZR94,ZR95,NB95,MO95,ZR96,MO96,ZR97,MO97,MO98,MO99,TC99,TC99M,MO100,RU100,RU101,RU102,RU103,RH103,RU104,PD104,RU105,RH105,PD105,RU106,PD106,PD107,PD108,PD109,AG109,PD110,AG110,CD110,AG111,CD111,CD112,CD113,CD113,CD114,IN115,SN115,CD116,SN116,SN117,SN118,SN119,SN120,SB121,SN122,TE122,SN123,SB123,SN124,SB124,TE124,SN125,SB125,TE125,TE125,SN126,TE126,SB127,TE127,I127,TE128,XE128,TE129,I129,TE130,XE130,TE131,I131,XE131,XE131,TE132,I132,XE132,I133,XE133,XE133,CS133,XE134,CS134,BA134,I135,XE135,CS135,BA135,XE136,CS136,BA136,CS137,BA137,BA138,LA138,LA139,BA140,LA140,CE140,LA141,CE141,PR141,CE142,ND142,CE143,PR143,ND143,CE144,ND144,PR145,ND145,ND146,ND147,PM147,SM147,ND148,PM148,SM148,PM149,SM149,ND150,SM150,PM151,SM151,SM152,SM153,EU153,SM154,EU154,GD154,EU155,GD155,EU156,GD156,GD157,GD158,TB159,GD160,TB160,DY160,DY161,DY162,DY163,DY164,HO165,ER166"
nucs_MA = nucs_Am + "," + nucs_Cm + "," + nucs_Np
def push_in_excel(matrix, sheet, column, start_row):
for i in range(len(matrix)):
cmd = "sed 's/\(r=\"" + column + str(start_row + i)
cmd += "\" .......<v>\)[^<]*/\\1"
cmd += str(matrix[i]) + "/' _tmp/xl/worksheets/" + \
sheet + " > _tmp/xl/worksheets/sheet_tmp.xml"
os.system(cmd)
os.system("mv _tmp/xl/worksheets/sheet_tmp.xml _tmp/xl/worksheets/" + sheet)
def save_xslm(filename):
cmd = 'cd _tmp; zip -r ../' + filename + ' *'
os.system(cmd + ' >/dev/null')
cmd = 'rm -rf _tmp'
os.system(cmd)
def open_xslm(filename):
cmd = 'unzip -o ' + filename + ' -d _tmp/'
output = subprocess.check_output(cmd.split())
def month2year(vals, mode=0, division=12):
out_val = []
val = 0
for i in range(len(vals)):
if mode == 0:
val = vals[i]
else:
val += vals[i]
if i % division == 0:
if mode == 2:
val *= 1. / double(division)
out_val.append(val)
val = 0
return out_val
def get_timestep():
timestep = get_val(ev, ['time'])
timestep = month2year(timestep, 0) / 12.
return timestep
def get_val(evaler, param):
key = param[2]
# clean empty parameters
for i in range(len(param))[3:]:
param[i] = param[i].split(',')
if param[i] == ['']:
param[i][:] = []
fac1 = param[3]
fac2 = param[4]
commods = param[5]
nucs = param[6]
if key == "inv":
val = cym.timeseries.inventories(evaler, facilities=fac1, nucs=nucs)
val = val['Mass'].copy() / mass_conversion
val = month2year(val.tolist(), 0, time_conversion)
elif key == "trans":
val = cym.timeseries.transactions(evaler, senders=fac1, receivers=fac2,
commodities=commods, nucs=nucs)
val = val['Mass'].copy() / mass_conversion
val = month2year(val.tolist(), 1, time_conversion)
elif key == "power":
val = cym.timeseries.get_power(evaler, facilities=fac1)
val = val['Value'].copy() / power_conversion
val = month2year(val.tolist(), 0, time_conversion)
elif key == "time":
val = evaler.eval('TimeList').copy() / time_conversion
val = val['Time'].tolist()
val = month2year(val, 0, time_conversion)
return val
def exec_line(evaler, line):
param = line.split(':')
if len(param) != 7:
print("Bad line: ", line)
quit()
vals = get_val(evaler, param)
push_in_excel(vals, param[0], param[1], start_row)
def process(evaler, file):
global start_row
global time_conversion
global mass_conversion
global power_conversion
template = 'template.xls'
outfile = 'outffile.xls'
time_conversion = 1.
mass_conversion = 1.
power_conversion = 1.
for line in file:
line = line.rstrip('\r\n')
if len(line) != 0:
if line[0] == '%':
param, val = line[1:].split(":", 1)
if param == 'template':
template = val
elif param == 'outfile':
outfile = val
elif param == 'startrow':
start_row = int(val)
elif param == 'time':
time_conversion = float(val)
elif param == 'mass':
mass_conversion = float(val)
elif param == 'power':
power_conversion = float(val)
open_xslm(template)
for line in file:
line = line.rstrip('\r\n')
if len(line) != 0:
if line[0] != '%' and line[0] != '#':
exec_line(evaler, line)
save_xslm(outfile)
def read_input(input):
f = open(input, 'r')
matrix = []
for line in f:
matrix.append(line)
return matrix
def main():
global timestep
if len(sys.argv) != 3:
print("missing argument !!")
quit()
db = cym.dbopen(sys.argv[2])
ev = cym.Evaluator(db=db, write=False)
input_list = read_input(sys.argv[1])
process(ev, input_list)
if __name__ == '__main__':
main()