diff options
Diffstat (limited to 'uv_magic_uv/utils')
-rw-r--r-- | uv_magic_uv/utils/__init__.py | 34 | ||||
-rw-r--r-- | uv_magic_uv/utils/bl_class_registry.py | 84 | ||||
-rw-r--r-- | uv_magic_uv/utils/property_class_registry.py | 72 |
3 files changed, 190 insertions, 0 deletions
diff --git a/uv_magic_uv/utils/__init__.py b/uv_magic_uv/utils/__init__.py new file mode 100644 index 00000000..4ce9d907 --- /dev/null +++ b/uv_magic_uv/utils/__init__.py @@ -0,0 +1,34 @@ +# <pep8-80 compliant> + +# ##### BEGIN GPL LICENSE BLOCK ##### +# +# This program is free software; you can redistribute it and/or +# modify it under the terms of the GNU General Public License +# as published by the Free Software Foundation; either version 2 +# of the License, or (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program; if not, write to the Free Software Foundation, +# Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. +# +# ##### END GPL LICENSE BLOCK ##### + +__author__ = "Nutti <nutti.metro@gmail.com>" +__status__ = "production" +__version__ = "5.2" +__date__ = "17 Nov 2018" + +if "bpy" in locals(): + import importlib + importlib.reload(bl_class_registry) + importlib.reload(property_class_registry) +else: + from . import bl_class_registry + from . import property_class_registry + +import bpy diff --git a/uv_magic_uv/utils/bl_class_registry.py b/uv_magic_uv/utils/bl_class_registry.py new file mode 100644 index 00000000..d1730615 --- /dev/null +++ b/uv_magic_uv/utils/bl_class_registry.py @@ -0,0 +1,84 @@ +# <pep8-80 compliant> + +# ##### BEGIN GPL LICENSE BLOCK ##### +# +# This program is free software; you can redistribute it and/or +# modify it under the terms of the GNU General Public License +# as published by the Free Software Foundation; either version 2 +# of the License, or (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program; if not, write to the Free Software Foundation, +# Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. +# +# ##### END GPL LICENSE BLOCK ##### + +__author__ = "Nutti <nutti.metro@gmail.com>" +__status__ = "production" +__version__ = "5.2" +__date__ = "17 Nov 2018" + +import bpy + +from .. import common + +__all__ = [ + 'BlClassRegistry', +] + + +class BlClassRegistry: + class_list = [] + + def __init__(self, *_, **kwargs): + self.legacy = kwargs.get('legacy', False) + + def __call__(self, cls): + if hasattr(cls, "bl_idname"): + BlClassRegistry.add_class(cls.bl_idname, cls, self.legacy) + else: + bl_idname = "{}{}{}{}".format(cls.bl_space_type, + cls.bl_region_type, + cls.bl_context, cls.bl_label) + BlClassRegistry.add_class(bl_idname, cls, self.legacy) + return cls + + @classmethod + def add_class(cls, bl_idname, op_class, legacy): + for class_ in cls.class_list: + if (class_["bl_idname"] == bl_idname) and \ + (class_["legacy"] == legacy): + raise RuntimeError("{} is already registered" + .format(bl_idname)) + + new_op = { + "bl_idname": bl_idname, + "class": op_class, + "legacy": legacy, + } + cls.class_list.append(new_op) + common.debug_print("{} is registered.".format(bl_idname)) + + @classmethod + def register(cls): + for class_ in cls.class_list: + bpy.utils.register_class(class_["class"]) + common.debug_print("{} is registered to Blender." + .format(class_["bl_idname"])) + + @classmethod + def unregister(cls): + for class_ in cls.class_list: + bpy.utils.unregister_class(class_["class"]) + common.debug_print("{} is unregistered from Blender." + .format(class_["bl_idname"])) + + @classmethod + def cleanup(cls): + cls.class_list = [] + common.debug_print("Cleanup registry.") diff --git a/uv_magic_uv/utils/property_class_registry.py b/uv_magic_uv/utils/property_class_registry.py new file mode 100644 index 00000000..20df0342 --- /dev/null +++ b/uv_magic_uv/utils/property_class_registry.py @@ -0,0 +1,72 @@ +# <pep8-80 compliant> + +# ##### BEGIN GPL LICENSE BLOCK ##### +# +# This program is free software; you can redistribute it and/or +# modify it under the terms of the GNU General Public License +# as published by the Free Software Foundation; either version 2 +# of the License, or (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program; if not, write to the Free Software Foundation, +# Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. +# +# ##### END GPL LICENSE BLOCK ##### + +__author__ = "Nutti <nutti.metro@gmail.com>" +__status__ = "production" +__version__ = "5.2" +__date__ = "17 Nov 2018" + +from .. import common + +__all__ = [ + 'PropertyClassRegistry', +] + + +class PropertyClassRegistry: + class_list = [] + + def __init__(self, *_, **kwargs): + self.legacy = kwargs.get('legacy', False) + + def __call__(self, cls): + PropertyClassRegistry.add_class(cls.idname, cls, self.legacy) + return cls + + @classmethod + def add_class(cls, idname, prop_class, legacy): + for class_ in cls.class_list: + if (class_["idname"] == idname) and (class_["legacy"] == legacy): + raise RuntimeError("{} is already registered".format(idname)) + + new_op = { + "idname": idname, + "class": prop_class, + "legacy": legacy, + } + cls.class_list.append(new_op) + common.debug_print("{} is registered.".format(idname)) + + @classmethod + def init_props(cls, scene): + for class_ in cls.class_list: + class_["class"].init_props(scene) + common.debug_print("{} is initialized.".format(class_["idname"])) + + @classmethod + def del_props(cls, scene): + for class_ in cls.class_list: + class_["class"].del_props(scene) + common.debug_print("{} is cleared.".format(class_["idname"])) + + @classmethod + def cleanup(cls): + cls.class_list = [] + common.debug_print("Cleanup registry.") |