mirror of
https://git.mirrors.martin98.com/https://github.com/prusa3d/PrusaSlicer.git
synced 2025-08-15 21:05:57 +08:00
implemented new comparator method, but gives wrong results; TODO fix
implemented debug exports
This commit is contained in:
parent
05fa9776fe
commit
55d79c1cb7
@ -9,55 +9,68 @@
|
||||
#include <boost/nowide/cstdio.hpp>
|
||||
#include <boost/log/trivial.hpp>
|
||||
|
||||
#define DEBUG_FILES
|
||||
|
||||
#ifdef DEBUG_FILES
|
||||
#include <boost/nowide/cstdio.hpp>
|
||||
#include "libslic3r/Utils.hpp"
|
||||
#endif
|
||||
|
||||
namespace Slic3r::GUI {
|
||||
|
||||
//TODO close in seprate namespace
|
||||
|
||||
static constexpr size_t empty { size_t(-1) };
|
||||
|
||||
struct Triangle {
|
||||
stl_triangle_vertex_indices indices;
|
||||
Vec3f normal;
|
||||
size_t index_in_its;
|
||||
Vec3i neighbours;
|
||||
Vec3f z_coords_sorted_asc;
|
||||
size_t order_by_z;
|
||||
|
||||
//members updated during algorithm
|
||||
float strength { 0.0 };
|
||||
bool supports = false;
|
||||
bool visited = false;
|
||||
bool gathering_supports = false;
|
||||
size_t group_id = 0;
|
||||
};
|
||||
|
||||
inline Vec3f value_to_rgbf(float minimum, float maximum, float value) {
|
||||
float ratio = 2.0f * (value - minimum) / (maximum - minimum);
|
||||
float b = std::max(0.0f, (1.0f - ratio));
|
||||
float r = std::max(0.0f, (ratio - 1.0f));
|
||||
float g = 1.0f - b - r;
|
||||
return Vec3f { r, g, b };
|
||||
}
|
||||
|
||||
inline float face_area(const stl_vertex vertex[3]) {
|
||||
return (vertex[1] - vertex[0]).cross(vertex[2] - vertex[1]).norm() / 2;
|
||||
}
|
||||
|
||||
inline float its_face_area(const indexed_triangle_set &its, const stl_triangle_vertex_indices face)
|
||||
{
|
||||
inline float its_face_area(const indexed_triangle_set &its, const stl_triangle_vertex_indices face) {
|
||||
const stl_vertex vertices[3] { its.vertices[face[0]], its.vertices[face[1]], its.vertices[face[2]] };
|
||||
return face_area(vertices);
|
||||
}
|
||||
inline float its_face_area(const indexed_triangle_set &its, const int face_idx)
|
||||
{
|
||||
inline float its_face_area(const indexed_triangle_set &its, const int face_idx) {
|
||||
return its_face_area(its, its.indices[face_idx]);
|
||||
}
|
||||
|
||||
struct SupportPlacerMesh {
|
||||
const float gather_phase_target = 3.0;
|
||||
const float gather_phase_target;
|
||||
const float limit_angle_cos;
|
||||
|
||||
indexed_triangle_set mesh;
|
||||
std::vector<Triangle> triangles;
|
||||
std::vector<size_t> triangle_indexes_by_z;
|
||||
|
||||
explicit SupportPlacerMesh(indexed_triangle_set &&t_mesh) :
|
||||
mesh(t_mesh) {
|
||||
explicit SupportPlacerMesh(indexed_triangle_set &&t_mesh, float dot_limit, float patch_size) :
|
||||
mesh(t_mesh), limit_angle_cos(dot_limit), gather_phase_target(patch_size) {
|
||||
auto neighbours = its_face_neighbors_par(mesh);
|
||||
|
||||
auto z_coords_sorted = [&](const size_t &face_index) {
|
||||
const auto &face = mesh.indices[face_index];
|
||||
Vec3f z_coords = Vec3f { mesh.vertices[face.x()].z(), mesh.vertices[face.y()].z(),
|
||||
mesh.vertices[face.z()].z() };
|
||||
Vec3f z_coords = Vec3f { mesh.vertices[face.x()].z(), mesh.vertices[face.y()].z(), mesh.vertices[face.z()].z() };
|
||||
;
|
||||
std::sort(z_coords.data(), z_coords.data() + 3);
|
||||
return z_coords;
|
||||
@ -65,80 +78,192 @@ struct SupportPlacerMesh {
|
||||
|
||||
for (size_t face_index = 0; face_index < mesh.indices.size(); ++face_index) {
|
||||
Vec3f normal = its_face_normal(mesh, face_index);
|
||||
triangles.push_back(Triangle { mesh.indices[face_index], normal, face_index, neighbours[face_index],
|
||||
z_coords_sorted(face_index) });
|
||||
triangles.push_back(
|
||||
Triangle { mesh.indices[face_index], normal, face_index, neighbours[face_index], z_coords_sorted(face_index) });
|
||||
triangle_indexes_by_z.push_back(face_index);
|
||||
}
|
||||
|
||||
std::sort(triangle_indexes_by_z.begin(), triangle_indexes_by_z.end(),
|
||||
[&](const size_t &left, const size_t &right) {
|
||||
for (int i = 0; i < triangles[left].z_coords_sorted_asc.size(); ++i) {
|
||||
if (triangles[left].z_coords_sorted_asc.data()[i]
|
||||
< triangles[right].z_coords_sorted_asc.data()[i]) {
|
||||
std::sort(triangle_indexes_by_z.begin(), triangle_indexes_by_z.end(), [&](const size_t &left, const size_t &right) {
|
||||
for (int i = 0; i < 2; ++i) {
|
||||
if (triangles[left].z_coords_sorted_asc.data()[i] < triangles[right].z_coords_sorted_asc.data()[i]) {
|
||||
return true;
|
||||
}
|
||||
if (triangles[left].z_coords_sorted_asc.data()[i]
|
||||
> triangles[right].z_coords_sorted_asc.data()[i]) {
|
||||
if (triangles[left].z_coords_sorted_asc.data()[i] > triangles[right].z_coords_sorted_asc.data()[i]) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
return false;
|
||||
// if two bottom points of both triangles have same z heights
|
||||
// assume neighbours (but do not rely on that), so one edge is probably common
|
||||
|
||||
// find the other edge of both triangles that originates in the lowest point
|
||||
|
||||
//extract vertices
|
||||
std::vector<Vec3f> vertices_left { mesh.vertices[triangles[left].indices[0]], mesh.vertices[triangles[left].indices[1]], mesh.vertices[triangles[left].indices[2]]};
|
||||
|
||||
std::vector<Vec3f> vertices_right { mesh.vertices[triangles[right].indices[0]], mesh.vertices[triangles[right].indices[1]],
|
||||
mesh.vertices[triangles[right].indices[2]] };
|
||||
|
||||
//sort vertices by z
|
||||
std::sort(vertices_left.begin(), vertices_left.end(), [](const Vec3f &a, const Vec3f &b) {
|
||||
return a.z() < b.z();
|
||||
});
|
||||
|
||||
std::sort(vertices_right.begin(), vertices_right.end(), [](const Vec3f &a, const Vec3f &b) {
|
||||
return a.z() < b.z();
|
||||
});
|
||||
|
||||
//get the other edge
|
||||
Vec3f lowest_edge_left = (vertices_left[2] - vertices_left[0]).normalized();
|
||||
Vec3f lowest_edge_right = (vertices_right[2] - vertices_right[0]).normalized();
|
||||
|
||||
Vec3f down = -Vec3f::UnitZ();
|
||||
|
||||
//Pick the one that is more downwards
|
||||
return lowest_edge_left.dot(down) > lowest_edge_right.dot(down);
|
||||
});
|
||||
|
||||
for (size_t order_index = 0; order_index < triangle_indexes_by_z.size(); ++order_index) {
|
||||
triangles[triangle_indexes_by_z[order_index]].order_by_z = order_index;
|
||||
}
|
||||
|
||||
for (Triangle &triangle : triangles) {
|
||||
std::sort(begin(triangle.neighbours), end(triangle.neighbours), [&](const size_t left, const size_t right) {
|
||||
return triangles[left].order_by_z < triangles[right].order_by_z;
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
const float calculate_support_cost(float dot_product, size_t index_in_its) const {
|
||||
float dot_pow = dot_product * dot_product;
|
||||
float weight = (1.0 / (-1.2 * dot_pow + 2)) - 0.5;
|
||||
float weight = std::max(0.0f, (dot_product - limit_angle_cos) / (1.0f - limit_angle_cos));
|
||||
return weight * its_face_area(mesh, index_in_its);
|
||||
}
|
||||
|
||||
void find_support_areas() {
|
||||
size_t next_group_id = 1;
|
||||
for (const size_t ¤t_index : triangle_indexes_by_z) {
|
||||
Triangle ¤t = triangles[current_index];
|
||||
float dot_product = current.normal.dot(-Vec3f::UnitZ());
|
||||
|
||||
if (dot_product < 0.2) {
|
||||
current.strength = gather_phase_target * 100.0;
|
||||
current.visited = true;
|
||||
current.supports = false;
|
||||
} else {
|
||||
float neighbours_strength_sum = 0;
|
||||
size_t group_id = 0;
|
||||
for (const auto &neighbour_index : current.neighbours) {
|
||||
const Triangle &neighbour = triangles[neighbour_index];
|
||||
if (!neighbour.visited) {
|
||||
//not visited yet, ignore
|
||||
continue;
|
||||
} else {
|
||||
if (group_id == 0) {
|
||||
group_id = neighbour.group_id;
|
||||
}
|
||||
|
||||
neighbours_strength_sum += neighbour.strength;
|
||||
float support_cost = calculate_support_cost(dot_product, current.index_in_its);
|
||||
if (neighbour.gathering_supports || neighbour.strength < support_cost) {
|
||||
continue;
|
||||
} else {
|
||||
current.strength = neighbour.strength - support_cost;
|
||||
current.strength = std::max(0.0f, neighbour.strength - support_cost);
|
||||
current.visited = true;
|
||||
current.supports = false;
|
||||
current.gathering_supports = false;
|
||||
current.group_id = neighbour.group_id;
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (!current.visited) {
|
||||
if (dot_product > 0.2) {
|
||||
current.supports = true;
|
||||
current.visited = true;
|
||||
current.strength = neighbours_strength_sum + its_face_area(mesh, current.index_in_its);
|
||||
current.gathering_supports = current.strength < gather_phase_target;
|
||||
current.group_id = next_group_id;
|
||||
next_group_id++;
|
||||
} else {
|
||||
current.strength = gather_phase_target;
|
||||
current.visited = true;
|
||||
current.supports = false;
|
||||
current.gathering_supports = false;
|
||||
current.group_id = group_id;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#ifdef DEBUG_FILES
|
||||
void debug_export() const {
|
||||
Slic3r::CNumericLocalesSetter locales_setter;
|
||||
|
||||
{
|
||||
std::string file_name = debug_out_path("strength.obj");
|
||||
|
||||
FILE *fp = boost::nowide::fopen(file_name.c_str(), "w");
|
||||
if (fp == nullptr) {
|
||||
BOOST_LOG_TRIVIAL(error)
|
||||
<< "stl_write_obj: Couldn't open " << file_name << " for writing";
|
||||
return;
|
||||
}
|
||||
|
||||
for (int i = 0; i < triangles.size(); ++i) {
|
||||
Vec3f color = value_to_rgbf(0.0f, 2.0f * gather_phase_target, triangles[i].strength);
|
||||
for (int index = 0; index < 3; ++index) {
|
||||
fprintf(fp, "v %f %f %f %f %f %f\n", mesh.vertices[triangles[i].indices[index]](0),
|
||||
mesh.vertices[triangles[i].indices[index]](1), mesh.vertices[triangles[i].indices[index]](2), color(0),
|
||||
color(1), color(2));
|
||||
}
|
||||
fprintf(fp, "f %d %d %d\n", i * 3 + 1, i * 3 + 2, i * 3 + 3);
|
||||
}
|
||||
fclose(fp);
|
||||
}
|
||||
{
|
||||
std::string file_name = debug_out_path("groups.obj");
|
||||
FILE *fp = boost::nowide::fopen(file_name.c_str(), "w");
|
||||
if (fp == nullptr) {
|
||||
BOOST_LOG_TRIVIAL(error)
|
||||
<< "stl_write_obj: Couldn't open " << file_name << " for writing";
|
||||
return;
|
||||
}
|
||||
|
||||
for (size_t i = 0; i < triangles.size(); ++i) {
|
||||
Vec3f color = value_to_rgbf(0.0f, 9.0f, float(triangles[i].group_id % 10));
|
||||
for (int index = 0; index < 3; ++index) {
|
||||
fprintf(fp, "v %f %f %f %f %f %f\n", mesh.vertices[triangles[i].indices[index]](0),
|
||||
mesh.vertices[triangles[i].indices[index]](1), mesh.vertices[triangles[i].indices[index]](2), color(0),
|
||||
color(1), color(2));
|
||||
}
|
||||
fprintf(fp, "f %d %d %d\n", i * 3 + 1, i * 3 + 2, i * 3 + 3);
|
||||
}
|
||||
fclose(fp);
|
||||
}
|
||||
{
|
||||
std::string file_name = debug_out_path("sorted.obj");
|
||||
FILE *fp = boost::nowide::fopen(file_name.c_str(), "w");
|
||||
if (fp == nullptr) {
|
||||
BOOST_LOG_TRIVIAL(error)
|
||||
<< "stl_write_obj: Couldn't open " << file_name << " for writing";
|
||||
return;
|
||||
}
|
||||
|
||||
for (size_t i = 0; i < triangle_indexes_by_z.size(); ++i) {
|
||||
const Triangle& triangle = triangles[triangle_indexes_by_z[i]];
|
||||
Vec3f color = value_to_rgbf(0.0f, 9999.0f, float(i % 10000));
|
||||
for (int index = 0; index < 3; ++index) {
|
||||
fprintf(fp, "v %f %f %f %f %f %f\n", mesh.vertices[triangles[i].indices[index]](0),
|
||||
mesh.vertices[triangles[i].indices[index]](1), mesh.vertices[triangles[i].indices[index]](2), color(0),
|
||||
color(1), color(2));
|
||||
}
|
||||
fprintf(fp, "f %d %d %d\n", i * 3 + 1, i * 3 + 2, i * 3 + 3);
|
||||
}
|
||||
fclose(fp);
|
||||
}
|
||||
}
|
||||
#endif
|
||||
|
||||
}
|
||||
;
|
||||
|
||||
inline void do_experimental_support_placement(indexed_triangle_set mesh,
|
||||
TriangleSelectorGUI *selector) {
|
||||
SupportPlacerMesh support_placer { std::move(mesh) };
|
||||
inline void do_experimental_support_placement(indexed_triangle_set mesh, TriangleSelectorGUI *selector, float dot_limit) {
|
||||
SupportPlacerMesh support_placer { std::move(mesh), dot_limit, 3.0 };
|
||||
|
||||
support_placer.find_support_areas();
|
||||
|
||||
@ -148,6 +273,8 @@ inline void do_experimental_support_placement(indexed_triangle_set mesh,
|
||||
selector->request_update_render_data();
|
||||
}
|
||||
}
|
||||
|
||||
support_placer.debug_export();
|
||||
}
|
||||
|
||||
}
|
||||
|
@ -350,7 +350,7 @@ void GLGizmoFdmSupports::select_facets_by_angle(float threshold_deg, bool block)
|
||||
|
||||
indexed_triangle_set mesh_triangles = mv->mesh().its;
|
||||
its_transform(mesh_triangles, mi->get_matrix(true));
|
||||
do_experimental_support_placement(std::move(mesh_triangles), m_triangle_selectors[mesh_id].get());
|
||||
do_experimental_support_placement(std::move(mesh_triangles), m_triangle_selectors[mesh_id].get(), dot_limit);
|
||||
|
||||
if (false) {
|
||||
// Now calculate dot product of vert_direction and facets' normals.
|
||||
|
Loading…
x
Reference in New Issue
Block a user