diff options
Diffstat (limited to 'space_view3d_brush_menus/brushes.py')
| -rw-r--r-- | space_view3d_brush_menus/brushes.py | 197 |
1 files changed, 105 insertions, 92 deletions
diff --git a/space_view3d_brush_menus/brushes.py b/space_view3d_brush_menus/brushes.py index 280b0367..9073eb48 100644 --- a/space_view3d_brush_menus/brushes.py +++ b/space_view3d_brush_menus/brushes.py @@ -5,149 +5,162 @@ from bpy.types import Menu from . import utils_core from bl_ui.properties_paint_common import UnifiedPaintPanel +# Particle Tools + +particle_tools = ( + ("None", 'NONE'), + ("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_SUBTRACT', + "SMOOTH": 'BRUSH_SMOOTH', + "SNAKE_HOOK": 'BRUSH_SNAKE_HOOK', + "THUMB": 'BRUSH_THUMB' + }, + + 'VERTEX_PAINT': { + "ADD": 'BRUSH_ADD', + "BLUR": 'BRUSH_BLUR', + "DARKEN": 'BRUSH_DARKEN', + "LIGHTEN": 'BRUSH_LIGHTEN', + "MIX": 'BRUSH_MIX', + "MUL": 'BRUSH_MULTIPLY', + "SUB": 'BRUSH_SUBTRACT' + }, + + 'WEIGHT_PAINT': { + "ADD": 'BRUSH_ADD', + "BLUR": 'BRUSH_BLUR', + "DARKEN": 'BRUSH_DARKEN', + "LIGHTEN": 'BRUSH_LIGHTEN', + "MIX": 'BRUSH_MIX', + "MUL": 'BRUSH_MULTIPLY', + "SUB": 'BRUSH_SUBTRACT' + }, + + '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 init(self): - if utils_core.get_mode() == utils_core.sculpt: - datapath = "tool_settings.sculpt.brush" - icon = {"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_SUBTRACT', - "SMOOTH": 'BRUSH_SMOOTH', - "SNAKE_HOOK": 'BRUSH_SNAKE_HOOK', - "THUMB": 'BRUSH_THUMB'} - elif utils_core.get_mode() == utils_core.vertex_paint: - datapath = "tool_settings.vertex_paint.brush" - icon = {"ADD": 'BRUSH_ADD', - "BLUR": 'BRUSH_BLUR', - "DARKEN": 'BRUSH_DARKEN', - "LIGHTEN": 'BRUSH_LIGHTEN', - "MIX": 'BRUSH_MIX', - "MUL": 'BRUSH_MULTIPLY', - "SUB": 'BRUSH_SUBTRACT'} - elif utils_core.get_mode() == utils_core.weight_paint: - datapath = "tool_settings.weight_paint.brush" - icon = {"ADD": 'BRUSH_ADD', - "BLUR": 'BRUSH_BLUR', - "DARKEN": 'BRUSH_DARKEN', - "LIGHTEN": 'BRUSH_LIGHTEN', - "MIX": 'BRUSH_MIX', - "MUL": 'BRUSH_MULTIPLY', - "SUB": 'BRUSH_SUBTRACT'} - elif utils_core.get_mode() == utils_core.texture_paint: - datapath = "tool_settings.image_paint.brush" - icon = {"CLONE": 'BRUSH_CLONE', - "DRAW": 'BRUSH_TEXDRAW', - "FILL": 'BRUSH_TEXFILL', - "MASK": 'BRUSH_TEXMASK', - "SMEAR": 'BRUSH_SMEAR', - "SOFTEN": 'BRUSH_SOFTEN'} - elif utils_core.get_mode() == utils_core.particle_edit: - datapath = "tool_settings.particle_edit.tool" - icon = None - else: - datapath = "" - - return datapath, icon - def draw(self, context): - datapath, icon = self.init() - menu = utils_core.Menu(self) + mode = utils_core.get_mode() + layout = self.layout settings = UnifiedPaintPanel.paint_settings(context) colum_n = utils_core.addon_settings(lists=False) - menu.add_item().label(text="Brush") - menu.add_item().separator() + layout.row().label(text="Brush") + layout.row().separator() has_brush = utils_core.get_brush_link(context, types="brush") - current_brush = eval("bpy.context.{}".format(datapath)) if has_brush else None + 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() != utils_core.particle_edit: + if current_brush and utils_core.get_mode() != 'PARTICLE_EDIT': current_brush = current_brush.name - if utils_core.get_mode() == utils_core.particle_edit: - particle_tools = [["None", 'NONE'], - ["Comb", 'COMB'], - ["Smooth", 'SMOOTH'], - ["Add", 'ADD'], - ["Length", 'LENGTH'], - ["Puff", 'PUFF'], - ["Cut", 'CUT'], - ["Weight", 'WEIGHT']] - + 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( - menu.add_item(), tool[0], tool[1], datapath, + layout.row(), tool[0], tool[1], brush_datapath[mode], icon='RADIOBUT_OFF', disable=True, disable_icon='RADIOBUT_ON' ) else: - column_flow = menu.add_item("column_flow", columns=colum_n) + column_flow = layout.column_flow(columns=colum_n) if utils_core.addon_settings(lists=True) == 'template': - self.layout.template_ID_preview(settings, "brush", - new="brush.add", rows=3, cols=colum_n) + layout.template_ID_preview(settings, "brush", + new="brush.add", rows=3, cols=colum_n) else: # iterate over all the brushes for item in bpy.data.brushes: - if utils_core.get_mode() == utils_core.sculpt: + 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( - menu.add_item(parent=column_flow), item.name, + column_flow.row(), item.name, 'bpy.data.brushes["%s"]' % item.name, - datapath, icon=icon[item.sculpt_tool], - disable=True, custom_disable_exp=[item.name, current_brush], + brush_datapath[mode], icon=brush_icon[mode][item.sculpt_tool], + disable=True, custom_disable_exp=(item.name, current_brush), path=True ) - if utils_core.get_mode() == utils_core.vertex_paint: + 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( - menu.add_item(parent=column_flow), item.name, + column_flow.row(), item.name, 'bpy.data.brushes["%s"]' % item.name, - datapath, icon=icon[item.vertex_tool], - disable=True, custom_disable_exp=[item.name, current_brush], + 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() == utils_core.weight_paint: + 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( - menu.add_item(parent=column_flow), item.name, + column_flow.row(), item.name, 'bpy.data.brushes["%s"]' % item.name, - datapath, icon=icon[item.vertex_tool], - disable=True, custom_disable_exp=[item.name, current_brush], + 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() == utils_core.texture_paint: + 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( - menu.add_item(parent=column_flow), item.name, + column_flow.row(), item.name, 'bpy.data.brushes["%s"]' % item.name, - datapath, icon=icon[item.image_tool], - disable=True, custom_disable_exp=[item.name, current_brush], + brush_datapath[mode], icon=brush_icon[mode][item.image_tool], + disable=True, custom_disable_exp=(item.name, current_brush), path=True ) |