yyc12345/BallanceBlenderHelper
1
0
Fork 0
Code Issues Pull Requests Releases Activity

fix: redirect some reference of bpy.context to context passed by operator argument

Browse Source
This commit is contained in:
yyc12345 2025-01-03 10:37:26 +08:00
parent 89a5e6367b
commit 4181096a9e
2 changed files with 16 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
bbp_ng/OP_OBJECT_virtools_group.py
View File

@ -50,6 +50,7 @@ class BBP_OT_select_object_by_virtools_group(bpy.types.Operator, PROP_virtools_g
def execute(self, context): def execute(self, context):
_select_object_by_virtools_group( _select_object_by_virtools_group(
context,
self.general_get_group_name(), self.general_get_group_name(),
_g_EnumHelper_SelectMode.get_selection(self.selection_mode) _g_EnumHelper_SelectMode.get_selection(self.selection_mode)
) )
@ -65,17 +66,17 @@ class BBP_OT_select_object_by_virtools_group(bpy.types.Operator, PROP_virtools_g
layout.label(text='Group Parameters') layout.label(text='Group Parameters')
self.draw_group_name_input(layout) self.draw_group_name_input(layout)
def _select_object_by_virtools_group(group_name: str, mode: SelectMode) -> None: def _select_object_by_virtools_group(context: bpy.types.Context, group_name: str, mode: SelectMode) -> None:
match(mode): match(mode):
case SelectMode.Set: case SelectMode.Set:
# iterate all objects and directly set # iterate all objects and directly set
for obj in bpy.context.scene.objects: for obj in context.scene.objects:
# check group and decide whether select this obj # check group and decide whether select this obj
with PROP_virtools_group.VirtoolsGroupsHelper(obj) as gp: with PROP_virtools_group.VirtoolsGroupsHelper(obj) as gp:
obj.select_set(gp.contain_group(group_name)) obj.select_set(gp.contain_group(group_name))
case SelectMode.Extend: case SelectMode.Extend:
# also iterate all objects # also iterate all objects
for obj in bpy.context.scene.objects: for obj in context.scene.objects:
# but only increase selection, for selected object, skip check # but only increase selection, for selected object, skip check
if obj.select_get(): continue if obj.select_get(): continue
# if not selected, check whether add it. # if not selected, check whether add it.
@ -86,7 +87,7 @@ def _select_object_by_virtools_group(group_name: str, mode: SelectMode) -> None:
# subtract only involving selected item. so we get selected objest first # subtract only involving selected item. so we get selected objest first
# and copy it (because we need modify it) # and copy it (because we need modify it)
# and iterate it to reduce useless operations # and iterate it to reduce useless operations
selected = bpy.context.selected_objects[:] selected = context.selected_objects[:]
for obj in selected: for obj in selected:
# remove matched only # remove matched only
with PROP_virtools_group.VirtoolsGroupsHelper(obj) as gp: with PROP_virtools_group.VirtoolsGroupsHelper(obj) as gp:
@ -94,16 +95,16 @@ def _select_object_by_virtools_group(group_name: str, mode: SelectMode) -> None:
obj.select_set(False) obj.select_set(False)
case SelectMode.Difference: case SelectMode.Difference:
# construct a selected obj set for convenient operations # construct a selected obj set for convenient operations
selected_set = set(bpy.context.selected_objects) selected_set = set(context.selected_objects)
# iterate all objects # iterate all objects
for obj in bpy.context.scene.objects: for obj in context.scene.objects:
with PROP_virtools_group.VirtoolsGroupsHelper(obj) as gp: with PROP_virtools_group.VirtoolsGroupsHelper(obj) as gp:
# use xor to select # use xor to select
# in_selected XOR in_group # in_selected XOR in_group
obj.select_set((obj in selected_set) ^ gp.contain_group(group_name)) obj.select_set((obj in selected_set) ^ gp.contain_group(group_name))
case SelectMode.Intersect: case SelectMode.Intersect:
# like subtract, only iterate selected obj # like subtract, only iterate selected obj
selected = bpy.context.selected_objects[:] selected = context.selected_objects[:]
for obj in selected: for obj in selected:
# but remove not matched # but remove not matched
with PROP_virtools_group.VirtoolsGroupsHelper(obj) as gp: with PROP_virtools_group.VirtoolsGroupsHelper(obj) as gp:
@ -124,7 +125,7 @@ class BBP_OT_add_objects_virtools_group(bpy.types.Operator, PROP_virtools_group.
@classmethod @classmethod
def poll(cls, context): def poll(cls, context):
return len(bpy.context.selected_objects) != 0 return len(context.selected_objects) != 0
def invoke(self, context, event): def invoke(self, context, event):
wm = context.window_manager wm = context.window_manager
@ -132,7 +133,7 @@ class BBP_OT_add_objects_virtools_group(bpy.types.Operator, PROP_virtools_group.
def execute(self, context): def execute(self, context):
group_name: str = self.general_get_group_name() group_name: str = self.general_get_group_name()
for obj in bpy.context.selected_objects: for obj in context.selected_objects:
with PROP_virtools_group.VirtoolsGroupsHelper(obj) as gp: with PROP_virtools_group.VirtoolsGroupsHelper(obj) as gp:
gp.add_group(group_name) gp.add_group(group_name)
self.report({'INFO'}, "Grouping objects successfully.") self.report({'INFO'}, "Grouping objects successfully.")
@ -149,7 +150,7 @@ class BBP_OT_rm_objects_virtools_group(bpy.types.Operator, PROP_virtools_group.S
@classmethod @classmethod
def poll(cls, context): def poll(cls, context):
return len(bpy.context.selected_objects) != 0 return len(context.selected_objects) != 0
def invoke(self, context, event): def invoke(self, context, event):
wm = context.window_manager wm = context.window_manager
@ -157,7 +158,7 @@ class BBP_OT_rm_objects_virtools_group(bpy.types.Operator, PROP_virtools_group.S
def execute(self, context): def execute(self, context):
group_name: str = self.general_get_group_name() group_name: str = self.general_get_group_name()
for obj in bpy.context.selected_objects: for obj in context.selected_objects:
with PROP_virtools_group.VirtoolsGroupsHelper(obj) as gp: with PROP_virtools_group.VirtoolsGroupsHelper(obj) as gp:
gp.remove_group(group_name) gp.remove_group(group_name)
self.report({'INFO'}, "Ungrouping objects successfully.") self.report({'INFO'}, "Ungrouping objects successfully.")
@ -174,7 +175,7 @@ class BBP_OT_clear_objects_virtools_group(bpy.types.Operator):
@classmethod @classmethod
def poll(cls, context): def poll(cls, context):
return len(bpy.context.selected_objects) != 0 return len(context.selected_objects) != 0
def invoke(self, context, event): def invoke(self, context, event):
wm = context.window_manager wm = context.window_manager
@ -182,7 +183,7 @@ class BBP_OT_clear_objects_virtools_group(bpy.types.Operator):
def execute(self, context): def execute(self, context):
# iterate object # iterate object
for obj in bpy.context.selected_objects: for obj in context.selected_objects:
with PROP_virtools_group.VirtoolsGroupsHelper(obj) as gp: with PROP_virtools_group.VirtoolsGroupsHelper(obj) as gp:
gp.clear_groups() gp.clear_groups()
self.report({'INFO'}, "Clear objects groups successfully.") self.report({'INFO'}, "Clear objects groups successfully.")

4
bbp_ng/OP_UV_flatten_uv.py
View File

@ -135,7 +135,7 @@ class BBP_OT_flatten_uv(bpy.types.Operator):
@classmethod @classmethod
def poll(cls, context): def poll(cls, context):
obj = bpy.context.active_object obj = context.active_object
if obj is None: if obj is None:
return False return False
if obj.type != 'MESH': if obj.type != 'MESH':
@ -162,7 +162,7 @@ class BBP_OT_flatten_uv(bpy.types.Operator):
return {'CANCELLED'} return {'CANCELLED'}
# do flatten uv and report # do flatten uv and report
failed: int = _flatten_uv_wrapper(bpy.context.active_object.data, flatten_param_) failed: int = _flatten_uv_wrapper(context.active_object.data, flatten_param_)
if failed != 0: if failed != 0:
print(f'[Flatten UV] {failed} faces are not be processed correctly because process failed.') print(f'[Flatten UV] {failed} faces are not be processed correctly because process failed.')
return {'FINISHED'} return {'FINISHED'}