def test_addDummyNuclidesToLibrary(self):
dummyNuclides = [XSNuclide(None, "U238AA")]
before = self.xsLib.getNuclides("")
self.assertEqual(len(self.xsLib.xsIDs), 0)
self.assertTrue(
gamiso.addDummyNuclidesToLibrary(self.xsLib, dummyNuclides))
self.assertEqual(len(self.xsLib.xsIDs), 1)
self.assertEqual(list(self.xsLib.xsIDs)[0], "38")
after = self.xsLib.getNuclides("")
self.assertGreater(len(after), len(before))
diff = set(after).difference(set(before))
self.assertEqual(len(diff), 1)
self.assertEqual(list(diff)[0].xsId, "38")
def mergeXSLibrariesInWorkingDirectory(lib,
xsLibrarySuffix="",
mergeGammaLibs=False):
"""
Merge neutron (ISOTXS) and gamma (GAMISO/PMATRX) library data into the provided library.
Parameters
----------
lib : obj
ISOTXS library object
xsLibrarySuffix : str, optional
XS library suffix used to determine which ISOTXS files are merged together,
typically something like `-doppler`. If empty string, will merge everything
without suffix (indicated by a `-`).
mergeGammaLibs : bool, optional
If True, the GAMISO and PMATRX files that correspond to the ISOTXS library will be merged. Note: if these
files do not exist this will fail.
"""
# pylint: disable=import-outside-toplevel) ; avoid cyclic import with isotxs bringing this in for data structure
from armi.nuclearDataIO.cccc import isotxs
from armi.nuclearDataIO.cccc import gamiso
from armi.nuclearDataIO.cccc import pmatrx
from armi import nuclearDataIO
xsLibFiles = getISOTXSLibrariesToMerge(
xsLibrarySuffix, [iso for iso in glob.glob(_ISOTXS_EXT + "*")])
librariesToMerge = []
neutronVelocities = {
} # Dictionary of neutron velocities from each ISOTXS file
for xsLibFilePath in sorted(xsLibFiles):
try:
xsID = re.search("ISO([A-Z0-9]{2})", xsLibFilePath).group(
1) # get XS ID from the cross section library name
except AttributeError:
# if glob has matched something that is not actually an ISOXX file,
# the .group() call will fail
runLog.debug(
f"Ignoring file {xsLibFilePath} in the merging of ISOXX files")
continue
xsFileTypes = "ISOTXS" if not mergeGammaLibs else "ISOTXS, GAMISO, and PMATRX"
runLog.info("Retrieving {} data for XS ID {}{}".format(
xsFileTypes, xsID, xsLibrarySuffix))
if xsLibFilePath in lib.isotxsMetadata.fileNames:
runLog.extra(
"Skipping merge of {} because data already exists in the library"
.format(xsLibFilePath))
continue
neutronLibrary = isotxs.readBinary(xsLibFilePath)
neutronVelocities[xsID] = neutronLibrary.neutronVelocity
librariesToMerge.append(neutronLibrary)
if mergeGammaLibs:
dummyNuclides = [
nuc for nuc in neutronLibrary.nuclides
if isinstance(nuc._base, nuclideBases.DummyNuclideBase)
]
gamisoLibraryPath = nuclearDataIO.getExpectedGAMISOFileName(
suffix=xsLibrarySuffix, xsID=xsID)
pmatrxLibraryPath = nuclearDataIO.getExpectedPMATRXFileName(
suffix=xsLibrarySuffix, xsID=xsID)
# Check if the gamiso and pmatrx data paths exist with the xs library suffix so that
# these are merged in. If they don't both exist then that is OK and we can just
# revert back to expecting the files just based on the XS ID.
if not (os.path.exists(gamisoLibraryPath)
and os.path.exists(pmatrxLibraryPath)):
runLog.warning(f"One of GAMISO or PMATRX data exist for "
f"XS ID {xsID} with suffix {xsLibrarySuffix}. "
f"Attempting to find GAMISO/PMATRX data with "
f"only XS ID {xsID} instead.")
gamisoLibraryPath = nuclearDataIO.getExpectedGAMISOFileName(
xsID=xsID)
pmatrxLibraryPath = nuclearDataIO.getExpectedPMATRXFileName(
xsID=xsID)
# GAMISO data
gammaLibrary = gamiso.readBinary(gamisoLibraryPath)
addedDummyData = gamiso.addDummyNuclidesToLibrary(
gammaLibrary,
dummyNuclides) # Add DUMMY nuclide data not produced by MC2-3
if addedDummyData:
gamisoDummyPath = os.path.abspath(
os.path.join(os.getcwd(), gamisoLibraryPath))
gamiso.writeBinary(gammaLibrary, gamisoDummyPath)
gammaLibraryDummyData = gamiso.readBinary(gamisoDummyPath)
librariesToMerge.append(gammaLibraryDummyData)
else:
librariesToMerge.append(gammaLibrary)
# PMATRX data
pmatrxLibrary = pmatrx.readBinary(pmatrxLibraryPath)
addedDummyData = pmatrx.addDummyNuclidesToLibrary(
pmatrxLibrary,
dummyNuclides) # Add DUMMY nuclide data not produced by MC2-3
if addedDummyData:
pmatrxDummyPath = os.path.abspath(
os.path.join(os.getcwd(), pmatrxLibraryPath))
pmatrx.writeBinary(pmatrxLibrary, pmatrxDummyPath)
pmatrxLibraryDummyData = pmatrx.readBinary(pmatrxDummyPath)
librariesToMerge.append(pmatrxLibraryDummyData)
else:
librariesToMerge.append(pmatrxLibrary)
for library in librariesToMerge:
lib.merge(library)
return neutronVelocities