crp/OpenMC
1
0
Fork 0
mirror of https://github.com/openmc-dev/openmc.git synced 2026-07-27 21:55:41 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

Add a dictionary to map local to global burnable materials

Browse source 
Removes the need to call self.burnable_mats.index
for every local material at every depletion step
This commit is contained in:
Andrew Johnson 2019-07-18 13:13:13 -05:00
parent 15c2bc2de3
commit d0d4a05c75
No known key found for this signature in database
GPG key ID: 253418E91B7F6FEB
1 changed files with 10 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

13
openmc/deplete/operator.py
View file

@ -138,6 +138,7 @@ class Operator(TransportOperator):
openmc.reset_auto_ids()
self.burnable_mats, volume, nuclides = self._get_burnable_mats()
self.local_mats = _distribute(self.burnable_mats)
self._mat_index_map = {}
# Determine which nuclides have incident neutron data
self.nuclides_with_data = self._get_nuclides_with_data()
@ -386,6 +387,10 @@ class Operator(TransportOperator):
self._energy_helper.prepare(
self.chain.nuclides, self.reaction_rates.index_nuc, materials)
# Generate map from local materials => material index
self._mat_index_map = {
lm: self.burnable_mats.index(lm) for lm in self.local_mats}
# Return number density vector
return list(self.number.get_mat_slice(np.s_[:]))
@ -519,7 +524,6 @@ class Operator(TransportOperator):
k_combined = openmc.capi.keff()
# Extract tally bins
materials = self.burnable_mats
nuclides = self._rate_helper.nuclides
# Form fast map
@ -541,7 +545,7 @@ class Operator(TransportOperator):
# Extract results
for i, mat in enumerate(self.local_mats):
# Get tally index
slab = materials.index(mat)
mat_index = self._mat_index_map[mat]
# Zero out reaction rates and nuclide numbers
number.fill(0.0)
@ -551,7 +555,7 @@ class Operator(TransportOperator):
number[i_nuc_results] = self.number[mat, nuc]
tally_rates = self._rate_helper.get_material_rates(
slab, nuc_ind, react_ind)
mat_index, nuc_ind, react_ind)
# Accumulate energy from fission
energy += self._energy_helper.get_fission_energy(
@ -561,7 +565,10 @@ class Operator(TransportOperator):
rates[i] = self._rate_helper.divide_by_adens(number)
# Reduce energy produced from all processes
print("Energy - {}: {:9.7e}".format(comm.rank, energy))
energy = comm.allreduce(energy)
if comm.rank == 0:
print("Energy: {:9.7e}".format(energy))
# Determine power in eV/s
power /= JOULE_PER_EV