Skip to content
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.

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

Updating pin-related functionality #1990

Merged
merged 46 commits into from
Nov 22, 2024
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
46 commits
Select commit Hold shift + click to select a range
76e36f3
Moving pin flux parameters to component level
zachmprince Oct 14, 2024
80b673e
add pinPercentBu component param
albeanth Oct 15, 2024
cf7ea6c
Revert "Moving pin flux parameters to component level"
zachmprince Oct 16, 2024
87a1c2a
Adding component method to retrieve pin fluxes
zachmprince Oct 16, 2024
6b69a6b
add c.p.molesHmBOL and populate it
albeanth Oct 16, 2024
7041ee4
add c.p.puFrac calc, move b.getPuMoles up the composite tree, add a p…
albeanth Oct 18, 2024
33ec45b
add defaults to pin level params
albeanth Oct 22, 2024
98c13f5
comment out print statements
albeanth Oct 23, 2024
567253f
Merge branch 'main' into component_flux_up
albeanth Oct 23, 2024
eda57f5
Merge branch 'main' into component_flux_up
albeanth Oct 25, 2024
29f5878
make history tracker respect control assemblies
albeanth Oct 25, 2024
07e0e7f
Merge branch 'main' into component_flux_up
albeanth Oct 27, 2024
67ced79
start rm'img things
albeanth Oct 30, 2024
9072ba1
Merge branch 'main' into component_flux_up
albeanth Oct 30, 2024
829b40b
rm call to non-existent method
albeanth Oct 30, 2024
7c32499
continue rm'img things
albeanth Oct 30, 2024
5319cb7
fix black and ruff
albeanth Oct 30, 2024
1408ac6
apparently "git add ." missed this....
albeanth Oct 30, 2024
fd4e9ff
fix unit tests
albeanth Oct 31, 2024
f7bf4a5
ugh. why is black kicking my butt today
albeanth Oct 31, 2024
ac34e2b
Update armi/reactor/components/component.py
john-science Oct 31, 2024
0e2659e
Merge branch 'main' into component_flux_up
albeanth Nov 1, 2024
ba0acf3
Using one-block reactor for component flux test
zachmprince Nov 1, 2024
9f7a1e0
No need to recast strings to strings
albeanth Nov 5, 2024
937057c
org imports
albeanth Nov 5, 2024
e657ce9
resolve fixme
albeanth Nov 5, 2024
fe246d9
beef up assertions
albeanth Nov 5, 2024
bb03ca2
Addressing reviewer comments for component pin mg fluxes
zachmprince Nov 5, 2024
1f869aa
release notes
albeanth Nov 6, 2024
fb66716
Merge branch 'main' into component_flux_up
albeanth Nov 6, 2024
9f7168d
fix linting
albeanth Nov 6, 2024
0cacf6c
Merge branch 'main' into component_flux_up
albeanth Nov 12, 2024
df62010
fix unit test
albeanth Nov 14, 2024
642f0f3
mv b.p.percentBuMin internal
albeanth Nov 18, 2024
9aa6895
Apply suggestions from code review
albeanth Nov 18, 2024
8c94b32
add testing for Block.completeInitialLoading
albeanth Nov 18, 2024
9c9fbf3
add test for total block HmBOL mass
albeanth Nov 18, 2024
89514da
move pin-specific params internal
albeanth Nov 19, 2024
b128d5f
move pinPercentBu back for incoming assembly rotation work
albeanth Nov 19, 2024
648c953
Update doc/release/0.4.rst
albeanth Nov 19, 2024
441382d
Merge branch 'main' into component_flux_up
albeanth Nov 20, 2024
8cb2c93
update release notes
albeanth Nov 20, 2024
43d3dd1
Merge branch 'main' into component_flux_up
albeanth Nov 21, 2024
35adca0
add pinNDens param, numpy32bit setter, update pinNDens with changeNDe…
albeanth Nov 21, 2024
938bf83
run black
albeanth Nov 21, 2024
223e5a6
Apply suggestions from code review
albeanth Nov 21, 2024
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
2 changes: 1 addition & 1 deletion armi/bookkeeping/db/tests/test_comparedb3.py
Original file line number Diff line number Diff line change
Expand Up @@ -181,7 +181,7 @@ def test_compareDatabaseSim(self):
dbs[1]._fullPath,
timestepCompare=[(0, 0), (0, 1)],
)
self.assertEqual(len(diffs.diffs), 480)
self.assertEqual(len(diffs.diffs), 504)
# Cycle length is only diff (x3)
self.assertEqual(diffs.nDiffs(), 3)

