def _run_openmoc(self): """Plot the flat source regions in the geometry.""" # Run an eigenvalue calculation to setup FSR centroids super(PlotFSRsTestHarness, self)._run_openmoc() # Create a series of Matplotlib Figures / PIL Images for different # plotting parameters and append to figures list self.figures.append( plot_flat_source_regions(self.input_set.geometry, gridsize=100, get_figure=True)) self.figures.append( plot_flat_source_regions(self.input_set.geometry, gridsize=100, get_figure=True, xlim=(0., 2.), ylim=(0., 2.))) self.figures.append( plot_flat_source_regions(self.input_set.geometry, gridsize=100, get_figure=True, centroids=True, marker_size=3)) self.figures.append( plot_flat_source_regions(self.input_set.geometry, gridsize=100, get_figure=True, centroids=True, library='pil'))
############################################################################### ######################## Creating the TrackGenerator ###################### ############################################################################### log.py_printf('NORMAL', 'Initializing the track generator...') track_generator = TrackGenerator(geometry, num_azim, track_spacing) track_generator.generateTracks() ############################################################################### ########################### Running a Simulation ########################## ############################################################################### solver = CPUSolver(geometry, track_generator) solver.setSourceConvergenceThreshold(tolerance) solver.setNumThreads(num_threads) solver.convergeSource(max_iters) solver.printTimerReport() ############################################################################### ############################ Generating Plots ############################# ############################################################################### log.py_printf('NORMAL', 'Plotting data...') plotter.plot_cells(geometry) plotter.plot_materials(geometry) plotter.plot_flat_source_regions(geometry) plotter.plot_fluxes(geometry, solver, energy_groups=[1,2,3,4,5,6,7,8])
# plot geometry materials and cells plotter.plot_cells(geometry, gridsize=1000) plotter.plot_materials(geometry, gridsize=1000) ############################################################################### ######################## Creating the TrackGenerator ###################### ############################################################################### log.py_printf('NORMAL', 'Initializing the track generator...') track_generator = ModularTrackGenerator(geometry, num_azim, track_spacing) track_generator.setLatticeStructure(25,25) track_generator.generateTracks() plotter.plot_flat_source_regions(geometry, gridsize=1000) ############################################################################### ########################### Running a Simulation ########################## ############################################################################### solver = ModularCPUSolver(geometry, track_generator) solver.setSourceConvergenceThreshold(tolerance) solver.setNumThreads(num_threads) solver.convergeSource(max_iters) solver.printTimerReport() ############################################################################### ############################ Generating Plots ############################# ###############################################################################