feat: add fast prediction menu

interface/__init__.py

@@ -25,6 +25,7 @@
 from . import main_menu
 from . import sketch_menu
 from . import utils_menu
+from . import fast_menu
 from . import boolean_menu
 from . import bevel_menu
 from . import extrude_menu
@@ -42,6 +43,7 @@
     main_menu,
     sketch_menu,
     utils_menu,
+    fast_menu,
     boolean_menu,
     bevel_menu,
     extrude_menu,

interface/fast_menu.py

@@ -0,0 +1,179 @@
+# ███╗   ██╗██████╗ 
+# ████╗  ██║██╔══██╗
+# ██╔██╗ ██║██║  ██║
+# ██║╚██╗██║██║  ██║
+# ██║ ╚████║██████╔╝
+# ╚═╝  ╚═══╝╚═════╝ 
+# 
+# “Commons Clause” License Condition v1.0
+# 
+# See LICENSE for license details. If you did not receive a copy of the license,
+# it may be obtained at https://github.com/hugemenace/nd/blob/main/LICENSE.
+# 
+# Software: ND Blender Addon
+# License: MIT
+# Licensor: T.S. & I.J. (HugeMenace)
+# 
+# ---
+# Contributors: Tristo (HM)
+# ---
+
+import bpy
+import bmesh
+from .. import bl_info
+
+
+keys = []
+
+
+class ND_MT_fast_menu(bpy.types.Menu):
+    bl_label = "ND v%s — Fast (Predict)" % ('.'.join([str(v) for v in bl_info['version']]))
+    bl_idname = "ND_MT_fast_menu"
+
+
+    def draw(self, context):
+        objs = context.selected_objects
+
+        if len(objs) == 0:
+            return self.draw_no_object_predictions(context)
+
+        if len(objs) == 1:
+            return self.draw_single_object_predictions(context)
+
+        if len(objs) == 2:
+            return self.draw_two_object_predictions(context)
+
+        if len(objs) > 2:
+            return self.draw_many_object_predictions(context)
+
+        layout.label(text="Sorry, ND could not make a prediction.", icon='LIBRARY_DATA_BROKEN')
+
+
+    def draw_no_object_predictions(self, context):
+        layout = self.layout
+        layout.operator_context = 'INVOKE_DEFAULT'
+
+        layout.operator("nd.single_vertex", icon='DOT')
+        layout.operator("nd.recon_poly", icon='SURFACE_NCURVE')
+
+
+    def draw_two_object_predictions(self, context):
+        layout = self.layout
+        layout.operator_context = 'INVOKE_DEFAULT'
+
+        layout.operator("nd.bool_vanilla", text="Difference", icon='MOD_BOOLEAN').mode = 'DIFFERENCE'
+        layout.operator("nd.bool_slice", icon='MOD_BOOLEAN')
+        layout.operator("nd.bool_inset", icon='MOD_BOOLEAN')
+        layout.operator("nd.mirror", icon='MOD_MIRROR')
+        layout.operator("nd.circular_array", icon='DRIVER_ROTATIONAL_DIFFERENCE')
+        layout.operator("nd.snap_align", icon='SNAP_ON')
+
+
+    def draw_single_object_predictions(self, context):
+        layout = self.layout
+        layout.operator_context = 'INVOKE_DEFAULT'
+
+        depsgraph = context.evaluated_depsgraph_get()
+        object_eval = context.object.evaluated_get(depsgraph)
+
+        bm = bmesh.new()
+        bm.from_mesh(object_eval.data)
+
+        self.verts = [vert for vert in bm.verts]
+        self.edges = [edge for edge in bm.edges]
+        self.faces = [face for face in bm.faces]
+        self.sketch = len(self.faces) == 1
+        self.profile = len(self.faces) == 0 and len(self.edges) > 0
+        self.form = len(self.faces) > 1
+
+        bm.free()
+
+        if context.mode == 'EDIT_MESH':
+            mesh = bmesh.from_edit_mesh(context.object.data)
+
+            if len([vert for vert in mesh.verts if vert.select]) >= 1:
+                layout.operator("nd.vertex_bevel", icon='VERTEXSEL')
+                layout.operator("nd.clear_vgs", icon='GROUP_VERTEX')
+
+            if len([edge for edge in mesh.edges if edge.select]) >= 1:
+                layout.operator("nd.edge_bevel", icon='EDGESEL')
+
+            if self.profile:
+                layout.operator("nd.make_manifold", icon='OUTLINER_DATA_SURFACE')
+
+            return
+
+        if context.mode == 'OBJECT':
+            mod_names = [mod.name for mod in context.object.modifiers]
+
+            has_mod_pe = False
+            has_mod_sol = False 
+            for name in mod_names:
+                if "— ND PE" in name:
+                    has_mod_pe = True
+                if "— ND SOL" in name:
+                    has_mod_sol = True
+
+            was_profile_extrude = has_mod_pe and not has_mod_sol
+
+            if "Bool —" in context.object.name:
+                layout.operator("nd.hydrate", icon='SHADING_RENDERED')
+
+                return
+
+            if was_profile_extrude or self.sketch:
+                layout.operator("nd.solidify", icon='MOD_SOLIDIFY')
+                layout.operator("nd.mirror", icon='MOD_MIRROR')
+
+                return
+
+            if self.profile:
+                layout.operator("nd.profile_extrude", icon='EMPTY_SINGLE_ARROW')
+                layout.operator("nd.screw", icon='MOD_SCREW')
+
+                return
+
+            if self.form:
+                layout.operator("nd.bevel", icon='MOD_BEVEL')
+                layout.operator("nd.weighted_normal_bevel", icon='MOD_BEVEL')
+                layout.operator("nd.array_cubed", icon='PARTICLES')
+                layout.operator("nd.screw_head", icon='CANCEL')
+                layout.operator("nd.flare", text="Flare (Lighting)", icon='LIGHT_AREA')
+
+                return
+
+
+    def draw_many_object_predictions(self, context):
+        layout = self.layout
+        layout.operator_context = 'INVOKE_DEFAULT'
+
+        layout.operator("nd.mirror", icon='MOD_MIRROR')
+        layout.operator("nd.triangulate", icon='MOD_TRIANGULATE')
+        layout.operator("nd.apply_modifiers", icon='ORPHAN_DATA')
+        layout.operator("nd.name_sync", icon='FILE_REFRESH')
+        layout.operator("nd.set_lod_suffix", text="Low LOD", icon='ALIASED').mode = 'LOW'
+        layout.operator("nd.set_lod_suffix", text="High LOD", icon='ANTIALIASED').mode = 'HIGH'
+
+
+def draw_item(self, context):
+    layout = self.layout
+    layout.menu(ND_MT_fast_menu.bl_idname)
+
+
+def register():
+    bpy.utils.register_class(ND_MT_fast_menu)
+
+    for mapping in [('3D View', 'VIEW_3D'), ('Mesh', 'EMPTY'), ('Object Mode', 'EMPTY')]:
+        keymap = bpy.context.window_manager.keyconfigs.addon.keymaps.new(name=mapping[0], space_type=mapping[1])
+        entry = keymap.keymap_items.new("wm.call_menu", 'F', 'PRESS', shift=True)
+        entry.properties.name = "ND_MT_fast_menu"
+        keys.append((keymap, entry))
+
+
+def unregister():
+    for keymap, entry in keys:
+        keymap.keymap_items.remove(entry)
+
+    keys.clear()
+
+    bpy.utils.unregister_class(ND_MT_fast_menu)