space_view3d_brush_menus/symmetry_menu.py

   1 # gpl author: Ryan Inch (Imaginer)
   2
   3 import bpy
   4 from bpy.types import Menu
   5 from . import utils_core
   6
   7
   8 class MasterSymmetryMenu(Menu):
   9     bl_label = "Symmetry Options"
  10     bl_idname = "VIEW3D_MT_sv3_master_symmetry_menu"
  11
  12     @classmethod
  13     def poll(self, context):
  14         return utils_core.get_mode() in (
  15                         'SCULPT',
  16                         'TEXTURE_PAINT'
  17                         )
  18
  19     def draw(self, context):
  20         layout = self.layout
  21
  22         if utils_core.get_mode() == 'TEXTURE_PAINT':
  23             layout.row().prop(context.tool_settings.image_paint,
  24                               "use_symmetry_x", toggle=True)
  25             layout.row().prop(context.tool_settings.image_paint,
  26                               "use_symmetry_y", toggle=True)
  27             layout.row().prop(context.tool_settings.image_paint,
  28                               "use_symmetry_z", toggle=True)
  29         else:
  30             layout.row().menu(SymmetryMenu.bl_idname)
  31             layout.row().menu(SymmetryRadialMenu.bl_idname)
  32             layout.row().prop(context.tool_settings.sculpt,
  33                               "use_symmetry_feather", toggle=True)
  34
  35
  36 class SymmetryMenu(Menu):
  37     bl_label = "Symmetry"
  38     bl_idname = "VIEW3D_MT_sv3_symmetry_menu"
  39
  40     def draw(self, context):
  41         layout = self.layout
  42
  43         layout.row().label(text="Symmetry")
  44         layout.row().separator()
  45
  46         layout.row().prop(context.tool_settings.sculpt,
  47                           "use_symmetry_x", toggle=True)
  48         layout.row().prop(context.tool_settings.sculpt,
  49                           "use_symmetry_y", toggle=True)
  50         layout.row().prop(context.tool_settings.sculpt,
  51                           "use_symmetry_z", toggle=True)
  52
  53
  54 class SymmetryRadialMenu(Menu):
  55     bl_label = "Radial"
  56     bl_idname = "VIEW3D_MT_sv3_symmetry_radial_menu"
  57
  58     def draw(self, context):
  59         layout = self.layout
  60
  61         layout.row().label(text="Radial")
  62         layout.row().separator()
  63
  64         layout.column().prop(context.tool_settings.sculpt,
  65                              "radial_symmetry", text="", slider=True)
  66
  67
  68 classes = (
  69     MasterSymmetryMenu,
  70     SymmetryMenu,
  71     SymmetryRadialMenu
  72     )
  73
  74 def register():
  75     for cls in classes:
  76         bpy.utils.register_class(cls)
  77
  78 def unregister():
  79     for cls in classes:
  80         bpy.utils.unregister_class(cls)