Fixed issue with lost selection in some preprocessing tools in some cases

Author: KaiKostack

kk_bullet_constraints_builder.zip

@@ -176,7 +176,7 @@ def tool_createGroupsFromNames(scene):
     except: pass
 
     # Backup selection
-    selection = [obj for obj in bpy.context.scene.objects if obj.select]
+    selection = selectionBak = [obj for obj in bpy.context.scene.objects if obj.select]
     # Find mesh objects in selection
     objs = [obj for obj in selection if obj.type == 'MESH' and not obj.hide and obj.is_visible(bpy.context.scene) and len(obj.data.vertices) > 0]
     if len(objs) == 0:
@@ -241,7 +241,7 @@ def tool_applyAllModifiers(scene):
     except: pass
 
     # Backup selection
-    selection = [obj for obj in bpy.context.scene.objects if obj.select]
+    selection = selectionBak = [obj for obj in bpy.context.scene.objects if obj.select]
     selectionActive = bpy.context.scene.objects.active
     # Find mesh objects in selection
     objs = [obj for obj in selection if (obj.type == 'MESH' or obj.type == 'CURVE') and not obj.hide and obj.is_visible(bpy.context.scene)]
@@ -352,7 +352,7 @@ def tool_centerModel(scene):
     except: pass
 
     # Backup selection
-    selection = [obj for obj in bpy.context.scene.objects if obj.select]
+    selection = selectionBak = [obj for obj in bpy.context.scene.objects if obj.select]
     selectionActive = bpy.context.scene.objects.active
     # Find mesh objects in selection
     objs = [obj for obj in selection if obj.type == 'MESH' and not obj.hide and obj.is_visible(bpy.context.scene) and len(obj.data.vertices) > 0]
@@ -402,7 +402,7 @@ def tool_separateLoose(scene):
     except: pass
 
     # Backup selection
-    selection = [obj for obj in bpy.context.scene.objects if obj.select]
+    selection = selectionBak = [obj for obj in bpy.context.scene.objects if obj.select]
     selectionActive = bpy.context.scene.objects.active
 
     ### Sort out user-defined objects (members of a specific group)
@@ -441,7 +441,7 @@ def tool_separateLoose(scene):
     kk_mesh_separate_loose.run()
 
     # Update selection list
-    selection = [obj for obj in bpy.context.scene.objects if obj.select]
+    selection = selectionBak = [obj for obj in bpy.context.scene.objects if obj.select]
 
     # Deselect objects with modifiers (in case they have not been applied yet) 
     for obj in selection:
@@ -471,7 +471,7 @@ def tool_remesh(scene):
     except: pass
 
     # Backup selection
-    selection = [obj for obj in bpy.context.scene.objects if obj.select]
+    selection = selectionBak = [obj for obj in bpy.context.scene.objects if obj.select]
     selectionActive = bpy.context.scene.objects.active
 
     ### Sort out user-defined objects (members of a specific group)
@@ -542,7 +542,7 @@ def tool_discretize(scene):
     except: pass
 
     # Backup selection
-    selection = [obj for obj in bpy.context.scene.objects if obj.select]
+    selection = selectionBak = [obj for obj in bpy.context.scene.objects if obj.select]
     selectionActive = bpy.context.scene.objects.active
 
     ### Sort out user-defined objects (members of a specific group)
@@ -561,7 +561,10 @@ def tool_discretize(scene):
     objs = [obj for obj in selection if obj.type == 'MESH' and not obj.hide and obj.is_visible(bpy.context.scene) and len(obj.data.vertices) > 0]
     if len(objs) == 0:
         print("No mesh objects selected.")
+        # Revert to start selection
+        for obj in selectionBak: obj.select = 1
         return
+    
     # Find empty objects with constraints in selection
     emptyObjs = [obj for obj in selection if obj.type == 'EMPTY' and not obj.hide and obj.is_visible(bpy.context.scene) and obj.rigid_body_constraint != None]
     # Remove mesh objects which are used within constraints (we want to leave them untouched)
@@ -577,8 +580,11 @@ def tool_discretize(scene):
     for obj in objs:
         if obj not in objsConst: objsNew.append(obj)
     objs = objsNew
+
     if len(objs) == 0:
         print("No mesh objects changed because of attached constraints.")
+        # Revert to start selection
+        for obj in selectionBak: obj.select = 1
         return
 
     ### Flip face normals for objects with negative scaling to avoid problems with booleans (will be reverted again when scaling is applied)
@@ -844,7 +850,7 @@ def tool_removeIntersections(scene, mode=1):
     except: pass
 
     # Backup selection
-    selection = [obj for obj in bpy.context.scene.objects if obj.select]
+    selection = selectionBak = [obj for obj in bpy.context.scene.objects if obj.select]
     selectionActive = bpy.context.scene.objects.active
     # Find mesh objects in selection
     objs = [obj for obj in selection if obj.type == 'MESH' and not obj.hide and obj.is_visible(bpy.context.scene) and len(obj.data.vertices) > 0]
@@ -898,7 +904,7 @@ def tool_enableRigidBodies(scene):
     except: pass
 
     # Backup selection
-    selection = [obj for obj in bpy.context.scene.objects if obj.select]
+    selection = selectionBak = [obj for obj in bpy.context.scene.objects if obj.select]
     selectionActive = bpy.context.scene.objects.active
     # Find mesh objects in selection
     objs = [obj for obj in selection if obj.type == 'MESH' and not obj.hide and obj.is_visible(bpy.context.scene) and len(obj.data.vertices) > 0]
@@ -977,7 +983,7 @@ def tool_fixFoundation(scene):
     except: pass
 
     # Backup selection
-    selection = [obj for obj in bpy.context.scene.objects if obj.select]
+    selection = selectionBak = [obj for obj in bpy.context.scene.objects if obj.select]
     selectionActive = bpy.context.scene.objects.active
 
     ### Sort out user-defined objects (members of a specific group)
@@ -996,6 +1002,8 @@ def tool_fixFoundation(scene):
     objs = [obj for obj in selection if obj.type == 'MESH' and not obj.hide and obj.is_visible(bpy.context.scene) and obj.rigid_body != None and obj.rigid_body.type == 'ACTIVE' and len(obj.data.vertices) > 0]
     if len(objs) == 0:
         print("No active rigid body objects selected.")
+        # Revert to start selection
+        for obj in selectionBak: obj.select = 1
         return
 
     props = bpy.context.window_manager.bcb
@@ -1295,7 +1303,7 @@ def tool_groundMotion(scene):
     except: pass
 
     # Backup selection
-    selection = [obj for obj in bpy.context.scene.objects if obj.select]
+    selection = selectionBak = [obj for obj in bpy.context.scene.objects if obj.select]
     selectionActive = bpy.context.scene.objects.active
 
     ### Sort out user-defined objects (members of a specific group)
@@ -2423,7 +2431,7 @@ def tool_cavityDetection(scene):
     except: pass
 
     # Backup selection
-    selection = [obj for obj in bpy.context.scene.objects if obj.select]
+    selection = selectionBak = [obj for obj in bpy.context.scene.objects if obj.select]
     selectionActive = bpy.context.scene.objects.active
     # Find mesh objects in selection
     objs = [obj for obj in selection if obj.type == 'MESH' and not obj.hide and obj.is_visible(bpy.context.scene) and len(obj.data.vertices) > 0]