Measuring: further separating frontend and backend

This commit is contained in:
Lukas Matena 2022-07-27 09:58:21 +02:00 committed by enricoturri1966
parent 9d5e9e8870
commit 7d6d33f92c
4 changed files with 233 additions and 122 deletions

View file

@ -39,16 +39,19 @@ public:
float area; float area;
}; };
const std::vector<SurfaceFeature*>& get_features() const; const std::vector<const SurfaceFeature*>& get_features() const;
const SurfaceFeature* get_feature(size_t face_idx, const Vec3d& point) const;
const std::vector<std::vector<int>> get_planes_triangle_indices() const;
private: private:
void update_planes(); void update_planes();
void extract_features(PlaneData& plane); void extract_features();
void save_features(); void save_features();
std::vector<PlaneData> m_planes; std::vector<PlaneData> m_planes;
std::vector<SurfaceFeature*> m_features; std::vector<size_t> m_face_to_plane;
std::vector<const SurfaceFeature*> m_features;
const indexed_triangle_set& m_its; const indexed_triangle_set& m_its;
}; };
@ -61,14 +64,7 @@ MeasuringImpl::MeasuringImpl(const indexed_triangle_set& its)
: m_its{its} : m_its{its}
{ {
update_planes(); update_planes();
extract_features();
for (PlaneData& plane : m_planes) {
extract_features(plane);
plane.borders.clear();
plane.borders.shrink_to_fit();
}
save_features(); save_features();
} }
@ -80,7 +76,7 @@ void MeasuringImpl::update_planes()
// Now we'll go through all the facets and append Points of facets sharing the same normal. // Now we'll go through all the facets and append Points of facets sharing the same normal.
// This part is still performed in mesh coordinate system. // This part is still performed in mesh coordinate system.
const size_t num_of_facets = m_its.indices.size(); const size_t num_of_facets = m_its.indices.size();
std::vector<size_t> face_to_plane(num_of_facets, size_t(-1)); m_face_to_plane.resize(num_of_facets, size_t(-1));
const std::vector<Vec3f> face_normals = its_face_normals(m_its); const std::vector<Vec3f> face_normals = its_face_normals(m_its);
const std::vector<Vec3i> face_neighbors = its_face_neighbors(m_its); const std::vector<Vec3i> face_neighbors = its_face_neighbors(m_its);
std::vector<int> facet_queue(num_of_facets, 0); std::vector<int> facet_queue(num_of_facets, 0);
@ -95,10 +91,10 @@ void MeasuringImpl::update_planes()
while (1) { while (1) {
// Find next unvisited triangle: // Find next unvisited triangle:
for (; seed_facet_idx < num_of_facets; ++ seed_facet_idx) for (; seed_facet_idx < num_of_facets; ++ seed_facet_idx)
if (face_to_plane[seed_facet_idx] == size_t(-1)) { if (m_face_to_plane[seed_facet_idx] == size_t(-1)) {
facet_queue[facet_queue_cnt ++] = seed_facet_idx; facet_queue[facet_queue_cnt ++] = seed_facet_idx;
normal_ptr = &face_normals[seed_facet_idx]; normal_ptr = &face_normals[seed_facet_idx];
face_to_plane[seed_facet_idx] = m_planes.size(); m_face_to_plane[seed_facet_idx] = m_planes.size();
m_planes.emplace_back(); m_planes.emplace_back();
break; break;
} }
@ -111,10 +107,10 @@ void MeasuringImpl::update_planes()
if (is_same_normal(this_normal, *normal_ptr)) { if (is_same_normal(this_normal, *normal_ptr)) {
const Vec3i& face = m_its.indices[facet_idx]; const Vec3i& face = m_its.indices[facet_idx];
face_to_plane[facet_idx] = m_planes.size() - 1; m_face_to_plane[facet_idx] = m_planes.size() - 1;
m_planes.back().facets.emplace_back(facet_idx); m_planes.back().facets.emplace_back(facet_idx);
for (int j = 0; j < 3; ++ j) for (int j = 0; j < 3; ++ j)
if (int neighbor_idx = face_neighbors[facet_idx][j]; neighbor_idx >= 0 && face_to_plane[neighbor_idx] == size_t(-1)) if (int neighbor_idx = face_neighbors[facet_idx][j]; neighbor_idx >= 0 && m_face_to_plane[neighbor_idx] == size_t(-1))
facet_queue[facet_queue_cnt ++] = neighbor_idx; facet_queue[facet_queue_cnt ++] = neighbor_idx;
} }
} }
@ -123,7 +119,7 @@ void MeasuringImpl::update_planes()
std::sort(m_planes.back().facets.begin(), m_planes.back().facets.end()); std::sort(m_planes.back().facets.begin(), m_planes.back().facets.end());
} }
assert(std::none_of(face_to_plane.begin(), face_to_plane.end(), [](size_t val) { return val == size_t(-1); })); assert(std::none_of(m_face_to_plane.begin(), m_face_to_plane.end(), [](size_t val) { return val == size_t(-1); }));
SurfaceMesh sm(m_its); SurfaceMesh sm(m_its);
for (int plane_id=0; plane_id < int(m_planes.size()); ++plane_id) { for (int plane_id=0; plane_id < int(m_planes.size()); ++plane_id) {
@ -133,9 +129,9 @@ void MeasuringImpl::update_planes()
std::vector<std::array<bool, 3>> visited(facets.size(), {false, false, false}); std::vector<std::array<bool, 3>> visited(facets.size(), {false, false, false});
for (int face_id=0; face_id<int(facets.size()); ++face_id) { for (int face_id=0; face_id<int(facets.size()); ++face_id) {
assert(face_to_plane[facets[face_id]] == plane_id); assert(m_face_to_plane[facets[face_id]] == plane_id);
for (int edge_id=0; edge_id<3; ++edge_id) { for (int edge_id=0; edge_id<3; ++edge_id) {
if (visited[face_id][edge_id] || face_to_plane[face_neighbors[facets[face_id]][edge_id]] == plane_id) { if (visited[face_id][edge_id] || m_face_to_plane[face_neighbors[facets[face_id]][edge_id]] == plane_id) {
visited[face_id][edge_id] = true; visited[face_id][edge_id] = true;
continue; continue;
} }
@ -161,7 +157,7 @@ void MeasuringImpl::update_planes()
do { do {
const Halfedge_index he_orig = he; const Halfedge_index he_orig = he;
he = sm.next_around_target(he); he = sm.next_around_target(he);
while ( face_to_plane[sm.face(he)] == plane_id && he != he_orig) while ( m_face_to_plane[sm.face(he)] == plane_id && he != he_orig)
he = sm.next_around_target(he); he = sm.next_around_target(he);
he = sm.opposite(he); he = sm.opposite(he);
@ -194,78 +190,89 @@ void MeasuringImpl::update_planes()
void MeasuringImpl::extract_features(PlaneData& plane) void MeasuringImpl::extract_features()
{ {
plane.surface_features.clear();
const Vec3d& normal = plane.normal;
const double edge_threshold = 25. * (M_PI/180.); const double edge_threshold = 25. * (M_PI/180.);
std::vector<double> angles; std::vector<double> angles;
Eigen::Quaterniond q;
q.setFromTwoVectors(plane.normal, Vec3d::UnitZ()); for (int i=0; i<m_planes.size(); ++i) {
Transform3d trafo = Transform3d::Identity(); PlaneData& plane = m_planes[i];
trafo.rotate(q); plane.surface_features.clear();
const Vec3d& normal = plane.normal;
Eigen::Quaterniond q;
q.setFromTwoVectors(plane.normal, Vec3d::UnitZ());
Transform3d trafo = Transform3d::Identity();
trafo.rotate(q);
for (const std::vector<Vec3d>& border : plane.borders) {
assert(border.size() > 1);
int start_idx = -1;
// First calculate angles at all the vertices.
angles.clear();
for (int i=0; i<int(border.size()); ++i) {
const Vec3d& v2 = (i == 0 ? border[0] - border[border.size()-1]
: border[i] - border[i-1]);
const Vec3d& v1 = i == border.size()-1 ? border[0] - border.back()
: border[i+1] - border[i];
double angle = -atan2(normal.dot(v1.cross(v2)), v1.dot(v2));
if (angle < -M_PI/2.)
angle += M_PI;
angles.push_back(angle);
}
assert(border.size() == angles.size());
bool circle = false;
for (const std::vector<Vec3d>& border : plane.borders) { std::vector<std::pair<size_t, size_t>> circles;
assert(border.size() > 1); for (int i=1; i<angles.size(); ++i) {
int start_idx = -1; if (angles[i] < edge_threshold && Slic3r::is_approx(angles[i], angles[i-1]) && i != angles.size()-1 ) {
// circle
if (! circle) {
// First calculate angles at all the vertices. circle = true;
angles.clear(); start_idx = std::max(0, i-2);
for (int i=0; i<int(border.size()); ++i) { }
const Vec3d& v2 = (i == 0 ? border[0] - border[border.size()-1] } else {
: border[i] - border[i-1]); if (circle) {
const Vec3d& v1 = i == border.size()-1 ? border[0] - border.back() circles.emplace_back(start_idx, i);
: border[i+1] - border[i]; circle = false;
double angle = -atan2(normal.dot(v1.cross(v2)), v1.dot(v2)); }
if (angle < -M_PI/2.)
angle += M_PI;
angles.push_back(angle);
}
assert(border.size() == angles.size());
bool circle = false;
std::vector<std::pair<size_t, size_t>> circles;
for (int i=1; i<angles.size(); ++i) {
if (angles[i] < edge_threshold && Slic3r::is_approx(angles[i], angles[i-1]) && i != angles.size()-1 ) {
// circle
if (! circle) {
circle = true;
start_idx = std::max(0, i-2);
}
} else {
if (circle) {
circles.emplace_back(start_idx, i);
circle = false;
} }
} }
// We have the circles. Now go around again and pick edges.
int cidx = 0; // index of next circle in the way
for (int i=1; i<int(border.size()); ++i) {
if (cidx < circles.size() && i > circles[cidx].first)
i = circles[cidx++].second;
else plane.surface_features.emplace_back(std::unique_ptr<SurfaceFeature>(
new Edge(border[i-1], border[i])));
}
// FIXME Throw away / do not create edges which are parts of circles or
// which lead to circle points (unless they belong to the same plane.)
// FIXME Check and merge first and last circle if needed.
// Now create the circle-typed surface features.
for (const auto& [start_idx, end_idx] : circles) {
std::pair<Vec3d, double> center_and_radius = get_center_and_radius(border, start_idx, end_idx, trafo);
plane.surface_features.emplace_back(std::unique_ptr<SurfaceFeature>(
new Circle(center_and_radius.first, center_and_radius.second)));
}
} }
// We have the circles. Now go around again and pick edges. // The last surface feature is the plane itself.
int cidx = 0; // index of next circle in the way plane.surface_features.emplace_back(std::unique_ptr<SurfaceFeature>(
for (int i=1; i<int(border.size()); ++i) { new Plane(i)));
if (cidx < circles.size() && i > circles[cidx].first)
i = circles[cidx++].second;
else plane.surface_features.emplace_back(std::unique_ptr<SurfaceFeature>(
new Edge(border[i-1], border[i])));
}
// FIXME Throw away / do not create edges which are parts of circles.
// FIXME Check and maybe merge first and last circle.
for (const auto& [start_idx, end_idx] : circles) {
std::pair<Vec3d, double> center_and_radius = get_center_and_radius(border, start_idx, end_idx, trafo);
plane.surface_features.emplace_back(std::unique_ptr<SurfaceFeature>(
new Circle(center_and_radius.first, center_and_radius.second)
));
}
plane.borders.clear();
plane.borders.shrink_to_fit();
} }
} }
@ -277,7 +284,7 @@ void MeasuringImpl::save_features()
for (PlaneData& plane : m_planes) for (PlaneData& plane : m_planes)
//PlaneData& plane = m_planes[0]; //PlaneData& plane = m_planes[0];
{ {
for (std::unique_ptr<SurfaceFeature>& feature : plane.surface_features) { for (const std::unique_ptr<SurfaceFeature>& feature : plane.surface_features) {
m_features.emplace_back(feature.get()); m_features.emplace_back(feature.get());
} }
} }
@ -285,13 +292,51 @@ void MeasuringImpl::save_features()
const std::vector<SurfaceFeature*>& MeasuringImpl::get_features() const const SurfaceFeature* MeasuringImpl::get_feature(size_t face_idx, const Vec3d& point) const
{
if (face_idx >= m_face_to_plane.size())
return nullptr;
const PlaneData& plane = m_planes[m_face_to_plane[face_idx]];
const SurfaceFeature* closest_feature = nullptr;
double min_dist = std::numeric_limits<double>::max();
for (const std::unique_ptr<SurfaceFeature>& feature : plane.surface_features) {
double dist = Measuring::get_distance(feature.get(), &point);
if (dist < 0.5 && dist < min_dist) {
min_dist = std::min(dist, min_dist);
closest_feature = feature.get();
}
}
if (closest_feature)
return closest_feature;
// Nothing detected, return the plane as a whole.
assert(plane.surface_features.back().get()->get_type() == SurfaceFeatureType::Plane);
return plane.surface_features.back().get();
}
const std::vector<const SurfaceFeature*>& MeasuringImpl::get_features() const
{ {
return m_features; return m_features;
} }
const std::vector<std::vector<int>> MeasuringImpl::get_planes_triangle_indices() const
{
std::vector<std::vector<int>> out;
for (const PlaneData& plane : m_planes)
out.emplace_back(plane.facets);
return out;
}
@ -309,12 +354,39 @@ Measuring::Measuring(const indexed_triangle_set& its)
Measuring::~Measuring() {} Measuring::~Measuring() {}
const std::vector<SurfaceFeature*>& Measuring::get_features() const const std::vector<const SurfaceFeature*>& Measuring::get_features() const
{ {
return priv->get_features(); return priv->get_features();
} }
const SurfaceFeature* Measuring::get_feature(size_t face_idx, const Vec3d& point) const
{
return priv->get_feature(face_idx, point);
}
const std::vector<std::vector<int>> Measuring::get_planes_triangle_indices() const
{
return priv->get_planes_triangle_indices();
}
double Measuring::get_distance(const SurfaceFeature* feature, const Vec3d* pt)
{
if (feature->get_type() == SurfaceFeatureType::Edge) {
const Edge* edge = static_cast<const Edge*>(feature);
const auto& [s,e] = edge->get_edge();
Eigen::ParametrizedLine<double, 3> line(s, (e-s).normalized());
return line.distance(*pt);
}
return std::numeric_limits<double>::max();
}

View file

@ -48,8 +48,12 @@ private:
class Plane : public SurfaceFeature { class Plane : public SurfaceFeature {
public: public:
Plane(int idx) : m_idx(idx) {}
SurfaceFeatureType get_type() const override { return SurfaceFeatureType::Plane; } SurfaceFeatureType get_type() const override { return SurfaceFeatureType::Plane; }
int get_plane_idx() const { return m_idx; } // index into vector provided by Measuring::get_plane_triangle_indices
private:
int m_idx;
}; };
@ -64,22 +68,23 @@ public:
~Measuring(); ~Measuring();
// Return a reference to a list of all features identified on the its. // Return a reference to a list of all features identified on the its.
const std::vector<SurfaceFeature*>& get_features() const; [[deprecated]]const std::vector<const SurfaceFeature*>& get_features() const;
// Given a face_idx where the mouse cursor points, return a feature that // Given a face_idx where the mouse cursor points, return a feature that
// should be highlighted or nullptr. // should be highlighted or nullptr.
const SurfaceFeature* get_feature(size_t face_idx, const Vec3d& point) const; const SurfaceFeature* get_feature(size_t face_idx, const Vec3d& point) const;
// Returns a list of triangle indices for each identified plane. Each
// Plane object contains an index into this vector.
const std::vector<std::vector<int>> get_planes_triangle_indices() const;
// Returns distance between two SurfaceFeatures. // Returns distance between two SurfaceFeatures.
static double get_distance(const SurfaceFeature* a, const SurfaceFeature* b); static double get_distance(const SurfaceFeature* a, const SurfaceFeature* b);
// Returns true if an x/y/z distance between features makes sense. // Returns distance between a SurfaceFeature and a point.
// If so, result contains the distances. static double get_distance(const SurfaceFeature* a, const Vec3d* pt);
static bool get_distances(const SurfaceFeature* a, const SurfaceFeature* b, std::array<double, 3>& result);
// Returns true if an x/y/z distance between feature and a point makes sense.
// If so, result contains the distances.
static bool get_axis_aligned_distances(const SurfaceFeature* feature, const Vec3d* pt, std::array<double, 3>& result);
// Returns true if measuring angles between features makes sense. // Returns true if measuring angles between features makes sense.
// If so, result contains the angle in radians. // If so, result contains the angle in radians.
@ -87,7 +92,6 @@ public:
private: private:
std::unique_ptr<MeasuringImpl> priv; std::unique_ptr<MeasuringImpl> priv;
}; };

View file

@ -144,6 +144,7 @@ void GLGizmoMeasure::on_render()
m_imgui->begin(std::string("DEBUG")); m_imgui->begin(std::string("DEBUG"));
m_imgui->checkbox(wxString("Show all features"), m_show_all); m_imgui->checkbox(wxString("Show all features"), m_show_all);
m_imgui->checkbox(wxString("Show all planes"), m_show_planes);
Vec3f pos; Vec3f pos;
Vec3f normal; Vec3f normal;
@ -157,37 +158,51 @@ void GLGizmoMeasure::on_render()
if (m_show_all) { std::vector<const Measure::SurfaceFeature*> features = {m_measuring->get_feature(facet_idx, pos.cast<double>())};
const std::vector<Measure::SurfaceFeature*> features = m_measuring->get_features(); if (m_show_all) {
for (const Measure::SurfaceFeature* feature : features) { features = m_measuring->get_features();
features.erase(std::remove_if(features.begin(), features.end(),
if (feature->get_type() == Measure::SurfaceFeatureType::Circle) { [](const Measure::SurfaceFeature* f) {
const auto* circle = static_cast<const Measure::Circle*>(feature); return f->get_type() == Measure::SurfaceFeatureType::Plane;
Transform3d view_feature_matrix = view_model_matrix * Transform3d(Eigen::Translation3d(circle->get_center())); }), features.end());
view_feature_matrix = view_model_matrix * Transform3d(Eigen::Translation3d(circle->get_center())); }
view_feature_matrix.scale(0.5);
shader->set_uniform("view_model_matrix", view_feature_matrix);
m_vbo_sphere.set_color(ColorRGBA(0.f, 1.f, 0.f, 1.f));
m_vbo_sphere.render();
}
else if (feature->get_type() == Measure::SurfaceFeatureType::Edge) { for (const Measure::SurfaceFeature* feature : features) {
const auto* edge = static_cast<const Measure::Edge*>(feature); if (! feature)
auto& [start, end] = edge->get_edge(); continue;
Transform3d view_feature_matrix = view_model_matrix * Transform3d(Eigen::Translation3d(start));
auto q = Eigen::Quaternion<double>::FromTwoVectors(Vec3d::UnitZ(), end - start);
view_feature_matrix *= q;
view_feature_matrix.scale(Vec3d(0.075, 0.075, (end - start).norm()));
shader->set_uniform("view_model_matrix", view_feature_matrix);
m_vbo_cylinder.set_color(ColorRGBA(0.7f, 0.7f, 0.f, 1.f));
m_vbo_cylinder.render();
}
if (feature->get_type() == Measure::SurfaceFeatureType::Circle) {
const auto* circle = static_cast<const Measure::Circle*>(feature);
Transform3d view_feature_matrix = view_model_matrix * Transform3d(Eigen::Translation3d(circle->get_center()));
view_feature_matrix = view_model_matrix * Transform3d(Eigen::Translation3d(circle->get_center()));
view_feature_matrix.scale(0.5);
shader->set_uniform("view_model_matrix", view_feature_matrix);
m_vbo_sphere.set_color(ColorRGBA(0.f, 1.f, 0.f, 1.f));
m_vbo_sphere.render();
}
else if (feature->get_type() == Measure::SurfaceFeatureType::Edge) {
const auto* edge = static_cast<const Measure::Edge*>(feature);
auto& [start, end] = edge->get_edge();
Transform3d view_feature_matrix = view_model_matrix * Transform3d(Eigen::Translation3d(start));
auto q = Eigen::Quaternion<double>::FromTwoVectors(Vec3d::UnitZ(), end - start);
view_feature_matrix *= q;
view_feature_matrix.scale(Vec3d(0.075, 0.075, (end - start).norm()));
shader->set_uniform("view_model_matrix", view_feature_matrix);
m_vbo_cylinder.set_color(ColorRGBA(0.8f, 0.2f, 0.2f, 1.f));
m_vbo_cylinder.render();
}
else if (feature->get_type() == Measure::SurfaceFeatureType::Plane) {
const auto* plane = static_cast<const Measure::Plane*>(feature);
assert(plane->get_plane_idx() < m_plane_models.size());
m_plane_models[plane->get_plane_idx()]->render();
} }
shader->set_uniform("view_model_matrix", view_model_matrix);
} }
shader->set_uniform("view_model_matrix", view_model_matrix);
if (m_show_planes)
for (const auto& glmodel : m_plane_models)
glmodel->render();
m_imgui->end(); m_imgui->end();
} }
@ -244,7 +259,25 @@ void GLGizmoMeasure::update_if_needed()
return; return;
UPDATE: UPDATE:
m_measuring.reset(new Measure::Measuring(mo->volumes.front()->mesh().its)); const indexed_triangle_set& its = mo->volumes.front()->mesh().its;
m_measuring.reset(new Measure::Measuring(its));
m_plane_models.clear();
const std::vector<std::vector<int>> planes_triangles = m_measuring->get_planes_triangle_indices();
for (const std::vector<int>& triangle_indices : planes_triangles) {
m_plane_models.emplace_back(std::unique_ptr<GLModel>(new GLModel()));
GUI::GLModel::Geometry init_data;
init_data.format = { GUI::GLModel::Geometry::EPrimitiveType::Triangles, GUI::GLModel::Geometry::EVertexLayout::P3 };
init_data.color = ColorRGBA(0.9f, 0.9f, 0.9f, 0.5f);
int i = 0;
for (int idx : triangle_indices) {
init_data.add_vertex(its.vertices[its.indices[idx][0]]);
init_data.add_vertex(its.vertices[its.indices[idx][1]]);
init_data.add_vertex(its.vertices[its.indices[idx][2]]);
init_data.add_triangle(i, i+1, i+2);
i+=3;
}
m_plane_models.back()->init_from(std::move(init_data));
}
// Let's save what we calculated it from: // Let's save what we calculated it from:
m_volumes_matrices.clear(); m_volumes_matrices.clear();

View file

@ -45,7 +45,9 @@ private:
int m_mouse_pos_x; int m_mouse_pos_x;
int m_mouse_pos_y; int m_mouse_pos_y;
bool m_show_all = true; bool m_show_all = false;
bool m_show_planes = false;
std::vector<std::unique_ptr<GLModel>> m_plane_models;
void update_if_needed(); void update_if_needed();
void set_flattening_data(const ModelObject* model_object); void set_flattening_data(const ModelObject* model_object);