Added flag for enabling/disabling domain decomposition in TRISO lattice

1 files changed, 2 insertions, 1 deletions

diff --git a/hpmr.py b/hpmr.py
index 770a8ba..460479b 100755
--- a/hpmr.py
+++ b/hpmr.py
@@ -36,6 +36,7 @@ cli_parser.add_argument("-g","--geometry-level",default="pin",choices=["pin","as
 cli_parser.add_argument("-d","--drums-in",default=False,action="store_true",help="Flag for specifying the control drums are rotated in, default false")
 cli_parser.add_argument("-p","--packing",default=False,action="store_true",help="Flag for specifying if packing information should be read from file")
 cli_parser.add_argument("-e","--execute",default=False,action="store_true",help="Flag for automatically running OpenMC, default false.")
+cli_parser.add_argument("--triso-lattice",default=False,action="store_true",help="Flag for enabling TRISO lattice domain decomposition in the compact, default false.")
 cli_args = cli_parser.parse_args()
 
 if cli_args.lclrs:
@@ -467,7 +468,7 @@ geo_triso_universe = openmc.Universe(name="TRISO Kernel Universe",cells=geo_tris
 
 ## TRISO compact ##
 geo_triso_compact_region = -openmc.ZCylinder(r=geo_const_triso_radius,x0=0.0,y0=0.0)&+geo_const_z_min&-geo_const_z_max
-geo_triso_compact_domain = (4,4,int(4*geo_const_height))
+geo_triso_compact_domain = (4,4,int(4*geo_const_height)) if cli_args.triso_lattice else (1,1,1)
 
 # Generate packing
 if cli_args.packing: