diff --git a/models/openmc/beavrs/builder.py b/models/openmc/beavrs/builder.py index 56e9919..d328cdb 100644 --- a/models/openmc/beavrs/builder.py +++ b/models/openmc/beavrs/builder.py @@ -106,6 +106,9 @@ class BEAVRS(object): nuclides.append(openmc.Nuclide(nuc)) self.depletion_nuclides = nuclides + # The OpenMC model container to be populated. + self.model = openmc.Model() + def set_boron_ppm(self, ppm): self.mats = openmc_materials(ppm=ppm) @@ -116,12 +119,15 @@ class BEAVRS(object): materials_file = openmc.Materials(self.mats.values()) materials_file.export_to_xml() + def get_openmc_plots(self): + self.plots = Plots(self.mats) + return openmc.Plots(self.plots.plots) + def write_openmc_plots(self): self.plots = Plots(self.mats) - plot_file = openmc.Plots(self.plots.plots) - plot_file.export_to_xml() + self.get_openmc_plots().export_to_xml() - def write_openmc_settings(self): + def get_openmc_settings_file(self): settings_file = openmc.Settings() settings_file.batches = self.settings_batches settings_file.inactive = self.settings_inactive @@ -139,16 +145,36 @@ class BEAVRS(object): output = {'tallies': self.settings_output_tallies, 'summary': self.settings_summary} settings_file.output = output - settings_file.export_to_xml() + return settings_file - def write_openmc_tallies(self): + def write_openmc_settings(self): + self.get_openmc_settings_file().export_to_xml() + + def get_openmc_tallies_file(self): if len(self.tallies) == 0: return tallies_file = openmc.Tallies() for mesh in self.tally_meshes: tallies_file.add_mesh(mesh) for tally in self.tallies: tallies_file.append(tally) - tallies_file.export_to_xml() + return tallies_file + + def write_openmc_tallies(self): + if len(self.tallies) == 0: return + self.get_openmc_tallies_file().export_to_xml() + + def build_openmc_model(self): + self.model = openmc.Model() + self.model.geometry = self.openmc_geometry + self.model.materials = openmc.Materials(self.mats.values()) + self.model.settings = self.get_openmc_settings_file() + if len(self.tallies) != 0: + self.model.tallies = self.get_openmc_tallies_file() + self.model.plots = self.get_openmc_plots() + return self.model + + def write_openmc_model(self): + self.build_openmc_model().export_to_model_xml() def set_params(self, particles=None, batches=None, inactive=None): if particles: self.settings_particles = particles diff --git a/models/openmc/make_beavrs.py b/models/openmc/make_beavrs.py index 2fa5234..d6fc2b6 100755 --- a/models/openmc/make_beavrs.py +++ b/models/openmc/make_beavrs.py @@ -23,8 +23,5 @@ if not len(args) == 0: p.print_help() b = BEAVRS(is_symmetric=options.is_symmetric, is_2d=options.is_2d) -b.write_openmc_geometry() -b.write_openmc_materials() -b.write_openmc_plots() -b.write_openmc_settings() +b.write_openmc_model()