Provide Component.getFuelMass

Directly calls getMass if the component is flagged with fuel

armi/reactor/components/component.py

@@ -28,6 +28,7 @@
 from armi.bookkeeping import report
 from armi.reactor import composites
 from armi.reactor import parameters
+from armi.reactor import flags
 from armi.reactor.components import componentParameters
 from armi.utils import densityTools
 from armi.utils.units import C_TO_K
@@ -1205,6 +1206,10 @@ def getPitchData(self):
             "Please implement if this component type can be a pitch defining component."
         )
 
+    def getFuelMass(self) -> float:
+        """Return the mass in grams if this is a fueled component"""
+        return self.getMass() if self.hasFlags(flags.Flags.FUEL) else 0.0
+
 
 class ShapedComponent(Component):
     """A component with well-defined dimensions."""

armi/reactor/tests/test_components.py

@@ -22,6 +22,7 @@
 
 from armi.materials.material import Material
 from armi.reactor import components
+from armi.reactor import flags
 from armi.reactor.components import (
     Component,
     UnshapedComponent,
@@ -516,6 +517,13 @@ def test_changeNumberDensities(self):
         self.component.changeNDensByFactor(3.0)
         self.assertEqual(self.component.getNumberDensity("NA23"), 3.0)
 
+    def test_fuelMass(self):
+        nominalMass = self.component.getMass()
+        self.component.p.flags = flags.Flags.FUEL
+        self.assertEqual(self.component.getFuelMass(), nominalMass)
+        self.component.p.flags = flags.Flags.MODERATOR
+        self.assertEqual(self.component.getFuelMass(), 0.0)
+
 
 class TestComponentExpansion(unittest.TestCase):
     tCold = 20