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

Optimization for complete list of bin indices

Browse source
This commit is contained in:
Patrick Shriwise 2022-02-11 21:29:37 -06:00
parent c8639e1630
commit ccec9ee7bf
1 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
openmc/tallies.py
View file

@ -1110,7 +1110,11 @@ class Tally(IDManagerMixin):
bins = self_filter.bins
# Add indices for each bin in this Filter to the list
indices = np.array([self_filter.get_bin_index(b) for b in bins])
if type(self_filter) in filters:
indices = np.array([self_filter.get_bin_index(b) for b in bins])
else:
indices = np.arange(len(bins))
filter_indices.append(indices)
# Account for stride in each of the previous filters