curve_tools/properties.py

   1 # SPDX-License-Identifier: GPL-2.0-or-later
   2
   3 import time
   4
   5 import bpy
   6 from bpy.props import *
   7
   8
   9
  10 class curvetoolsSelectedObjectHeader(bpy.types.Header):
  11     bl_label = "Selection"
  12     bl_space_type = "VIEW_3D"
  13
  14     def __init__(self):
  15         self.update()
  16
  17
  18     def update(self):
  19         blenderSelectedObjects = bpy.context.selected_objects
  20         selectedObjects = bpy.context.scene.curvetools.SelectedObjects
  21
  22         selectedObjectsToRemove = []
  23         for selectedObject in selectedObjects:
  24             if not selectedObject.IsElementOf(blenderSelectedObjects): selectedObjectsToRemove.append(selectedObject)
  25         for selectedObject in selectedObjectsToRemove: selectedObjects.remove(selectedObject)
  26
  27         blenderObjectsToAdd = []
  28         for blenderObject in blenderSelectedObjects:
  29             if not curvetoolsSelectedObject.ListContains(selectedObjects, blenderObject): blenderObjectsToAdd.append(blenderObject)
  30         for blenderObject in blenderObjectsToAdd:
  31             newSelectedObject = curvetoolsSelectedObject(blenderObject)
  32             selectedObjects.append(newSelectedObject)
  33
  34
  35     def draw(self, context):
  36         selectedObjects = bpy.context.scene.curvetools.SelectedObjects
  37         nrSelectedObjects = len(selectedObjects)
  38
  39         layout = self.layout
  40         row = layout.row()
  41         row.label(text="Sel: " + str(nrSelectedObjects))
  42
  43
  44 class curvetoolsSelectedObject(bpy.types.PropertyGroup):
  45     name: StringProperty(name = "name", default = "??")
  46
  47
  48     @staticmethod
  49     def UpdateThreadTarget(lock, sleepTime, selectedObjectNames, selectedBlenderObjectNames):
  50         time.sleep(sleepTime)
  51
  52         newSelectedObjectNames = []
  53
  54         for name in selectedObjectNames:
  55             if name in selectedBlenderObjectNames: newSelectedObjectNames.append(name)
  56
  57         for name in selectedBlenderObjectNames:
  58             if not (name in selectedObjectNames): newSelectedObjectNames.append(name)
  59
  60         # sometimes it still complains about the context
  61         try:
  62             nrNewSelectedObjects = len(newSelectedObjectNames)
  63             bpy.context.scene.curvetools.NrSelectedObjects = nrNewSelectedObjects
  64
  65             selectedObjects = bpy.context.scene.curvetools.SelectedObjects
  66             selectedObjects.clear()
  67             for i in range(nrNewSelectedObjects): selectedObjects.add()
  68             for i, newSelectedObjectName in enumerate(newSelectedObjectNames):
  69                 selectedObjects[i].name = newSelectedObjectName
  70         except: pass
  71
  72
  73     @staticmethod
  74     def GetSelectedObjectNames():
  75         selectedObjects = bpy.context.scene.curvetools.SelectedObjects
  76
  77         rvNames = []
  78         selectedObjectValues = selectedObjects.values()
  79         for selectedObject in selectedObjectValues: rvNames.append(selectedObject.name)
  80
  81         return rvNames
  82
  83
  84     @staticmethod
  85     def GetSelectedBlenderObjectNames():
  86         blenderSelectedObjects = bpy.context.selected_objects
  87
  88         rvNames = []
  89         for blObject in blenderSelectedObjects: rvNames.append(blObject.name)
  90
  91         return rvNames
  92
  93 def register():
  94     for cls in classes:
  95         bpy.utils.register_class(operators)
  96
  97 def unregister():
  98     for cls in classes:
  99         bpy.utils.unregister_class(operators)
 100
 101 if __name__ == "__main__":
 102     register()
 103
 104 operators = [
 105     curvetoolsSelectedObject,
 106     ]