crp/OpenMC
1
0
Fork 0
mirror of https://github.com/openmc-dev/openmc.git synced 2026-07-26 21:25:36 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add new variables and functions to query global surface source bank size and index

Browse source
This commit is contained in:
YoungHui Park 2020-07-22 04:54:35 -05:00
parent b06725b4fb
commit 5d9655256b
2 changed files with 32 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

26
src/simulation.cpp
View file

@ -270,6 +270,8 @@ const RegularMesh* ufs_mesh {nullptr};
std::vector<double> k_generation;
std::vector<int64_t> work_index;
int64_t total_surf_banks;
std::vector<int64_t> surf_src_index;
} // namespace simulation
@ -385,6 +387,9 @@ void finalize_batch()
// Write out surface source if requested.
if (settings::surface_source && simulation::current_batch == settings::n_batches) {
query_surf_src_size();
auto filename = settings::path_output + "surface_source.h5";
write_source_point(filename.c_str(), true); //!!
}
@ -630,6 +635,27 @@ void initialize_data()
data::energy_min[neutron]) / settings::n_log_bins;
}
void query_surf_src_size()
{
int64_t i_bank = 0;
simulation::surf_src_index.resize(mpi::n_procs + 1);
simulation::surf_src_index[0] = 0;
for (int i = 0; i < mpi::n_procs; ++i) {
int64_t work_i = 0;
// Set number of particles
if (mpi::rank == i) work_i = simulation::surf_src_bank.size();
#ifdef OPENMC_MPI
MPI_Reduce(&work_i, &simulation::total_surf_banks, 1, MPI_INT,
MPI_SUM, 0, mpi::intracomm);
#endif
// Set index into source bank for rank i
i_bank += work_i;
simulation::surf_src_index[i + 1] = i_bank;
}
}
#ifdef OPENMC_MPI
void broadcast_results() {
// Broadcast tally results so that each process has access to results