workspacetools.py

from bpy.types import WorkSpaceTool
from . import operators, gizmos
import os

tool_prefix = "sketcher."


def get_addon_icon_path(icon_name):
    return os.path.join(os.path.dirname(__file__), "icons", icon_name)


def tool_invoke_kmi(button, tool, operator):
    return (
        operators.View3D_OT_invoke_tool.bl_idname,
        {"type": button, "value": "PRESS"},
        {"properties": [("tool_name", tool), ("operator", operator)]},
    )


tool_access = (
    tool_invoke_kmi(
        "P",
        "sketcher.slvs_add_point2d",
        operators.View3D_OT_slvs_add_point2d.bl_idname,
    ),
    tool_invoke_kmi(
        "L", "sketcher.slvs_add_line2d", operators.View3D_OT_slvs_add_line2d.bl_idname
    ),
    tool_invoke_kmi(
        "C",
        "sketcher.slvs_add_circle2d",
        operators.View3D_OT_slvs_add_circle2d.bl_idname,
    ),
    tool_invoke_kmi(
        "A", "sketcher.slvs_add_arc2d", operators.View3D_OT_slvs_add_arc2d.bl_idname
    ),
    tool_invoke_kmi(
        "R",
        "sketcher.slvs_add_rectangle",
        operators.View3D_OT_slvs_add_rectangle.bl_idname,
    ),
    (
        operators.View3D_OT_slvs_add_sketch.bl_idname,
        {"type": "S", "value": "PRESS"},
        {"properties": [("wait_for_input", True), ]}
    )
)


def tool_numeric_invoke_km(operator):
    km = []
    for event in operators.numeric_events:
        km.append(
            (
                operator,
                {"type": event, "value": "PRESS"},
                None,
            )
        )
    return km


class VIEW3D_T_slvs_select(WorkSpaceTool):
    bl_space_type = "VIEW_3D"
    bl_context_mode = "OBJECT"
    bl_idname = tool_prefix + "slvs_select"
    bl_label = "Solvespace Select"
    bl_description = "Select Solvespace Entities"
    bl_icon = "ops.generic.select"
    bl_widget = gizmos.VIEW3D_GGT_slvs_preselection.bl_idname
    bl_keymap = (
        (
            operators.View3D_OT_slvs_select_all.bl_idname,
            {"type": "ESC", "value": "PRESS"},
            {"properties": [("deselect", True)]},
        ),
        (
            operators.View3D_OT_slvs_select_all.bl_idname,
            {"type": "A", "value": "PRESS", "ctrl": True},
            {"properties": [("deselect", False)]},
        ),
        (
            operators.View3D_OT_slvs_select.bl_idname,
            {"type": "LEFTMOUSE", "value": "CLICK"},
            None,
        ),
        (
            operators.View3D_OT_slvs_tweak.bl_idname,
            {"type": "LEFTMOUSE", "value": "CLICK_DRAG"},
            None,
        ),
        (
            operators.View3D_OT_slvs_tweak.bl_idname,
            {"type": "LEFTMOUSE", "value": "PRESS", "ctrl": True},
            None,
        ),
        (
            operators.View3D_OT_slvs_context_menu.bl_idname,
            {"type": "RIGHTMOUSE", "value": "PRESS"},
            None,
        ),
        (
            operators.View3D_OT_slvs_delete_entity.bl_idname,
            {"type": "DEL", "value": "PRESS"},
            None,
        ),
        *tool_access,
    )

    def draw_settings(context, layout, tool):
        props = tool.operator_properties(operators.View3D_OT_slvs_select.bl_idname)


tool_keymap = (
    (
        "wm.tool_set_by_id",
        {"type": "ESC", "value": "PRESS"},
        {"properties": [("name", VIEW3D_T_slvs_select.bl_idname)]},
    ),
    (
        "wm.tool_set_by_id",
        {"type": "RIGHTMOUSE", "value": "PRESS"},
        {"properties": [("name", VIEW3D_T_slvs_select.bl_idname)]},
    ),
)


def operator_access(operator):
    return (
        *tool_numeric_invoke_km(operator),
        (
            operator,
            {"type": "LEFTMOUSE", "value": "PRESS"},
            {"properties": [("wait_for_input", False)]},
        ),
    )


