Skip to content
Snippets Groups Projects
Commit 92cda885 authored by Campbell Barton's avatar Campbell Barton
Browse files

- make pep8 compliant 

- fix error when a mesh had a None material
- no need to set the object as avtive when assigning materials
parent 537ad49d
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
space_view3d_materials_utils.py
+ 218
− 211
View file @ 92cda885
...@@ -29,36 +29,44 @@ bl_info = { ...@@ -29,36 +29,44 @@ bl_info = {
"blender": (2, 5, 6), "blender": (2, 5, 6),
"api": 35324, "api": 35324,
"location": "View3D > Q key", "location": "View3D > Q key",
"description": "Menu of material tools (assign, select by etc) in the 3D View", "description": "Menu of material tools (assign, select..) in the 3D View",
"warning": "", "warning": "",
"wiki_url": "http://wiki.blender.org/index.php/Extensions:2.5/Py/"\ "wiki_url": "http://wiki.blender.org/index.php/Extensions:2.5/Py/"
"Scripts/3D interaction/Materials Utils", "Scripts/3D interaction/Materials Utils",
"tracker_url": "https://projects.blender.org/tracker/index.php?"\ "tracker_url": "https://projects.blender.org/tracker/index.php?"
"func=detail&aid=22140", "func=detail&aid=22140",
"category": "3D View"} "category": "3D View"}
""" """
This script has several functions and operators... grouped for convenience This script has several functions and operators... grouped for convenience
* assign material: * assign material:
offers the user a list of ALL the materials in the blend file and an additional "new" entry offers the user a list of ALL the materials in the blend file and an
the chosen material will be assigned to all the selected objects in object mode. additional "new" entry the chosen material will be assigned to all the
selected objects in object mode.
in edit mode the selected faces get the selected material applied. in edit mode the selected faces get the selected material applied.
if the user chose "new" the new material can be renamed using the "last operator" section of the toolbox if the user chose "new" the new material can be renamed using the
After assigning the material "clean material slots" and "material to texface" are auto run to keep things tidy (see description bellow) "last operator" section of the toolbox.
After assigning the material "clean material slots" and
"material to texface" are auto run to keep things tidy
(see description bellow)
* select by material * select by material
in object mode this offers the user a menu of all materials in the blend file in object mode this offers the user a menu of all materials in the blend
any objects using the selected material will become selected, any objects without the material will be removed from selection. file any objects using the selected material will become selected, any
objects without the material will be removed from selection.
in edit mode: the menu offers only the materials attached to the current object. It will select the faces that use the material and deselect those that do not.
in edit mode: the menu offers only the materials attached to the current
object. It will select the faces that use the material and deselect those
that do not.
* clean material slots * clean material slots
for all selected objects any empty material slots or material slots with materials that are not used by the mesh faces will be removed. for all selected objects any empty material slots or material slots with
materials that are not used by the mesh faces will be removed.
* Any un-used materials and slots will be removed * Any un-used materials and slots will be removed
""" """
...@@ -66,92 +74,92 @@ import bpy ...@@ -66,92 +74,92 @@ import bpy
from bpy.props import* from bpy.props import*
def replace_material(m1 , m2, all_objects = False): def replace_material(m1, m2, all_objects=False):
#replace material named m1 with material named m2 # replace material named m1 with material named m2
#m1 is the name of original material # m1 is the name of original material
#m2 is the name of the material to replace it with # m2 is the name of the material to replace it with
#'all' will replace throughout the blend file # 'all' will replace throughout the blend file
try:
matorg = bpy.data.materials[m1] matorg = bpy.data.materials.get(m1)
matrep = bpy.data.materials[m2] matrep = bpy.data.materials.get(m2)
if matorg and matrep:
#store active object #store active object
scn = bpy.context.scene scn = bpy.context.scene
ob_active = bpy.context.active_object ob_active = bpy.context.active_object
if all_objects: if all_objects:
objs = bpy.data.objects objs = bpy.data.objects
else: else:
objs = bpy.context.selected_editable_objects objs = bpy.context.selected_editable_objects
for ob in objs: for ob in objs:
if ob.type == 'MESH': if ob.type == 'MESH':
scn.objects.active = ob scn.objects.active = ob
print(ob.name)
ms = ob.material_slots.values() for m in ob.material_slots.values():
for m in ms:
if m.material == matorg: if m.material == matorg:
m.material = matrep m.material = matrep
#don't break the loop as the material can be # don't break the loop as the material can be
# ref'd more than once # ref'd more than once
#restore active object else:
scn.objects.active = ob_active
except:
print('no match to replace') print('no match to replace')
def select_material_by_name(find_mat):
#in object mode selects all objects with material find_mat def select_material_by_name(find_mat_name):
#in edit mode selects all faces with material find_mat #in object mode selects all objects with material find_mat_name
#in edit mode selects all faces with material find_mat_name
find_mat = bpy.data.materials.get(find_mat_name)
if find_mat is None:
return
#check for editmode #check for editmode
editmode = False editmode = False
scn = bpy.context.scene scn = bpy.context.scene
#set selection mode to faces #set selection mode to faces
scn.tool_settings.mesh_select_mode =[False,False,True] scn.tool_settings.mesh_select_mode = False, False, True
actob = bpy.context.active_object actob = bpy.context.active_object
if actob.mode == 'EDIT': if actob.mode == 'EDIT':
editmode =True editmode = True
bpy.ops.object.mode_set() bpy.ops.object.mode_set()
if not editmode: if not editmode:
objs = bpy.data.objects objs = bpy.data.objects
for ob in objs: for ob in objs:
typ = ['MESH','CURVE', 'SURFACE', 'FONT', 'META'] if ob.type in {'MESH', 'CURVE', 'SURFACE', 'FONT', 'META'}:
if ob.type in typ:
ms = ob.material_slots.values() ms = ob.material_slots.values()
for m in ms: for m in ms:
if m.material.name == find_mat: if m.material == find_mat:
ob.select = True ob.select = True
#the active object may not have the mat! # the active object may not have the mat!
#set it to one that does! # set it to one that does!
scn.objects.active = ob scn.objects.active = ob
break break
else: else:
ob.select = False ob.select = False
#deselect non-meshes #deselect non-meshes
else: else:
ob.select = False ob.select = False
else: else:
#it's editmode, so select the faces #it's editmode, so select the faces
ob = actob ob = actob
ms = ob.material_slots.values() ms = ob.material_slots.values()
#same material can be on multiple slots #same material can be on multiple slots
slot_indeces =[] slot_indeces = []
i = 0 i = 0
# found = False # UNUSED # found = False # UNUSED
for m in ms: for m in ms:
if m.material.name == find_mat: if m.material == find_mat:
slot_indeces.append(i) slot_indeces.append(i)
# found = True # UNUSED # found = True # UNUSED
i += 1 i += 1
...@@ -161,29 +169,30 @@ def select_material_by_name(find_mat): ...@@ -161,29 +169,30 @@ def select_material_by_name(find_mat):
f.select = True f.select = True
else: else:
f.select = False f.select = False
me.update() me.update()
if editmode: if editmode:
bpy.ops.object.mode_set(mode = 'EDIT') bpy.ops.object.mode_set(mode='EDIT')
def mat_to_texface(): def mat_to_texface():
#assigns the first image in each material to the faces in the active uvlayer # assigns the first image in each material to the faces in the active
#for all selected objects # uvlayer for all selected objects
#check for editmode #check for editmode
editmode = False editmode = False
actob = bpy.context.active_object actob = bpy.context.active_object
if actob.mode == 'EDIT': if actob.mode == 'EDIT':
editmode =True editmode = True
bpy.ops.object.mode_set() bpy.ops.object.mode_set()
for ob in bpy.context.selected_editable_objects: for ob in bpy.context.selected_editable_objects:
if ob.type == 'MESH': if ob.type == 'MESH':
#get the materials from slots #get the materials from slots
ms = ob.material_slots.values() ms = ob.material_slots.values()
#build a list of images, one per material #build a list of images, one per material
images=[] images = []
#get the textures from the mats #get the textures from the mats
for m in ms: for m in ms:
gotimage = False gotimage = False
...@@ -195,17 +204,16 @@ def mat_to_texface(): ...@@ -195,17 +204,16 @@ def mat_to_texface():
if tex.type == 'IMAGE': if tex.type == 'IMAGE':
img = tex.image img = tex.image
images.append(img) images.append(img)
gotimage =True gotimage = True
break break
if not gotimage: if not gotimage:
print('noimage on', m.name) print('noimage on', m.name)
images.append(None) images.append(None)
#now we have the images # now we have the images
#applythem to the uvlayer # applythem to the uvlayer
me = ob.data me = ob.data
#got uvs? #got uvs?
if not me.uv_textures: if not me.uv_textures:
...@@ -213,7 +221,7 @@ def mat_to_texface(): ...@@ -213,7 +221,7 @@ def mat_to_texface():
scn.objects.active = ob scn.objects.active = ob
bpy.ops.mesh.uv_texture_add() bpy.ops.mesh.uv_texture_add()
scn.objects.active = actob scn.objects.active = actob
#get active uvlayer #get active uvlayer
for t in me.uv_textures: for t in me.uv_textures:
if t.active: if t.active:
...@@ -224,14 +232,12 @@ def mat_to_texface(): ...@@ -224,14 +232,12 @@ def mat_to_texface():
uvtex[f.index].image = images[f.material_index] uvtex[f.index].image = images[f.material_index]
else: else:
uvtex[f.index].image = None uvtex[f.index].image = None
me.update() me.update()
if editmode: if editmode:
bpy.ops.object.mode_set(mode = 'EDIT') bpy.ops.object.mode_set(mode='EDIT')
def assignmatslots(ob, matlist): def assignmatslots(ob, matlist):
#given an object and a list of material names #given an object and a list of material names
...@@ -246,15 +252,14 @@ def assignmatslots(ob, matlist): ...@@ -246,15 +252,14 @@ def assignmatslots(ob, matlist):
for s in ob.material_slots: for s in ob.material_slots:
bpy.ops.object.material_slot_remove() bpy.ops.object.material_slot_remove()
# re-add them and assign material
#re-add them and assign material
i = 0 i = 0
for m in matlist: for m in matlist:
mat = bpy.data.materials[m] mat = bpy.data.materials[m]
ob.data.materials.append(mat) ob.data.materials.append(mat)
i += 1 i += 1
#restore active object: # restore active object:
scn.objects.active = ob_active scn.objects.active = ob_active
...@@ -263,52 +268,49 @@ def cleanmatslots(): ...@@ -263,52 +268,49 @@ def cleanmatslots():
editmode = False editmode = False
actob = bpy.context.active_object actob = bpy.context.active_object
if actob.mode == 'EDIT': if actob.mode == 'EDIT':
editmode =True editmode = True
bpy.ops.object.mode_set() bpy.ops.object.mode_set()
objs = bpy.context.selected_editable_objects objs = bpy.context.selected_editable_objects
for ob in objs: for ob in objs:
if ob.type == 'MESH': if ob.type == 'MESH':
mats = ob.material_slots.keys() mats = ob.material_slots.keys()
#check the faces on the mesh to build a list of used materials #check the faces on the mesh to build a list of used materials
usedMatIndex =[] #we'll store used materials indices here usedMatIndex = [] # we'll store used materials indices here
faceMats =[] faceMats = []
me = ob.data me = ob.data
for f in me.faces: for f in me.faces:
#get the material index for this face... #get the material index for this face...
faceindex = f.material_index faceindex = f.material_index
#indices will be lost: Store face mat use by name #indices will be lost: Store face mat use by name
currentfacemat = mats[faceindex] currentfacemat = mats[faceindex]
faceMats.append(currentfacemat) faceMats.append(currentfacemat)
# check if index is already listed as used or not
#check if index is already listed as used or not
found = 0 found = 0
for m in usedMatIndex: for m in usedMatIndex:
if m == faceindex: if m == faceindex:
found = 1 found = 1
#break #break
if found == 0: if found == 0:
#add this index to the list #add this index to the list
usedMatIndex.append(faceindex) usedMatIndex.append(faceindex)
#re-assign the used mats to the mesh and leave out the unused #re-assign the used mats to the mesh and leave out the unused
ml = [] ml = []
mnames = [] mnames = []
for u in usedMatIndex: for u in usedMatIndex:
ml.append( mats[u] ) ml.append(mats[u])
#we'll need a list of names to get the face indices... #we'll need a list of names to get the face indices...
mnames.append(mats[u]) mnames.append(mats[u])
assignmatslots(ob, ml) assignmatslots(ob, ml)
# restore face indices:
#restore face indices:
i = 0 i = 0
for f in me.faces: for f in me.faces:
matindex = mnames.index(faceMats[i]) matindex = mnames.index(faceMats[i])
...@@ -316,81 +318,73 @@ def cleanmatslots(): ...@@ -316,81 +318,73 @@ def cleanmatslots():
i += 1 i += 1
if editmode: if editmode:
bpy.ops.object.mode_set(mode = 'EDIT') bpy.ops.object.mode_set(mode='EDIT')
def assign_mat(matname="Default"): def assign_mat(matname="Default"):
#get active object so we can restore it later # get active object so we can restore it later
actob = bpy.context.active_object actob = bpy.context.active_object
#check if material exists, if it doesn't then create it # check if material exists, if it doesn't then create it
mats =bpy.data.materials
found = False found = False
for m in mats: for m in bpy.data.materials:
if m.name == matname: if m.name == matname:
target = m target = m
found = True found = True
break break
if not found: if not found:
target = bpy.data.materials.new(matname) target = bpy.data.materials.new(matname)
# if objectmode then set all faces
#if objectmode then set all faces
editmode = False editmode = False
allfaces = True allfaces = True
if actob.mode == 'EDIT': if actob.mode == 'EDIT':
editmode =True editmode = True
allfaces = False allfaces = False
bpy.ops.object.mode_set() bpy.ops.object.mode_set()
objs = bpy.context.selected_editable_objects objs = bpy.context.selected_editable_objects
for ob in objs: for ob in objs:
#set the active object to our object # set the active object to our object
scn = bpy.context.scene scn = bpy.context.scene
scn.objects.active = ob scn.objects.active = ob
if ob.type in {'CURVE', 'SURFACE', 'FONT', 'META'}:
other = ['CURVE', 'SURFACE', 'FONT', 'META'] found = False
if ob.type in other:
found=False
i = 0 i = 0
mats = bpy.data.materials for m in bpy.data.materials:
for m in mats:
if m.name == matname: if m.name == matname:
found =True found = True
index = i index = i
break break
i += 1 i += 1
if not found: if not found:
index = i-1 index = i - 1
targetlist =[index] targetlist = [index]
assignmatslots(ob, targetlist) assignmatslots(ob, targetlist)
elif ob.type =='MESH': elif ob.type == 'MESH':
#check material slots for matname material # check material slots for matname material
found=False found = False
i = 0 i = 0
mats = ob.material_slots mats = ob.material_slots
for m in mats: for m in mats:
if m.name == matname: if m.name == matname:
found =True found = True
index = i index = i
#make slot active #make slot active
ob.active_material_index = i ob.active_material_index = i
break break
i += 1 i += 1
if not found: if not found:
index=i index = i
#the material is not attached to the object #the material is not attached to the object
ob.data.materials.append(target) ob.data.materials.append(target)
#now assign the material: #now assign the material:
me =ob.data me = ob.data
if allfaces: if allfaces:
for f in me.faces: for f in me.faces:
f.material_index = index f.material_index = index
...@@ -399,17 +393,14 @@ def assign_mat(matname="Default"): ...@@ -399,17 +393,14 @@ def assign_mat(matname="Default"):
if f.select: if f.select:
f.material_index = index f.material_index = index
me.update() me.update()
#restore the active object #restore the active object
bpy.context.scene.objects.active = actob bpy.context.scene.objects.active = actob
if editmode: if editmode:
bpy.ops.object.mode_set(mode = 'EDIT') bpy.ops.object.mode_set(mode='EDIT')
def check_texture(img, mat):
def check_texture(img,mat):
#finds a texture from an image #finds a texture from an image
#makes a texture if needed #makes a texture if needed
#adds it to the material if it isn't there already #adds it to the material if it isn't there already
...@@ -434,11 +425,12 @@ def check_texture(img,mat): ...@@ -434,11 +425,12 @@ def check_texture(img,mat):
mtex.texture_coords = 'UV' mtex.texture_coords = 'UV'
mtex.use_map_color_diffuse = True mtex.use_map_color_diffuse = True
def texface_to_mat(): def texface_to_mat():
# editmode check here! # editmode check here!
editmode = False editmode = False
ob = bpy.context.object ob = bpy.context.object
if ob.mode =='EDIT': if ob.mode == 'EDIT':
editmode = True editmode = True
bpy.ops.object.mode_set() bpy.ops.object.mode_set()
...@@ -446,11 +438,11 @@ def texface_to_mat(): ...@@ -446,11 +438,11 @@ def texface_to_mat():
faceindex = [] faceindex = []
unique_images = [] unique_images = []
# get the texface images and store indices # get the texface images and store indices
if (ob.data.uv_textures): if (ob.data.uv_textures):
for f in ob.data.uv_textures.active.data: for f in ob.data.uv_textures.active.data:
if f.image: if f.image:
img = f.image img = f.image
#build list of unique images #build list of unique images
if img not in unique_images: if img not in unique_images:
...@@ -459,30 +451,26 @@ def texface_to_mat(): ...@@ -459,30 +451,26 @@ def texface_to_mat():
else: else:
img = None img = None
faceindex.append(None) faceindex.append(None)
#check materials for images exist; create if needed # check materials for images exist; create if needed
matlist = [] matlist = []
for i in unique_images: for i in unique_images:
if i: if i:
print(i.name)
try: try:
m = bpy.data.materials[i.name] m = bpy.data.materials[i.name]
except: except:
m = bpy.data.materials.new(name = i.name) m = bpy.data.materials.new(name=i.name)
continue continue
finally: finally:
matlist.append(m.name) matlist.append(m.name)
# add textures if needed # add textures if needed
check_texture(i,m) check_texture(i, m)
#set up the object material slots # set up the object material slots
assignmatslots(ob, matlist) assignmatslots(ob, matlist)
#set texface indices to material slot indices.. #set texface indices to material slot indices..
me = ob.data me = ob.data
...@@ -492,11 +480,11 @@ def texface_to_mat(): ...@@ -492,11 +480,11 @@ def texface_to_mat():
me.faces[i].material_index = f me.faces[i].material_index = f
i += 1 i += 1
if editmode: if editmode:
bpy.ops.object.mode_set(mode = 'EDIT') bpy.ops.object.mode_set(mode='EDIT')
#operator classes: # -----------------------------------------------------------------------------
#--------------------------------------------------------------------- # operator classes:
class VIEW3D_OT_texface_to_material(bpy.types.Operator): class VIEW3D_OT_texface_to_material(bpy.types.Operator):
'''''' ''''''
...@@ -513,19 +501,24 @@ class VIEW3D_OT_texface_to_material(bpy.types.Operator): ...@@ -513,19 +501,24 @@ class VIEW3D_OT_texface_to_material(bpy.types.Operator):
texface_to_mat() texface_to_mat()
return {'FINISHED'} return {'FINISHED'}
else: else:
self.report({'WARNING'}, "No editable selected objects, could not finish") self.report({'WARNING'},
"No editable selected objects, could not finish")
return {'CANCELLED'} return {'CANCELLED'}
class VIEW3D_OT_assign_material(bpy.types.Operator): class VIEW3D_OT_assign_material(bpy.types.Operator):
'''assign a material to the selection''' '''assign a material to the selection'''
bl_idname = "view3d.assign_material" bl_idname = "view3d.assign_material"
bl_label = "MW Assign Material" bl_label = "MW Assign Material"
bl_options = {'REGISTER', 'UNDO'} bl_options = {'REGISTER', 'UNDO'}
matname = StringProperty(name = 'Material Name', matname = StringProperty(
description = 'Name of Material to Assign', name='Material Name',
default = "", maxlen = 21) description='Name of Material to Assign',
default="",
maxlen=21,
)
@classmethod @classmethod
def poll(cls, context): def poll(cls, context):
return context.active_object != None return context.active_object != None
...@@ -538,8 +531,9 @@ class VIEW3D_OT_assign_material(bpy.types.Operator): ...@@ -538,8 +531,9 @@ class VIEW3D_OT_assign_material(bpy.types.Operator):
mat_to_texface() mat_to_texface()
return {'FINISHED'} return {'FINISHED'}
class VIEW3D_OT_clean_material_slots(bpy.types.Operator): class VIEW3D_OT_clean_material_slots(bpy.types.Operator):
'''removes any material slots from the '''removes any material slots from the
selected objects that are not used by the mesh''' selected objects that are not used by the mesh'''
bl_idname = "view3d.clean_material_slots" bl_idname = "view3d.clean_material_slots"
bl_label = "MW Clean Material Slots" bl_label = "MW Clean Material Slots"
...@@ -553,6 +547,7 @@ class VIEW3D_OT_clean_material_slots(bpy.types.Operator): ...@@ -553,6 +547,7 @@ class VIEW3D_OT_clean_material_slots(bpy.types.Operator):
cleanmatslots() cleanmatslots()
return {'FINISHED'} return {'FINISHED'}
class VIEW3D_OT_material_to_texface(bpy.types.Operator): class VIEW3D_OT_material_to_texface(bpy.types.Operator):
'''''' ''''''
bl_idname = "view3d.material_to_texface" bl_idname = "view3d.material_to_texface"
...@@ -567,14 +562,17 @@ class VIEW3D_OT_material_to_texface(bpy.types.Operator): ...@@ -567,14 +562,17 @@ class VIEW3D_OT_material_to_texface(bpy.types.Operator):
mat_to_texface() mat_to_texface()
return {'FINISHED'} return {'FINISHED'}
class VIEW3D_OT_select_material_by_name(bpy.types.Operator): class VIEW3D_OT_select_material_by_name(bpy.types.Operator):
'''''' ''''''
bl_idname = "view3d.select_material_by_name" bl_idname = "view3d.select_material_by_name"
bl_label = "MW Select Material By Name" bl_label = "MW Select Material By Name"
bl_options = {'REGISTER', 'UNDO'} bl_options = {'REGISTER', 'UNDO'}
matname = StringProperty(name = 'Material Name', matname = StringProperty(
description = 'Name of Material to Select', name='Material Name',
default = "", maxlen = 21) description='Name of Material to Select',
maxlen=21,
)
@classmethod @classmethod
def poll(cls, context): def poll(cls, context):
...@@ -592,18 +590,21 @@ class VIEW3D_OT_replace_material(bpy.types.Operator): ...@@ -592,18 +590,21 @@ class VIEW3D_OT_replace_material(bpy.types.Operator):
bl_label = "MW Replace Material" bl_label = "MW Replace Material"
bl_options = {'REGISTER', 'UNDO'} bl_options = {'REGISTER', 'UNDO'}
matorg = StringProperty(name = 'Material to Replace', matorg = StringProperty(
description = 'Name of Material to Assign', name='Material to Replace',
default = "", maxlen = 21) description="Name of Material to Assign",
maxlen=21,
matrep = StringProperty(name = 'Replacement material', )
description = 'Name of Material to Assign', matrep = StringProperty(name="Replacement material",
default = "", maxlen = 21) description='Name of Material to Assign',
maxlen=21,
)
all_objects = BoolProperty(
name="all_objects",
description="replace for all objects in this blend file",
default=True,
)
all_objects = BoolProperty(name ='all_objects',
description="replace for all objects in this blend file",
default = True)
@classmethod @classmethod
def poll(cls, context): def poll(cls, context):
return context.active_object != None return context.active_object != None
...@@ -612,11 +613,13 @@ class VIEW3D_OT_replace_material(bpy.types.Operator): ...@@ -612,11 +613,13 @@ class VIEW3D_OT_replace_material(bpy.types.Operator):
m1 = self.matorg m1 = self.matorg
m2 = self.matrep m2 = self.matrep
all = self.all_objects all = self.all_objects
replace_material(m1,m2,all) replace_material(m1, m2, all)
return {'FINISHED'} return {'FINISHED'}
#menu classes
#------------------------------------------------------------------------------- # -----------------------------------------------------------------------------
# menu classes
class VIEW3D_MT_master_material(bpy.types.Menu): class VIEW3D_MT_master_material(bpy.types.Menu):
bl_label = "Master Material Menu" bl_label = "Master Material Menu"
...@@ -627,17 +630,20 @@ class VIEW3D_MT_master_material(bpy.types.Menu): ...@@ -627,17 +630,20 @@ class VIEW3D_MT_master_material(bpy.types.Menu):
layout.menu("VIEW3D_MT_assign_material", icon='ZOOMIN') layout.menu("VIEW3D_MT_assign_material", icon='ZOOMIN')
layout.menu("VIEW3D_MT_select_material", icon='HAND') layout.menu("VIEW3D_MT_select_material", icon='HAND')
layout.separator() layout.separator()
layout.operator("view3d.clean_material_slots", layout.operator("view3d.clean_material_slots",
text = 'Clean Material Slots', icon='CANCEL') text='Clean Material Slots',
icon='CANCEL')
layout.operator("view3d.material_to_texface", layout.operator("view3d.material_to_texface",
text = 'Material to Texface',icon='FACESEL_HLT') text='Material to Texface',
icon='FACESEL_HLT')
layout.operator("view3d.texface_to_material", layout.operator("view3d.texface_to_material",
text = 'Texface to Material',icon='FACESEL_HLT') text="Texface to Material",
icon='FACESEL_HLT')
layout.separator() layout.separator()
layout.operator("view3d.replace_material", layout.operator("view3d.replace_material",
text = 'Replace Material', icon='ARROW_LEFTRIGHT') text='Replace Material',
icon='ARROW_LEFTRIGHT')
class VIEW3D_MT_assign_material(bpy.types.Menu): class VIEW3D_MT_assign_material(bpy.types.Menu):
...@@ -646,14 +652,15 @@ class VIEW3D_MT_assign_material(bpy.types.Menu): ...@@ -646,14 +652,15 @@ class VIEW3D_MT_assign_material(bpy.types.Menu):
def draw(self, context): def draw(self, context):
layout = self.layout layout = self.layout
layout.operator_context = 'INVOKE_REGION_WIN' layout.operator_context = 'INVOKE_REGION_WIN'
for i in range (len(bpy.data.materials)): for material_name in bpy.data.materials.keys():
layout.operator("view3d.assign_material", layout.operator("view3d.assign_material",
text=bpy.data.materials[i].name, text=material_name,
icon='MATERIAL_DATA').matname = bpy.data.materials[i].name icon='MATERIAL_DATA').matname = material_name
layout.operator("view3d.assign_material",
text="Add New",
icon='ZOOMIN')
layout.operator("view3d.assign_material",text="Add New",
icon='ZOOMIN')
class VIEW3D_MT_select_material(bpy.types.Menu): class VIEW3D_MT_select_material(bpy.types.Menu):
bl_label = "Select by Material" bl_label = "Select by Material"
...@@ -666,30 +673,31 @@ class VIEW3D_MT_select_material(bpy.types.Menu): ...@@ -666,30 +673,31 @@ class VIEW3D_MT_select_material(bpy.types.Menu):
layout.label layout.label
if ob.mode == 'OBJECT': if ob.mode == 'OBJECT':
#show all used materials in entire blend file #show all used materials in entire blend file
for i in range (len(bpy.data.materials)): for material_name, material in bpy.data.materials.items():
if bpy.data.materials[i].users > 0: if material.users > 0:
layout.operator("view3d.select_material_by_name", layout.operator("view3d.select_material_by_name",
text=bpy.data.materials[i].name, text=material_name,
icon='MATERIAL_DATA').matname = bpy.data.materials[i].name icon='MATERIAL_DATA',
).matname = material_name
elif ob.mode == 'EDIT': elif ob.mode == 'EDIT':
#show only the materials on this object #show only the materials on this object
mats = ob.material_slots.keys() mats = ob.material_slots.keys()
for m in mats: for m in mats:
layout.operator("view3d.select_material_by_name", layout.operator("view3d.select_material_by_name",
text=m, text=m,
icon='MATERIAL_DATA').matname = m icon='MATERIAL_DATA').matname = m
def register(): def register():
bpy.utils.register_module(__name__) bpy.utils.register_module(__name__)
kc = bpy.context.window_manager.keyconfigs.addon kc = bpy.context.window_manager.keyconfigs.addon
km = kc.keymaps.new(name="3D View", space_type="VIEW_3D") km = kc.keymaps.new(name="3D View", space_type="VIEW_3D")
kmi = km.keymap_items.new('wm.call_menu', 'Q', 'PRESS') kmi = km.keymap_items.new('wm.call_menu', 'Q', 'PRESS')
kmi.properties.name = "VIEW3D_MT_master_material" kmi.properties.name = "VIEW3D_MT_master_material"
def unregister(): def unregister():
bpy.utils.unregister_module(__name__) bpy.utils.unregister_module(__name__)
...@@ -697,10 +705,9 @@ def unregister(): ...@@ -697,10 +705,9 @@ def unregister():
km = kc.keymaps["3D View"] km = kc.keymaps["3D View"]
for kmi in km.keymap_items: for kmi in km.keymap_items:
if kmi.idname == 'wm.call_menu': if kmi.idname == 'wm.call_menu':
if kmi.properties.name == "VIEW3D_MT_master_material": if kmi.properties.name == "VIEW3D_MT_master_material":
km.keymap_items.remove(kmi) km.keymap_items.remove(kmi)
break break
if __name__ == "__main__": if __name__ == "__main__":
register() register()
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment