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

Expand Particle::tracks_ to include full state

Browse source
This commit is contained in:
Paul Romano 2022-05-12 12:45:20 -05:00
parent 7752afb554
commit a6bd42ee8e
3 changed files with 42 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/track_output.cpp
View file

@ -30,7 +30,7 @@ void add_particle_track(Particle& p)
void write_particle_track(Particle& p)
{
p.tracks().back().push_back(p.r());
p.tracks().back().push_back(p.get_track_state());
}
void finalize_particle_track(Particle& p)
@ -57,9 +57,9 @@ void finalize_particle_track(Particle& p)
size_t n = t.size();
xt::xtensor<double, 2> data({n, 3});
for (int j = 0; j < n; ++j) {
data(j, 0) = t[j].x;
data(j, 1) = t[j].y;
data(j, 2) = t[j].z;
data(j, 0) = t[j].r.x;
data(j, 1) = t[j].r.y;
data(j, 2) = t[j].r.z;
}
std::string name = fmt::format("coordinates_{}", i);
write_dataset(file_id, name.c_str(), data);