cursor_to_transform.py

"""
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 3 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, see <https://www.gnu.org/licenses/>.
"""

import bpy
import bmesh
import math
import mathutils as mu
import numpy as np

OLD_DRAW = None
OLD_OBJECT_ADD = None

bl_info = {
    "name": "3D Cursor Enhancements",
    "author": "Tommi Hyppänen (ambi)",
    "version": (1, 0, 6),
    "blender": (2, 80, 0),
    "location": "View3D > Sidebar > View > 3D Cursor",
    "description": "Various tools to use 3D cursor rotation for more purposes.",
    "category": "Object",
}


def rotate_object(obj, q, point):
    R = q.to_matrix().to_4x4()
    T = mu.Matrix.Translation(point)
    M = T @ R @ T.inverted()
    obj.location = M @ obj.location
    obj.rotation_euler.rotate(M)


class TFC_OT_SeriousTableFlip(bpy.types.Operator):
    bl_idname = "object.serious_table_flip"
    bl_label = "Serious table flip"
    bl_description = "Transform all selected objects according to the 3D cursor"
    bl_options = {"REGISTER", "UNDO"}

    def execute(self, context):
        mode = context.object.mode
        bpy.ops.object.mode_set(mode="OBJECT")

        to_qt = context.scene.cursor.rotation_euler.to_quaternion()
        to_qt.invert()

        to_tf = context.scene.cursor.location

        for obj in context.selected_objects:
            rotate_object(obj, to_qt, to_tf)
            obj.location -= to_tf

        context.scene.cursor.location = mu.Vector((0, 0, 0))
        context.scene.cursor.rotation_euler = mu.Vector((0, 0, 0))

        bpy.ops.object.mode_set(mode=mode)

        return {"FINISHED"}


class TFC_OT_TransformFromCursor(bpy.types.Operator):
    bl_idname = "object.transform_from_cursor"
    bl_label = "Transform from cursor"
    bl_description = "Copy 3D cursor rotation to object transform rotation"
    bl_options = {"REGISTER", "UNDO"}

    def execute(self, context):
        mode = context.object.mode
        bpy.ops.object.mode_set(mode="OBJECT")

        bpy.ops.object.transform_apply(location=False, rotation=True, scale=False)
        # obj = context.active_object

        to_qt = context.scene.cursor.rotation_euler.to_quaternion()

        to_qt.invert()
        context.active_object.rotation_euler = to_qt.to_euler()
        bpy.ops.object.transform_apply(location=False, rotation=True, scale=False)
        to_qt.invert()

        context.active_object.rotation_euler = to_qt.to_euler()

        bpy.ops.object.mode_set(mode=mode)

        return {"FINISHED"}


# classes = (TFC_OT_TransformFromCursor, TFC_OT_SeriousTableFlip)
classes = (TFC_OT_TransformFromCursor, )


