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

Merge pull request #1388 from paulromano/voxel-color-by

Browse source 
Allow voxel plots to be colored by material
This commit is contained in:
Patrick Shriwise 2019-10-29 07:34:53 -05:00 • committed by GitHub
commit 4e2fe16795
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
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);