keymaps.py

import bpy

from .declarations import Macros, Operators, WorkSpaceTools
from .stateful_operator.utilities.keymap import tool_invoke_kmi

constraint_access = (
    (
        Operators.AddCoincident,
        {"type": "C", "value": "PRESS", "shift": True},
        {
            "properties": [
                ("wait_for_input", True),
            ]
        },
    ),
    (
        Operators.AddVertical,
        {"type": "V", "value": "PRESS", "shift": True},
        {
            "properties": [
                ("wait_for_input", True),
            ]
        },
    ),
    (
        Operators.AddHorizontal,
        {"type": "H", "value": "PRESS", "shift": True},
        {
            "properties": [
                ("wait_for_input", True),
            ]
        },
    ),
    (
        Operators.AddEqual,
        {"type": "E", "value": "PRESS", "shift": True},
        {
            "properties": [
                ("wait_for_input", True),
            ]
        },
    ),
    (
        Operators.AddParallel,
        {"type": "A", "value": "PRESS", "shift": True},
        {
            "properties": [
                ("wait_for_input", True),
            ]
        },
    ),
    (
        Operators.AddPerpendicular,
        {"type": "P", "value": "PRESS", "shift": True},
        {
            "properties": [
                ("wait_for_input", True),
            ]
        },
    ),
    (
        Operators.AddTangent,
        {"type": "T", "value": "PRESS", "shift": True},
        {
            "properties": [
                ("wait_for_input", True),
            ]
        },
    ),
    (
        Operators.AddMidPoint,
        {"type": "M", "value": "PRESS", "shift": True},
        {
            "properties": [
                ("wait_for_input", True),
            ]
        },
    ),
    (
        Operators.AddRatio,
        {"type": "R", "value": "PRESS", "shift": True},
        {
            "properties": [
                ("wait_for_input", True),
            ]
        },
    ),
    # Dimensional Constraints
    (
        Operators.AddDistance,
        {"type": "D", "value": "PRESS", "alt": True},
        {
            "properties": [
                ("wait_for_input", True),
            ]
        },
    ),
    (
        Operators.AddDistance,
        {"type": "V", "value": "PRESS", "alt": True},
        {"properties": [("wait_for_input", True), ("align", "VERTICAL")]},
    ),
    (
        Operators.AddDistance,
        {"type": "H", "value": "PRESS", "alt": True},
        {"properties": [("wait_for_input", True), ("align", "HORIZONTAL")]},
    ),
    (
        Operators.AddAngle,
        {"type": "A", "value": "PRESS", "alt": True},
        {
            "properties": [
                ("wait_for_input", True),
            ]
        },
    ),
    (
        Operators.AddDiameter,
        {"type": "O", "value": "PRESS", "alt": True},
        {
            "properties": [
                ("wait_for_input", True),
            ]
        },
    ),
    (
        Operators.AddDiameter,
        {"type": "R", "value": "PRESS", "alt": True},
        {"properties": [("wait_for_input", True), ("setting", True)]},
    ),
)

tool_access = (
    tool_invoke_kmi(
        "P",
        WorkSpaceTools.AddPoint2D,
        Operators.AddPoint2D,
    ),
    tool_invoke_kmi(
        "L",
        WorkSpaceTools.AddLine2D,
        Operators.AddLine2D,
    ),
    tool_invoke_kmi(
        "C",
        WorkSpaceTools.AddCircle2D,
        Operators.AddCircle2D,
    ),
    tool_invoke_kmi(
        "A",
        WorkSpaceTools.AddArc2D,
        Operators.AddArc2D,
    ),
    tool_invoke_kmi(
        "R",
        WorkSpaceTools.AddRectangle,
        Operators.AddRectangle,
    ),
    tool_invoke_kmi(
        "Y",
        WorkSpaceTools.Trim,
        Operators.Trim,
    ),
    tool_invoke_kmi(
        "B",
        WorkSpaceTools.Bevel,
        Operators.Bevel,
    ),
    (
        Operators.AddSketch,
        {"type": "S", "value": "PRESS"},
        {
            "properties": [
                ("wait_for_input", True),
            ]
        },
    ),
    *constraint_access,
)

disable_gizmos = (
    # Disabling gizmos when pressing ctrl + shift
    # Add two entries so it doesn't matter which key is pressed first
    # NOTE: This cannot be done as a normal modifier key to selection since it has to toggle a global property
    (
        "wm.context_set_boolean",
        {"type": "LEFT_SHIFT", "value": "PRESS", "ctrl": True},
        {
            "properties": [
                ("data_path", "scene.sketcher.selectable_constraints"),
                ("value", False),
            ]
        },
    ),
    (
        "wm.context_set_boolean",
        {"type": "LEFT_CTRL", "value": "PRESS", "shift": True},
        {
            "properties": [
                ("data_path", "scene.sketcher.selectable_constraints"),
                ("value", False),
            ]
        },
    ),
    (
        "wm.context_set_boolean",
        {"type": "LEFT_SHIFT", "value": "RELEASE", "any": True},
        {
            "properties": [
                ("data_path", "scene.sketcher.selectable_constraints"),
                ("value", True),
            ]
        },
    ),
)

