crp/OpenMC
1
0
Fork 0
mirror of https://github.com/openmc-dev/openmc.git synced 2026-07-28 22:26:08 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

TTB implementation fixes

Browse source
This commit is contained in:
amandalund 2018-02-19 22:27:47 -06:00
parent 62985f6198
commit 150bedcb50
8 changed files with 136 additions and 55 deletions
Download patch file Download diff file Expand all files Collapse all files

9
openmc/filter.py
View file

@ -565,7 +565,7 @@ class ParticleFilter(WithIDFilter):
Attributes
----------
bins : Iterable of Integral
openmc.Materi IDs.
openmc.Material IDs.
id : int
Unique identifier for the filter
num_bins : Integral
@ -582,7 +582,12 @@ class ParticleFilter(WithIDFilter):
@bins.setter
def bins(self, bins):
bins = np.atleast_1d(bins)
cv.check_iterable_type('filter bins', bins, str)
cv.check_iterable_type('filter bins', bins, (Integral, str))
for edge in bins:
if isinstance(edge, Integral):
cv.check_value('filter bin', edge, _PARTICLE_IDS.values())
else:
cv.check_value('filter bin', edge, _PARTICLE_IDS.keys())
bins = np.atleast_1d([b if isinstance(b, Integral) else _PARTICLE_IDS[b]
for b in bins])
self._bins = bins