blender/blender-addons
21
47
Fork 103
Code Issues 592 Pull Requests 28 Projects 1 Releases Wiki Activity

Collection Manager: Improve filtering 2. Task: T69577

Browse Source 
Allow filters to be combined with each other.
This commit is contained in:
Ryan Inch 2020-10-20 03:08:47 -04:00
parent 52fb8e51ac
commit b66e1363a0
2 changed files with 36 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
object_collection_manager/__init__.py
View File

@ -22,7 +22,7 @@ bl_info = {
"name": "Collection Manager",
"description": "Manage collections and their objects",
"author": "Ryan Inch",
"version": (2, 16, 0),
"version": (2, 17, 0),
"blender": (2, 80, 0),
"location": "View3D - Object Mode (Shortcut - M)",
"warning": '', # used for warning icon and text in addons panel

50
object_collection_manager/ui.py
View File

@ -772,44 +772,56 @@ class CM_UL_items(UIList):
subrow.prop(self, "filter_by_qcd", text="", icon='EVENT_Q')
def filter_items(self, context, data, propname):
CM_UL_items.filtering = True
CM_UL_items.filtering = False
flt_flags = []
flt_neworder = []
list_items = getattr(data, propname)
if self.filter_name:
flt_flags = filter_items_by_name_custom(self.filter_name, self.bitflag_filter_item, list_items)
elif self.filter_by_selected:
flt_flags = [0] * len(list_items)
if self.filter_name:
CM_UL_items.filtering = True
new_flt_flags = filter_items_by_name_custom(self.filter_name, self.bitflag_filter_item, list_items)
flt_flags = merge_flt_flags(flt_flags, new_flt_flags)
if self.filter_by_selected:
CM_UL_items.filtering = True
new_flt_flags = [0] * len(list_items)
for idx, item in enumerate(list_items):
collection = layer_collections[item.name]["ptr"].collection
# check if any of the selected objects are in the collection
if not set(context.selected_objects).isdisjoint(collection.objects):
flt_flags[idx] |= self.bitflag_filter_item
new_flt_flags[idx] = self.bitflag_filter_item
# add in any recently created collections
if item.name in CM_UL_items.new_collections:
flt_flags[idx] |= self.bitflag_filter_item
new_flt_flags[idx] = self.bitflag_filter_item
elif self.filter_by_qcd:
flt_flags = [0] * len(list_items)
flt_flags = merge_flt_flags(flt_flags, new_flt_flags)
if self.filter_by_qcd:
CM_UL_items.filtering = True
new_flt_flags = [0] * len(list_items)
for idx, item in enumerate(list_items):
if item.qcd_slot_idx:
flt_flags[idx] |= self.bitflag_filter_item
new_flt_flags[idx] = self.bitflag_filter_item
# add in any recently created collections
if item.name in CM_UL_items.new_collections:
flt_flags[idx] |= self.bitflag_filter_item
new_flt_flags[idx] = self.bitflag_filter_item
else: # display as treeview
CM_UL_items.filtering = False
flt_flags = merge_flt_flags(flt_flags, new_flt_flags)
if not CM_UL_items.filtering: # display as treeview
CM_UL_items.new_collections.clear()
flt_flags = [self.bitflag_filter_item] * len(list_items)
for idx, item in enumerate(list_items):
@ -817,9 +829,11 @@ class CM_UL_items(UIList):
flt_flags[idx] = 0
if self.use_filter_invert:
CM_UL_items.filtering = True # invert can act as pseudo filtering
for idx, flag in enumerate(flt_flags):
flt_flags[idx] = 0 if flag else self.bitflag_filter_item
# update visible items list
CM_UL_items.visible_items.clear()
CM_UL_items.visible_items.extend(flt_flags)
@ -1107,3 +1121,9 @@ def filter_items_by_name_custom(pattern, bitflag, items, propname="name", flags=
flags[i] |= bitflag
return flags
def merge_flt_flags(l1, l2):
for idx, _ in enumerate(l1):
l1[idx] &= l2.pop(0)
return l1 + l2