use_construction = (
    (
        "wm.context_toggle",
        {"type": "C", "value": "PRESS", "alt": True, "shift": True},
        {
            "properties": [
                ("data_path", "scene.sketcher.use_construction"),
            ]
        },
    )
)

tool_use_select = (
    (
        "wm.tool_set_by_id",
        {"type": "ESC", "value": "PRESS"},
        {"properties": [("name", WorkSpaceTools.Select)]},
    ),
    (
        "wm.tool_set_by_id",
        {"type": "RIGHTMOUSE", "value": "PRESS"},
        {"properties": [("name", WorkSpaceTools.Select)]},
    ),
)

tool_generic = (
    *disable_gizmos,
    use_construction,
    *tool_use_select,
    *tool_access,
    (
        Operators.DeleteEntity,
        {"type": "DEL", "value": "PRESS"},
        None,
    ),
    (
        Operators.Copy,
        {"type": "C", "value": "PRESS", "ctrl": True},
        None,
    ),
    (
        Operators.Paste,
        {"type": "V", "value": "PRESS", "ctrl": True},
        None,
    ),
    (
        Macros.DuplicateMove,
        {"type": "D", "value": "PRESS", "shift": True},
        None,
    ),
    (
        Operators.Move,
        {"type": "G", "value": "PRESS"},
        None,
    ),
)

tool_select = (
    *disable_gizmos,
    *tool_access,
    (
        Operators.SelectAll,
        {"type": "ESC", "value": "PRESS"},
        {"properties": [("deselect", True)]},
    ),
    (
        Operators.SelectAll,
        {"type": "A", "value": "PRESS", "ctrl": True},
        {"properties": [("deselect", False)]},
    ),
    (
        Operators.Select,
        {"type": "LEFTMOUSE", "value": "CLICK", "any": True},
        None,
    ),
    (
        Operators.Select,
        {"type": "LEFTMOUSE", "value": "CLICK", "shift": True},
        {"properties": [("mode", "EXTEND")]},
    ),
    (
        Operators.Select,
        {"type": "LEFTMOUSE", "value": "CLICK", "ctrl": True},
        {"properties": [("mode", "SUBTRACT")]},
    ),
    (
        Operators.SelectInvert,
        {"type": "I", "value": "PRESS", "ctrl": True},
        None,
    ),
    (
        Operators.SelectExtend,
        {"type": "E", "value": "PRESS", "ctrl": True},
        None,
    ),
    (
        Operators.SelectExtendAll,
        {"type": "E", "value": "PRESS", "ctrl": True, "shift": True},
        None,
    ),
    (
        Operators.SelectBox,
        {"type": "LEFTMOUSE", "value": "CLICK_DRAG"},
        None,
    ),
    (
        Operators.SelectBox,
        {"type": "LEFTMOUSE", "value": "CLICK_DRAG", "ctrl": True},
        {"properties": [("mode", "SUBTRACT")]},
    ),
    (
        Operators.SelectBox,
        {"type": "LEFTMOUSE", "value": "CLICK_DRAG", "shift": True},
        {"properties": [("mode", "EXTEND")]},
    ),
    (
        Operators.Tweak,
        {"type": "LEFTMOUSE", "value": "CLICK_DRAG"},
        None,
    ),
    (
        Operators.ContextMenu,
        {"type": "RIGHTMOUSE", "value": "PRESS"},
        {"properties": [("delayed", True)]},
    ),
)

addon_keymaps = []


def register():
    wm = bpy.context.window_manager
    if wm.keyconfigs.addon:
        km = wm.keyconfigs.addon.keymaps.new(name="Object Mode", space_type="EMPTY")

        # Select
        kmi = km.keymap_items.new("wm.tool_set_by_id", "ESC", "PRESS", shift=True)
        kmi.properties.name = WorkSpaceTools.Select
        addon_keymaps.append((km, kmi))

        # Add Sketch
        kmi = km.keymap_items.new(
            Operators.AddSketch, "A", "PRESS", ctrl=True, shift=True
        )
        kmi.properties.wait_for_input = True
        addon_keymaps.append((km, kmi))


def unregister():
    wm = bpy.context.window_manager
    kc = wm.keyconfigs.addon
    if kc:
        for km, kmi in addon_keymaps:
            km.keymap_items.remove(kmi)
            addon_keymaps.clear()