Expand Down
54 changes: 29 additions & 25 deletions armi/bookkeeping/historyTracker.py
Original file line number Diff line number Diff line change
Expand Up @@ -67,7 +67,7 @@ def getHistoryParams(self):
See :ref:`detail-assems`.

"""
from typing import Tuple
from typing import TYPE_CHECKING
import traceback

from armi import interfaces
Expand All @@ -79,6 +79,10 @@ def getHistoryParams(self):

ORDER = 2 * interfaces.STACK_ORDER.BEFORE + interfaces.STACK_ORDER.BOOKKEEPING

if TYPE_CHECKING:
from armi.reactor.blocks import Block
from armi.reactor.assemblies import Assembly


def describeInterfaces(cs):
"""Function for exposing interface(s) to other code."""
Expand Down Expand Up @@ -120,6 +124,8 @@ class HistoryTrackerInterface(interfaces.Interface):

name = "history"

DETAILED_ASSEMBLY_FLAGS = [Flags.FUEL, Flags.CONTROL]
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Did we check with the users of this to ensure that this doesn't need to be configurable? Specifically for pin performance evaluations? @sammiller11235?

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Yes. We were told that, for now, we are only concerned with fuel and control assemblies.

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

prior to this we only got fuel.


def __init__(self, r, cs):
"""
HistoryTracker that uses the database to look up parameter history rather than
Expand All @@ -146,7 +152,7 @@ def interactBOC(self, cycle=None):
"""Look for any new assemblies that are asked for and add them to tracking."""
self.addDetailAssemsByAssemNums()
if self.cs["detailAllAssems"]:
self.addAllFuelAssems()
self.addAllDetailedAssems()

def interactEOL(self):
"""Generate the history reports."""
Expand All @@ -171,16 +177,16 @@ def addDetailAssembliesBOL(self):
self.addDetailAssembly(a)

if self.cs["detailAllAssems"]:
self.addAllFuelAssems()
self.addAllDetailedAssems()

# This also gets called at BOC but we still
# do it here for operators that do not call BOC.
self.addDetailAssemsByAssemNums()

def addAllFuelAssems(self):
"""Add all fuel assems as detail assems."""
def addAllDetailedAssems(self):
"""Add all assems who have the DETAILED_ASSEMBLY_FLAGS as detail assems."""
for a in self.r.core:
if a.hasFlags(Flags.FUEL):
if a.hasFlags(self.DETAILED_ASSEMBLY_FLAGS):
self.addDetailAssembly(a)

def addDetailAssemsByAssemNums(self):
Expand Down Expand Up @@ -236,13 +242,13 @@ def getTrackedParams(self):
trackedParams.add(newParam)
return sorted(trackedParams)

def addDetailAssembly(self, a):
def addDetailAssembly(self, a: "Assembly"):
"""Track the name of assemblies that are flagged for detailed treatment."""
aName = a.getName()
if aName not in self.detailAssemblyNames:
self.detailAssemblyNames.append(aName)

def getDetailAssemblies(self):
def getDetailAssemblies(self) -> list["Assembly"]:
"""Returns the assemblies that have been signaled as detail assemblies."""
assems = []
if not self.detailAssemblyNames:
Expand All @@ -259,7 +265,7 @@ def getDetailAssemblies(self):
)
return assems

def getDetailBlocks(self):
def getDetailBlocks(self) -> list["Block"]:
"""Get all blocks in all detail assemblies."""
return [block for a in self.getDetailAssemblies() for block in a]

Expand All @@ -281,7 +287,7 @@ def filterTimeIndices(self, timeIndices, boc=False, moc=False, eoc=False):

return filtered

def writeAssemHistory(self, a, fName=""):
def writeAssemHistory(self, a: "Assembly", fName: str = ""):
"""Write the assembly history report to a text file."""
fName = fName or self._getAssemHistoryFileName(a)
dbi = self.getInterface("database")
Expand Down Expand Up @@ -373,7 +379,7 @@ def unloadBlockHistoryVals(self):
"""Remove all cached db reads."""
self._preloadedBlockHistory = None