def register():
    for c in classes:
        bpy.utils.register_class(c)

    bpy.types.Scene.new_objects_to_cursor = bpy.props.BoolProperty(default=False)
    bpy.types.Scene.new_objects_to_cursor_offset = bpy.props.FloatProperty(default=0.0, min=-100.0, max=100.0)

    # ----- HACK: 3D cursor panel additions -----
    global OLD_DRAW
    if OLD_DRAW is None:
        OLD_DRAW = bpy.types.VIEW3D_PT_view3d_cursor.draw

    def draw(self, context):
        layout = self.layout

        cursor = context.scene.cursor

        layout.column().prop(cursor, "location", text="Location")
        rotation_mode = cursor.rotation_mode
        if rotation_mode == "QUATERNION":
            layout.column().prop(cursor, "rotation_quaternion", text="Rotation")
        elif rotation_mode == "AXIS_ANGLE":
            layout.column().prop(cursor, "rotation_axis_angle", text="Rotation")
        else:
            layout.column().prop(cursor, "rotation_euler", text="Rotation")

        layout.prop(cursor, "rotation_mode", text="")

        row = layout.row()
        row.operator(TFC_OT_TransformFromCursor.bl_idname, text="Copy to object")

        # row = layout.row()
        # row.operator(TFC_OT_SeriousTableFlip.bl_idname, text="Serious table flip")

        # row = layout.row()
        # row.prop(context.scene, "new_objects_to_cursor", text="Align new objects to cursor")

        # if context.scene.new_objects_to_cursor is True:
        #     row = layout.row()
        #     row.prop(context.scene, "new_objects_to_cursor_offset", text="Surface offset")

    bpy.types.VIEW3D_PT_view3d_cursor.draw = draw

    # # ----- HACK: Object add menu additions -----
    # global OLD_OBJECT_ADD
    # if OLD_OBJECT_ADD is None:
    #     OLD_OBJECT_ADD = bpy.types.VIEW3D_MT_mesh_add.draw

    # def draw(self, context):
    #     layout = self.layout

    #     layout.operator_context = "INVOKE_REGION_WIN"

    #     if context.scene.new_objects_to_cursor is True:
    #         oft = context.scene.new_objects_to_cursor_offset
    #         step_1 = mu.Vector((0.0, 0.0, 1.0))
    #         cursor_rotation = context.scene.cursor.rotation_euler
    #         cursor_matrix = cursor_rotation.to_matrix()

    #         def op_sizes():
    #             sizes = {
    #                 "plane": 2,
    #                 "cube": 2,
    #                 "circle": 1,
    #                 "uv_sphere": 1,
    #                 "ico_sphere": 1,
    #                 "cylinder": 2,
    #                 "cone": 2,
    #                 "torus": 1,
    #                 "grid": 2,
    #                 "monkey": 2,
    #             }

    #             for entry in sizes:
    #                 op_name = "MESH_OT_primitive_" + entry + "_add"
    #                 op = bpy.data.window_managers[0].operator_properties_last(op_name)
    #                 if op is not None:
    #                     if entry in ["plane", "cube", "grid", "monkey"]:
    #                         sizes[entry] = op.size / sizes[entry]
    #                     if entry in ["circle", "uv_sphere", "ico_sphere"]:
    #                         sizes[entry] = op.radius / sizes[entry]
    #                     if entry in ["cylinder", "cone"]:
    #                         sizes[entry] = op.depth / sizes[entry]
    #                     if entry == "torus":
    #                         sizes[entry] = op.major_radius / sizes[entry]

    #             return sizes

    #         sz = op_sizes()
    #         # print(sz)
    #         # print(bpy.data.window_managers[0].operators.keys())
    #         # print(bpy.data.window_managers[0].operators)

    #         op = layout.operator("mesh.primitive_plane_add", text="Plane", icon="MESH_PLANE")
    #         op.rotation = cursor_rotation
    #         op.location = context.scene.cursor.location + cursor_matrix @ step_1 * oft * sz["plane"]
    #         op = layout.operator("mesh.primitive_cube_add", text="Cube", icon="MESH_CUBE")
    #         op.rotation = cursor_rotation
    #         op.location = context.scene.cursor.location + cursor_matrix @ step_1 * oft * sz["cube"]
    #         op = layout.operator("mesh.primitive_circle_add", text="Circle", icon="MESH_CIRCLE")
    #         op.rotation = cursor_rotation
    #         op.location = context.scene.cursor.location + cursor_matrix @ step_1 * oft * sz["circle"]
    #         op = layout.operator("mesh.primitive_uv_sphere_add", text="UV Sphere", icon="MESH_UVSPHERE")
    #         op.rotation = cursor_rotation
    #         op.location = context.scene.cursor.location + cursor_matrix @ step_1 * oft * sz["uv_sphere"]
    #         op = layout.operator("mesh.primitive_ico_sphere_add", text="Ico Sphere", icon="MESH_ICOSPHERE")
    #         op.rotation = cursor_rotation
    #         op.location = context.scene.cursor.location + cursor_matrix @ step_1 * oft * sz["ico_sphere"]
    #         op = layout.operator("mesh.primitive_cylinder_add", text="Cylinder", icon="MESH_CYLINDER")
    #         op.rotation = cursor_rotation
    #         op.location = context.scene.cursor.location + cursor_matrix @ step_1 * oft * sz["cylinder"]
    #         op = layout.operator("mesh.primitive_cone_add", text="Cone", icon="MESH_CONE")
    #         op.rotation = cursor_rotation
    #         op.location = context.scene.cursor.location + cursor_matrix @ step_1 * oft * sz["cone"]
    #         op = layout.operator("mesh.primitive_torus_add", text="Torus", icon="MESH_TORUS")
    #         op.rotation = cursor_rotation
    #         op.location = context.scene.cursor.location + cursor_matrix @ step_1 * oft * sz["torus"]

    #         layout.separator()

    #         op = layout.operator("mesh.primitive_grid_add", text="Grid", icon="MESH_GRID")
    #         op.rotation = cursor_rotation
    #         op.location = context.scene.cursor.location + cursor_matrix @ step_1 * oft * sz["grid"]
    #         op = layout.operator("mesh.primitive_monkey_add", text="Monkey", icon="MESH_MONKEY")
    #         op.rotation = cursor_rotation
    #         op.location = context.scene.cursor.location + cursor_matrix @ step_1 * oft * sz["monkey"]

    #     else:
    #         op = layout.operator("mesh.primitive_plane_add", text="Plane", icon="MESH_PLANE")
    #         op = layout.operator("mesh.primitive_cube_add", text="Cube", icon="MESH_CUBE")
    #         op = layout.operator("mesh.primitive_circle_add", text="Circle", icon="MESH_CIRCLE")
    #         op = layout.operator("mesh.primitive_uv_sphere_add", text="UV Sphere", icon="MESH_UVSPHERE")
    #         op = layout.operator("mesh.primitive_ico_sphere_add", text="Ico Sphere", icon="MESH_ICOSPHERE")
    #         op = layout.operator("mesh.primitive_cylinder_add", text="Cylinder", icon="MESH_CYLINDER")
    #         op = layout.operator("mesh.primitive_cone_add", text="Cone", icon="MESH_CONE")
    #         op = layout.operator("mesh.primitive_torus_add", text="Torus", icon="MESH_TORUS")

    #         layout.separator()

    #         op = layout.operator("mesh.primitive_grid_add", text="Grid", icon="MESH_GRID")
    #         op = layout.operator("mesh.primitive_monkey_add", text="Monkey", icon="MESH_MONKEY")

    # bpy.types.VIEW3D_MT_mesh_add.draw = draw


def unregister():
    for c in reversed(classes):
        bpy.utils.unregister_class(c)

    del bpy.types.Scene.new_objects_to_cursor
    del bpy.types.Scene.new_objects_to_cursor_offset

    global OLD_DRAW
    if OLD_DRAW is not None:
        bpy.types.VIEW3D_PT_view3d_cursor.draw = OLD_DRAW

    # global OLD_OBJECT_ADD
    # if OLD_OBJECT_ADD is not None:
    #     bpy.types.VIEW3D_MT_mesh_add.draw = OLD_OBJECT_ADD


if __name__ == "__main__":
    register()