From 7805e62674d51ebc0ee0b806268d100307c02138 Mon Sep 17 00:00:00 2001
From: Paul Romano <paul.k.romano@gmail.com>
Date: Wed, 8 Jun 2022 09:10:38 -0500
Subject: [PATCH] Fix assembly assignment in build-assembly.py

---
 smr/build-assembly-long.py | 15 +++++++--------
 smr/build-assembly.py      | 13 ++++---------
 2 files changed, 11 insertions(+), 17 deletions(-)

diff --git a/smr/build-assembly-long.py b/smr/build-assembly-long.py
index 98ade14..623907c 100644
--- a/smr/build-assembly-long.py
+++ b/smr/build-assembly-long.py
@@ -174,14 +174,13 @@ settings.particles = 10000
 settings.output = {'tallies': False, 'summary': False}
 settings.source = source
 settings.sourcepoint = {'write': False}
-
-if args.multipole:
-    settings.temperature = {
-        'multipole': True,
-        'tolerance': 1000,
-        'default': 531.5,
-        'method': 'interpolation',
+settings.temperature = {
+    'default': 531.5,
+    'method': 'interpolation',
         'range': (500.0, 1300.0)
-    }
+}
+if args.multipole:
+    settings.temperature['multipole'] = True
+    settings.temperature['tolerance'] = 1000
 
 settings.export_to_xml(str(directory / 'settings.xml'))
diff --git a/smr/build-assembly.py b/smr/build-assembly.py
index 123fa91..1b2682e 100644
--- a/smr/build-assembly.py
+++ b/smr/build-assembly.py
@@ -10,7 +10,6 @@ import openmc
 from smr.materials import materials, clone
 from smr.surfaces import surfs, lattice_pitch, bottom_fuel_stack, top_active_core, pellet_OR
 from smr.assemblies import assembly_universes
-from smr.plots import assembly_plots
 from smr import inlet_temperature
 
 
@@ -48,10 +47,10 @@ if args.rings > 1:
 else:
     ring_radii = None
 assembly = assembly_universes(ring_radii, args.axial, args.depleted)
-lattice_sides = openmc.model.get_rectangular_prism(lattice_pitch, lattice_pitch,
-                                                   boundary_type='reflective')
+lattice_sides = openmc.model.rectangular_prism(lattice_pitch, lattice_pitch,
+                                               boundary_type='reflective')
 main_cell = openmc.Cell(
-    fill=assembly['Assembly (3.1%) 16BA'],
+    fill=assembly['Assembly (3.1%)'],
     region=lattice_sides & +surfs['lower bound'] & -surfs['upper bound']
 )
 root_univ = openmc.Universe(cells=[main_cell])
@@ -98,7 +97,7 @@ settings.inactive = 100
 settings.particles = 10000
 settings.output = {'tallies': False, 'summary': False}
 settings.source = source
-settings.sourcepoint_write = False
+settings.sourcepoint = {'write': False}
 settings.temperature = {
     'default': inlet_temperature,
     'method': 'interpolation',
@@ -140,7 +139,3 @@ elif args.tallies == 'mat':
     tallies.append(tally)
 
 tallies.export_to_xml(str(directory / 'tallies.xml'))
-
-# Create plots
-plots = assembly_plots(main_cell.fill)
-plots.export_to_xml(str(directory / 'plots.xml'))