Added enum for material libraries types.

montepy/__init__.py

@@ -28,7 +28,7 @@
 from .input_parser.input_reader import read_input
 
 # top level
-from montepy.particle import Particle
+from montepy.particle import Particle, LibraryType
 from montepy.universe import Universe
 from montepy.cell import Cell
 from montepy.mcnp_problem import MCNP_Problem

montepy/particle.py

@@ -1,9 +1,9 @@
 # Copyright 2024, Battelle Energy Alliance, LLC All Rights Reserved.
-from enum import Enum, unique
+from enum import StrEnum, unique
 
 
 @unique
-class Particle(Enum):
+class Particle(StrEnum):
  """
  Supported MCNP supported particles.
 
@@ -53,3 +53,33 @@ def __lt__(self, other):
 
  def __str__(self):
  return self.name.lower()
+
+
+@unique
+class LibraryType(StrEnum):
+ """
+
+ Taken from section of 5.6.1 of LA-UR-22-30006
+ """
+
+ def __new__(cls, value, particle=None):
+ obj = str.__new__(cls)
+ obj._value_ = value
+ obj._particle_ = particle
+ return obj
+
+ NEUTRON = ("NLIB", Particle.NEUTRON)
+ PHOTO_ATOMIC = ("PLIB", None)
+ PHOTO_NUCLEAR = ("PNLIB", None)
+ ELECTRON = ("ELIB", Particle.ELECTRON)
+ PROTON = ("HLIB", Particle.PROTON)
+ ALPHA_PARTICLE = ("ALIB", Particle.ALPHA_PARTICLE)
+ HELION = ("SLIB", Particle.HELION)
+ TRITON = ("TLIB", Particle.TRITON)
+ DEUTERON = ("DLIB", Particle.DEUTERON)
+
+ def __str__(self):
+ return self.name.lower()
+
+ def __lt__(self, other):
+ return self.value < other.value

montepy/surfaces/surface_type.py

@@ -1,9 +1,9 @@
 # Copyright 2024, Battelle Energy Alliance, LLC All Rights Reserved.
-from enum import unique, Enum
+from enum import unique, StrEnum
 
 
-# @unique
-class SurfaceType(str, Enum):
+@unique
+class SurfaceType(StrEnum):
  """
  An enumeration of the surface types allowed.