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

Adding a function to determine the instance of a cell at any level of Particle::coords.

Browse source
This commit is contained in:
Patrick Shriwise 2021-07-05 12:35:14 -05:00
parent c373d9548f
commit 4e3350ebcf
2 changed files with 36 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

11
include/openmc/geometry.h
View file

@ -40,6 +40,17 @@ inline bool coincident(double d1, double d2) {
bool check_cell_overlap(Particle& p, bool error=true);
//==============================================================================
//! Get the cell instance for a particle at the specified universe level
//!
//! \param p A particle for which to compute the instance using
//! it's vector of LocalCoord.
//! \param level The level (zero indexed) of the geometry where the instance should be computed.
//! \return The instance of the cell at the specified level.
//==============================================================================
int cell_instance_at_level(const Particle& p, int level);
//==============================================================================
//! Locate a particle in the geometry tree and set its geometry data fields.
//!

39
src/geometry.cpp
View file

@ -65,6 +65,30 @@ bool check_cell_overlap(Particle& p, bool error)
//==============================================================================
int cell_instance_at_level(const Particle& p, int level) {
if (level > p.n_coord()) {
fatal_error(fmt::format("Cell instance at level {} requested, but only {} levels exist in the model.", level, p.n_coord()));
}
// determine the cell instance
Cell& c {*model::cells[p.coord(level).cell]};
int instance = 0;
for (int i = 0; i < level; i++) {
const auto& c_i {*model::cells[p.coord(i).cell]};
if (c_i.type_ == Fill::UNIVERSE) {
instance += c_i.offset_[c.distribcell_index_];
} else if (c_i.type_ == Fill::LATTICE) {
auto& lat {*model::lattices[p.coord(i + 1).lattice]};
const auto& i_xyz {p.coord(i + 1).lattice_i};
if (lat.are_valid_indices(i_xyz)) {
instance += lat.offset(c.distribcell_index_, i_xyz);
}
}
}
return instance;
}
//==============================================================================
bool
find_cell_inner(Particle& p, const NeighborList* neighbor_list)
{
@ -130,22 +154,9 @@ find_cell_inner(Particle& p, const NeighborList* neighbor_list)
// Found a material cell which means this is the lowest coord level.
// Find the distribcell instance number.
int offset = 0;
if (c.distribcell_index_ >= 0) {
for (int i = 0; i < p.n_coord(); i++) {
const auto& c_i {*model::cells[p.coord(i).cell]};
if (c_i.type_ == Fill::UNIVERSE) {
offset += c_i.offset_[c.distribcell_index_];
} else if (c_i.type_ == Fill::LATTICE) {
auto& lat {*model::lattices[p.coord(i + 1).lattice]};
const auto& i_xyz {p.coord(i + 1).lattice_i};
if (lat.are_valid_indices(i_xyz)) {
offset += lat.offset(c.distribcell_index_, i_xyz);
}
}
}
p.cell_instance() = cell_instance_at_level(p, p.n_coord() - 1);
}
p.cell_instance() = offset;
// Set the material and temperature.
p.material_last() = p.material();