[Bf-blender-cvs] SVN commit: /data/svn/bf-blender [11553] trunk/blender/release/scripts/ mesh_vgroup_average.py: script submitted by Johnny Matthews (guitargeek) - but I rewrote to make use of some existing functions in BPyMesh.py

[Campbell Barton]

Revision: 11553
          http://projects.blender.org/plugins/scmsvn/viewcvs.php?view=rev&root=bf-blender&revision=11553
Author:   campbellbarton
Date:     2007-08-12 03:30:14 +0200 (Sun, 12 Aug 2007)

Log Message:
-----------
script submitted by Johnny Matthews (guitargeek) - but I rewrote to make use of some existing functions in BPyMesh.py
This averages the weights of vert groups per island (based on face connectivity)
Nice for making scales and works well with new dual-quat interpolation

Added Paths:
-----------
    trunk/blender/release/scripts/mesh_vgroup_average.py

Added: trunk/blender/release/scripts/mesh_vgroup_average.py
===================================================================
--- trunk/blender/release/scripts/mesh_vgroup_average.py	                        (rev 0)
+++ trunk/blender/release/scripts/mesh_vgroup_average.py	2007-08-12 01:30:14 UTC (rev 11553)
@@ -0,0 +1,104 @@
+#!BPY
+"""
+Name: 'Vertex Groups Island Average'
+Blender: 243
+Group: 'Mesh'
+Tooltip: 'Average the vertex weights for each connected set of verts'
+"""
+
+# -------------------------------------------------------------------------- 
+# ***** 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 ***** 
+# --------------------------------------------------------------------------
+import Blender
+from Blender import Scene, Mesh, Window, sys
+from BPyMesh import meshWeight2List, list2MeshWeight, mesh2linkedFaces
+
+import BPyMessages
+import bpy
+
+def faceGroups2VertSets(face_groups):
+	'''	Return the face groups as sets of vert indicies	'''
+	return [set([v.index for f in fg for v in f]) for fg in face_groups]
+
+
+def vgroup_average(ob_orig, me, sce):
+	weight_names, weight_list = meshWeight2List(me)
+	
+	if not weight_names:
+		return
+		
+	weight_names_len = len(weight_names)
+	vgroup_dummy = [0.0] * weight_names_len
+	vgroup_range = range(weight_names_len)
+	
+	for vert_set in faceGroups2VertSets( mesh2linkedFaces(me) ):
+		if not vert_set:
+			continue
+		
+		collected_group = vgroup_dummy[:]
+		
+		# We need to average the vgroups
+		for i in vert_set:
+			vert_group = weight_list[i]			# get the original weight
+			weight_list[i] = collected_group	# replace with the collected group
+			
+			for j in vgroup_range: # iter through the vgroups
+				print collected_group, vert_group[j]
+				collected_group[j] += vert_group[j]
+		
+		for j in vgroup_range:
+			collected_group[j] /= len(vert_set)
+	
+	list2MeshWeight(me, weight_names, weight_list)
+
+def main():
+	
+	# Gets the current scene, there can be many scenes in 1 blend file.
+	sce = bpy.data.scenes.active
+	
+	# Get the active object, there can only ever be 1
+	# and the active object is always the editmode object.
+	ob_act = sce.objects.active
+	
+	if not ob_act or ob_act.type != 'Mesh':
+		BPyMessages.Error_NoMeshActive()
+		return 
+	
+	# Saves the editmode state and go's out of 
+	# editmode if its enabled, we cant make
+	# changes to the mesh data while in editmode.
+	is_editmode = Window.EditMode()
+	Window.EditMode(0)
+	
+	Window.WaitCursor(1)
+	me = ob_act.getData(mesh=1) # old NMesh api is default
+	t = sys.time()
+	
+	# Run the mesh editing function
+	vgroup_average(ob_act, me, sce)
+	
+	# Timing the script is a good way to be aware on any speed hits when scripting
+	print 'Average VGroups in %.2f seconds' % (sys.time()-t)
+	Window.WaitCursor(0)
+	if is_editmode: Window.EditMode(1)
+	
+	
+# This lets you can import the script without running it
+if __name__ == '__main__':
+	main()
\ No newline at end of file