class GenericStateTool():
    @classmethod
    def bl_description(cls, context, item, keymap):
        op_name = cls.bl_operator if hasattr(cls, "bl_operator") else ""
        desc = ""

        if op_name:
            import _bpy
            desc = _bpy.ops.get_rna_type(op_name).description

        for kmi in tool_access:
            if kmi[0] == operators.View3D_OT_invoke_tool.bl_idname:
                props = kmi[2]["properties"]
                i = [prop[0] for prop in props].index("tool_name")
                if props[i][1] == cls.bl_idname:
                    button = kmi[1]["type"]
                    desc += "\n\nInvoke Shortcut: " + button

        return desc

class View3D_T_slvs_add_point3d(GenericStateTool, WorkSpaceTool):
    bl_space_type = "VIEW_3D"
    bl_context_mode = "OBJECT"
    bl_idname = tool_prefix + "slvs_add_point3d"
    bl_label = "Add 3D Point"
    bl_operator = operators.View3D_OT_slvs_add_point3d.bl_idname
    bl_icon = get_addon_icon_path("ops.bgs.add_point")
    bl_widget = gizmos.VIEW3D_GGT_slvs_preselection.bl_idname
    bl_keymap = (
        *tool_keymap,
        *tool_access,
        *operator_access(operators.View3D_OT_slvs_add_point3d.bl_idname),
    )


class View3D_T_slvs_add_point2d(GenericStateTool, WorkSpaceTool):
    bl_space_type = "VIEW_3D"
    bl_context_mode = "OBJECT"
    bl_idname = tool_prefix + "slvs_add_point2d"
    bl_label = "Add 2D Point"
    bl_operator = operators.View3D_OT_slvs_add_point2d.bl_idname
    bl_icon = get_addon_icon_path("ops.bgs.add_point")
    bl_widget = gizmos.VIEW3D_GGT_slvs_preselection.bl_idname
    bl_keymap = (
        *tool_keymap,
        *tool_access,
        *operator_access(operators.View3D_OT_slvs_add_point2d.bl_idname),
    )


class View3D_T_slvs_add_line3d(GenericStateTool, WorkSpaceTool):
    bl_space_type = "VIEW_3D"
    bl_context_mode = "OBJECT"
    bl_idname = tool_prefix + "slvs_add_line3d"
    bl_label = "Add 3D Line"
    bl_operator = operators.View3D_OT_slvs_add_line3d.bl_idname
    bl_icon = "ops.gpencil.primitive_line"
    bl_widget = gizmos.VIEW3D_GGT_slvs_preselection.bl_idname
    bl_keymap = (
        *tool_keymap,
        *tool_access,
        *operator_access(operators.View3D_OT_slvs_add_line3d.bl_idname),
    )

    def draw_settings(context, layout, tool):
        props = tool.operator_properties(operators.View3D_OT_slvs_add_line3d.bl_idname)
        layout.prop(props, "continuose_draw")


class View3D_T_slvs_add_line2d(GenericStateTool, WorkSpaceTool):
    bl_space_type = "VIEW_3D"
    bl_context_mode = "OBJECT"
    bl_idname = tool_prefix + "slvs_add_line2d"
    bl_label = "Add 2D Line"
    bl_operator = operators.View3D_OT_slvs_add_line2d.bl_idname
    bl_icon = "ops.gpencil.primitive_line"
    bl_widget = gizmos.VIEW3D_GGT_slvs_preselection.bl_idname
    bl_keymap = (
        *tool_keymap,
        *tool_access,
        *operator_access(operators.View3D_OT_slvs_add_line2d.bl_idname),
    )

    def draw_settings(context, layout, tool):
        props = tool.operator_properties(operators.View3D_OT_slvs_add_line2d.bl_idname)
        layout.prop(props, "continuose_draw")


class View3D_T_slvs_add_circle2d(GenericStateTool, WorkSpaceTool):
    bl_space_type = "VIEW_3D"
    bl_context_mode = "OBJECT"
    bl_idname = tool_prefix + "slvs_add_circle2d"
    bl_label = "Add 2D Circle"
    bl_operator = operators.View3D_OT_slvs_add_circle2d.bl_idname
    bl_icon = "ops.gpencil.primitive_circle"
    bl_widget = gizmos.VIEW3D_GGT_slvs_preselection.bl_idname
    bl_keymap = (
        *tool_keymap,
        *tool_access,
        *operator_access(operators.View3D_OT_slvs_add_circle2d.bl_idname),
    )