def getBlockHistoryVal(self, name: str, paramName: str, ts: Tuple[int, int]):
def getBlockHistoryVal(self, name: str, paramName: str, ts: tuple[int, int]):
"""
Use the database interface to return the parameter values for the supplied block
names, and timesteps.
Expand Down Expand Up @@ -422,28 +428,28 @@ def getBlockHistoryVal(self, name: str, paramName: str, ts: Tuple[int, int]):
raise
return val

def _isCurrentTimeStep(self, ts: Tuple[int, int]):
def _isCurrentTimeStep(self, ts: tuple[int, int]) -> bool:
"""Return True if the timestep requested is the current time step."""
return ts == (self.r.p.cycle, self.r.p.timeNode)

def _databaseHasDataForTimeStep(self, ts):
def _databaseHasDataForTimeStep(self, ts) -> bool:
"""Return True if the database has data for the requested time step."""
dbi = self.getInterface("database")
return ts in dbi.database.genTimeSteps()

def getTimeSteps(self, a=None):
r"""
Return list of time steps values (in years) that are available.
def getTimeSteps(self, a: "Assembly" = None) -> list[float]:
"""
Given a fuel assembly, return list of time steps values (in years) that are available.

Parameters
----------
a : Assembly object, optional
An assembly object designated a detail assem. If passed, only timesteps
a
A fuel assembly that has been designated a detail assem. If passed, only timesteps
where this assembly is in the core will be tracked.

Returns
-------
timeSteps : list
timeSteps
times in years that are available in the history

See Also
Expand All @@ -462,15 +468,13 @@ def getTimeSteps(self, a=None):
return timeInYears

@staticmethod
def _getBlockInAssembly(a):
"""Get a representative fuel block from an assembly."""
def _getBlockInAssembly(a: "Assembly") -> "Block":
"""Get a representative fuel block from a fuel assembly."""
b = a.getFirstBlock(Flags.FUEL)
if not b:
# there is a problem, it doesn't look like we have a fueled assembly
# but that is all we track... what is it? Throw an error
runLog.warning("Assembly {} does not contain fuel".format(a))
runLog.error("Assembly {} does not contain fuel".format(a))
for b in a:
runLog.warning("Block {}".format(b))
runLog.error("Block {}".format(b))
raise RuntimeError(
"A tracked assembly does not contain fuel and has caused this error, see the details in stdout."
)
Expand Down
4 changes: 2 additions & 2 deletions armi/bookkeeping/tests/test_historyTracker.py
Original file line number Diff line number Diff line change
Expand Up @@ -231,8 +231,8 @@ def test_historyReport(self):

# test that detailAssemblyNames() is working
self.assertEqual(len(history.detailAssemblyNames), 1)
history.addAllFuelAssems()
self.assertEqual(len(history.detailAssemblyNames), 51)
history.addAllDetailedAssems()
self.assertEqual(len(history.detailAssemblyNames), 54)

def test_getBlockInAssembly(self):
history = self.o.getInterface("history")
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,6 @@ def test_buReducingAssemblyRotation(self):

# apply dummy pin-level data to allow intelligent rotation
for b in assem.getBlocks(Flags.FUEL):
b.breakFuelComponentsIntoIndividuals()
b.initializePinLocations()
b.p.percentBuMaxPinLocation = 10
b.p.percentBuMax = 5
Expand Down
7 changes: 0 additions & 7 deletions armi/reactor/blockParameters.py
Original file line number Diff line number Diff line change
Expand Up @@ -151,13 +151,6 @@ def getBlockParameterDefinitions():
location=ParamLocation.MAX,
)

pb.defParam(
"percentBuMin",
units=units.PERCENT_FIMA,
description="Minimum percentage of the initial heavy metal atoms that have been fissioned",
location=ParamLocation.MAX,
)

pb.defParam(
"residence",
units=units.DAYS,
Expand Down
83 changes: 12 additions & 71 deletions armi/reactor/blocks.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,6 @@
from armi import nuclideBases
from armi import runLog
from armi.bookkeeping import report
from armi.nucDirectory import elements
from armi.nuclearDataIO import xsCollections
from armi.physics.neutronics import GAMMA
from armi.physics.neutronics import NEUTRON
Expand Down Expand Up @@ -794,11 +793,22 @@ def completeInitialLoading(self, bolBlock=None):
massHmBOL = 0.0
sf = self.getSymmetryFactor()
for child in self:
# multiplying by sf ends up cancelling out the symmetry factor used in
# Component.getMass(). So massHmBOL does not respect the symmetry factor.
hmMass = child.getHMMass() * sf
massHmBOL += hmMass
# Components have a massHmBOL parameter but not every composite will
# Components have the following parameters but not every composite will
# massHmBOL, molesHmBOL, puFrac
if isinstance(child, components.Component):
child.p.massHmBOL = hmMass
# to stay consistent with massHmBOL, molesHmBOL and puFrac should be
# independent of sf. As such, the need to be scaled by 1/sf.
child.p.molesHmBOL = child.getHMMoles() / sf
child.p.puFrac = (
self.getPuMoles() / sf / child.p.molesHmBOL
if child.p.molesHmBOL > 0.0
else 0.0
)

self.p.massHmBOL = massHmBOL

Expand Down Expand Up @@ -1432,63 +1442,6 @@ def updateComponentDims(self):
except NotImplementedError:
runLog.warning("{0} has no updatedDims method -- skipping".format(c))

def breakFuelComponentsIntoIndividuals(self):
john-science marked this conversation as resolved.
Show resolved Hide resolved
"""
Split block-level components (in fuel blocks) into pin-level components.

