new extension for vertex color threshold

Author: RenaudRohlinger

__init__.py

@@ -9,9 +9,11 @@
     import importlib
     importlib.reload(apply_all)
     importlib.reload(apply_multi_user)
+    importlib.reload(vertex_color_picker)
 else:
     from . import apply_all
     from . import apply_multi_user
+    from . import vertex_color_picker
 
 import bpy
 
@@ -452,6 +454,23 @@ def execute(self, context):
             # raise the exception again
             raise e
 
+class VertexColorPicker(bpy.types.Operator):
+    bl_idname = "object.color_picker"
+    bl_label = "Vertex color picker"
+
+    def execute(self, context):
+        try:
+            self.report(
+                {"INFO"}, "----- ----- ----- Sougen Scripts ----- ----- -----")
+            self.report({"INFO"}, "Vertex color processing processing")
+            vertex_color_picker.vertex_color_picker(self, context)
+            return {"FINISHED"}
+        except Exception as e:
+            print("Something went wrong")
+            self.report({"ERROR"}, "Something went wrong")
+            # raise the exception again
+            raise e
+
 
 def curves_export(self, context):
     bpy.context.window.cursor_set("WAIT")
@@ -604,6 +623,16 @@ def draw(self, context):
         else:
             col2.prop(context.scene, "unlit")
 
+            box3 = layout.box()
+            col3 = box3.column()
+            row3 = col3.row()
+            row4 = col3.row()
+
+
+            row3.prop(context.scene, "color_treshold")
+            row4.operator(
+                "object.color_picker", icon="MOD_VERTEX_WEIGHT", depress=loading
+            )
             layout.operator(
                 "object.simple_operator", icon="VIEW_CAMERA", depress=loading
             )
@@ -620,8 +649,9 @@ def draw(self, context):
                             icon="EDITMODE_HLT", depress=loading)
 
 
+
 blender_classes = [BakeCamera, SimpleGLTF, GLTF_Collider,
-                   GLTF_PT_Panel, GLTF_Instance, Curves_Export, ApplyMultiUser]
+                   GLTF_PT_Panel, GLTF_Instance, Curves_Export, ApplyMultiUser, VertexColorPicker]
 
 
 def register():
@@ -658,6 +688,14 @@ def register():
         min=0,
         max=10,
     )
+    bpy.types.Scene.color_treshold = bpy.props.FloatProperty(
+        name="Color threshold:",
+        description="Vertex color treshold",
+        default=0.2,
+        min=0.01,
+        max=1,
+    )
+
     bpy.types.Scene.unlit = bpy.props.BoolProperty(
         name="Use KHR Unlit",
         description="Convert materials from metal/rough to unlit",
@@ -675,5 +713,6 @@ def unregister():
     del bpy.types.Scene.filename_path
     del bpy.types.Scene.draco
     del bpy.types.Scene.draco_level
+    del bpy.types.Scene.color_treshold
     del bpy.types.Scene.unlit
     del bpy.types.Scene.gltf_sys

__pycache__/__init__.cpython-39.pyc

@@ -23,7 +23,7 @@ def applyModifierToMultiUser(self, context):
                 linked.append(obj)
 
 
-    bpy.ops.object.make_single_user(object=True, obdata=True, material=False, animation=False, obdata_animation=False)
+    bpy.ops.object.make_single_user(object=True, obdata=True, material=False)
     bpy.ops.object.modifier_apply(modifier="Decimate")
 
 

__pycache__/apply_multi_user.cpython-39.pyc

@@ -0,0 +1,61 @@
+import bpy
+import pprint
+import bmesh
+
+from mathutils import Color
+
+from collections import defaultdict
+
+def avg_col(cols):
+    avg_col = Color((0.0, 0.0, 0.0))
+
+    for col in cols:
+        tempcol = Color((col[0], col[1], col[2]))
+
+        avg_col += tempcol/len(cols)
+    return avg_col
+
+def vertex_color_picker(self, context):
+    tk = defaultdict(list)
+
+    mesh = bpy.context.active_object.data
+    color_layer = mesh.vertex_colors['Col']
+
+    i = 0
+    for poly in mesh.polygons:
+        for idx in poly.loop_indices:
+            loop = mesh.loops[idx]
+            color = color_layer.data[i].color
+            tk[loop.vertex_index].append(color)
+            i += 1
+
+    vcol_averages = {k: avg_col(v) for k, v in tk.items()}
+
+    vertices_of_interest = [k for k, v in vcol_averages.items() if v.r < context.scene.color_treshold]
+
+
+    bpy.ops.object.mode_set(mode='EDIT')
+
+
+    obj = bpy.context.object
+    me = obj.data
+    bm = bmesh.from_edit_mesh(me)
+
+    vertices= [e for e in bm.verts]
+    oa = bpy.context.active_object
+
+    for vert in vertices:
+        if vert.index in vertices_of_interest:
+            vert.select_set(True)
+        else:
+            vert.select_set(False)
+
+
+    bmesh.update_edit_mesh(me)      
+
+    pprint.pprint(vertices_of_interest)
+
+    # you must cast this as a dict manually like  tk = dict(tk) , 
+    # if you need it as a dict, but defaultdict will behave mostly the same
+
+    #pprint.pprint(tk)