space_view3d_brush_menus/dyntopo_menu.py

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