The fuel component will be broken up according to its multiplicity.

Order matters! The first pin component will be located at a particular (x, y), which
will be used in the fluxRecon module to determine the interpolated flux.

The fuel will become fuel001 through fuel169 if there are 169 pins.
"""
fuels = self.getChildrenWithFlags(Flags.FUEL)
if len(fuels) != 1:
runLog.error(
"This block contains {0} fuel components: {1}".format(len(fuels), fuels)
)
raise RuntimeError(
"Cannot break {0} into multiple fuel components b/c there is not a single fuel"
" component.".format(self)
)

fuel = fuels[0]
fuelFlags = fuel.p.flags
nPins = self.getNumPins()
runLog.info(
"Creating {} individual {} components on {}".format(nPins, fuel, self)
)

# Handle all other components that may be linked to the fuel multiplicity
# by unlinking them and setting them directly.
# TODO: What about other (actual) dimensions? This is a limitation in that only fuel
# components are duplicated, and not the entire pin. It is also a reasonable assumption with
# current/historical usage of ARMI.
for comp, dim in self.getComponentsThatAreLinkedTo(fuel, "mult"):
comp.setDimension(dim, nPins)

# finish the first pin as a single pin
fuel.setDimension("mult", 1)
fuel.setName("fuel001")
fuel.p.pinNum = 1

# create all the new pin components and add them to the block with 'fuel001' names
for i in range(nPins - 1):
# wow, only use of a non-deepcopy
newC = copy.copy(fuel)
newC.setName("fuel{0:03d}".format(i + 2)) # start with 002.
newC.p.pinNum = i + 2
self.add(newC)

# update moles at BOL for each pin
self.p.molesHmBOLByPin = []
for pin in self.iterComponents(Flags.FUEL):
# Update the fuel component flags to be the same as before the split (i.e., DEPLETABLE)
pin.p.flags = fuelFlags
self.p.molesHmBOLByPin.append(pin.getHMMoles())
pin.p.massHmBOL /= nPins

def getIntegratedMgFlux(self, adjoint=False, gamma=False):
"""
Return the volume integrated multigroup neutron tracklength in [n-cm/s].
Expand Down Expand Up @@ -1739,18 +1692,6 @@ def getBoronMassEnrich(self):
return 0.0
return b10 / total

def getPuMoles(self):
drewj-tp marked this conversation as resolved.
Show resolved Hide resolved
"""Returns total number of moles of Pu isotopes."""
nucNames = [nuc.name for nuc in elements.byZ[94].nuclides]
puN = sum(self.getNuclideNumberDensities(nucNames))

return (
puN
/ units.MOLES_PER_CC_TO_ATOMS_PER_BARN_CM
* self.getVolume()
* self.getSymmetryFactor()
)

def getUraniumMassEnrich(self):
"""Returns U-235 mass fraction assuming U-235 and U-238 only."""
u5 = self.getMass("U235")
Expand Down
Loading
Loading