blender
/
blender
134
400
Fork 584
Code Issues 6.3k Pull Requests 758 Projects 19 Wiki Activity

Fix tool-system index error with mode-specific tools added to groups 

It was possible to have invalid indices when a script registered
a tool into an existing group of tools.

Address issue raised by D7436
This commit is contained in:
Campbell Barton 2020-09-25 17:42:41 +10:00
parent b3e1620e21
commit 56cd92413f
1 changed files with 21 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
release/scripts/startup/bl_ui/space_toolsystem_common.py
View File

@ -288,7 +288,6 @@ class ToolSelectPanelHelper:
else:
yield item
@classmethod
def _tool_get_active(cls, context, space_type, mode, with_icon=False):
"""
@ -326,7 +325,7 @@ class ToolSelectPanelHelper:
if item is not None:
if type(item) is tuple:
if item[0].idname == idname:
index = cls._tool_group_active.get(item[0].idname, 0)
index = cls._tool_group_active_get_from_item(item)
return (item[index], index)
else:
if item.idname == idname:
@ -342,7 +341,7 @@ class ToolSelectPanelHelper:
if item is not None:
if type(item) is tuple:
if item[0].idname == idname:
index = cls._tool_group_active.get(item[0].idname, 0)
index = cls._tool_group_active_get_from_item(item)
return (item[index], index, item)
else:
if item.idname == idname:
@ -395,7 +394,7 @@ class ToolSelectPanelHelper:
if item is not None:
if i == tool_index:
if type(item) is tuple:
index = cls._tool_group_active.get(item[0].idname, 0)
index = cls._tool_group_active_get_from_item(item)
item = item[index]
else:
index = -1
@ -403,6 +402,19 @@ class ToolSelectPanelHelper:
i += 1
return None, -1
@classmethod
def _tool_group_active_get_from_item(cls, item):
index = cls._tool_group_active.get(item[0].idname, 0)
# Can happen in the case a group is dynamic.
#
# NOTE(Campbell): that in this case it's possible the order could change too,
# So if we want to support this properly we will need to switch away from using
# an index and instead use an ID.
# Currently this is such a rare case occurrence that a range check is OK for now.
if index >= len(item):
index = 0
return index
@classmethod
def _tool_group_active_set_by_id(cls, context, idname_group, idname):
item_group = cls._tool_get_group_by_id(context, idname_group, coerce=True)
@ -664,7 +676,7 @@ class ToolSelectPanelHelper:
# not ideal, write this every time :S
cls._tool_group_active[item[0].idname] = index
else:
index = cls._tool_group_active.get(item[0].idname, 0)
index = cls._tool_group_active_get_from_item(item)
item = item[index]
use_menu = True
@ -821,7 +833,8 @@ class ToolSelectPanelHelper:
if is_active_tool:
index_current = -1
else:
index_current = cls._tool_group_active.get(item_group[0].idname, 0)
index_current = cls._tool_group_active_get_from_item(item_group)
for i, sub_item in enumerate(item_group):
is_active = (i == index_current)
@ -871,7 +884,7 @@ class ToolSelectPanelHelper:
if is_active_tool:
index_current = -1
else:
index_current = cls._tool_group_active.get(item_group[0].idname, 0)
index_current = cls._tool_group_active_get_from_item(item_group)
for i, sub_item in enumerate(item_group):
is_active = (i == index_current)
props = pie.operator(
@ -1023,7 +1036,7 @@ def activate_by_id_or_cycle(context, space_type, idname, *, offset=1, as_fallbac
id_current = ""
for item_group in cls.tools_from_context(context):
if type(item_group) is tuple:
index_current = cls._tool_group_active.get(item_group[0].idname, 0)
index_current = cls._tool_group_active_get_from_item(item_group)
for sub_item in item_group:
if sub_item.idname == idname:
id_current = item_group[index_current].idname