space_view3d_pie_menus/pie_apply_transform_menu.py

   1 # SPDX-License-Identifier: GPL-2.0-or-later
   2
   3 bl_info = {
   4     "name": "Hotkey: 'Ctrl A'",
   5     "description": "Apply Transform Menu",
   6     "author": "pitiwazou, meta-androcto",
   7     "version": (0, 1, 1),
   8     "blender": (2, 80, 0),
   9     "location": "3D View",
  10     "warning": "",
  11     "doc_url": "",
  12     "category": "Apply Transform Pie"
  13 }
  14
  15 import bpy
  16 from bpy.types import (
  17     Menu,
  18     Operator,
  19 )
  20 from bpy.props import EnumProperty
  21
  22
  23 # Pie Apply Transforms - Ctrl + A
  24 class PIE_MT_PieApplyTransforms(Menu):
  25     bl_idname = "PIE_MT_applytransforms"
  26     bl_label = "Pie Apply Transforms"
  27
  28     def draw(self, context):
  29         layout = self.layout
  30         pie = layout.menu_pie()
  31         # 4 - LEFT
  32         pie.operator("object.visual_transform_apply", text="Apply Visual")
  33         # 6 - RIGHT
  34         props = pie.operator("object.transform_apply", text="Apply All")
  35         props.location, props.rotation, props.scale = (True, True, True)
  36         # 2 - BOTTOM
  37         props = pie.operator("object.transform_apply", text="Rotation/Scale")
  38         props.location, props.rotation, props.scale = (False, True, True)
  39         # 8 - TOP
  40         props = pie.operator("object.transform_apply", text="Rotation")
  41         props.location, props.rotation, props.scale = (False, True, False)
  42         # 7 - TOP - LEFT
  43         props = pie.operator("object.transform_apply", text="Location")
  44         props.location, props.rotation, props.scale = (True, False, False)
  45         # 9 - TOP - RIGHT
  46         props = pie.operator("object.transform_apply", text="Scale")
  47         props.location, props.rotation, props.scale = (False, False, True)
  48         # 1 - BOTTOM - LEFT
  49         pie.operator("object.duplicates_make_real", text="Make Instances Real")
  50         # 3 - BOTTOM - RIGHT
  51         pie.menu("PIE_MT_clear_menu", text="Clear Transform Menu")
  52
  53
  54 # Clear Menu
  55 class PIE_MT_ClearMenu(Menu):
  56     bl_idname = "PIE_MT_clear_menu"
  57     bl_label = "Clear Menu"
  58
  59     def draw(self, context):
  60         layout = self.layout
  61         layout.operator("clear.all", text="Clear All", icon='NONE')
  62         layout.operator("object.location_clear", text="Clear Location", icon='NONE')
  63         layout.operator("object.rotation_clear", text="Clear Rotation", icon='NONE')
  64         layout.operator("object.scale_clear", text="Clear Scale", icon='NONE')
  65         layout.operator("object.origin_clear", text="Clear Origin", icon='NONE')
  66
  67
  68 # Clear all
  69 class PIE_OT_ClearAll(Operator):
  70     bl_idname = "clear.all"
  71     bl_label = "Clear All"
  72     bl_description = "Clear All Transforms"
  73     bl_options = {'REGISTER', 'UNDO'}
  74
  75     def execute(self, context):
  76         bpy.ops.object.location_clear()
  77         bpy.ops.object.rotation_clear()
  78         bpy.ops.object.scale_clear()
  79         return {'FINISHED'}
  80
  81
  82 classes = (
  83     PIE_MT_PieApplyTransforms,
  84     PIE_MT_ClearMenu,
  85     PIE_OT_ClearAll,
  86 )
  87
  88 addon_keymaps = []
  89
  90
  91 def register():
  92     for cls in classes:
  93         bpy.utils.register_class(cls)
  94     wm = bpy.context.window_manager
  95
  96     if wm.keyconfigs.addon:
  97         # Apply Transform
  98         km = wm.keyconfigs.addon.keymaps.new(name='Object Mode')
  99         kmi = km.keymap_items.new('wm.call_menu_pie', 'A', 'PRESS', ctrl=True)
 100         kmi.properties.name = "PIE_MT_applytransforms"
 101         addon_keymaps.append((km, kmi))
 102
 103
 104 def unregister():
 105     for cls in classes:
 106         bpy.utils.unregister_class(cls)
 107
 108     wm = bpy.context.window_manager
 109     kc = wm.keyconfigs.addon
 110     if kc:
 111         for km, kmi in addon_keymaps:
 112             km.keymap_items.remove(kmi)
 113     addon_keymaps.clear()
 114
 115
 116 if __name__ == "__main__":
 117     register()