Changeset View
Changeset View
Standalone View
Standalone View
release/scripts/modules/bpy_types.py
| Show First 20 Lines • Show All 761 Lines • ▼ Show 20 Lines | def _dyn_ui_initialize(cls): | ||||
| if draw_funcs is None: | if draw_funcs is None: | ||||
| def draw_ls(self, context): | def draw_ls(self, context): | ||||
| # ensure menus always get default context | # ensure menus always get default context | ||||
| operator_context_default = self.layout.operator_context | operator_context_default = self.layout.operator_context | ||||
| for func in draw_ls._draw_funcs: | for func in draw_ls._draw_funcs: | ||||
| # Incase this is wrapped | |||||
| func = getattr(func, "_draw_func_wrap", func) | |||||
| # so bad menu functions don't stop | # so bad menu functions don't stop | ||||
| # the entire menu from drawing | # the entire menu from drawing | ||||
| try: | try: | ||||
| func(self, context) | func(self, context) | ||||
| except: | except: | ||||
| import traceback | import traceback | ||||
| traceback.print_exc() | traceback.print_exc() | ||||
| self.layout.operator_context = operator_context_default | self.layout.operator_context = operator_context_default | ||||
| draw_funcs = draw_ls._draw_funcs = [cls.draw] | draw_funcs = draw_ls._draw_funcs = [cls.draw] | ||||
| cls.draw = draw_ls | cls.draw = draw_ls | ||||
| return draw_funcs | return draw_funcs | ||||
| @staticmethod | |||||
| def _dyn_bl_origin_apply(draw_func): | |||||
| from _bpy import _bl_origin_get | |||||
| bl_origin = _bl_origin_get() | |||||
| if bl_origin is None: | |||||
| return | |||||
| def draw_func_wrap(self, context): | |||||
| workspace = context.workspace | |||||
| if workspace.use_bl_origins: | |||||
| bl_origin_name = draw_func_wrap.bl_origin | |||||
| if not any( | |||||
| True for bl_origin in workspace.bl_origins | |||||
| if bl_origin_name == bl_origin.name | |||||
| ): | |||||
| return | |||||
| draw_func_wrap._draw_func_real(self, context) | |||||
| draw_func_wrap.bl_origin = bl_origin | |||||
| draw_func_wrap._draw_func_real = draw_func | |||||
| draw_func._draw_func_wrap = draw_func_wrap | |||||
| @classmethod | @classmethod | ||||
| def is_extended(cls): | def is_extended(cls): | ||||
| return bool(getattr(cls.draw, "_draw_funcs", None)) | return bool(getattr(cls.draw, "_draw_funcs", None)) | ||||
| @classmethod | @classmethod | ||||
| def append(cls, draw_func): | def append(cls, draw_func): | ||||
| """ | """ | ||||
| Append a draw function to this menu, | Append a draw function to this menu, | ||||
| takes the same arguments as the menus draw function | takes the same arguments as the menus draw function | ||||
| """ | """ | ||||
| draw_funcs = cls._dyn_ui_initialize() | draw_funcs = cls._dyn_ui_initialize() | ||||
| cls._dyn_bl_origin_apply(draw_func) | |||||
| draw_funcs.append(draw_func) | draw_funcs.append(draw_func) | ||||
| @classmethod | @classmethod | ||||
| def prepend(cls, draw_func): | def prepend(cls, draw_func): | ||||
| """ | """ | ||||
| Prepend a draw function to this menu, takes the same arguments as | Prepend a draw function to this menu, takes the same arguments as | ||||
| the menus draw function | the menus draw function | ||||
| """ | """ | ||||
| draw_funcs = cls._dyn_ui_initialize() | draw_funcs = cls._dyn_ui_initialize() | ||||
| cls._dyn_bl_origin_apply(draw_func) | |||||
| draw_funcs.insert(0, draw_func) | draw_funcs.insert(0, draw_func) | ||||
| @classmethod | @classmethod | ||||
| def remove(cls, draw_func): | def remove(cls, draw_func): | ||||
| """Remove a draw function that has been added to this menu""" | """Remove a draw function that has been added to this menu""" | ||||
| draw_funcs = cls._dyn_ui_initialize() | draw_funcs = cls._dyn_ui_initialize() | ||||
| try: | try: | ||||
| draw_funcs.remove(draw_func) | draw_funcs.remove(draw_func) | ||||
| ▲ Show 20 Lines • Show All 169 Lines • Show Last 20 Lines | |||||