object_mesh_versions: changed cleanup mode behaviour

object_mesh_versions.py

@@ -16,18 +16,21 @@
 #
 # ##### END GPL LICENSE BLOCK #####
 
+import bpy
+import time
+from datetime import datetime
+from bpy.types import Menu, Panel
+from bpy.props import StringProperty, BoolProperty
+
+
 bl_info = {
     "name": "KTX Mesh Versions",
     "author": "Roel Koster, @koelooptiemanna, irc:kostex",
-    "version": (1, 4, 2),
+    "version": (1, 4, 3),
     "blender": (2, 7, 0),
     "location": "View3D > Properties",
     "category": "Object"}
 
-import bpy,time
-from datetime import datetime
-from bpy.types import Menu, Panel
-from bpy.props import StringProperty, BoolProperty
 
 class KTX_MeshInit(bpy.types.Operator):
     bl_label = "Initialise Mesh Versioning"
@@ -39,6 +42,7 @@ class KTX_MeshInit(bpy.types.Operator):
         context.object.data.ktx_mesh_id = context.object.ktx_object_id = unique_id
         return {'FINISHED'}
 
+
 class KTX_MeshSelect(bpy.types.Operator):
     bl_label = "Select Mesh"
     bl_idname = "ktx.meshversions_select"
@@ -55,6 +59,7 @@ class KTX_MeshSelect(bpy.types.Operator):
         bpy.ops.object.mode_set(mode=c_mode)
         return {'FINISHED'}
 
+
 class KTX_MeshRemove(bpy.types.Operator):
     bl_label = "Remove Mesh"
     bl_idname = "ktx.meshversions_remove"
@@ -66,6 +71,19 @@ class KTX_MeshRemove(bpy.types.Operator):
         bpy.data.meshes.remove(bpy.data.meshes[self.m_index])
         return {'FINISHED'}
 
+
+class KTX_Cleanup(bpy.types.Operator):
+    bl_label = "Cleanup Mode"
+    bl_idname = "ktx.cleanup"
+    bl_description = "Cleanup Mode"
+
+    def execute(self, context):
+        for o in bpy.data.objects:
+            o.select = False
+        context.scene.objects.active = None
+        return {'FINISHED'}
+
+
 class KTX_MeshCreate(bpy.types.Operator):
     bl_label = "Create Mesh Version"
     bl_idname = "ktx.meshversions_create"
@@ -86,6 +104,7 @@ class KTX_MeshCreate(bpy.types.Operator):
             bpy.ops.object.mode_set(mode=c_mode)
         return {'FINISHED'}
 
+
 class KTX_Mesh_Versions(bpy.types.Panel):
     bl_label = "KTX Mesh Versions"
     bl_idname = "ktx.meshversions"
@@ -96,9 +115,10 @@ class KTX_Mesh_Versions(bpy.types.Panel):
         scene = context.scene
         obj = context.object
         layout = self.layout
-        if obj == None:
-            layout.label('No Object Selected')
-        else:
+
+        meshes_exist = bool(bpy.data.meshes.items() != [])
+        if meshes_exist:
+            if obj != None:
                 if obj.type == 'MESH':
                     if obj.ktx_object_id != '' and (obj.data.ktx_mesh_id == obj.ktx_object_id):
                         icon = 'PINNED' if scene.ktx_defpin else 'UNPINNED'
@@ -119,10 +139,15 @@ class KTX_Mesh_Versions(bpy.types.Panel):
                                 icon = 'PINNED' if m.use_fake_user else 'UNPINNED'
                                 row.prop(m, "use_fake_user", text="", icon=icon)
                         box.label()
+                        row = layout.row(align=True)
+                        row.operator("ktx.cleanup", text="Cleanup Mode")
                     else:
                         layout.operator("ktx.meshversions_init")
                 else:
-                if bpy.data.meshes.items() != []:
+                    layout.label("Select a Mesh Object")
+                    layout.operator("ktx.cleanup", text="Cleanup Mode")
+
+            else:
                 layout.label('All Meshes (Cleanup/Pin):')
                 box = layout.box()
                 box.scale_y = 0.65
@@ -135,6 +160,9 @@ class KTX_Mesh_Versions(bpy.types.Panel):
                     icon = 'PINNED' if m.use_fake_user else 'UNPINNED'
                     row.prop(m, "use_fake_user", text="", icon=icon)
                 box.label()
+        else:
+            layout.label("No Meshes Exist")
+
 
 def register():
     bpy.types.Object.ktx_object_id = bpy.props.StringProperty(name="KTX Object ID", description="Unique ID to 'link' one object to multiple meshes")
@@ -149,5 +177,6 @@ def unregister():
     del bpy.types.Object.ktx_object_id
     del bpy.types.Scene.ktx_defpin
 
+
 if __name__ == "__main__":
     register()