Skip to content
Snippets Groups Projects
__init__.py 11.3 KiB
Newer Older
  • Learn to ignore specific revisions
  • Brendon Murphy's avatar
    fix
    Brendon Murphy committed
    # ##### 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 #####
    
    # Contributed to by:
    
    Brendon Murphy's avatar
    fix
    Brendon Murphy committed
    # testscreenings, Alejandro Omar Chocano Vasquez, Jimmy Hazevoet, meta-androcto #
    
    # Cmomoney, Jared Forsyth, Adam Newgas, Spivak Vladimir, Jared Forsyth, Atom    #
    # Antonio Osprite, Marius Giurgi (DolphinDream)
    
    Brendon Murphy's avatar
    fix
    Brendon Murphy committed
    
    bl_info = {
        "name": "Extra Objects",
        "author": "Multiple Authors",
    
        "version": (0, 1, 2),
    
    Brendon Murphy's avatar
    fix
    Brendon Murphy committed
        "location": "View3D > Add > Curve > Extra Objects",
        "description": "Add extra curve object types",
        "warning": "",
    
        "wiki_url": "https://wiki.blender.org/index.php/Extensions:2.6/Py/"
    
    Brendon Murphy's avatar
    fix
    Brendon Murphy committed
                    "Scripts/Curve/Curve_Objects",
    
    Brendon Murphy's avatar
    fix
    Brendon Murphy committed
    
    if "bpy" in locals():
    
        import importlib
        importlib.reload(add_curve_aceous_galore)
        importlib.reload(add_curve_spirals)
        importlib.reload(add_curve_torus_knots)
        importlib.reload(add_surface_plane_cone)
    
        importlib.reload(add_curve_curly)
        importlib.reload(beveltaper_curve)
        importlib.reload(add_curve_celtic_links)
        importlib.reload(add_curve_braid)
        importlib.reload(add_curve_simple)
    
        importlib.reload(add_curve_spirofit_bouncespline)
    
    Brendon Murphy's avatar
    fix
    Brendon Murphy committed
    
    else:
        from . import add_curve_aceous_galore
        from . import add_curve_spirals
        from . import add_curve_torus_knots
    
        from . import add_surface_plane_cone
    
        from . import add_curve_curly
        from . import beveltaper_curve
        from . import add_curve_celtic_links
        from . import add_curve_braid
        from . import add_curve_simple
    
        from . import add_curve_spirofit_bouncespline
    
    Brendon Murphy's avatar
    fix
    Brendon Murphy committed
    
    import bpy
    
    from bpy.types import (
            Menu,
            AddonPreferences,
            )
    from bpy.props import (
            StringProperty,
            BoolProperty,
            )
    
    Brendon Murphy's avatar
    fix
    Brendon Murphy committed
    
    
    
    def convert_old_presets(data_path, msg_data_path, old_preset_subdir,
                            new_preset_subdir, fixdic={}, ext=".py"):
        """
        convert old presets
        """
    
    
        def convert_presets(self, context):
            if not getattr(self, data_path, False):
                return None
            import os
    
            target_path = os.path.join("presets", old_preset_subdir)
            target_path = bpy.utils.user_resource('SCRIPTS',
                                                  target_path)
    
            # created an anytype op to run against preset
            op = type('', (), {})()
    
            files = [f for f in os.listdir(target_path) if f.endswith(ext)]
            if not files:
                print("No old presets in %s" % target_path)
                setattr(self, msg_data_path, "No old presets")
                return None
    
            new_target_path = os.path.join("presets", new_preset_subdir)
            new_target_path = bpy.utils.user_resource('SCRIPTS',
                                                  new_target_path,
                                                  create=True)
            for f in files:
                file = open(os.path.join(target_path, f))
                for line in file:
                    if line.startswith("op."):
                        exec(line)
                file.close()
                for key, items in fixdic.items():
                    if hasattr(op, key) and isinstance(getattr(op, key), int):
                        setattr(op, key, items[getattr(op, key)])
                # create a new one
                new_file_path = os.path.join(new_target_path, f)
                if os.path.isfile(new_file_path):
                    # do nothing
                    print("Preset %s already exists, passing..." % f)
                    continue
                file_preset = open(new_file_path, 'w')
                file_preset.write("import bpy\n")
                file_preset.write("op = bpy.context.active_operator\n")
    
                for prop, value in vars(op).items():
    
                    if isinstance(value, str):
                        file_preset.write("op.%s = '%s'\n" % (prop, str(value)))
                    else:
                        file_preset.write("op.%s = %s\n" % (prop, str(value)))
    
                file_preset.close()
                print("Writing new preset to %s" % new_file_path)
    
            setattr(self, msg_data_path, "Converted %d old presets" % len(files))
            return None
    
    class CurveExtraObjectsAddonPreferences(AddonPreferences):
        bl_idname = __name__
    
    
        spiral_fixdic = {
                "spiral_type": ['ARCH', 'ARCH', 'LOG', 'SPHERE', 'TORUS'],
                "curve_type": ['POLY', 'NURBS'],
                "spiral_direction": ['COUNTER_CLOCKWISE', 'CLOCKWISE']
                }
        update_spiral_presets_msg = StringProperty(
                default="Nothing to do"
                )
    
        update_spiral_presets = BoolProperty(
                name="Update Old Presets",
                description="Update presets to reflect data changes",
                default=False,
    
                update=convert_old_presets(
                        "update_spiral_presets",  # this props name
                        "update_spiral_presets_msg",  # message prop
                        "operator/curve.spirals",
                        "curve_extras/curve.spirals",
                        fixdic=spiral_fixdic
                        )
                )
        show_menu_list = BoolProperty(
                name="Menu List",
                description="Show/Hide the Add Menu items",
                default=False
                )
        show_panel_list = BoolProperty(
                name="Panels List",
                description="Show/Hide the Panel items",
                default=False
    
                )
    
        def draw(self, context):
            layout = self.layout
    
            box = layout.box()
            box.label(text="Spirals:")
    
    
            if self.update_spiral_presets:
    
                box.label(self.update_spiral_presets_msg, icon="FILE_TICK")
    
                box.prop(self, "update_spiral_presets")
    
            icon_1 = "TRIA_RIGHT" if not self.show_menu_list else "TRIA_DOWN"
            box = layout.box()
            box.prop(self, "show_menu_list", emboss=False, icon=icon_1)
    
            if self.show_menu_list:
                box.label(text="Items located in the Add Menu > Curve (default shortcut Ctrl + A):",
                          icon="LAYER_USED")
                box.label(text="2D Objects:", icon="LAYER_ACTIVE")
                box.label(text="Angle, Arc, Circle, Distance, Ellipse, Line, Point, Polygon,",
                          icon="LAYER_USED")
                box.label(text="Polygon ab, Rectangle, Rhomb, Sector, Segment, Trapezoid",
                          icon="LAYER_USED")
                box.label(text="Curve Profiles:", icon="LAYER_ACTIVE")
                box.label(text="Arc, Arrow, Cogwheel, Cycloid, Flower, Helix (3D),",
                          icon="LAYER_USED")
                box.label(text="Noise (3D), Nsided, Profile, Rectangle, Splat, Star",
                          icon="LAYER_USED")
                box.label(text="Curve Spirals:", icon="LAYER_ACTIVE")
                box.label(text="Archemedian, Logarithmic, Spheric, Torus",
                          icon="LAYER_USED")
                box.label(text="Knots:", icon="LAYER_ACTIVE")
                box.label(text="Torus Knots Plus, Celtic Links, Braid Knot",
                          icon="LAYER_USED")
                box.label(text="Curly Curve", icon="LAYER_ACTIVE")
                box.label(text="Bevel/Taper:", icon="LAYER_ACTIVE")
                box.label(text="Add Curve as Bevel, Add Curve as Taper",
                          icon="LAYER_USED")
    
                box.label(text="Items located in the Add Menu > Surface (default shortcut Ctrl + A):",
                          icon="LAYER_USED")
                box.label(text="Wedge, Cone, Star, Plane",
                          icon="LAYER_ACTIVE")
    
            icon_2 = "TRIA_RIGHT" if not self.show_panel_list else "TRIA_DOWN"
            box = layout.box()
            box.prop(self, "show_panel_list", emboss=False, icon=icon_2)
    
            if self.show_panel_list:
                box.label(text="Panel located in 3D View Tools Region > Create:",
                          icon="LAYER_ACTIVE")
                box.label(text="Spline:", icon="LAYER_ACTIVE")
                box.label(text="SpiroFit, Bounce Spline, Catenary", icon="LAYER_USED")
                box.label(text="Panel located in 3D View Tools Region > Tools:",
                          icon="LAYER_ACTIVE")
                box.label(text="Simple Curve:", icon="LAYER_ACTIVE")
                box.label(text="Available if the Active Object is a Curve was created with 2D Objects",
                         icon="LAYER_USED")
    
    
    class INFO_MT_curve_knots_add1(Menu):
    
    Brendon Murphy's avatar
    fix
    Brendon Murphy committed
        # Define the "Extras" menu
    
        bl_idname = "curve_knots_add"
        bl_label = "Plants"
    
    Brendon Murphy's avatar
    fix
    Brendon Murphy committed
    
        def draw(self, context):
            layout = self.layout
            layout.operator_context = 'INVOKE_REGION_WIN'
    
            layout.operator("curve.torus_knot_plus", text="Torus Knot Plus")
            layout.operator("curve.celtic_links", text="Celtic Links")
            layout.operator("mesh.add_braid", text="Braid Knot")
    
    Brendon Murphy's avatar
    fix
    Brendon Murphy committed
    def menu_func(self, context):
    
        if context.mode != 'OBJECT':
            # fix in D2142 will allow to work in EDIT_CURVE
            return None
    
        layout.operator_menu_enum("mesh.curveaceous_galore", "ProfileType",
                                  icon='CURVE_DATA')
        layout.operator_menu_enum("curve.spirals", "spiral_type",
                                  icon='CURVE_DATA')
    
        layout.separator()
    
        layout.menu("curve_knots_add", text="Knots", icon='CURVE_DATA')
        layout.separator()
    
        layout.operator("curve.curlycurve", text="Curly Curve",
                        icon='CURVE_DATA')
        layout.menu("OBJECT_MT_bevel_taper_curve_menu", text="Bevel/Taper",
                    icon='CURVE_DATA')
    
    Brendon Murphy's avatar
    fix
    Brendon Murphy committed
    
    
    def menu_surface(self, context):
        self.layout.separator()
    
        if context.mode == 'EDIT_SURFACE':
    
            self.layout.operator("curve.smooth_x_times",
                                 text="Special Smooth", icon="MOD_CURVE")
    
        elif context.mode == 'OBJECT':
    
            self.layout.operator("object.add_surface_wedge", text="Wedge",
                                 icon="SURFACE_DATA")
            self.layout.operator("object.add_surface_cone", text="Cone",
                                 icon="SURFACE_DATA")
            self.layout.operator("object.add_surface_star", text="Star",
                                 icon="SURFACE_DATA")
            self.layout.operator("object.add_surface_plane", text="Plane",
                                 icon="SURFACE_DATA")
    
    
    Brendon Murphy's avatar
    fix
    Brendon Murphy committed
    def register():
    
        add_curve_simple.register()
    
    Brendon Murphy's avatar
    fix
    Brendon Murphy committed
        bpy.utils.register_module(__name__)
    
        # Add "Extras" menu to the "Add Curve" menu
        bpy.types.INFO_MT_curve_add.append(menu_func)
    
        # Add "Extras" menu to the "Add Surface" menu
        bpy.types.INFO_MT_surface_add.append(menu_surface)
    
    Brendon Murphy's avatar
    fix
    Brendon Murphy committed
    
    
    
        add_curve_simple.unregister()
    
    Brendon Murphy's avatar
    fix
    Brendon Murphy committed
        # Remove "Extras" menu from the "Add Curve" menu.
        bpy.types.INFO_MT_curve_add.remove(menu_func)
    
        # Remove "Extras" menu from the "Add Surface" menu.
        bpy.types.INFO_MT_surface_add.remove(menu_surface)
    
    Brendon Murphy's avatar
    fix
    Brendon Murphy committed
    
    
        bpy.utils.unregister_module(__name__)
    
    
    Brendon Murphy's avatar
    fix
    Brendon Murphy committed
    if __name__ == "__main__":
        register()