Add new blur operator for isolate mode

Author: andyp123

README.md

@@ -145,6 +145,16 @@ Copy the value of the 'Src' channel to the uv layer in the 'Dst' layer.
 ## Misc Operations
 The 'Misc Operations' section contains features unrelated to the main features of the addon.
 
+### Blur Channel Values
+Blurs the values of all the vertices in the mesh to smooth them out. It uses the smooth weights function available in weight paint mode, but allows it to be used interactively on vertex color channel data.
+__Note__: This feature is currently only available when a channel has been isolated.
+
++ __Factor__ (0.5) - The amount of blur to apply.
+
++ __Iterations__ (1) - The number of times to apply the blur. Increasing this will result in a smoother blur.
+
++ __Expand/Contract__ (0.0) - Use this to adjust the balance of dark/light values as the blur is applied. It is similar to brightness/contrast adjustment, or the remap operator.
+
 ### Randomize Mesh Island Colors
 All separate parts of a mesh (islands), will be assigned random colors based on various randomization parameters.
 

vertex_color_master/__init__.py

@@ -37,7 +37,7 @@
 bl_info = {
     "name": "Vertex Color Master",
     "author": "Andrew Palmer (with contributions from Bartosz Styperek)",
-    "version": (0, 82),
+    "version": (0, 83),
     "blender": (2, 80, 0),
     "location": "Vertex Paint | View3D > Vertex Color Master",
     "description": "Tools for manipulating vertex color data.",
@@ -65,9 +65,9 @@
     vcm_ops.VERTEXCOLORMASTER_OT_IsolateChannel,
     vcm_ops.VERTEXCOLORMASTER_OT_ApplyIsolatedChannel,
     vcm_ops.VERTEXCOLORMASTER_OT_RandomizeMeshIslandColors,
-    vcm_ops.VERTEXCOLORMASTER_OT_AdjustHSV,
     vcm_ops.VERTEXCOLORMASTER_OT_FlipBrushColors,
     vcm_ops.VERTEXCOLORMASTER_OT_Gradient,
+    vcm_ops.VERTEXCOLORMASTER_OT_BlurChannel,
     vcm_menus.VERTEXCOLORMASTER_PT_MainPanel,
     vcm_menus.VERTEXCOLORMASTER_MT_PieMain,
 )

vertex_color_master/vcm_helpers.py

@@ -182,7 +182,7 @@ def color_to_uvs(mesh, src_vcol, dst_uv, src_u_idx=0, src_v_idx=1):
     mesh.update()
 
 
-def weights_to_color(mesh, src_vgroup_idx, dst_vcol, dst_channel_idx):
+def weights_to_color(mesh, src_vgroup_idx, dst_vcol, dst_channel_idx, all_channels=False):
     vertex_weights = [0.0] * len(mesh.vertices)
 
     # build list of weights for vertex indices
@@ -193,9 +193,14 @@ def weights_to_color(mesh, src_vgroup_idx, dst_vcol, dst_channel_idx):
                 break
 
     # copy weights to channel of dst color layer
-    for loop_index, loop in enumerate(mesh.loops):
-        weight = vertex_weights[loop.vertex_index]
-        dst_vcol.data[loop_index].color[dst_channel_idx] = weight
+    if not all_channels:
+        for loop_index, loop in enumerate(mesh.loops):
+            weight = vertex_weights[loop.vertex_index]
+            dst_vcol.data[loop_index].color[dst_channel_idx] = weight
+    else:
+        for loop_index, loop in enumerate(mesh.loops):
+            weight = vertex_weights[loop.vertex_index]
+            dst_vcol.data[loop_index].color[:3] = [weight]*3
 
     mesh.update()
 

vertex_color_master/vcm_menus.py

