diff options
Diffstat (limited to 'space_view3d_brush_menus/brushes.py')
| -rw-r--r-- | space_view3d_brush_menus/brushes.py | 163 |
1 files changed, 163 insertions, 0 deletions
diff --git a/space_view3d_brush_menus/brushes.py b/space_view3d_brush_menus/brushes.py new file mode 100644 index 00000000..3478d575 --- /dev/null +++ b/space_view3d_brush_menus/brushes.py @@ -0,0 +1,163 @@ +# gpl author: Ryan Inch (Imaginer) + +import bpy +from bpy.types import Menu +from . import utils_core +from bl_ui.properties_paint_common import UnifiedPaintPanel + +# Particle Tools + +particle_tools = ( + ("Comb", 'COMB'), + ("Smooth", 'SMOOTH'), + ("Add", 'ADD'), + ("Length", 'LENGTH'), + ("Puff", 'PUFF'), + ("Cut", 'CUT'), + ("Weight", 'WEIGHT') +) + +# Brush Datapaths + +brush_datapath = { + 'SCULPT': "tool_settings.sculpt.brush", + 'VERTEX_PAINT': "tool_settings.vertex_paint.brush", + 'WEIGHT_PAINT': "tool_settings.weight_paint.brush", + 'TEXTURE_PAINT': "tool_settings.image_paint.brush", + 'PARTICLE_EDIT': "tool_settings.particle_edit.tool" +} + +# Brush Icons + +brush_icon = { + 'SCULPT': { + "BLOB": 'BRUSH_BLOB', + "CLAY": 'BRUSH_CLAY', + "CLAY_STRIPS": 'BRUSH_CLAY_STRIPS', + "CREASE": 'BRUSH_CREASE', + "DRAW": 'BRUSH_SCULPT_DRAW', + "FILL": 'BRUSH_FILL', + "FLATTEN": 'BRUSH_FLATTEN', + "GRAB": 'BRUSH_GRAB', + "INFLATE": 'BRUSH_INFLATE', + "LAYER": 'BRUSH_LAYER', + "MASK": 'BRUSH_MASK', + "NUDGE": 'BRUSH_NUDGE', + "PINCH": 'BRUSH_PINCH', + "ROTATE": 'BRUSH_ROTATE', + "SCRAPE": 'BRUSH_SCRAPE', + "SIMPLIFY": 'BRUSH_DATA', + "SMOOTH": 'BRUSH_SMOOTH', + "SNAKE_HOOK": 'BRUSH_SNAKE_HOOK', + "THUMB": 'BRUSH_THUMB' + }, + + 'VERTEX_PAINT': { + "AVERAGE": 'BRUSH_BLUR', + "BLUR": 'BRUSH_BLUR', + "DRAW": 'BRUSH_MIX', + "SMEAR": 'BRUSH_BLUR' + }, + + 'WEIGHT_PAINT': { + "AVERAGE": 'BRUSH_BLUR', + "BLUR": 'BRUSH_BLUR', + "DRAW": 'BRUSH_MIX', + "SMEAR": 'BRUSH_BLUR' + }, + + 'TEXTURE_PAINT': { + "CLONE": 'BRUSH_CLONE', + "DRAW": 'BRUSH_TEXDRAW', + "FILL": 'BRUSH_TEXFILL', + "MASK": 'BRUSH_TEXMASK', + "SMEAR": 'BRUSH_SMEAR', + "SOFTEN": 'BRUSH_SOFTEN' + } +} + + +class BrushesMenu(Menu): + bl_label = "Brush" + bl_idname = "VIEW3D_MT_sv3_brushes_menu" + + def draw(self, context): + mode = utils_core.get_mode() + layout = self.layout + settings = UnifiedPaintPanel.paint_settings(context) + colum_n = utils_core.addon_settings() + + layout.row().label(text="Brush") + layout.row().separator() + + has_brush = utils_core.get_brush_link(context, types="brush") + current_brush = eval("bpy.context.{}".format(brush_datapath[mode])) if has_brush else None + + # get the current brush's name + if current_brush and utils_core.get_mode() != 'PARTICLE_EDIT': + current_brush = current_brush.name + + if mode == 'PARTICLE_EDIT': + # if you are in particle edit mode add the menu items for particle mode + for tool in particle_tools: + utils_core.menuprop( + layout.row(), tool[0], tool[1], brush_datapath[mode], + icon='RADIOBUT_OFF', disable=True, + disable_icon='RADIOBUT_ON' + ) + else: + column_flow = layout.column_flow(columns=colum_n) + + # iterate over all the brushes + for item in bpy.data.brushes: + if mode == 'SCULPT': + if item.use_paint_sculpt: + # if you are in sculpt mode and the brush + # is a sculpt brush add the brush to the menu + utils_core.menuprop( + column_flow.row(), item.name, + 'bpy.data.brushes["%s"]' % item.name, + brush_datapath[mode], icon=brush_icon[mode][item.sculpt_tool], + disable=True, custom_disable_exp=(item.name, current_brush), + path=True + ) + if mode == 'VERTEX_PAINT': + if item.use_paint_vertex: + # if you are in vertex paint mode and the brush + # is a vertex paint brush add the brush to the menu + utils_core.menuprop( + column_flow.row(), item.name, + 'bpy.data.brushes["%s"]' % item.name, + brush_datapath[mode], icon=brush_icon[mode][item.vertex_tool], + disable=True, custom_disable_exp=(item.name, current_brush), + path=True + ) + if mode == 'WEIGHT_PAINT': + if item.use_paint_weight: + # if you are in weight paint mode and the brush + # is a weight paint brush add the brush to the menu + utils_core.menuprop( + column_flow.row(), item.name, + 'bpy.data.brushes["%s"]' % item.name, + brush_datapath[mode], icon=brush_icon[mode][item.vertex_tool], + disable=True, custom_disable_exp=(item.name, current_brush), + path=True + ) + if utils_core.get_mode() == 'TEXTURE_PAINT': + if item.use_paint_image: + # if you are in texture paint mode and the brush + # is a texture paint brush add the brush to the menu + utils_core.menuprop( + column_flow.row(), item.name, + 'bpy.data.brushes["%s"]' % item.name, + brush_datapath[mode], icon=brush_icon[mode][item.image_tool], + disable=True, custom_disable_exp=(item.name, current_brush), + path=True + ) + + +def register(): + bpy.utils.register_class(BrushesMenu) + +def unregister(): + bpy.utils.unregister_class(BrushesMenu) |