class View3D_T_slvs_add_arc2d(GenericStateTool, WorkSpaceTool):
    bl_space_type = "VIEW_3D"
    bl_context_mode = "OBJECT"
    bl_idname = tool_prefix + "slvs_add_arc2d"
    bl_label = "Add 2D Arc"
    bl_operator = operators.View3D_OT_slvs_add_arc2d.bl_idname
    bl_icon = "ops.gpencil.primitive_arc"
    bl_widget = gizmos.VIEW3D_GGT_slvs_preselection.bl_idname
    bl_keymap = (
        *tool_keymap,
        *tool_access,
        *operator_access(operators.View3D_OT_slvs_add_arc2d.bl_idname),
    )


class View3D_T_slvs_add_rectangle(GenericStateTool, WorkSpaceTool):
    bl_space_type = "VIEW_3D"
    bl_context_mode = "OBJECT"
    bl_idname = tool_prefix + "slvs_add_rectangle"
    bl_label = "Add Rectangle"
    bl_operator = operators.View3D_OT_slvs_add_rectangle.bl_idname
    bl_icon = "ops.gpencil.primitive_box"
    bl_widget = gizmos.VIEW3D_GGT_slvs_preselection.bl_idname
    bl_keymap = (
        *tool_keymap,
        *tool_access,
        *operator_access(operators.View3D_OT_slvs_add_rectangle.bl_idname),
    )

class View3D_T_slvs_add_workplane_face(GenericStateTool, WorkSpaceTool):
    bl_space_type = "VIEW_3D"
    bl_context_mode = "OBJECT"
    bl_idname = tool_prefix + "slvs_add_workplane_face"
    bl_label = "Add Workplane on mesh face"
    bl_operator = operators.View3D_OT_slvs_add_workplane_face.bl_idname
    bl_icon = "ops.mesh.primitive_grid_add_gizmo"
    bl_widget = gizmos.VIEW3D_GGT_slvs_preselection.bl_idname
    bl_keymap = (
        *tool_keymap,
        *tool_access,
        *operator_access(operators.View3D_OT_slvs_add_workplane_face.bl_idname),
    )

class View3D_T_slvs_add_workplane(GenericStateTool, WorkSpaceTool):
    bl_space_type = "VIEW_3D"
    bl_context_mode = "OBJECT"
    bl_idname = tool_prefix + "slvs_add_workplane"
    bl_label = "Add Workplane"
    bl_operator = operators.View3D_OT_slvs_add_workplane.bl_idname
    bl_icon = "ops.mesh.primitive_grid_add_gizmo"
    bl_widget = gizmos.VIEW3D_GGT_slvs_preselection.bl_idname
    bl_keymap = (
        *tool_keymap,
        *tool_access,
        *operator_access(operators.View3D_OT_slvs_add_workplane.bl_idname),
    )


tools = (
    (VIEW3D_T_slvs_select, {"separator": True, "group": False}),
    (View3D_T_slvs_add_point2d, {"separator": True, "group": True}),
    (
        View3D_T_slvs_add_point3d,
        {
            "after": {View3D_T_slvs_add_point2d.bl_idname},
        },
    ),
    (View3D_T_slvs_add_line2d, {"separator": False, "group": True}),
    (
        View3D_T_slvs_add_line3d,
        {
            "after": {View3D_T_slvs_add_line2d.bl_idname},
        },
    ),
    (View3D_T_slvs_add_circle2d, {"separator": False, "group": False}),
    (View3D_T_slvs_add_arc2d, {"separator": False, "group": False}),
    (View3D_T_slvs_add_rectangle, {"separator": False, "group": False}),
    (View3D_T_slvs_add_workplane_face, {"separator": True, "group": True}),
    (View3D_T_slvs_add_workplane, {"after": {View3D_T_slvs_add_workplane_face.bl_idname}}),
)

import bpy
def register():
    if bpy.app.background:
        return

    from bpy.utils import register_tool

    for tool in tools:
        register_tool(tool[0], **tool[1])


def unregister():
    if bpy.app.background:
        return

    from bpy.utils import unregister_tool

    for tool in reversed(tools):
        unregister_tool(tool[0])