ant landscape, fix for last commit

1 files changed, 13 insertions, 0 deletions

diff --git a/ant_landscape/ant_functions.py b/ant_landscape/ant_functions.py
index 826d693e..01354295 100644
--- a/ant_landscape/ant_functions.py
+++ b/ant_landscape/ant_functions.py
@@ -1122,12 +1122,23 @@ class Eroder(bpy.types.Operator):
             default=False
             )
 
+    weight_mode = BoolProperty(
+            name="Weight Paint Mode",
+            default=False,
+            description="Enter weightpaint mode"
+            )
+
     stats = Stats()
     counts= {}
 
     # add poll function to restrict action to mesh object in object mode
     
     def execute(self, context):
+
+        # Enter WeightPaintMode
+        if self.weight_mode:
+            bpy.ops.paint.weight_paint_toggle()
+
         ob = context.active_object
         #obwater = bpy.data.objects["water"]
         me = ob.data
@@ -1322,6 +1333,8 @@ class Eroder(bpy.types.Operator):
 
         layout.prop(self,'smooth')
 
+        layout.prop(self,'weight_mode')
+
         #if numexpr_available:
         #  layout.prop(self, 'numexpr')
         #else: