mirror of
https://github.com/openmc-dev/openmc.git
synced 2026-07-25 12:35:29 -04:00
178 lines
6.7 KiB
Python
178 lines
6.7 KiB
Python
from openmc.filter import *
|
|
from openmc.filter_expansion import *
|
|
from openmc import RegularMesh, Tally
|
|
|
|
from tests.testing_harness import HashedPyAPITestHarness
|
|
|
|
|
|
def test_tallies():
|
|
harness = HashedPyAPITestHarness('statepoint.5.h5')
|
|
model = harness._model
|
|
|
|
# Set settings explicitly
|
|
model.settings.batches = 5
|
|
model.settings.inactive = 0
|
|
model.settings.particles = 400
|
|
model.settings.source = openmc.IndependentSource(space=openmc.stats.Box(
|
|
[-160, -160, -183], [160, 160, 183]))
|
|
|
|
azimuthal_bins = (-3.14159, -1.8850, -0.6283, 0.6283, 1.8850, 3.14159)
|
|
azimuthal_filter = AzimuthalFilter(azimuthal_bins)
|
|
azimuthal_tally1 = Tally()
|
|
azimuthal_tally1.filters = [azimuthal_filter]
|
|
azimuthal_tally1.scores = ['flux']
|
|
azimuthal_tally1.estimator = 'tracklength'
|
|
|
|
azimuthal_tally2 = Tally()
|
|
azimuthal_tally2.filters = [azimuthal_filter]
|
|
azimuthal_tally2.scores = ['flux']
|
|
azimuthal_tally2.estimator = 'analog'
|
|
|
|
mesh_2x2 = RegularMesh(mesh_id=1)
|
|
mesh_2x2.lower_left = [-182.07, -182.07]
|
|
mesh_2x2.upper_right = [182.07, 182.07]
|
|
mesh_2x2.dimension = [2, 2]
|
|
mesh_filter = MeshFilter(mesh_2x2)
|
|
azimuthal_tally3 = Tally()
|
|
azimuthal_tally3.filters = [azimuthal_filter, mesh_filter]
|
|
azimuthal_tally3.scores = ['flux']
|
|
azimuthal_tally3.estimator = 'tracklength'
|
|
|
|
cellborn_tally = Tally()
|
|
cellborn_tally.filters = [
|
|
CellBornFilter((model.geometry.get_all_cells()[10],
|
|
model.geometry.get_all_cells()[21],
|
|
22, 23))] # Test both Cell objects and ids
|
|
cellborn_tally.scores = ['total']
|
|
|
|
dg_tally = Tally()
|
|
dg_tally.filters = [DelayedGroupFilter((1, 2, 3, 4, 5, 6))]
|
|
dg_tally.scores = ['delayed-nu-fission', 'decay-rate']
|
|
dg_tally.nuclides = ['U235', 'O16', 'total']
|
|
|
|
four_groups = (0.0, 0.253, 1.0e3, 1.0e6, 20.0e6)
|
|
energy_filter = EnergyFilter(four_groups)
|
|
energy_tally = Tally()
|
|
energy_tally.filters = [energy_filter]
|
|
energy_tally.scores = ['total']
|
|
|
|
energyout_filter = EnergyoutFilter(four_groups)
|
|
energyout_tally = Tally()
|
|
energyout_tally.filters = [energyout_filter]
|
|
energyout_tally.scores = ['scatter']
|
|
|
|
transfer_tally = Tally()
|
|
transfer_tally.filters = [energy_filter, energyout_filter]
|
|
transfer_tally.scores = ['scatter', 'nu-fission']
|
|
|
|
material_tally = Tally()
|
|
material_tally.filters = [
|
|
MaterialFilter((model.geometry.get_materials_by_name('UOX fuel')[0],
|
|
model.geometry.get_materials_by_name('Zircaloy')[0],
|
|
3, 4))] # Test both Material objects and ids
|
|
material_tally.scores = ['total']
|
|
|
|
mu_bins = (-1.0, -0.5, 0.0, 0.5, 1.0)
|
|
mu_filter = MuFilter(mu_bins)
|
|
mu_tally1 = Tally()
|
|
mu_tally1.filters = [mu_filter]
|
|
mu_tally1.scores = ['scatter', 'nu-scatter']
|
|
|
|
mu_tally2 = Tally()
|
|
mu_tally2.filters = [mu_filter, mesh_filter]
|
|
mu_tally2.scores = ['scatter', 'nu-scatter']
|
|
|
|
polar_bins = (0.0, 0.6283, 1.2566, 1.8850, 2.5132, 3.14159)
|
|
polar_filter = PolarFilter(polar_bins)
|
|
polar_tally1 = Tally()
|
|
polar_tally1.filters = [polar_filter]
|
|
polar_tally1.scores = ['flux']
|
|
polar_tally1.estimator = 'tracklength'
|
|
|
|
polar_tally2 = Tally()
|
|
polar_tally2.filters = [polar_filter]
|
|
polar_tally2.scores = ['flux']
|
|
polar_tally2.estimator = 'analog'
|
|
|
|
polar_tally3 = Tally()
|
|
polar_tally3.filters = [polar_filter, mesh_filter]
|
|
polar_tally3.scores = ['flux']
|
|
polar_tally3.estimator = 'tracklength'
|
|
|
|
legendre_filter = LegendreFilter(order=4)
|
|
legendre_tally = Tally()
|
|
legendre_tally.filters = [legendre_filter]
|
|
legendre_tally.scores = ['scatter', 'nu-scatter']
|
|
legendre_tally.estimator = 'analog'
|
|
|
|
harmonics_filter = SphericalHarmonicsFilter(order=4)
|
|
harmonics_tally = Tally()
|
|
harmonics_tally.filters = [harmonics_filter]
|
|
harmonics_tally.scores = ['scatter', 'nu-scatter', 'flux', 'total']
|
|
harmonics_tally.estimator = 'analog'
|
|
|
|
harmonics_tally2 = Tally()
|
|
harmonics_tally2.filters = [harmonics_filter]
|
|
harmonics_tally2.scores = ['flux', 'total']
|
|
harmonics_tally2.estimator = 'collision'
|
|
|
|
harmonics_tally3 = Tally()
|
|
harmonics_tally3.filters = [harmonics_filter]
|
|
harmonics_tally3.scores = ['flux', 'total']
|
|
harmonics_tally3.estimator = 'tracklength'
|
|
|
|
universe_tally = Tally()
|
|
universe_tally.filters = [
|
|
UniverseFilter((model.geometry.get_all_universes()[1],
|
|
model.geometry.get_all_universes()[2],
|
|
3, 4, 6, 8))] # Test both Universe objects and ids
|
|
universe_tally.scores = ['total']
|
|
|
|
cell_filter = CellFilter((model.geometry.get_all_cells()[10],
|
|
model.geometry.get_all_cells()[21],
|
|
22, 23, 60)) # Test both Cell objects and ids
|
|
score_tallies = [Tally() for i in range(6)]
|
|
for t in score_tallies:
|
|
t.filters = [cell_filter]
|
|
t.scores = ['absorption', 'delayed-nu-fission', 'events', 'fission',
|
|
'inverse-velocity', 'kappa-fission', '(n,2n)', '(n,n1)',
|
|
'(n,gamma)', 'nu-fission', 'scatter', 'elastic',
|
|
'total', 'prompt-nu-fission', 'fission-q-prompt',
|
|
'fission-q-recoverable', 'decay-rate']
|
|
for t in score_tallies[0:2]: t.estimator = 'tracklength'
|
|
for t in score_tallies[2:4]: t.estimator = 'analog'
|
|
for t in score_tallies[4:6]: t.estimator = 'collision'
|
|
for t in score_tallies[1::2]:
|
|
t.nuclides = ['U235', 'O16', 'total']
|
|
|
|
cell_filter2 = CellFilter((21, 22, 23, 27, 28, 29, 60))
|
|
flux_tallies = [Tally() for i in range(3)]
|
|
for t in flux_tallies:
|
|
t.filters = [cell_filter2]
|
|
t.scores = ['flux']
|
|
flux_tallies[0].estimator = 'tracklength'
|
|
flux_tallies[1].estimator = 'analog'
|
|
flux_tallies[2].estimator = 'collision'
|
|
|
|
fusion_tally = Tally()
|
|
fusion_tally.scores = ['H1-production', 'H2-production', 'H3-production',
|
|
'He3-production', 'He4-production', 'heating', 'damage-energy']
|
|
|
|
n_collision = (1, 2, 5, 3, 6)
|
|
collision_filter = CollisionFilter(n_collision)
|
|
collision_tally = Tally()
|
|
collision_tally.filters = [collision_filter]
|
|
collision_tally.scores = ['scatter']
|
|
|
|
model.tallies += [
|
|
azimuthal_tally1, azimuthal_tally2, azimuthal_tally3,
|
|
cellborn_tally, dg_tally, energy_tally, energyout_tally,
|
|
transfer_tally, material_tally, mu_tally1, mu_tally2,
|
|
polar_tally1, polar_tally2, polar_tally3, legendre_tally,
|
|
harmonics_tally, harmonics_tally2, harmonics_tally3,
|
|
universe_tally, collision_tally]
|
|
model.tallies += score_tallies
|
|
model.tallies += flux_tallies
|
|
model.tallies.append(fusion_tally)
|
|
|
|
harness.main()
|