space_view3d_brush_menus/dyntopo_menu.py

   1 # SPDX-FileCopyrightText: 2017-2022 Blender Foundation
   2 #
   3 # SPDX-License-Identifier: GPL-2.0-or-later
   4
   5 import bpy
   6 from bpy.types import Menu
   7 from . import utils_core
   8
   9
  10 class DynTopoMenu(Menu):
  11     bl_label = "Dyntopo"
  12     bl_idname = "VIEW3D_MT_sv3_dyntopo"
  13
  14     @classmethod
  15     def poll(self, context):
  16         return utils_core.get_mode() == 'SCULPT'
  17
  18     def draw(self, context):
  19         layout = self.layout
  20
  21         if context.object.use_dynamic_topology_sculpting:
  22             layout.row().operator("sculpt.dynamic_topology_toggle",
  23                                      text="Disable Dynamic Topology")
  24
  25             layout.row().separator()
  26
  27             layout.row().menu(DynDetailMenu.bl_idname)
  28             layout.row().menu(DetailMethodMenu.bl_idname)
  29
  30             layout.row().separator()
  31
  32             layout.row().operator("sculpt.optimize")
  33             if context.tool_settings.sculpt.detail_type_method == 'CONSTANT':
  34                 layout.row().operator("sculpt.detail_flood_fill")
  35
  36             layout.row().menu(SymmetrizeMenu.bl_idname)
  37             layout.row().prop(context.tool_settings.sculpt,
  38                                  "use_smooth_shading", toggle=True)
  39
  40         else:
  41             row = layout.row()
  42             row.operator_context = 'INVOKE_DEFAULT'
  43             row.operator("sculpt.dynamic_topology_toggle",
  44                                         text="Enable Dynamic Topology")
  45
  46
  47 class DynDetailMenu(Menu):
  48     bl_label = "Detail Size"
  49     bl_idname = "VIEW3D_MT_sv3_dyn_detail"
  50
  51     def init(self):
  52         settings = (("40", 40),
  53                     ("30", 30),
  54                     ("20", 20),
  55                     ("10", 10),
  56                     ("5", 5),
  57                     ("1", 1))
  58
  59         if bpy.context.tool_settings.sculpt.detail_type_method == 'RELATIVE':
  60             datapath = "tool_settings.sculpt.detail_size"
  61             slider_setting = "detail_size"
  62
  63         elif bpy.context.tool_settings.sculpt.detail_type_method == 'CONSTANT':
  64             datapath = "tool_settings.sculpt.constant_detail_resolution"
  65             slider_setting = "constant_detail_resolution"
  66         else:
  67             datapath = "tool_settings.sculpt.detail_percent"
  68             slider_setting = "detail_percent"
  69             settings = (("100", 100),
  70                         ("75", 75),
  71                         ("50", 50),
  72                         ("25", 25),
  73                         ("10", 10),
  74                         ("5", 5))
  75
  76         return settings, datapath, slider_setting
  77
  78     def draw(self, context):
  79         settings, datapath, slider_setting = self.init()
  80         layout = self.layout
  81
  82         # add the top slider
  83         layout.row().prop(context.tool_settings.sculpt,
  84                              slider_setting, slider=True)
  85         layout.row().separator()
  86
  87         # add the rest of the menu items
  88         for i in range(len(settings)):
  89             utils_core.menuprop(
  90                     layout.row(), settings[i][0], settings[i][1], datapath,
  91                     icon='RADIOBUT_OFF', disable=True,
  92                     disable_icon='RADIOBUT_ON'
  93                     )
  94
  95
  96 class DetailMethodMenu(Menu):
  97     bl_label = "Detail Method"
  98     bl_idname = "VIEW3D_MT_sv3_detail_method_menu"
  99
 100     def draw(self, context):
 101         layout = self.layout
 102         refine_path = "tool_settings.sculpt.detail_refine_method"
 103         type_path = "tool_settings.sculpt.detail_type_method"
 104
 105         refine_items = (("Subdivide Edges", 'SUBDIVIDE'),
 106                         ("Collapse Edges", 'COLLAPSE'),
 107                         ("Subdivide Collapse", 'SUBDIVIDE_COLLAPSE'))
 108
 109         type_items = (("Relative Detail", 'RELATIVE'),
 110                       ("Constant Detail", 'CONSTANT'),
 111                       ("Brush Detail", 'BRUSH'))
 112
 113         layout.row().label(text="Refine")
 114         layout.row().separator()
 115
 116         # add the refine menu items
 117         for item in refine_items:
 118             utils_core.menuprop(
 119                     layout.row(), item[0], item[1],
 120                     refine_path, disable=True,
 121                     icon='RADIOBUT_OFF',
 122                     disable_icon='RADIOBUT_ON'
 123                     )
 124
 125         layout.row().label(text="")
 126
 127         layout.row().label(text="Type")
 128         layout.row().separator()
 129
 130         # add the type menu items
 131         for item in type_items:
 132             utils_core.menuprop(
 133                     layout.row(), item[0], item[1],
 134                     type_path, disable=True,
 135                     icon='RADIOBUT_OFF', disable_icon='RADIOBUT_ON'
 136                     )
 137
 138
 139 class SymmetrizeMenu(Menu):
 140     bl_label = "Symmetrize"
 141     bl_idname = "VIEW3D_MT_sv3_symmetrize_menu"
 142
 143     def draw(self, context):
 144         layout = self.layout
 145         path = "tool_settings.sculpt.symmetrize_direction"
 146
 147         # add the the symmetrize operator to the menu
 148         layout.row().operator("sculpt.symmetrize")
 149         layout.row().separator()
 150
 151         # add the rest of the menu items
 152         for item in context.tool_settings.sculpt. \
 153           bl_rna.properties['symmetrize_direction'].enum_items:
 154             utils_core.menuprop(
 155                     layout.row(), item.name, item.identifier,
 156                     path, disable=True,
 157                     icon='RADIOBUT_OFF', disable_icon='RADIOBUT_ON'
 158                     )
 159
 160
 161 classes = (
 162     DynTopoMenu,
 163     DynDetailMenu,
 164     DetailMethodMenu,
 165     SymmetrizeMenu
 166     )
 167
 168 def register():
 169     for cls in classes:
 170         bpy.utils.register_class(cls)
 171
 172 def unregister():
 173     for cls in classes:
 174         bpy.utils.unregister_class(cls)