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

Writing out element centroids to the statepoint file.

Browse source
This commit is contained in:
Patrick Shriwise 2020-02-17 16:34:59 -06:00
parent 22836ed611
commit 862c20752e
1 changed files with 7 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/mesh.cpp
View file

@ -14,6 +14,7 @@
#include "xtensor/xmath.hpp"
#include "xtensor/xsort.hpp"
#include "xtensor/xtensor.hpp"
#include "xtensor/xview.hpp"
#include "openmc/capi.h"
#include "openmc/constants.h"
@ -1868,10 +1869,15 @@ UnstructuredMesh::to_hdf5(hid_t group) const
// write volume of each tet
std::vector<double> tet_vols;
for (const auto& eh : ehs_) {
xt::xtensor<double, 2> centroids({n_bins(), 3});
for (int i = 0; i < ehs_.size(); i++) {
const auto& eh = ehs_[i];
tet_vols.emplace_back(tet_volume(eh));
Position c = centroid(eh);
xt::view(centroids, i, xt::all()) = xt::xarray<double>({c.x, c.y, c.z});
}
write_dataset(mesh_group, "volumes", tet_vols);
write_dataset(mesh_group, "centroids", centroids);
close_group(mesh_group);
}