Add Clockmender's Precision Drawing Tools v1.1.5

Accepted for inclusion per T70238

1 files changed, 539 insertions, 0 deletions

diff --git a/precision_drawing_tools/__init__.py b/precision_drawing_tools/__init__.py
new file mode 100644
index 00000000..b3c3024d
--- /dev/null
+++ b/precision_drawing_tools/__init__.py
@@ -0,0 +1,539 @@
+# ##### 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 #####
+#
+# <pep8 compliant>
+#
+# -----------------------------------------------------------------------
+# Author: Alan Odom (Clockmender), Rune Morling (ermo) Copyright (c) 2019
+# -----------------------------------------------------------------------
+#
+# ----------------------------------------------
+# Define Addon info
+# ----------------------------------------------
+#
+bl_info = {
+    "name": "Precision Drawing Tools (PDT)",
+    "author": "Alan Odom (Clockmender), Rune Morling (ermo)",
+    "version": (1, 1, 5),
+    "blender": (2, 80, 0),
+    "location": "View3D > UI > PDT",
+    "description": "Precision Drawing Tools for Acccurate Modelling",
+    "warning": "",
+    "wiki_url": "https://github.com/Clockmender/Precision-Drawing-Tools/wiki",
+    "category": "3D View",
+}
+
+
+# ----------------------------------------------
+# Import modules
+# ----------------------------------------------
+if "bpy" in locals():
+    import importlib
+
+    importlib.reload(pdt_design)
+    importlib.reload(pdt_pivot_point)
+    importlib.reload(pdt_menus)
+    importlib.reload(pdt_library)
+    importlib.reload(pdt_view)
+    importlib.reload(pdt_xall)
+    importlib.reload(pdt_bix)
+    importlib.reload(pdt_etof)
+else:
+    from . import pdt_design
+    from . import pdt_pivot_point
+    from . import pdt_menus
+    from . import pdt_library
+    from . import pdt_view
+    from . import pdt_xall
+    from . import pdt_bix
+    from . import pdt_etof
+
+import bpy
+import os
+from pathlib import Path
+from bpy.types import AddonPreferences, PropertyGroup, Scene, WindowManager
+from bpy.props import (
+    BoolProperty,
+    CollectionProperty,
+    EnumProperty,
+    FloatProperty,
+    FloatVectorProperty,
+    IntProperty,
+    PointerProperty,
+    StringProperty,
+)
+from .pdt_msg_strings import (
+    PDT_DES_COORDS,
+    PDT_DES_FILLETPROF,
+    PDT_DES_FILLETRAD,
+    PDT_DES_FILLETSEG,
+    PDT_DES_FILLETVERTS,
+    PDT_DES_FLIPANG,
+    PDT_DES_FLIPPER,
+    PDT_DES_LIBCOLS,
+    PDT_DES_LIBMATS,
+    PDT_DES_LIBMODE,
+    PDT_DES_LIBOBS,
+    PDT_DES_LIBSER,
+    PDT_DES_MOVESEL,
+    PDT_DES_OBORDER,
+    PDT_DES_OFFANG,
+    PDT_DES_OFFDIS,
+    PDT_DES_OFFPER,
+    PDT_DES_OPMODE,
+    PDT_DES_PIVOTDIS,
+    PDT_DES_PPLOC,
+    PDT_DES_PPSCALEFAC,
+    PDT_DES_PPSIZE,
+    PDT_DES_PPTRANS,
+    PDT_DES_PPWIDTH,
+    PDT_DES_ROTMOVAX,
+    PDT_DES_TRIM,
+    PDT_DES_VALIDLET,
+    PDT_DES_WORPLANE
+)
+from .pdt_command import command_run
+from .pdt_functions import scale_set
+
+
+# Declare enum items variables
+#
+_pdt_obj_items = []
+_pdt_col_items = []
+_pdt_mat_items = []
+
+
+def enumlist_objects(self, context):
+    """Populate Objects List from Parts Library.
+
+    Creates list of objects that optionally have search string contained in them
+    to populate variable pdt_lib_objects enumerator.
+
+    Args:
+        context: Blender bpy.context instance.
+
+    Returns:
+        list of Object Names.
+    """
+
+    scene = context.scene
+    pg = scene.pdt_pg
+    file_path = context.preferences.addons[__package__].preferences.pdt_library_path
+    path = Path(file_path)
+    _pdt_obj_items.clear()
+
+    if path.is_file() and ".blend" in str(path):
+        with bpy.data.libraries.load(str(path)) as (data_from, _):
+            if len(pg.object_search_string) == 0:
+                object_names = [ob for ob in data_from.objects]
+            else:
+                object_names = [ob for ob in data_from.objects if pg.object_search_string in ob]
+        for object_name in object_names:
+            _pdt_obj_items.append((object_name, object_name, ""))
+    else:
+        _pdt_obj_items.append(("MISSING", "Library is Missing", ""))
+    return _pdt_obj_items
+
+
+def enumlist_collections(self, context):
+    """Populate Collections List from Parts Library.
+
+    Creates list of collections that optionally have search string contained in them
+    to populate variable pg.lib_collections enumerator
+
+    Args:
+        context: Blender bpy.context instance.
+
+    Returns:
+        list of Collections Names.
+    """
+
+    scene = context.scene
+    pg = scene.pdt_pg
+    file_path = context.preferences.addons[__package__].preferences.pdt_library_path
+    path = Path(file_path)
+    _pdt_col_items.clear()
+
+    if path.is_file() and ".blend" in str(path):
+        with bpy.data.libraries.load(str(path)) as (data_from, _):
+            if len(pg.collection_search_string) == 0:
+                object_names = [ob for ob in data_from.collections]
+            else:
+                object_names = [ob for ob in data_from.collections if pg.collection_search_string in ob]
+        for object_name in object_names:
+            _pdt_col_items.append((object_name, object_name, ""))
+    else:
+        _pdt_col_items.append(("MISSING", "Library is Missing", ""))
+    return _pdt_col_items
+
+
+def enumlist_materials(self, context):
+    """Populate Materials List from Parts Library.
+
+    Creates list of materials that optionally have search string contained in them
+    to populate variable pg.lib_materials enumerator.
+
+    Args:
+        context: Blender bpy.context instance.
+
+    Returns:
+        list of Object Names.
+    """
+
+    scene = context.scene
+    pg = scene.pdt_pg
+    file_path = context.preferences.addons[__package__].preferences.pdt_library_path
+    path = Path(file_path)
+    _pdt_mat_items.clear()
+
+    if path.is_file() and ".blend" in str(path):
+        with bpy.data.libraries.load(str(path)) as (data_from, _):
+            if len(pg.material_search_string) == 0:
+                object_names = [ob for ob in data_from.materials]
+            else:
+                object_names = [ob for ob in data_from.materials if pg.material_search_string in ob]
+        for object_name in object_names:
+            _pdt_mat_items.append((object_name, object_name, ""))
+    else:
+        _pdt_mat_items.append(("MISSING", "Library is Missing", ""))
+    return _pdt_mat_items
+
+
+class PDTSceneProperties(PropertyGroup):
+    """Contains all PDT related properties."""
+
+    object_search_string : StringProperty(
+        name="Search", default="", description=PDT_DES_LIBSER
+    )
+    collection_search_string : StringProperty(
+        name="Search", default="", description=PDT_DES_LIBSER
+    )
+    material_search_string : StringProperty(
+        name="Search", default="", description=PDT_DES_LIBSER
+    )
+
+    cartesian_coords : FloatVectorProperty(
+        name="Coords",
+        default=(0.0, 0.0, 0.0),
+        subtype="XYZ",
+        description=PDT_DES_COORDS
+    )
+    distance : FloatProperty(
+        name="Distance", default=0.0, precision=5, description=PDT_DES_OFFDIS, unit="LENGTH"
+    )
+    angle : FloatProperty(
+        name="Angle", min=-180, max=180, default=0.0, precision=5, description=PDT_DES_OFFANG
+    )
+    percent : FloatProperty(
+        name="Percent", default=0.0, precision=5, description=PDT_DES_OFFPER
+    )
+    plane : EnumProperty(
+        items=(
+            ("XZ", "Front(X-Z)", "Use X-Z Plane"),
+            ("XY", "Top(X-Y)", "Use X-Y Plane"),
+            ("YZ", "Right(Y-Z)", "Use Y-Z Plane"),
+            ("LO", "View", "Use View Plane"),
+        ),
+        name="Working Plane",
+        default="XZ",
+        description=PDT_DES_WORPLANE,
+    )
+    select : EnumProperty(
+        items=(
+            ("REL", "Current", "Moved Relative to Current Position"),
+            (
+                "SEL",
+                "Selected",
+                "Moved Relative to Selected Object, or Vertex, Cursor & Pivot Only",
+            ),
+        ),
+        name="Move Mode",
+        default="SEL",
+        description=PDT_DES_MOVESEL,
+    )
+    operation : EnumProperty(
+        items=(
+            ("CU", "Cursor", "This Function will Move the Cursor"),
+            ("PP", "Pivot", "This Function will Move the Pivot Point"),
+            ("MV", "Move", "This function will Move Vertices, or Objects"),
+            ("NV", "New Vertex", "This function will Add a New Vertex"),
+            ("EV", "Extrude Vertices", "This function will Extrude Vertices Only in EDIT Mode"),
+            ("SE", "Split Edges", "This function will Split Edges Only in EDIT Mode"),
+            (
+                "DG",
+                "Duplicate Geometry",
+                "This function will Duplicate Geometry in EDIT Mode (Delta & Direction Only)",
+            ),
+            (
+                "EG",
+                "Extrude Geometry",
+                "This function will Extrude Geometry in EDIT Mode (Delta & Direction Only)",
+            ),
+        ),
+        name="Operation",
+        default="CU",
+        description=PDT_DES_OPMODE,
+    )
+    taper : EnumProperty(
+        items=(
+            ("RX-MY", "RotX-MovY", "Rotate X - Move Y"),
+            ("RX-MZ", "RotX-MovZ", "Rotate X - Move Z"),
+            ("RY-MX", "RotY-MovX", "Rotate Y - Move X"),
+            ("RY-MZ", "RotY-MovZ", "Rotate Y - Move Z"),
+            ("RZ-MX", "RotZ-MovX", "Rotate Z - Move X"),
+            ("RZ-MY", "RotZ-MovY", "Rotate Z - Move Y"),
+        ),
+        name="Axes",
+        default="RX-MY",
+        description=PDT_DES_ROTMOVAX,
+    )
+
+    flip_angle : BoolProperty(
+        name="Flip Angle", default=False, description=PDT_DES_FLIPANG
+    )
+    flip_percent : BoolProperty(
+        name="Flip %", default=False, description=PDT_DES_FLIPPER
+    )
+
+    extend : BoolProperty(
+        name="Trim/Extend All", default=False, description=PDT_DES_TRIM
+    )
+
+    lib_objects : EnumProperty(
+        items=enumlist_objects, name="Objects", description=PDT_DES_LIBOBS
+    )
+    lib_collections : EnumProperty(
+        items=enumlist_collections, name="Collections", description=PDT_DES_LIBCOLS
+    )
+    lib_materials : EnumProperty(
+        items=enumlist_materials, name="Materials", description=PDT_DES_LIBMATS
+    )
+    lib_mode : EnumProperty(
+        items=(
+            ("OBJECTS", "Objects", "Use Objects"),
+            ("COLLECTIONS", "Collections", "Use Collections"),
+            ("MATERIALS", "Materials", "Use Materials"),
+        ),
+        name="Mode",
+        default="OBJECTS",
+        description=PDT_DES_LIBMODE,
+    )
+
+    rotation_coords : FloatVectorProperty(
+        name="Rotation",
+        default=(0.0, 0.0, 0.0),
+        subtype="XYZ",
+        description="Rotation Coordinates"
+    )
+
+    object_order : EnumProperty(
+        items=(
+            ("1,2,3,4", "1,2,3,4", "Objects 1 & 2 are First Line"),
+            ("1,3,2,4", "1,3,2,4", "Objects 1 & 3 are First Line"),
+            ("1,4,2,3", "1,4,2,3", "Objects 1 & 4 are First Line"),
+        ),
+        name="Order",
+        default="1,2,3,4",
+        description=PDT_DES_OBORDER,
+    )
+    vrotangle : FloatProperty(name="View Rotate Angle", default=10, max=180, min=-180)
+    command : StringProperty(
+        name="Command",
+        default="CA0,0,0",
+        update=command_run,
+        description=PDT_DES_VALIDLET,
+    )
+    error : StringProperty(name="Error", default="")
+
+    # Was pivot* -- is now pivot_*
+    pivot_loc : FloatVectorProperty(
+        name="Pivot Location",
+        default=(0.0, 0.0, 0.0),
+        subtype="XYZ",
+        description=PDT_DES_PPLOC,
+    )
+    pivot_scale : FloatVectorProperty(
+        name="Pivot Scale", default=(1.0, 1.0, 1.0), subtype="XYZ", description=PDT_DES_PPSCALEFAC
+    )
+    pivot_size : FloatProperty(
+        name="Pivot Factor", min=0.4, max=10, default=1, precision=1, description=PDT_DES_PPSIZE
+    )
+    pivot_width : IntProperty(
+        name="Width", min=1, max=5, default=2, description=PDT_DES_PPWIDTH
+    )
+    # FIXME: might as well become pivot_angle
+    pivot_ang : FloatProperty(name="Pivot Angle", min=-180, max=180, default=0.0)
+    # FIXME: pivot_dist for consistency?
+    pivot_dis : FloatProperty(name="Pivot Dist",
+        default=0.0,
+        min = 0,
+        update=scale_set,
+        description=PDT_DES_PIVOTDIS,
+    )
+    pivot_alpha : FloatProperty(
+        name="Alpha",
+        min=0.2,
+        max=1,
+        default=0.6,
+        precision=1,
+        description=PDT_DES_PPTRANS,
+    )
+    pivot_show : BoolProperty()
+
+    # Was filletrad
+    fillet_radius : FloatProperty(
+        name="Fillet Radius", min=0.0, default=1.0, description=PDT_DES_FILLETRAD
+    )
+    # Was filletnum
+    fillet_segments : IntProperty(
+        name="Fillet Segments", min=1, default=4, description=PDT_DES_FILLETSEG
+    )
+    # Was filletpro
+    fillet_profile : FloatProperty(
+        name="Fillet Profile", min=0.0, max=1.0, default=0.5, description=PDT_DES_FILLETPROF
+    )
+    # Was filletbool
+    fillet_vertices_only : BoolProperty(
+        name="Fillet Vertices Only",
+        default=True,
+        description=PDT_DES_FILLETVERTS,
+    )
+
+
+class PDTPreferences(AddonPreferences):
+    # This must match the addon name, use '__package__' when defining this in a submodule of a python package.
+
+    bl_idname = __name__
+
+    pdt_library_path : StringProperty(
+        name="Parts Library", default="", description="Parts Library File",
+        maxlen=1024, subtype='FILE_PATH'
+    )
+
+    debug : BoolProperty(
+        name="Enable console debug output from PDT scripts", default=False,
+        description="NOTE: Does not enable debugging globally in Blender (only in PDT scripts)"
+    )
+
+    def draw(self, context):
+        layout = self.layout
+
+        box = layout.box()
+        row1 = box.row()
+        row2 = box.row()
+        row1.prop(self, "debug")
+        row2.prop(self, "pdt_library_path")
+
+
+# List of All Classes in the Add-on to register
+#
+# Due to the way PropertyGroups work, this needs to be listed/loaded first
+# (and unloaded last)
+#
+classes = (
+    PDTSceneProperties,
+    PDTPreferences,
+    pdt_bix.PDT_OT_LineOnBisection,
+    pdt_design.PDT_OT_PlacementAbs,
+    pdt_design.PDT_OT_PlacementDelta,
+    pdt_design.PDT_OT_PlacementDis,
+    pdt_design.PDT_OT_PlacementCen,
+    pdt_design.PDT_OT_PlacementPer,
+    pdt_design.PDT_OT_PlacementNormal,
+    pdt_design.PDT_OT_PlacementInt,
+    pdt_design.PDT_OT_JoinVerts,
+    pdt_design.PDT_OT_Angle2,
+    pdt_design.PDT_OT_Angle3,
+    pdt_design.PDT_OT_Origin,
+    pdt_design.PDT_OT_Taper,
+    pdt_design.PDT_OT_Fillet,
+    pdt_etof.PDT_OT_EdgeToFace,
+    pdt_library.PDT_OT_Append,
+    pdt_library.PDT_OT_Link,
+    pdt_library.PDT_OT_LibShow,
+    pdt_menus.PDT_PT_PanelDesign,
+    pdt_menus.PDT_PT_PanelCommandLine,
+    pdt_menus.PDT_PT_PanelViewControl,
+    pdt_menus.PDT_PT_PanelPivotPoint,
+    pdt_menus.PDT_PT_PanelPartsLibrary,
+    pdt_pivot_point.PDT_OT_ModalDrawOperator,
+    pdt_pivot_point.PDT_OT_ViewPlaneRotate,
+    pdt_pivot_point.PDT_OT_ViewPlaneScale,
+    pdt_pivot_point.PDT_OT_PivotToCursor,
+    pdt_pivot_point.PDT_OT_CursorToPivot,
+    pdt_pivot_point.PDT_OT_PivotSelected,
+    pdt_pivot_point.PDT_OT_PivotOrigin,
+    pdt_pivot_point.PDT_OT_PivotWrite,
+    pdt_pivot_point.PDT_OT_PivotRead,
+    pdt_view.PDT_OT_ViewRot,
+    pdt_view.PDT_OT_vRotL,
+    pdt_view.PDT_OT_vRotR,
+    pdt_view.PDT_OT_vRotU,
+    pdt_view.PDT_OT_vRotD,
+    pdt_view.PDT_OT_vRoll,
+    pdt_view.PDT_OT_viso,
+    pdt_xall.PDT_OT_IntersectAllEdges,
+)
+
+
+def register():
+    """Register Classes and Create Scene Variables.
+
+    Operates on the classes list defined above.
+    """
+
+    from bpy.utils import register_class
+
+    for cls in classes:
+        register_class(cls)
+
+    # OpenGL flag
+    #
+    wm = WindowManager
+    # Register Internal OpenGL Property
+    #
+    wm.pdt_run_opengl = BoolProperty(default=False)
+
+    Scene.pdt_pg = PointerProperty(type=PDTSceneProperties)
+
+
+def unregister():
+    """Unregister Classes and Delete Scene Variables.
+
+    Operates on the classes list defined above.
+    """
+
+    from bpy.utils import unregister_class
+
+    # remove OpenGL data
+    pdt_pivot_point.PDT_OT_ModalDrawOperator.handle_remove(
+        pdt_pivot_point.PDT_OT_ModalDrawOperator, bpy.context
+    )
+    wm = bpy.context.window_manager
+    p = "pdt_run_opengl"
+    if p in wm:
+        del wm[p]
+
+    for cls in reversed(classes):
+        unregister_class(cls)
+
+    del Scene.pdt_pg
+
+
+if __name__ == "__main__":
+    register()