adding a script that copies material colors to vertex colors,

usefull when consolidating 3d models from formats that use many materials for different colors, into 1 object with fewer materials.

1 files changed, 99 insertions, 0 deletions

diff --git a/release/scripts/vertexpaint_from_material.py b/release/scripts/vertexpaint_from_material.py
new file mode 100644
index 00000000000..6ecde41347d
--- /dev/null
+++ b/release/scripts/vertexpaint_from_material.py
@@ -0,0 +1,99 @@
+#!BPY
+"""
+Name: 'Copy from Material...'
+Blender: 242
+Group: 'VertexPaint'
+Tooltip: 'Writes material diffuse color as vertex colors.'
+"""
+
+__author__ = "Campbell Barton"
+__url__ = ("www.blender.org", "blenderartists.org")
+__version__ = "1.0"
+
+__bpydoc__ = """\
+This script copies material colors to vertex colors.
+Optionaly you can operate on all faces and many objects as well as multiplying with the current color.
+"""
+
+
+from Blender import *
+
+def matcol(mat):
+	'''
+	Returns the material color as a tuple 3 from 0 to 255
+	'''
+	if mat:
+		return \
+		int(mat.R*255),\
+		int(mat.G*255),\
+		int(mat.B*255)
+		return None
+
+def mat2vcol(PREF_SEL_FACES_ONLY, PREF_ACTOB_ONLY, PREF_MULTIPLY_COLOR):
+	scn= Scene.GetCurrent()
+	if PREF_ACTOB_ONLY:
+		obs= [scn.getActiveObject()]
+	else:
+		obs= Object.GetSelected()
+		ob= scn.getActiveObject()
+		if ob not in obs:
+			obs.append(ob)
+	
+	
+	for ob in obs:
+		if ob.type != 'Mesh':
+			continue
+		
+		me= ob.getData(mesh=1)
+		
+		try:
+			me.faceUV=True
+		except: # no faces
+			continue
+		
+		matcols= [matcol(mat) for mat in me.materials]
+		len_matcols= len(matcols)
+		FSEL= Mesh.FaceFlags.SELECT
+		
+		for f in me.faces:
+			if not PREF_SEL_FACES_ONLY or f.flag & FSEL:
+				f_mat= f.mat
+				if f_mat < len_matcols:
+					mat= matcols[f.mat]
+					if mat:
+						if PREF_MULTIPLY_COLOR:
+							for c in f.col:
+								c.r= int(((c.r/255.0) * (mat[0]/255.0)) * 255.0)
+								c.g= int(((c.g/255.0) * (mat[1]/255.0)) * 255.0)
+								c.b= int(((c.b/255.0) * (mat[2]/255.0)) * 255.0)
+						else:
+							for c in f.col:
+								c.r=mat[0]
+								c.g=mat[1]
+								c.b=mat[2]
+		me.update()
+
+def main():
+	# Create the variables.
+	
+	PREF_SEL_FACES_ONLY= Draw.Create(1)
+	PREF_ACTOB_ONLY= Draw.Create(1)
+	PREF_MULTIPLY_COLOR = Draw.Create(0)
+	
+	pup_block = [\
+	('Sel Faces Only', PREF_SEL_FACES_ONLY, 'Only apply to selected faces.'),\
+	('Active Only', PREF_ACTOB_ONLY, 'Operate on all selected objects.'),\
+	('Multiply Existing', PREF_MULTIPLY_COLOR, 'Multiplies material color with existing.'),\
+	]
+	
+	if not Draw.PupBlock('VCols from Material', pup_block):
+		return
+	
+	PREF_SEL_FACES_ONLY= PREF_SEL_FACES_ONLY.val
+	PREF_ACTOB_ONLY= PREF_ACTOB_ONLY.val
+	PREF_MULTIPLY_COLOR= PREF_MULTIPLY_COLOR.val
+
+	mat2vcol(PREF_SEL_FACES_ONLY, PREF_ACTOB_ONLY, PREF_MULTIPLY_COLOR)
+
+if __name__=='__main__':
+	main()
\ No newline at end of file