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

Copy/simplify source bank writing functions to make temp surface source bank writing functions

Browse source
This commit is contained in:
YoungHui Park 2020-03-17 00:18:35 -05:00
parent 209394082e
commit 2ebf9d844f
2 changed files with 59 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
include/openmc/state_point.h
View file

@ -14,6 +14,8 @@ namespace openmc {
void load_state_point();
void write_source_point(const char* filename);
void write_source_bank(hid_t group_id);
void write_surf_src_point(const char* filename);
void write_surf_src_bank(hid_t group_id);
void read_source_bank(hid_t group_id, std::vector<Particle::Bank>& sites, bool distribute);
void write_tally_results_nr(hid_t file_id);
void restart_set_keff();

57
src/state_point.cpp
View file

@ -653,6 +653,63 @@ write_source_bank(hid_t group_id)
}
void
write_surf_src_point(const char* filename)
{
std::string filename_;
filename_ = filename;
hid_t file_id;
file_id = file_open(filename_, 'w', true);
write_attribute(file_id, "filetype", "source");
write_surf_src_bank(file_id);
file_close(file_id);
}
void
write_surf_src_bank(hid_t group_id)
{
hid_t banktype = h5banktype();
if (mpi::master) {
// Create dataset big enough to hold all source sites
hsize_t dims[] {static_cast<hsize_t>(settings::n_particles)};
hid_t dspace = H5Screate_simple(1, dims, nullptr);
hid_t dset = H5Dcreate(group_id, "source_bank", banktype, dspace,
H5P_DEFAULT, H5P_DEFAULT, H5P_DEFAULT);
for (int i = 0; i < mpi::n_procs; ++i) {
// Create memory space
hsize_t count[] {static_cast<hsize_t>(simulation::work_index[i+1] -
simulation::work_index[i])};
hid_t memspace = H5Screate_simple(1, count, nullptr);
// Select hyperslab for this dataspace
dspace = H5Dget_space(dset);
hsize_t start[] {static_cast<hsize_t>(simulation::work_index[i])};
H5Sselect_hyperslab(dspace, H5S_SELECT_SET, start, nullptr, count, nullptr);
// Write data to hyperslab
H5Dwrite(dset, banktype, memspace, dspace, H5P_DEFAULT,
simulation::surf_src_bank.data());
H5Sclose(memspace);
H5Sclose(dspace);
}
// Close all ids
H5Dclose(dset);
} else {
}
H5Tclose(banktype);
}
void read_source_bank(hid_t group_id, std::vector<Particle::Bank>& sites, bool distribute)
{
hid_t banktype = h5banktype();