Rearrange code a little bit to make it more reusable

This commit is contained in:
Noisyfox
2026-05-08 22:54:20 +08:00
parent fad680fcc1
commit 2f160ad8bc
3 changed files with 39 additions and 28 deletions

View File

@@ -14,17 +14,8 @@ namespace Slic3r {
using namespace Geometry; using namespace Geometry;
// Saved painting data for remapping after mesh change.
struct SavedPainting {
indexed_triangle_set its; // Original mesh
TriangleSelector::TriangleSplittingData supported;
TriangleSelector::TriangleSplittingData seam;
TriangleSelector::TriangleSplittingData mmu;
TriangleSelector::TriangleSplittingData fuzzy;
};
// Remap painting data from saved source to a cut result mesh, and set on a volume. // Remap painting data from saved source to a cut result mesh, and set on a volume.
static void remap_and_set_painting(ModelVolume* vol, const SavedPainting* saved) static void remap_and_set_painting(ModelVolume* vol, const std::optional<TriangleSelector::SavedPainting>& saved)
{ {
if (!saved) { if (!saved) {
return; return;
@@ -34,7 +25,7 @@ static void remap_and_set_painting(ModelVolume* vol, const SavedPainting* saved)
FacetsAnnotation& target_facets) { FacetsAnnotation& target_facets) {
if (src_data.bitstream.empty()) if (src_data.bitstream.empty())
return; return;
auto result = TriangleSelector::remap_painting(saved->its, src_data, vol->mesh().its, translation_transform(vol->mesh().get_init_shift())); auto result = TriangleSelector::remap_painting(saved->mesh.its, src_data, vol->mesh().its, translation_transform(vol->mesh().get_init_shift()));
if (!result.bitstream.empty()) if (!result.bitstream.empty())
target_facets.set_data(result); target_facets.set_data(result);
}; };
@@ -211,7 +202,7 @@ static void process_modifier_cut(ModelVolume* volume, const Transform3d& instanc
static void process_solid_part_cut(ModelVolume* volume, const Transform3d& instance_matrix, const Transform3d& cut_matrix, static void process_solid_part_cut(ModelVolume* volume, const Transform3d& instance_matrix, const Transform3d& cut_matrix,
ModelObjectCutAttributes attributes, ModelObject* upper, ModelObject* lower, ModelObjectCutAttributes attributes, ModelObject* upper, ModelObject* lower,
const SavedPainting* saved_painting = nullptr) const std::optional<TriangleSelector::SavedPainting>& saved_painting)
{ {
// Perform cut // Perform cut
TriangleMesh upper_mesh, lower_mesh; TriangleMesh upper_mesh, lower_mesh;
@@ -362,20 +353,14 @@ const ModelObjectPtrs& Cut::perform_with_plane()
for (ModelVolume* volume : mo->volumes) { for (ModelVolume* volume : mo->volumes) {
// Save painting data before reset_extra_facets() discards it. // Save painting data before reset_extra_facets() discards it.
// Only for model parts that will be cut (not modifiers/connectors). std::optional<TriangleSelector::SavedPainting> saved_painting;
std::optional<SavedPainting> saved_painting; if (m_attributes.has(ModelObjectCutAttribute::KeepPaint)) {
if (m_attributes.has(ModelObjectCutAttribute::KeepPaint) && volume->is_any_painted() && volume->is_model_part() && !volume->mesh().empty()) { saved_painting = volume->save_painting();
// Get mesh in cut space (same transform as process_volume_cut applies) if (saved_painting) {
TriangleMesh mesh(volume->mesh()); // Transform mesh to cut space (same transform as process_volume_cut applies)
const auto volume_matrix = volume->get_matrix(); const auto volume_matrix = volume->get_matrix();
mesh.transform(instance_matrix * volume_matrix, true); saved_painting->mesh.transform(instance_matrix * volume_matrix, true);
SavedPainting sp; }
sp.its = std::move(mesh.its);
sp.supported = volume->supported_facets.get_data();
sp.seam = volume->seam_facets.get_data();
sp.mmu = volume->mmu_segmentation_facets.get_data();
sp.fuzzy = volume->fuzzy_skin_facets.get_data();
saved_painting = std::move(sp);
} }
volume->reset_extra_facets(); volume->reset_extra_facets();
@@ -387,8 +372,7 @@ const ModelObjectPtrs& Cut::perform_with_plane()
process_connector_cut(volume, instance_matrix, m_cut_matrix, m_attributes, upper, lower, dowels); process_connector_cut(volume, instance_matrix, m_cut_matrix, m_attributes, upper, lower, dowels);
} }
else if (!volume->mesh().empty()) else if (!volume->mesh().empty())
process_solid_part_cut(volume, instance_matrix, m_cut_matrix, m_attributes, upper, lower, process_solid_part_cut(volume, instance_matrix, m_cut_matrix, m_attributes, upper, lower, saved_painting);
saved_painting ? &*saved_painting : nullptr);
} }
// Post-process cut parts // Post-process cut parts

View File

@@ -878,6 +878,24 @@ public:
// List of mesh facets painted for fuzzy skin. // List of mesh facets painted for fuzzy skin.
FacetsAnnotation fuzzy_skin_facets; FacetsAnnotation fuzzy_skin_facets;
// Save painting data before reset_extra_facets() discards it.
// Used for replacing mesh without losing painting data.
// Only for model parts (not modifiers/connectors).
std::optional<TriangleSelector::SavedPainting> save_painting() const
{
if (is_any_painted() && is_model_part() && !mesh().empty()) {
TriangleSelector::SavedPainting sp;
sp.mesh = mesh();
sp.supported = supported_facets.get_data();
sp.seam = seam_facets.get_data();
sp.mmu = mmu_segmentation_facets.get_data();
sp.fuzzy = fuzzy_skin_facets.get_data();
return sp;
}
return {};
}
// BBS: quick access for volume extruders, 1 based // BBS: quick access for volume extruders, 1 based
mutable std::vector<int> mmuseg_extruders; mutable std::vector<int> mmuseg_extruders;
mutable Timestamp mmuseg_ts; mutable Timestamp mmuseg_ts;

View File

@@ -372,6 +372,15 @@ public:
// The operation may merge split triangles if they are being assigned the same color. // The operation may merge split triangles if they are being assigned the same color.
void seed_fill_apply_on_triangles(EnforcerBlockerType new_state); void seed_fill_apply_on_triangles(EnforcerBlockerType new_state);
// Saved painting data for remapping after mesh change.
struct SavedPainting {
TriangleMesh mesh; // Original mesh
TriangleSplittingData supported;
TriangleSplittingData seam;
TriangleSplittingData mmu;
TriangleSplittingData fuzzy;
};
// Remap painting data from source mesh to target mesh using spatial mapping. // Remap painting data from source mesh to target mesh using spatial mapping.
// `target_transform` should transform the target mesh into source's coordinate space. // `target_transform` should transform the target mesh into source's coordinate space.
static TriangleSplittingData remap_painting( static TriangleSplittingData remap_painting(