Nathan
130 lines
3.3 KiB
Python
130 lines
3.3 KiB
Python
import bpy
|
|
|
|
from ..functions.canvas import (
|
|
is_canvas,
|
|
)
|
|
|
|
from .common import (
|
|
boolean_operators_menu,
|
|
boolean_extras_menu,
|
|
)
|
|
|
|
|
|
#### ------------------------------ /boolean_menu/ ------------------------------ ####
|
|
|
|
# 3D Viewport (Object Mode) -> Object
|
|
class VIEW3D_MT_boolean(bpy.types.Menu):
|
|
bl_label = "Boolean"
|
|
bl_idname = "VIEW3D_MT_boolean"
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
layout.menu("VIEW3D_MT_carve")
|
|
layout.separator()
|
|
boolean_operators_menu(self, context)
|
|
boolean_extras_menu(self, context)
|
|
|
|
|
|
def object_mode_menu(self, context):
|
|
layout = self.layout
|
|
layout.separator()
|
|
layout.menu("VIEW3D_MT_boolean")
|
|
|
|
|
|
|
|
#### ------------------------------ /popup_menu/ ------------------------------ ####
|
|
|
|
# Shift-Ctrl-B Menu
|
|
class VIEW3D_MT_boolean_popup(bpy.types.Menu):
|
|
bl_label = "Boolean"
|
|
bl_idname = "VIEW3D_MT_boolean_popup"
|
|
|
|
def draw(self, context):
|
|
boolean_operators_menu(self, context)
|
|
boolean_extras_menu(self, context)
|
|
|
|
|
|
|
|
#### ------------------------------ /carve_menu/ ------------------------------ ####
|
|
|
|
class VIEW3D_MT_carve(bpy.types.Menu):
|
|
bl_label = "Carve"
|
|
bl_idname = "VIEW3D_MT_carve"
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
layout.operator("object.carve_box", text="Box Carve")
|
|
layout.operator("object.carve_circle", text="Circle Carve")
|
|
layout.operator("object.carve_polyline", text="Polyline Carve")
|
|
|
|
|
|
# Separate "Carve" menu in Edit Mode (where we don't have "Boolean" menu).
|
|
def edit_mode_menu(self, context):
|
|
layout = self.layout
|
|
layout.separator()
|
|
layout.menu("VIEW3D_MT_carve")
|
|
|
|
|
|
|
|
#### ------------------------------ /select_menu/ ------------------------------ ####
|
|
|
|
def select_menu(self, context):
|
|
layout = self.layout
|
|
obj = context.active_object
|
|
|
|
if not obj:
|
|
return
|
|
|
|
if is_canvas(obj) or obj.booleans.cutter:
|
|
layout.separator()
|
|
|
|
if is_canvas(obj):
|
|
layout.operator("object.boolean_select_all", text="Boolean Cutters")
|
|
if obj.booleans.cutter:
|
|
layout.operator("object.select_cutter_canvas", text="Boolean Canvases")
|
|
|
|
|
|
|
|
#### ------------------------------ REGISTRATION ------------------------------ ####
|
|
|
|
addon_keymaps = []
|
|
|
|
classes = (
|
|
VIEW3D_MT_boolean,
|
|
VIEW3D_MT_boolean_popup,
|
|
VIEW3D_MT_carve,
|
|
)
|
|
|
|
def register():
|
|
for cls in classes:
|
|
bpy.utils.register_class(cls)
|
|
|
|
# MENU
|
|
bpy.types.VIEW3D_MT_object.append(object_mode_menu)
|
|
bpy.types.VIEW3D_MT_edit_mesh.append(edit_mode_menu)
|
|
bpy.types.VIEW3D_MT_select_object.append(select_menu)
|
|
|
|
# KEYMAP
|
|
addon = bpy.context.window_manager.keyconfigs.addon
|
|
km = addon.keymaps.new(name="Object Mode")
|
|
|
|
kmi = km.keymap_items.new("wm.call_menu", 'B', 'PRESS', ctrl=True, shift=True)
|
|
kmi.properties.name = "VIEW3D_MT_boolean_popup"
|
|
kmi.active = True
|
|
addon_keymaps.append((km, kmi))
|
|
|
|
|
|
def unregister():
|
|
for cls in reversed(classes):
|
|
bpy.utils.unregister_class(cls)
|
|
|
|
# MENU
|
|
bpy.types.VIEW3D_MT_object.remove(object_mode_menu)
|
|
bpy.types.VIEW3D_MT_edit_mesh.remove(edit_mode_menu)
|
|
bpy.types.VIEW3D_MT_select_object.remove(select_menu)
|
|
|
|
# KEYMAP
|
|
for km, kmi in addon_keymaps:
|
|
km.keymap_items.remove(kmi)
|
|
addon_keymaps.clear()
|