rewrote this script, it was the most out of date script and had some errors that are solved in the new script.

- moved to the weightpaint menu
- Use a group for envelope objects, no special names.
- Can use non mesh data for envalopes (anything that can be converted into a mesh - metaballs, text, surface)
- Dosnt rely on applying loc/size/rot for it to work properly
- Uses a good point-inside-mesh method (optimized with mesh and face bounding checks)
- uses Mesh instead of NMesh

todo, only update active vgroup

1 files changed, 204 insertions, 0 deletions

diff --git a/release/scripts/weightpaint_envelope_assign.py b/release/scripts/weightpaint_envelope_assign.py
new file mode 100644
index 00000000000..cdd2c3ad21e
--- /dev/null
+++ b/release/scripts/weightpaint_envelope_assign.py
@@ -0,0 +1,204 @@
+#!BPY
+
+"""
+Name: 'Envelope via Group Objects'
+Blender: 242
+Group: 'WeightPaint'
+Tooltip: 'Assigns weights to vertices via object envelopes'
+"""
+
+__author__ = ["Campbell Barton"]
+__url__ = ("blender", "blenderartist.org")
+__version__ = "0.1"
+
+# ***** BEGIN GPL LICENSE BLOCK *****
+#
+# This program is free software; you can redistribute it and/or
+# modify it under the terms of the GNU General Public License
+# as published by the Free Software Foundation; either version 2
+# of the License, or (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software Foundation,
+# Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
+#
+# ***** END GPL LICENCE BLOCK *****
+# --------------------------------------------------------------------------
+
+def intersection_data(ob):
+	'''
+	Takes an object and returns raw triangles and tri bounds.
+	returns a list of (v1,v2,v3, xmin, ymin, xmax, ymax)
+	... to be used for ray intersection
+	'''
+	global tot_ymin, tot_xmin, tot_ymax, tot_xmax, tot_zmin, tot_zmax
+	tot_zmin = tot_ymin = tot_xmin =  100000000.0
+	tot_zmax = tot_ymax = tot_xmax = -100000000.0
+	
+	def tribounds_xy(v1,v2,v3):
+		'''
+		return the 3 vecs with bounds, also update tot_*min/max
+		'''
+		global tot_ymin, tot_xmin, tot_ymax, tot_xmax, tot_zmin, tot_zmax
+		xmin = xmax = v1.x
+		ymin = ymax = v1.y
+		
+		x = v2.x
+		y = v2.y
+		if x>xmax: xmax = x
+		if y>ymax: ymax = y
+		if x<xmin: xmin = x
+		if y<ymin: ymin = y
+		
+			
+		x = v3.x
+		y = v3.y
+		if x>xmax: xmax = x
+		if y>ymax: ymax = y
+		if x<xmin: xmin = x
+		if y<ymin: ymin = y
+		
+		# totbounds
+		for z in v1.z, v2.z, v3.z:
+			if z>tot_zmax: tot_zmax = z
+			if z<tot_zmin: tot_zmin = z
+		
+		# maintain global min/max for x and y
+		if xmax>tot_xmax: tot_xmax = xmax
+		if xmin<tot_xmin: tot_xmin = xmin
+		
+		if ymax>tot_ymax: tot_ymax = ymax
+		if ymin<tot_ymin: tot_ymin = ymin
+		
+		return v1,v2,v3, xmin, ymin, xmax, ymax
+	
+	me = BPyMesh.getMeshFromObject(ob)
+	
+	if not me:
+		return []
+	
+	me.transform(ob.matrixWorld)
+	
+	mesh_data = []
+	
+	# copy all the vectors so when the mesh is free'd we still have access.
+	mesh_vert_cos= [v.co.copy() for v in me.verts]
+	
+	for f in me.faces:
+		f_v = [mesh_vert_cos[v.index] for v in f]
+		mesh_data.append( tribounds_xy(f_v[0], f_v[1], f_v[2]) )
+		if len(f) == 4:
+			mesh_data.append( tribounds_xy(f_v[0], f_v[2], f_v[3]) )
+	
+	me.verts = None # free some memory
+	
+	return ob.name, mesh_data, tot_xmin, tot_ymin, tot_zmin, tot_xmax, tot_ymax, tot_zmax
+	
+from Blender import *
+ray = Mathutils.Vector(0,0,-1)
+def point_in_data(point, mesh_data_tuple):
+	ob_name, mesh_data, tot_xmin, tot_ymin, tot_zmin, tot_xmax, tot_ymax, tot_zmax = mesh_data_tuple
+	
+	Intersect = Mathutils.Intersect
+	x = point.x
+	y = point.y
+	z = point.z
+	
+	# Bouds check for all mesh data
+	if not (\
+	tot_zmin < z < tot_zmax and\
+	tot_ymin < y < tot_ymax and\
+	tot_xmin < x < tot_xmax\
+	):
+		return False
+	
+	def isect(tri_data):
+		v1,v2,v3, xmin, ymin, xmax, ymax = tri_data
+		if not (xmin < x < xmax and ymin < y < ymax):
+			return False
+		
+		if v1.z < z and v2.z < z and v3.z < z:
+			return False
+		
+		isect = Intersect(v1, v2,v3, ray, point, 1) # Clipped.
+		if isect and isect.z > z: # This is so the ray only counts if its above the point. 
+			return True
+		else:
+			return False
+	
+	# return len([None for tri_data in mesh_data if isect(tri_data)]) % 2
+	return len( filter(isect, mesh_data) ) % 2
+
+import BPyMesh
+def env_from_group(ob_act, grp, PREF_OVERWRITE=True):
+	# get intersection data
+	# group_isect_data = [intersection_data(ob) for ob in group.objects]
+	group_isect_data = []
+	for ob in grp.objects:
+		if ob != ob_act: # in case we're in the group.
+			gid = intersection_data(ob)
+			if gid[1]: # has some triangles?
+				group_isect_data.append( gid )
+	
+	# sort by name
+	group_isect_data.sort()
+	
+	group_names = [gid[0] for gid in group_isect_data]
+	
+	me = ob_act.getData(mesh=1)
+	len_group_names= len(group_names)
+	vweight_list= [[0.0]*len_group_names for i in xrange(len(me.verts))]
+	
+	ob_act_mat = ob_act.matrixWorld
+	for vi, v in enumerate(me.verts):
+		# Get all the groups for this vert
+		co = v.co * ob_act_mat
+		for group_index, group_isect in enumerate(group_isect_data):
+			if point_in_data(co, group_isect):
+				vweight_list[vi][group_index] = 1.0
+	
+	BPyMesh.list2MeshWeight(me, group_names, vweight_list)
+
+
+import BPyMessages
+def main():
+	
+	scn= Scene.GetCurrent()
+	ob_act = scn.objects.active
+	if not ob_act or ob_act.type != 'Mesh':
+		BPyMessages.Error_NoMeshActive()
+		return
+	
+	PREF_ENV_GROUPNAME= Draw.Create('')	
+	PREF_OVERWRITE= Draw.Create(False)
+	pup_block= [\
+	'Group Name',\
+	('GR:', PREF_ENV_GROUPNAME, 0, 21, 'The name of an existing groups to '),\
+	#('Overwrite', PREF_OVERWRITE, 'Overwrite existing vertex groups.'),\
+	]
+	
+	if not Draw.PupBlock('Envalope From Group...', pup_block):
+		return
+	
+	try:
+		grp = Group.Get(PREF_ENV_GROUPNAME.val)
+	except:	
+		Draw.PupMenu('Error%t|Group "' + PREF_ENV_GROUPNAME.val + '" does not exist.')
+		return
+	
+	PREF_ENV_GROUPNAME= PREF_ENV_GROUPNAME.val
+	PREF_OVERWRITE= PREF_OVERWRITE.val
+	
+	Window.WaitCursor(1)
+	t = sys.time()
+	env_from_group(ob_act, grp, PREF_OVERWRITE)
+	print 'assigned envelopes in:', sys.time() - t
+	Window.WaitCursor(0)
+
+if __name__ == '__main__':
+	main()