Skip to content

Generalized search_mesh that takes a functor to execute at element boundary or destination #130

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 4 commits into from
Nov 21, 2024
Merged

Generalized search_mesh that takes a functor to execute at element boundary or destination #130

Changes from 1 commit
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
fc8dbc7
generalized search mesh with functor
Fuad-HH Nov 11, 2024
e702d3c
modified functor signature to remove requireIntersection
Fuad-HH Nov 13, 2024
59d0fd2
rename particle_search method and add documentation
Fuad-HH Nov 18, 2024
3d11b9c
Merge branch 'master' into functor_parameter_for_search_mesh
jacobmerson Nov 21, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
61 changes: 51 additions & 10 deletions src/pumipic_adjacency.tpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -395,16 +395,25 @@ namespace pumipic {
printf("Min area is: %.15f, Planned tol is %.15f\n", min_area, tol);
return tol;
}
template <class ParticleType, typename Segment3d, typename SegmentInt>
bool search_mesh(o::Mesh& mesh, ParticleStructure<ParticleType>* ptcls,

template <class ParticleType, typename Segment3d, typename SegmentInt, typename Func>
bool particle_search(o::Mesh& mesh, ParticleStructure<ParticleType>* ptcls,
Segment3d x_ps_orig, Segment3d x_ps_tgt, SegmentInt pids,
o::Write<o::LO>& elem_ids,
bool requireIntersection,
o::Write<o::LO>& inter_faces,
o::Write<o::Real>& inter_points,
int looplimit,
int debug) {
bool debug,
Func& func) {
static_assert(
std::is_invocable_r_v<
void, Func, o::Mesh &, ParticleStructure<ParticleType> *,
o::Write<o::LO> &, o::Write<o::LO> &, o::Write<o::LO> &,
o::Write<o::Real> &, o::Write<o::LO> &, bool,
Segment3d, Segment3d>,
"Functional must accept <mesh> <ps> <elem_ids> <inter_faces> <lastExit> <inter_points> <ptcl_done> <requireIntersection> <x_ps_orig> <x_ps_tgt>\n");

//Initialize timer
const auto btime = pumipic_prebarrier();
Kokkos::Profiling::pushRegion("pumipic_search_mesh");
Expand Down Expand Up @@ -489,12 +498,7 @@ namespace pumipic {
//Find intersection face
find_exit_face(mesh, ptcls, x_ps_orig, x_ps_tgt, elem_ids, ptcl_done, elmArea, useBcc, lastExit, inter_points, tol);
//Check if intersection face is exposed
check_model_intersection(mesh, ptcls, x_ps_orig, x_ps_tgt, elem_ids, ptcl_done, lastExit, side_is_exposed,
requireIntersection, inter_faces);

//Move to next element
set_new_element(mesh, ptcls, elem_ids, ptcl_done, lastExit);

func(mesh, ptcls, elem_ids, inter_faces, lastExit, inter_points, ptcl_done, requireIntersection, x_ps_orig, x_ps_tgt);
//Check if all particles are found
found = true;
o::LOs ptcl_done_r(ptcl_done);
Expand Down Expand Up @@ -544,5 +548,42 @@ namespace pumipic {
Kokkos::Profiling::popRegion();
return found;
}

template <typename ParticleType, typename Segment3d>
struct NativeParticleHandlerAtElemBdry {
NativeParticleHandlerAtElemBdry () {}

void operator()(o::Mesh& mesh, ParticleStructure<ParticleType>* ptcls,
o::Write<o::LO>& elem_ids, o::Write<o::LO>& inter_faces,
o::Write<o::LO>& lastExit, o::Write<o::Real>& inter_points,
o::Write<o::LO>& ptcl_done,
bool requireIntersection,
Segment3d x_ps_orig,
Segment3d x_ps_tgt) const {
const auto side_is_exposed = mark_exposed_sides(&mesh);
// Check if intersection face is exposed
check_model_intersection(mesh, ptcls, x_ps_orig, x_ps_tgt, elem_ids,
ptcl_done, lastExit, side_is_exposed,
requireIntersection, inter_faces);

// Move to next element
set_new_element(mesh, ptcls, elem_ids, ptcl_done, lastExit);
}
};

template <class ParticleType, typename Segment3d, typename SegmentInt>
bool search_mesh(o::Mesh& mesh, ParticleStructure<ParticleType>* ptcls,
Segment3d x_ps_orig, Segment3d x_ps_tgt, SegmentInt pids,
o::Write<o::LO>& elem_ids,
bool requireIntersection,
o::Write<o::LO>& inter_faces,
o::Write<o::Real>& inter_points,
int looplimit,
int debug) {
NativeParticleHandlerAtElemBdry<ParticleType, Segment3d> native_handler;

return particle_search(mesh, ptcls, x_ps_orig, x_ps_tgt, pids, elem_ids, requireIntersection,
inter_faces, inter_points, looplimit, debug, native_handler);
}
}
#endif