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

Merge pull request #1645 from pshriwise/umesh_scores

Browse source 
Fix for data crossover in VTK filess when using multiple tallies w/ unstructured mesh.
This commit is contained in:
Paul Romano 2020-08-26 21:22:21 -05:00 • committed by GitHub
commit bf34c4bcad
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 36 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
include/openmc/mesh.h
View file

@ -298,6 +298,9 @@ public:
//! Add a score to the mesh instance
void add_score(std::string score) const;
//! Remove a score from the mesh instance
void remove_score(std::string score) const;
//! Set data for a score
void set_score_data(const std::string& score,
std::vector<double> values,

28
src/mesh.cpp
View file

@ -2015,6 +2015,34 @@ UnstructuredMesh::add_score(std::string score) const {
auto score_tags = this->get_score_tags(score);
}
void UnstructuredMesh::remove_score(std::string score) const {
auto value_name = score + "_mean";
moab::Tag tag;
moab::ErrorCode rval = mbi_->tag_get_handle(value_name.c_str(), tag);
if (rval != moab::MB_SUCCESS) return;
rval = mbi_->tag_delete(tag);
if (rval != moab::MB_SUCCESS) {
auto msg = fmt::format("Failed to delete mesh tag for the score {}"
" on unstructured mesh {}", score, id_);
fatal_error(msg);
}
auto std_dev_name = score + "_std_dev";
rval = mbi_->tag_get_handle(std_dev_name.c_str(), tag);
if (rval != moab::MB_SUCCESS) {
auto msg = fmt::format("Std. Dev. mesh tag does not exist for the score {}"
" on unstructured mesh {}", score, id_);
}
rval = mbi_->tag_delete(tag);
if (rval != moab::MB_SUCCESS) {
auto msg = fmt::format("Failed to delete mesh tag for the score {}"
" on unstructured mesh {}", score, id_);
fatal_error(msg);
}
}
void
UnstructuredMesh::set_score_data(const std::string& score,
std::vector<double> values,

5
src/state_point.cpp
View file

@ -687,6 +687,8 @@ void read_source_bank(hid_t group_id)
void write_unstructured_mesh_results() {
for (auto& tally : model::tallies) {
std::vector<std::string> tally_scores;
for (auto filter_idx : tally->filters()) {
auto& filter = model::tally_filters[filter_idx];
if (filter->type() != "mesh") continue;
@ -741,6 +743,7 @@ void write_unstructured_mesh_results() {
std::string score_name = tally->score_name(i_score);
auto score_str = fmt::format("{}_{}", score_name, nuclide_name);
tally_scores.push_back(score_str);
umesh->set_score_data(score_str, mean_vec, std_dev_vec);
}
}
@ -754,6 +757,8 @@ void write_unstructured_mesh_results() {
w);
// Write the unstructured mesh and data to file
umesh->write(filename);
for (const auto& score : tally_scores) { umesh->remove_score(score); }
}
}
}