mesh_snap_utilities_line/__init__.py

   1 # SPDX-License-Identifier: GPL-2.0-or-later
   2
   3 # Contact for more information about the Addon:
   4 # Email: germano.costa@ig.com.br
   5 # Twitter: wii_mano @mano_wii
   6 bl_info = {
   7     "name": "Snap_Utilities_Line",
   8     "author": "Germano Cavalcante",
   9     "version": (6, 9, 5),
  10     "blender": (3, 0, 0),
  11     "location": "View3D > TOOLS > Line Tool",
  12     "description": "Extends Blender Snap controls",
  13     "doc_url" : "{BLENDER_MANUAL_URL}/addons/mesh/snap_utilities_line.html",
  14     "category": "Mesh",
  15 }
  16
  17 if "bpy" in locals():
  18     import importlib
  19     importlib.reload(navigation_ops)
  20     importlib.reload(widgets)
  21     importlib.reload(preferences)
  22     importlib.reload(op_line)
  23     importlib.reload(keys)
  24 else:
  25     from . import navigation_ops
  26     from . import widgets
  27     from . import preferences
  28     from . import op_line
  29     from . import keys
  30
  31 import bpy
  32 from bpy.utils.toolsystem import ToolDef
  33
  34 if not __package__:
  35     __package__ = "mesh_snap_utilities_line"
  36
  37 @ToolDef.from_fn
  38 def tool_line():
  39     import os
  40     def draw_settings(context, layout, tool):
  41         addon_prefs = context.preferences.addons[__package__].preferences
  42
  43         layout.prop(addon_prefs, "incremental")
  44         layout.prop(addon_prefs, "increments_grid")
  45         layout.prop(addon_prefs, "intersect")
  46         layout.prop(addon_prefs, "create_face")
  47         if context.mode == 'EDIT_MESH':
  48             layout.prop(addon_prefs, "outer_verts")
  49         #props = tool.operator_properties("mesh.snap_utilities_line")
  50         #layout.prop(props, "radius")
  51
  52     icons_dir = os.path.join(os.path.dirname(__file__), "icons")
  53
  54     return dict(
  55         idname="snap_utilities.line",
  56         label="Make Line",
  57         description=(
  58             "Make Lines\n"
  59             "Connect them to split faces"
  60         ),
  61         icon=os.path.join(icons_dir, "ops.mesh.snap_utilities_line"),
  62         widget="MESH_GGT_snap_point",
  63         #operator="mesh.snap_utilities_line",
  64         keymap=keys.km_tool_snap_utilities_line,
  65         draw_settings=draw_settings,
  66     )
  67
  68
  69 # -----------------------------------------------------------------------------
  70 # Tool Registraion
  71
  72
  73 def get_tool_list(space_type, context_mode):
  74     from bl_ui.space_toolsystem_common import ToolSelectPanelHelper
  75     cls = ToolSelectPanelHelper._tool_class_from_space_type(space_type)
  76     return cls._tools[context_mode]
  77
  78
  79 def register_snap_tools():
  80     tools = get_tool_list('VIEW_3D', 'EDIT_MESH')
  81
  82     for index, tool in enumerate(tools, 1):
  83         if isinstance(tool, ToolDef) and tool.label == "Measure":
  84             break
  85
  86     tools[:index] += None, tool_line
  87
  88     del tools
  89
  90
  91 def unregister_snap_tools():
  92     tools = get_tool_list('VIEW_3D', 'EDIT_MESH')
  93
  94     index = tools.index(tool_line) - 1 #None
  95     tools.pop(index)
  96     tools.remove(tool_line)
  97
  98     del tools
  99     del index
 100
 101
 102 def register_keymaps():
 103     keyconfigs = bpy.context.window_manager.keyconfigs
 104     kc_defaultconf = keyconfigs.default
 105     kc_addonconf = keyconfigs.addon
 106     # In background mode.
 107     if kc_defaultconf is None or kc_addonconf is None:
 108         return
 109
 110     # TODO: find the user defined tool_mouse.
 111     from bl_keymap_utils.io import keyconfig_init_from_data
 112     keyconfig_init_from_data(kc_defaultconf, keys.generate_empty_snap_utilities_tools_keymaps())
 113     keyconfig_init_from_data(kc_addonconf, keys.generate_snap_utilities_keymaps())
 114
 115     #snap_modalkeymap = kc_addonconf.keymaps.find(keys.km_snap_utilities_modal_keymap)
 116     #snap_modalkeymap.assign("MESH_OT_snap_utilities_line")
 117 def unregister_keymaps():
 118     keyconfigs = bpy.context.window_manager.keyconfigs
 119     defaultmap = getattr(keyconfigs.get("Blender"), "keymaps", None)
 120     addonmap = getattr(keyconfigs.get("Blender addon"), "keymaps", None)
 121     # In background mode.
 122     if defaultmap is None or addonmap is None:
 123         return
 124
 125     for keyconfig_data in keys.generate_snap_utilities_global_keymaps():
 126         km_name, km_args, km_content = keyconfig_data
 127         keymap = addonmap.find(km_name, **km_args)
 128         keymap_items = keymap.keymap_items
 129         for item in km_content['items']:
 130             item_id = keymap_items.find(item[0])
 131             if item_id != -1:
 132                 keymap_items.remove(keymap_items[item_id])
 133
 134     for keyconfig_data in keys.generate_snap_utilities_tools_keymaps():
 135         km_name, km_args, km_content = keyconfig_data
 136         addonmap.remove(addonmap.find(km_name, **km_args))
 137
 138     for keyconfig_data in keys.generate_empty_snap_utilities_tools_keymaps():
 139         km_name, km_args, km_content = keyconfig_data
 140         defaultmap.remove(defaultmap.find(km_name, **km_args))
 141
 142
 143 # -----------------------------------------------------------------------------
 144 # Addon Registraion
 145
 146 classes = (
 147     preferences.SnapUtilitiesPreferences,
 148     op_line.SnapUtilitiesLine,
 149     navigation_ops.VIEW3D_OT_rotate_custom_pivot,
 150     navigation_ops.VIEW3D_OT_zoom_custom_target,
 151     widgets.SnapPointWidget,
 152     widgets.SnapPointWidgetGroup,
 153 )
 154
 155 def register():
 156     for cls in classes:
 157         bpy.utils.register_class(cls)
 158
 159     register_snap_tools()
 160     register_keymaps()
 161
 162
 163 def unregister():
 164     unregister_keymaps()
 165     unregister_snap_tools()
 166
 167     for cls in reversed(classes):
 168         bpy.utils.unregister_class(cls)
 169
 170
 171 if __name__ == "__main__":
 172     register()