FIX: Clean unassociated cliusters adds the colors as well

diffCheckOrg · Sep 20, 2024 · 42f0bd2 · 42f0bd2
1 parent d229d02
commit 42f0bd2
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/diffCheck/segmentation/DFSegmentation.cc b/src/diffCheck/segmentation/DFSegmentation.cc
@@ -327,6 +327,10 @@ namespace diffCheck::segmentation
                 {
                     completed_segment->Points.push_back(point);
                     completed_segment->Normals.push_back(cluster->Normals[std::distance(cluster->Points.begin(), std::find(cluster->Points.begin(), cluster->Points.end(), point))]);
+                    if (cluster->GetNumColors() > 0)
+                    {
+                        completed_segment->Colors.push_back(cluster->Colors[std::distance(cluster->Points.begin(), std::find(cluster->Points.begin(), cluster->Points.end(), point))]);
+                    }
                 }
             }
             std::vector<int> indicesToRemove;