Start creating SourceRateHelper

This commit is contained in:
Paul Romano 2020-07-17 14:19:49 -05:00
parent 14fdcf9d4d
commit baecead5da
2 changed files with 14 additions and 4 deletions

View file

@ -18,8 +18,8 @@ from .abc import (
TalliedFissionYieldHelper)
__all__ = (
"DirectReactionRateHelper", "ChainFissionHelper",
"ConstantFissionYieldHelper", "FissionYieldCutoffHelper",
"DirectReactionRateHelper", "ChainFissionHelper", "EnergyScoreHelper"
"SourceRateHelper", "ConstantFissionYieldHelper", "FissionYieldCutoffHelper",
"AveragedFissionYieldHelper")
# -------------------------------------
@ -205,6 +205,14 @@ class EnergyScoreHelper(NormalizationHelper):
if comm.rank == 0:
self._energy = self._tally.results[0, 0, 1]
class SourceRateHelper(NormalizationHelper):
def prepare(self, *args, **kwargs):
pass
def factor(self, source_rate):
return source_rate
# ------------------------------------
# Helper for collapsing fission yields
# ------------------------------------

View file

@ -26,7 +26,8 @@ from .reaction_rates import ReactionRates
from .results_list import ResultsList
from .helpers import (
DirectReactionRateHelper, ChainFissionHelper, ConstantFissionYieldHelper,
FissionYieldCutoffHelper, AveragedFissionYieldHelper, EnergyScoreHelper)
FissionYieldCutoffHelper, AveragedFissionYieldHelper, EnergyScoreHelper,
SourceRateHelper)
__all__ = ["Operator", "OperatorResult"]
@ -238,13 +239,14 @@ class Operator(TransportOperator):
# Get classes to assist working with tallies
self._rate_helper = DirectReactionRateHelper(
self.reaction_rates.n_nuc, self.reaction_rates.n_react)
if normalization_mode == "fission-q":
self._normalization_helper = ChainFissionHelper()
elif normalization_mode == "energy-deposition":
score = "heating" if settings.photon_transport else "heating-local"
self._normalization_helper = EnergyScoreHelper(score)
else:
self._normalization_helper = ...
self._normalization_helper = SourceRateHelper()
# Select and create fission yield helper
fission_helper = self._fission_helpers[fission_yield_mode]