Try to increase query size with each failed attempt

TODO: check performance gain
This commit is contained in:
tamasmeszaros 2022-06-09 17:22:57 +02:00
parent e6d49b75de
commit de3cbd483d
2 changed files with 22 additions and 14 deletions

View file

@ -11,14 +11,15 @@ namespace Slic3r { namespace branchingtree {
void build_tree(PointCloud &nodes, Builder &builder)
{
constexpr size_t ReachablesToExamine = 5;
constexpr size_t initK = 5;
auto ptsqueue = nodes.start_queue();
auto &properties = nodes.properties();
struct NodeDistance { size_t node_id = Node::ID_NONE; float distance = NaNf; };
auto distances = reserve_vector<NodeDistance>(ReachablesToExamine);
auto distances = reserve_vector<NodeDistance>(initK);
double prev_dist_max = 0.;
size_t K = initK;
while (!ptsqueue.empty() && builder.is_valid()) {
size_t node_id = ptsqueue.top();
@ -27,13 +28,13 @@ void build_tree(PointCloud &nodes, Builder &builder)
nodes.mark_unreachable(node_id);
distances.clear();
distances.reserve(K);
nodes.foreach_reachable<ReachablesToExamine>(
nodes.foreach_reachable(
node.pos,
[&distances](size_t id, float distance) {
distances.emplace_back(NodeDistance{id, distance});
},
prev_dist_max);
}, K, prev_dist_max);
std::sort(distances.begin(), distances.end(),
[](auto &a, auto &b) { return a.distance < b.distance; });
@ -41,11 +42,13 @@ void build_tree(PointCloud &nodes, Builder &builder)
if (distances.empty()) {
builder.report_unroutable(node);
ptsqueue.pop();
K = initK;
prev_dist_max = 0.;
continue;
}
prev_dist_max = distances.back().distance;
K *= 2;
auto closest_it = distances.begin();
bool routed = false;
@ -146,6 +149,7 @@ void build_tree(PointCloud &nodes, Builder &builder)
if (routed) {
ptsqueue.pop();
prev_dist_max = 0.;
K = initK;
}
}

View file

@ -185,8 +185,11 @@ public:
size_t reachable_count() const { return m_reachable_cnt; }
template<size_t K, class Fn>
void foreach_reachable(const Vec3f &pos, Fn &&visitor, double min_dist = 0.)
template<class Fn>
void foreach_reachable(const Vec3f &pos,
Fn &&visitor,
size_t k,
double min_dist = 0.)
{
// Fake output iterator
struct Output {
@ -203,14 +206,15 @@ public:
namespace bgi = boost::geometry::index;
size_t cnt = 0;
auto filter = bgi::satisfies([this, &pos, min_dist, &cnt](const PointIndexEl &e) {
double d = get_distance(pos, e.second);
cnt++;
return m_searchable_indices[e.second] && !isinf(d) && d > min_dist;
});
size_t cnt = 0;
auto filter = bgi::satisfies(
[this, &pos, min_dist, &cnt](const PointIndexEl &e) {
cnt++;
double d = get_distance(pos, e.second);
return m_searchable_indices[e.second] && !isinf(d) && d > min_dist;
});
m_ktree.query(bgi::nearest(pos, K) && filter, Output{this, pos, visitor});
m_ktree.query(bgi::nearest(pos, k) && filter, Output{this, pos, visitor});
}
auto start_queue()