Skip to content

Speed up depletion with transfer rates #3839

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
GuySten wants to merge 2 commits into
base: develop
Choose a base branch
from
+21 −29
Open

Speed up depletion with transfer rates #3839

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
df26e43
simplify form_rr_term
GuySten Feb 26, 2026
297a928
fix missing import
GuySten Feb 26, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
50 changes: 21 additions & 29 deletions openmc/deplete/chain.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@

import lxml.etree as ET

from openmc.checkvalue import check_type, check_greater_than, PathLike
from openmc.checkvalue import check_type, check_length, check_greater_than, PathLike
from openmc.data import gnds_name, zam
from openmc.exceptions import DataError
from .nuclide import FissionYieldDistribution, Nuclide
Expand Down Expand Up @@ -807,37 +807,29 @@ def form_rr_term(self, tr_rates, current_timestep, mats):
# Use DOK as intermediate representation
n = len(self)
matrix = dok_array((n, n))

check_type("mats", mats, (tuple, str))
if not isinstance(mats, str):
check_type("mats", mats, tuple, str)
check_length("mats", mats, 2, 2)

# Build transfer terms (nuclide transfer only)
if isinstance(mats, str):
mat = mats
dest_mat = None
components = tr_rates.get_components(mat, current_timestep)
# Build transfer terms (transfer from one material into another)
elif isinstance(mats, tuple):
dest_mat, mat = mats
components = tr_rates.get_components(mat, current_timestep, dest_mat)

for i, nuc in enumerate(self.nuclides):
elm = re.split(r'\d+', nuc.name)[0]
# Build transfer terms (nuclide transfer only)
if isinstance(mats, str):
mat = mats
components = tr_rates.get_components(mat, current_timestep)
if not components:
break
if elm in components:
matrix[i, i] = sum(
tr_rates.get_external_rate(mat, elm, current_timestep))
elif nuc.name in components:
matrix[i, i] = sum(
tr_rates.get_external_rate(mat, nuc.name, current_timestep))
else:
matrix[i, i] = 0.0

# Build transfer terms (transfer from one material into another)
elif isinstance(mats, tuple):
dest_mat, mat = mats
components = tr_rates.get_components(mat, current_timestep, dest_mat)
if elm in components:
matrix[i, i] = tr_rates.get_external_rate(
mat, elm, current_timestep, dest_mat)[0]
elif nuc.name in components:
matrix[i, i] = tr_rates.get_external_rate(
mat, nuc.name, current_timestep, dest_mat)[0]
else:
matrix[i, i] = 0.0

if elm in components:
matrix[i, i] = sum(tr_rates.get_external_rate(mat, elm, current_timestep, dest_mat))
elif nuc.name in components:
matrix[i, i] = sum(tr_rates.get_external_rate(mat, nuc.name, current_timestep, dest_mat))

# Return CSC instead of DOK
return matrix.tocsc()

Expand Down
Loading