crp/MSRE
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

remove dependency to onion

Browse source
This commit is contained in:
church89 2023-06-09 15:16:50 +02:00
parent cb0f0eaee2
commit 72a07a7d24
1 changed files with 16 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

18
msre_power_history_restart.py
View file

@ -10,9 +10,23 @@ import pandas as pd
import math
import numpy as np
import re
import onion
from pathlib import Path
def get_geom_level_from_res(depletion_path, index):
if depletion_path is None:
depletion_path = Path(os.getcwd())
with h5py.File(depletion_path / 'msr_results.h5','r') as f:
items = {}
for key in f.keys():
if re.split(r'_', key)[0] == 'geometry':
items[int(re.split(r'_', key)[1])] = np.array(f.get(key))
items = OrderedDict(sorted(items.items()))
res = list(items.items())[index][1].mean()
return res
salt_temp = 648.9
salt = openmc.Material(name="salt", temperature = salt_temp + 273.15)
salt.add_nuclide('Li6',1.31480070E-05)
@ -293,7 +307,7 @@ model.plots.append(plot)
depletion_path = Path(os.getcwd())
cell = model.geometry.get_cells_by_name('CR1')[0]
last_level = onion.utils.get_geom_level_from_res(depletion_path, index=-1)
last_level = get_geom_level_from_res(depletion_path, index=-1)
setattr(cell, 'translation', [0, 0, last_level])
res_path = depletion_path / 'depletion_results.h5'