@@ -297,6 +297,9 @@ def draw_misc_operations(context, layout, obj, settings, mode='STANDARD', pie=Fa
     if mode == 'STANDARD':
         row = col.row(align=True)
         row.operator('paint.vertex_color_hsv', text="Adjust HSV")
+    else:
+        row = col.row(align=True)
+        row.operator('vertexcolormaster.blur_channel', text="Blur Channel Values")
     row = col.row(align=True)
     row.operator('vertexcolormaster.randomize_mesh_island_colors', text="Random Mesh Island Colors")
     row = col.row(align=True)

vertex_color_master/vcm_ops.py

@@ -441,52 +441,70 @@ def execute(self, context):
         return {'FINISHED'}
 
 
-class VERTEXCOLORMASTER_OT_AdjustHSV(bpy.types.Operator):
-    """Adjust the Hue, Saturation and Value of the the active vertex colors"""
-    bl_idname = 'vertexcolormaster.adjust_hsv'
-    bl_label = 'VCM Adjust HSV'
+class VERTEXCOLORMASTER_OT_BlurChannel(bpy.types.Operator):
+    """Blur values of a particular channel"""
+    bl_idname = 'vertexcolormaster.blur_channel'
+    bl_label = 'VCM Blur Channel'
     bl_options = {'REGISTER', 'UNDO'}
 
-    colorize: BoolProperty(
-        name="Colorize",
-        description="Colorize the mesh instead of adjusting hue.",
-        default=False
-    )
+    factor: FloatProperty(
+        name="Factor",
+        description="Amount of blur to apply.",
+        default=0.5,
+        min=0.0,
+        max=1.0
+    )  
 
-    hue_adjust: FloatProperty(
-        name="Hue",
-        description="Hue adjustment.",
-        default=0.0,
-        min=-0.5,
-        max=0.5
-    )
+    iterations: IntProperty(
+        name="Iterations",
+        description="Number of iterations to blur values.",
+        default=1,
+        min=1,
+        max=200
+    ) 
 
-    sat_adjust: FloatProperty(
-        name="Saturation",
-        description="Saturation adjustment.",
+    expand: FloatProperty(
+        name="Expand/Contract",
+        description="Alter how the blur affects the distribution of dark/light values.",
         default=0.0,
         min=-1.0,
         max=1.0
-    )
+    ) 
 
-    val_adjust: FloatProperty(
-        name="Value",
-        description="Value adjustment.",
-        default=0.0,
-        min=-1.0,
-        max=1.0
-    )
+    @classmethod
+    def poll(cls, context):
+        obj = context.active_object
+        return bpy.context.object.mode == 'VERTEX_PAINT' and obj is not None and obj.type == 'MESH'
 
     def execute(self, context):
-        settings = context.scene.vertex_color_master_settings
-        mesh = context.active_object.data
-        vcol = mesh.vertex_colors.active
+        obj = context.active_object
+        mesh = obj.data
+        vcol = mesh.vertex_colors.active if mesh.vertex_colors else mesh.vertex_colors.new()
+        isolate = get_isolated_channel_ids(vcol)
 
-        if vcol is None:
-            self.report({'ERROR'}, "Can't modify HSV when no vertex color data exists.")
-            return {'FINISHED'}
+        if isolate is None:
+            self.report({'ERROR'}, "Blur only works with an isolated channel")
+            return {'CANCELLED'}
+
+        vgroup_id = 'vcm_temp_weights'
+        vgroup = obj.vertex_groups.new(name=vgroup_id)
+        obj.vertex_groups.active_index = vgroup.index
+
+        channel_idx = channel_id_to_idx(isolate[1])
+        color_to_weights(obj, vcol, channel_idx, vgroup.index)
+
+        bpy.ops.object.mode_set(mode='WEIGHT_PAINT', toggle=False)
+        bpy.ops.object.vertex_group_smooth(
+            group_select_mode='ACTIVE',
+            factor=self.factor,
+            repeat=self.iterations,
+            expand=self.expand
+        )
+        bpy.ops.object.mode_set(mode='VERTEX_PAINT', toggle=False)
+
+        weights_to_color(mesh, vgroup.index, vcol, channel_idx, all_channels=True)
 
-        adjust_hsv(mesh, vcol, self.hue_adjust, self.sat_adjust, self.val_adjust, self.colorize)
+        obj.vertex_groups.remove(vgroup)
 
         return {'FINISHED'}