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

Allow voxel plots to be colored by material

Browse source
This commit is contained in:
Paul Romano 2019-10-28 21:43:26 -05:00
parent 27e3f25bc2
commit 4d4c174a1b
1 changed files with 5 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/plot.cpp
View file

@ -908,11 +908,13 @@ void create_voxel(Plot pl)
// generate ids using plotbase
IdData ids = pltbase.get_map<IdData>();
// select only cell ID data and flip the y-axis
xt::xtensor<int32_t, 2> data1 = xt::flip(xt::view(ids.data_, xt::all(), xt::all(), 0), 0);
// select only cell/material ID data and flip the y-axis
int idx = pl.color_by_ == PlotColorBy::cells ? 0 : 1;
xt::xtensor<int32_t, 2> data_slice = xt::view(ids.data_, xt::all(), xt::all(), idx);
xt::xtensor<int32_t, 2> data_flipped = xt::flip(data_slice, 0);
// Write to HDF5 dataset
voxel_write_slice(z, dspace, dset, memspace, &(data1(0,0)));
voxel_write_slice(z, dspace, dset, memspace, data_flipped.data());
}
voxel_finalize(dspace, dset, memspace);