materials_utils/menus.py

   1 # SPDX-FileCopyrightText: 2019-2023 Blender Foundation
   2 #
   3 # SPDX-License-Identifier: GPL-2.0-or-later
   4
   5 import bpy
   6
   7 from .functions import *
   8 from .operators import *
   9 from .preferences import *
  10
  11 # -----------------------------------------------------------------------------
  12 # menu classes
  13
  14 class VIEW3D_MT_materialutilities_assign_material(bpy.types.Menu):
  15     """Menu for choosing which material should be assigned to current selection"""
  16     # The menu is filled programmatically with available materials
  17
  18     bl_idname = "VIEW3D_MT_materialutilities_assign_material"
  19     bl_label = "Assign Material"
  20
  21     def draw(self, context):
  22         layout = self.layout
  23         layout.operator_context = 'INVOKE_REGION_WIN'
  24         edit_mode = False
  25
  26         materials = bpy.data.materials.items()
  27
  28         bl_id = VIEW3D_OT_materialutilities_assign_material_object.bl_idname
  29         obj = context.object
  30         mu_prefs = materialutilities_get_preferences(context)
  31
  32         if (not obj is None) and obj.mode == 'EDIT':
  33             bl_id = VIEW3D_OT_materialutilities_assign_material_edit.bl_idname
  34             edit_mode = True
  35
  36         if len(materials) > mu_prefs.search_show_limit:
  37             op = layout.operator(bl_id,
  38                             text = 'Search',
  39                             icon = 'VIEWZOOM')
  40             op.material_name = ""
  41             op.new_material = False
  42             op.show_dialog = True
  43             if not edit_mode:
  44                 op.override_type = mu_prefs.override_type
  45
  46         op = layout.operator(bl_id,
  47                 text = "Add New Material",
  48                 icon = 'ADD')
  49         op.material_name = mu_new_material_name(mu_prefs.new_material_name)
  50         op.new_material = True
  51         op.show_dialog = True
  52         if not edit_mode:
  53             op.override_type = mu_prefs.override_type
  54
  55         layout.separator()
  56
  57         for material_name, material in materials:
  58             material.preview_ensure()
  59             op = layout.operator(bl_id,
  60                     text = material_name,
  61                     icon_value = material.preview.icon_id)
  62             op.material_name = material_name
  63             op.new_material = False
  64             op.show_dialog = False
  65             if not edit_mode:
  66                 op.override_type = mu_prefs.override_type
  67
  68
  69 class VIEW3D_MT_materialutilities_clean_slots(bpy.types.Menu):
  70     """Menu for cleaning up the material slots"""
  71
  72     bl_idname = "VIEW3D_MT_materialutilities_clean_slots"
  73     bl_label = "Clean Slots"
  74
  75     def draw(self, context):
  76         layout = self.layout
  77
  78         layout.label
  79         layout.operator(VIEW3D_OT_materialutilities_clean_material_slots.bl_idname,
  80                         text = "Clean Material Slots",
  81                         icon = 'X')
  82         layout.separator()
  83         layout.operator(VIEW3D_OT_materialutilities_remove_material_slot.bl_idname,
  84                         text = "Remove Active Material Slot",
  85                         icon = 'REMOVE')
  86         layout.operator(VIEW3D_OT_materialutilities_remove_all_material_slots.bl_idname,
  87                         text = "Remove All Material Slots",
  88                         icon = 'CANCEL')
  89
  90
  91 class VIEW3D_MT_materialutilities_select_by_material(bpy.types.Menu):
  92     """Menu for choosing which material should be used for selection"""
  93     # The menu is filled programmatically with available materials
  94
  95     bl_idname = "VIEW3D_MT_materialutilities_select_by_material"
  96     bl_label = "Select by Material"
  97
  98     def draw(self, context):
  99         layout = self.layout
 100
 101         bl_id = VIEW3D_OT_materialutilities_select_by_material_name.bl_idname
 102         obj = context.object
 103         mu_prefs = materialutilities_get_preferences(context)
 104
 105         layout.label
 106
 107         if obj is None or obj.mode == 'OBJECT':
 108             materials = bpy.data.materials.items()
 109
 110             if len(materials) > mu_prefs.search_show_limit:
 111                 layout.operator(bl_id,
 112                                 text = 'Search',
 113                                 icon = 'VIEWZOOM'
 114                                 ).show_dialog = True
 115
 116                 layout.separator()
 117
 118             #show all used materials in entire blend file
 119             for material_name, material in materials:
 120                 # There's no point in showing materials with 0 users
 121                 #  (It will still show materials with fake user though)
 122                 if material.users > 0:
 123                     material.preview_ensure()
 124                     op = layout.operator(bl_id,
 125                                     text = material_name,
 126                                     icon_value = material.preview.icon_id
 127                                     )
 128                     op.material_name = material_name
 129                     op.show_dialog = False
 130
 131         elif obj.mode == 'EDIT':
 132             objects = context.selected_editable_objects
 133             materials_added = []
 134
 135             for obj in objects:
 136                 #show only the materials on this object
 137                 material_slots = obj.material_slots
 138                 for material_slot in material_slots:
 139                     material = material_slot.material
 140
 141                     # Don't add a material that's already in the menu
 142                     if material.name in materials_added:
 143                         continue
 144
 145                     material.preview_ensure()
 146                     op = layout.operator(bl_id,
 147                                     text = material.name,
 148                                     icon_value = material.preview.icon_id
 149                                     )
 150                     op.material_name = material.name
 151                     op.show_dialog = False
 152
 153                     materials_added.append(material.name)
 154
 155 class VIEW3D_MT_materialutilities_specials(bpy.types.Menu):
 156     """Spcials menu for Material Utilities"""
 157
 158     bl_idname = "VIEW3D_MT_materialutilities_specials"
 159     bl_label = "Specials"
 160
 161     def draw(self, context):
 162         mu_prefs = materialutilities_get_preferences(context)
 163         layout = self.layout
 164
 165         #layout.operator(VIEW3D_OT_materialutilities_set_new_material_name.bl_idname, icon = "SETTINGS")
 166
 167         #layout.separator()
 168
 169         layout.operator(MATERIAL_OT_materialutilities_merge_base_names.bl_idname,
 170                         text = "Merge Base Names",
 171                         icon = "GREASEPENCIL")
 172
 173         layout.operator(MATERIAL_OT_materialutilities_join_objects.bl_idname,
 174                         text = "Join by material",
 175                         icon = "OBJECT_DATAMODE")
 176
 177
 178 class VIEW3D_MT_materialutilities_main(bpy.types.Menu):
 179     """Main menu for Material Utilities"""
 180
 181     bl_idname = "VIEW3D_MT_materialutilities_main"
 182     bl_label = "Material Utilities"
 183
 184     def draw(self, context):
 185         obj = context.object
 186         mu_prefs = materialutilities_get_preferences(context)
 187
 188         layout = self.layout
 189         layout.operator_context = 'INVOKE_REGION_WIN'
 190
 191         layout.menu(VIEW3D_MT_materialutilities_assign_material.bl_idname,
 192                      icon = 'ADD')
 193         layout.menu(VIEW3D_MT_materialutilities_select_by_material.bl_idname,
 194                      icon = 'VIEWZOOM')
 195         layout.separator()
 196
 197         layout.operator(VIEW3D_OT_materialutilities_copy_material_to_others.bl_idname,
 198                          text = 'Copy Materials to Selected',
 199                          icon = 'COPY_ID')
 200
 201         layout.separator()
 202
 203         layout.menu(VIEW3D_MT_materialutilities_clean_slots.bl_idname,
 204                     icon = 'NODE_MATERIAL')
 205
 206         layout.separator()
 207         layout.operator(VIEW3D_OT_materialutilities_replace_material.bl_idname,
 208                         text = 'Replace Material',
 209                         icon = 'OVERLAY')
 210
 211         op = layout.operator(VIEW3D_OT_materialutilities_fake_user_set.bl_idname,
 212                        text = 'Set Fake User',
 213                        icon = 'FAKE_USER_OFF')
 214         op.fake_user = mu_prefs.fake_user
 215         op.affect = mu_prefs.fake_user_affect
 216
 217         op = layout.operator(VIEW3D_OT_materialutilities_change_material_link.bl_idname,
 218                        text = 'Change Material Link',
 219                        icon = 'LINKED')
 220         op.link_to = mu_prefs.link_to
 221         op.affect = mu_prefs.link_to_affect
 222         layout.separator()
 223
 224         layout.menu(VIEW3D_MT_materialutilities_specials.bl_idname,
 225                         icon = 'SOLO_ON')
 226
 227
 228
 229 def materialutilities_specials_menu(self, contxt):
 230     self.layout.separator()
 231     self.layout.menu(VIEW3D_MT_materialutilities_main.bl_idname)
 232
 233
 234 def materialutilities_menu_move(self, context):
 235     layout = self.layout
 236     layout.operator_context = 'INVOKE_REGION_WIN'
 237
 238     layout.operator(MATERIAL_OT_materialutilities_material_slot_move.bl_idname,
 239                     icon = 'TRIA_UP_BAR',
 240                     text = 'Move Slot to the Top').movement = 'TOP'
 241     layout.operator(MATERIAL_OT_materialutilities_material_slot_move.bl_idname,
 242                     icon = 'TRIA_DOWN_BAR',
 243                     text = 'Move Slot to the Bottom').movement = 'BOTTOM'
 244     layout.separator()
 245
 246 def materialutilities_menu_functions(self, context):
 247     layout = self.layout
 248     layout.operator_context = 'INVOKE_REGION_WIN'
 249
 250     layout.separator()
 251
 252     layout.menu(VIEW3D_MT_materialutilities_assign_material.bl_idname,
 253                  icon = 'ADD')
 254     layout.menu(VIEW3D_MT_materialutilities_select_by_material.bl_idname,
 255                  icon = 'VIEWZOOM')
 256     layout.separator()
 257
 258     layout.separator()
 259
 260     layout.menu(VIEW3D_MT_materialutilities_clean_slots.bl_idname,
 261                 icon = 'NODE_MATERIAL')
 262
 263     layout.separator()
 264     layout.operator(VIEW3D_OT_materialutilities_replace_material.bl_idname,
 265                     text = 'Replace Material',
 266                     icon = 'OVERLAY')
 267
 268     layout.operator(VIEW3D_OT_materialutilities_fake_user_set.bl_idname,
 269                    text = 'Set Fake User',
 270                    icon = 'FAKE_USER_OFF')
 271
 272     layout.operator(VIEW3D_OT_materialutilities_change_material_link.bl_idname,
 273                    text = 'Change Material Link',
 274                    icon = 'LINKED')
 275     layout.separator()
 276
 277     layout.menu(VIEW3D_MT_materialutilities_specials.bl_idname,
 278                     icon = 'SOLO_ON')