forked from bartvdbraak/blender
8537a4d471
Added support for "Weighted Collapse" Before an edge could only collapse into its middle, Now the edge collapses into a point bias'd by the 2 verts Concave/Convec "Pointyness" value as well as boundry weighting. This works much better for boundry verts. - UV's Vcols and Weights are correctly interpolated into the new location. Added a tool in the mesh menu for accessing the poly reduction tool.
554 lines
16 KiB
Python
554 lines
16 KiB
Python
# ***** BEGIN GPL LICENSE BLOCK *****
|
|
#
|
|
# (C) Copyright 2006 MetaVR, Inc.
|
|
# http://www.metavr.com
|
|
# Written by Campbell Barton
|
|
#
|
|
# 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
|
|
Vector= Blender.Mathutils.Vector
|
|
Ang= Blender.Mathutils.AngleBetweenVecs
|
|
LineIntersect= Blender.Mathutils.LineIntersect
|
|
CrossVecs= Blender.Mathutils.CrossVecs
|
|
import BPyMesh
|
|
|
|
|
|
def uv_key(uv):
|
|
return round(uv.x, 5), round(uv.y, 5)
|
|
|
|
def uv_key_mix(uv1, uv2, w1, w2):
|
|
# Weighted mix. w1+w2==1.0
|
|
return w1*uv1[0]+w2*uv2[0], w1*uv1[1]+w2*uv2[1]
|
|
|
|
def col_key(col):
|
|
return col.r, col.g, col.b
|
|
|
|
def col_key_mix(col1, col2, w1, w2):
|
|
# Weighted mix. w1+w2==1.0
|
|
return int(w1*col1[0] + w2*col2[0]), int(w1*col1[1] + w2*col2[1]), int(w1*col1[2]+col2[2]*w2)
|
|
|
|
def ed_key(ed):
|
|
i1= ed.v1.index
|
|
i2= ed.v2.index
|
|
if i1<i2: return i1,i2
|
|
return i2,i1
|
|
|
|
def redux(ob, REDUX=0.5, BOUNDRY_WEIGHT=5.0, FACE_AREA_WEIGHT=1.0, FACE_TRIANGULATE=True, DO_UV=True, DO_VCOL=True, DO_WEIGHTS=True):
|
|
'''
|
|
BOUNDRY_WEIGHT - 0 is no boundry weighting. 2.0 will make them twice as unlikely to collapse.
|
|
FACE_AREA_WEIGHT - 0 is no weight. 1 is normal, 2.0 is higher.
|
|
'''
|
|
|
|
""" # DEBUG!
|
|
if Blender.Get('rt') == 1000:
|
|
DEBUG=True
|
|
else:
|
|
DEBUG= False
|
|
"""
|
|
me= ob.getData(mesh=1)
|
|
|
|
if REDUX>1.0 or REDUX<0.0 or len(me.faces)<4:
|
|
return
|
|
|
|
if FACE_TRIANGULATE:
|
|
me.quadToTriangle()
|
|
|
|
if (not me.getVertGroupNames()) and DO_WEIGHTS:
|
|
DO_WEIGHTS= False
|
|
|
|
OLD_MESH_MODE= Blender.Mesh.Mode()
|
|
Blender.Mesh.Mode(Blender.Mesh.SelectModes.VERTEX)
|
|
|
|
if (DO_UV or DO_VCOL) and not me.faceUV:
|
|
DO_VCOL= DO_UV= False
|
|
|
|
current_face_count= len(me.faces)
|
|
target_face_count= int(current_face_count * REDUX)
|
|
# % of the collapseable faces to collapse per pass.
|
|
#collapse_per_pass= 0.333 # between 0.1 - lots of small nibbles, slow but high q. and 0.9 - big passes and faster.
|
|
collapse_per_pass= 0.333 # between 0.1 - lots of small nibbles, slow but high q. and 0.9 - big passes and faster.
|
|
|
|
"""# DEBUG!
|
|
if DEBUG:
|
|
COUNT= [0]
|
|
def rd():
|
|
if COUNT[0]< 330:
|
|
COUNT[0]+=1
|
|
return
|
|
me.update()
|
|
Blender.Window.RedrawAll()
|
|
print 'Press key for next, count "%s"' % COUNT[0]
|
|
try: input()
|
|
except KeyboardInterrupt:
|
|
raise "Error"
|
|
except:
|
|
pass
|
|
|
|
COUNT[0]+=1
|
|
"""
|
|
|
|
class collapseEdge(object):
|
|
__slots__ = 'length', 'key', 'faces', 'collapse_loc', 'v1', 'v2','uv1', 'uv2', 'col1', 'col2', 'collapse_weight'
|
|
def __init__(self, ed):
|
|
self.key= ed_key(ed)
|
|
self.length= ed.length
|
|
self.faces= []
|
|
self.v1= ed.v1
|
|
self.v2= ed.v2
|
|
if DO_UV or DO_VCOL:
|
|
self.uv1= []
|
|
self.uv2= []
|
|
self.col1= []
|
|
self.col2= []
|
|
|
|
# self.collapse_loc= None # new collapse location.
|
|
# Basic weighting.
|
|
#self.collapse_weight= self.length * (1+ ((ed.v1.no-ed.v2.no).length**2))
|
|
self.collapse_weight= 1.0
|
|
|
|
class collapseFace(object):
|
|
__slots__ = 'verts', 'normal', 'area', 'index', 'orig_uv', 'orig_col', 'uv', 'col' # , 'collapse_edge_count'
|
|
def __init__(self, f):
|
|
self.verts= f.v
|
|
self.normal= f.no
|
|
self.area= f.area
|
|
self.index= f.index
|
|
if DO_UV or DO_VCOL:
|
|
self.orig_uv= [uv_key(uv) for uv in f.uv]
|
|
self.uv= f.uv
|
|
self.orig_col= [col_key(col) for col in f.col]
|
|
self.col= f.col
|
|
|
|
#self.collapse_edge_count= 0 # used so we know how many edges of the face are collapsed.
|
|
|
|
for v in me.verts:
|
|
v.hide=0
|
|
|
|
collapse_edges= collapse_faces= None
|
|
|
|
while target_face_count <= len(me.faces):
|
|
BPyMesh.meshCalcNormals(me)
|
|
|
|
if DO_WEIGHTS:
|
|
groupNames, vWeightDict= BPyMesh.meshWeight2Dict(me)
|
|
|
|
# THIS CRASHES
|
|
#verts= list(me.verts)
|
|
#edges= list(me.edges)
|
|
#faces= list(me.faces)
|
|
|
|
# THIS WORKS
|
|
verts= me.verts
|
|
edges= me.edges
|
|
faces= me.faces
|
|
|
|
DOUBLE_CHECK= [0]*len(verts)
|
|
|
|
for v in verts:
|
|
v.sel= False
|
|
|
|
collapse_faces= [collapseFace(f) for f in faces]
|
|
collapse_edges= [collapseEdge(ed) for ed in edges]
|
|
collapse_edges_dict= dict( [(ced.key, ced) for ced in collapse_edges] )
|
|
|
|
# Store verts edges.
|
|
vert_ed_users= [[] for i in xrange(len(verts))]
|
|
for ced in collapse_edges:
|
|
vert_ed_users[ced.v1.index].append(ced)
|
|
vert_ed_users[ced.v2.index].append(ced)
|
|
|
|
# Store face users
|
|
vert_face_users= [[] for i in xrange(len(verts))]
|
|
|
|
# Have decieded not to use this. area is better.
|
|
#face_perim= [0.0]* len(me.faces)
|
|
|
|
for ii, cfa in enumerate(collapse_faces):
|
|
for i, v1 in enumerate(cfa.verts):
|
|
vert_face_users[v1.index].append( (i,cfa) )
|
|
|
|
# add the uv coord to the vert
|
|
v2 = cfa.verts[i-1]
|
|
i1= v1.index
|
|
i2= v2.index
|
|
|
|
if i1>i2: ced= collapse_edges_dict[i2,i1]
|
|
else: ced= collapse_edges_dict[i1,i2]
|
|
|
|
ced.faces.append(cfa)
|
|
if DO_UV or DO_VCOL:
|
|
# if the edge is flipped from its order in the face then we need to flip the order indicies.
|
|
if cfa.verts[i]==ced.v1: i1,i2 = i, i-1
|
|
else: i1,i2 = i-1, i
|
|
|
|
if DO_UV:
|
|
ced.uv1.append( cfa.orig_uv[i1] )
|
|
ced.uv2.append( cfa.orig_uv[i2] )
|
|
|
|
if DO_VCOL:
|
|
ced.col1.append( cfa.orig_col[i1] )
|
|
ced.col2.append( cfa.orig_col[i2] )
|
|
|
|
|
|
# PERIMITER
|
|
#face_perim[ii]+= ced.length
|
|
|
|
|
|
|
|
# How weight the verts by the area of their faces * the normal difference.
|
|
# when the edge collapses, to vert weights are taken into account
|
|
|
|
vert_weights= [0.5] * len(verts)
|
|
|
|
for ii, vert_faces in enumerate(vert_face_users):
|
|
for f in vert_faces:
|
|
try:
|
|
no_ang= (Ang(verts[ii].no, f[1].normal)/180) * f[1].area
|
|
except:
|
|
no_ang= 1.0
|
|
|
|
vert_weights[ii] += no_ang
|
|
|
|
|
|
# BOUNDRY CHECKING AND WEIGHT EDGES. CAN REMOVE
|
|
# Now we know how many faces link to an edge. lets get all the boundry verts
|
|
if BOUNDRY_WEIGHT > 0:
|
|
verts_boundry= [1] * len(verts)
|
|
#for ed_idxs, faces_and_uvs in edge_faces_and_uvs.iteritems():
|
|
for ced in collapse_edges:
|
|
if len(ced.faces) < 2:
|
|
verts_boundry[ced.key[0]]= 2
|
|
verts_boundry[ced.key[1]]= 2
|
|
|
|
for ced in collapse_edges:
|
|
if verts_boundry[ced.v1.index] != verts_boundry[ced.v2.index]:
|
|
# Edge has 1 boundry and 1 non boundry vert. weight higher
|
|
ced.collapse_weight= BOUNDRY_WEIGHT
|
|
|
|
# weight the verts by their boundry status
|
|
|
|
for ii, boundry in enumerate(verts_boundry):
|
|
if boundry==2:
|
|
vert_weights[ii] *= BOUNDRY_WEIGHT
|
|
|
|
vert_collapsed= verts_boundry
|
|
del verts_boundry
|
|
else:
|
|
vert_collapsed= [1] * len(verts)
|
|
|
|
|
|
|
|
|
|
def ed_set_collapse_loc(ced):
|
|
v1co= ced.v1.co
|
|
v2co= ced.v2.co
|
|
v1no= ced.v1.co
|
|
v2no= ced.v2.co
|
|
|
|
# Use the vertex weights to bias the new location.
|
|
w1= vert_weights[ced.v1.index]
|
|
w2= vert_weights[ced.v2.index]
|
|
|
|
# normalize the weights of each vert - se we can use them as scalers.
|
|
wscale= w1+w2
|
|
if not wscale: # no scale?
|
|
w1=w2= 0.5
|
|
else:
|
|
w1/=wscale
|
|
w2/=wscale
|
|
|
|
length= ced.length
|
|
|
|
if 0:
|
|
between= ((v1co*w1) + (v2co*w2))
|
|
ced.collapse_loc= between
|
|
else:
|
|
between= (v1co+v2co) * 0.5
|
|
|
|
# Collapse
|
|
# new_location = between # Replace tricky code below. this code predicts the best collapse location.
|
|
|
|
# Make lines at right angles to the normals- these 2 lines will intersect and be
|
|
# the point of collapsing.
|
|
|
|
# Enlarge so we know they intersect: ced.length*2
|
|
cv1= CrossVecs(v1no, CrossVecs(v1no, v1co-v2co))
|
|
cv2= CrossVecs(v2no, CrossVecs(v2no, v2co-v1co))
|
|
|
|
# Scale to be less then the edge lengths.
|
|
cv1.normalize()
|
|
cv2.normalize()
|
|
cv1 = cv1 * (length* 0.4)
|
|
cv2 = cv2 * (length* 0.4)
|
|
|
|
smart_offset_loc= between + (cv1 + cv2)
|
|
|
|
# Now we need to blend between smart_offset_loc and w1/w2
|
|
# you see were blending between a vert and the edges midpoint, so we cant use a normal weighted blend.
|
|
if w1 > 0.5: # between v1 and smart_offset_loc
|
|
#ced.collapse_loc= v1co*(w2+0.5) + smart_offset_loc*(w1-0.5)
|
|
w2*=2
|
|
w1= 1-w2
|
|
|
|
|
|
ced.collapse_loc= v1co*w1 + smart_offset_loc*w2
|
|
else: # w between v2 and smart_offset_loc
|
|
w1*=2
|
|
w2= 1-w1
|
|
ced.collapse_loc= v2co*w2 + smart_offset_loc*w1
|
|
|
|
if ced.collapse_loc.x != ced.collapse_loc.x: # NAN LOCATION, revert to between
|
|
ced.collapse_loc= between
|
|
|
|
|
|
# Best method, no quick hacks here, Correction. Should be the best but needs tweaks.
|
|
def ed_set_collapse_error(ced):
|
|
i1= ced.v1.index
|
|
i2= ced.v1.index
|
|
|
|
test_faces= set()
|
|
for i in (i1,i2): # faster then LC's
|
|
for f in vert_face_users[i]:
|
|
test_faces.add(f[1].index)
|
|
|
|
for f in ced.faces:
|
|
test_faces.remove(f.index)
|
|
|
|
# test_faces= tuple(test_faces) # keep order
|
|
|
|
v1_orig= Vector(ced.v1.co)
|
|
v2_orig= Vector(ced.v2.co)
|
|
|
|
ced.v1.co= ced.v2.co= ced.collapse_loc
|
|
|
|
new_nos= [faces[i].no for i in test_faces]
|
|
|
|
ced.v1.co= v1_orig
|
|
ced.v2.co= v2_orig
|
|
|
|
# now see how bad the normals are effected
|
|
angle_diff= 1.0
|
|
|
|
for ii, i in enumerate(test_faces): # local face index, global face index
|
|
cfa= collapse_faces[i] # this collapse face
|
|
try:
|
|
# can use perim, but area looks better.
|
|
if FACE_AREA_WEIGHT:
|
|
angle_diff+= (Ang(cfa.normal, new_nos[ii])/180) * (1+(cfa.area * FACE_AREA_WEIGHT)) # 4 is how much to influence area
|
|
else:
|
|
angle_diff+= (Ang(cfa.normal, new_nos[ii])/180)
|
|
|
|
except:
|
|
pass
|
|
|
|
# This is very arbirary, feel free to modify
|
|
try: no_ang= (Ang(ced.v1.no, ced.v2.no)/180) + 1
|
|
except: no_ang= 2.0
|
|
|
|
# do *= because we face the boundry weight to initialize the weight. 1.0 default.
|
|
ced.collapse_weight*= ((no_ang * ced.length) * (1-(1/angle_diff)))# / max(len(test_faces), 1)
|
|
|
|
# We can calculate the weights on __init__ but this is higher qualuity.
|
|
for ced in collapse_edges:
|
|
if ced.faces: # dont collapse faceless edges.
|
|
ed_set_collapse_loc(ced)
|
|
ed_set_collapse_error(ced)
|
|
|
|
# Wont use the function again.
|
|
del ed_set_collapse_error
|
|
del ed_set_collapse_loc
|
|
# END BOUNDRY. Can remove
|
|
|
|
# sort by collapse weight
|
|
collapse_edges.sort(lambda ced1, ced2: cmp(ced1.collapse_weight, ced2.collapse_weight)) # edges will be used for sorting
|
|
|
|
vert_collapsed= [0]*len(verts)
|
|
|
|
collapse_edges_to_collapse= []
|
|
|
|
# Make a list of the first half edges we can collapse,
|
|
# these will better edges to remove.
|
|
collapse_count=0
|
|
for ced in collapse_edges:
|
|
if ced.faces:
|
|
i1= ced.v1.index
|
|
i2= ced.v2.index
|
|
# Use vert selections
|
|
if vert_collapsed[i1] or vert_collapsed[i2]:
|
|
pass
|
|
else:
|
|
# Now we know the verts havnyt been collapsed.
|
|
vert_collapsed[i2]= vert_collapsed[i1]= 1 # Dont collapse again.
|
|
collapse_count+=1
|
|
collapse_edges_to_collapse.append(ced)
|
|
|
|
# Get a subset of the entire list- the first "collapse_per_pass", that are best to collapse.
|
|
if collapse_count > 4:
|
|
collapse_count = int(collapse_count*collapse_per_pass)
|
|
else:
|
|
collapse_count = len(collapse_edges)
|
|
# We know edge_container_list_collapse can be removed.
|
|
for ced in collapse_edges_to_collapse:
|
|
"""# DEBUG!
|
|
if DEBUG:
|
|
if DOUBLE_CHECK[ced.v1.index] or\
|
|
DOUBLE_CHECK[ced.v2.index]:
|
|
raise 'Error'
|
|
else:
|
|
DOUBLE_CHECK[ced.v1.index]=1
|
|
DOUBLE_CHECK[ced.v2.index]=1
|
|
|
|
tmp= (ced.v1.co+ced.v2.co)*0.5
|
|
Blender.Window.SetCursorPos(tmp.x, tmp.y, tmp.z)
|
|
Blender.Window.RedrawAll()
|
|
"""
|
|
|
|
# Chech if we have collapsed our quota.
|
|
collapse_count-=1
|
|
if not collapse_count:
|
|
break
|
|
|
|
current_face_count -= len(ced.faces)
|
|
|
|
# Interpolate the bone weights.
|
|
if DO_WEIGHTS:
|
|
i1= ced.v1.index
|
|
i2= ced.v2.index
|
|
w1= vert_weights[i1]
|
|
w2= vert_weights[i2]
|
|
|
|
# Normalize weights
|
|
wscale= w1+w2
|
|
if not wscale: # no scale?
|
|
w1=w2= 0.5
|
|
else:
|
|
w1/= wscale
|
|
w2/= wscale
|
|
wd= vWeightDict[i1] # v1 weight dict
|
|
for group_key, weight_value in wd.iteritems():
|
|
wd[group_key]= weight_value*w1
|
|
|
|
wd= vWeightDict[i2] # v1 weight dict
|
|
for group_key, weight_value in wd.iteritems():
|
|
wd[group_key]= weight_value*w2
|
|
|
|
|
|
if DO_UV or DO_VCOL:
|
|
# Handel UV's and vert Colors!
|
|
for v, my_weight, other_weight, edge_my_uvs, edge_other_uvs, edge_my_cols, edge_other_cols in (\
|
|
( ced.v1, vert_weights[ced.v1.index], vert_weights[ced.v2.index], ced.uv1, ced.uv2, ced.col1, ced.col2),\
|
|
( ced.v2, vert_weights[ced.v2.index], vert_weights[ced.v1.index], ced.uv2, ced.uv1, ced.col2, ced.col1)\
|
|
):
|
|
|
|
# Normalize weights
|
|
wscale= my_weight+other_weight
|
|
if not wscale: # no scale?
|
|
my_weight=other_weight= 0.5
|
|
else:
|
|
my_weight/= wscale
|
|
other_weight/= wscale
|
|
|
|
uvs_mixed= [ uv_key_mix(edge_my_uvs[iii], edge_other_uvs[iii], my_weight, other_weight) for iii in xrange(len(edge_my_uvs)) ]
|
|
cols_mixed= [ col_key_mix(edge_my_cols[iii], edge_other_cols[iii], my_weight, other_weight) for iii in xrange(len(edge_my_cols)) ]
|
|
#print 'FACE, USWERS', len(vert_face_users[v.index])
|
|
for face_vert_index, cfa in vert_face_users[v.index]:
|
|
if len(cfa.verts)==3 and cfa not in ced.faces: # if the face is apart of this edge then dont bother finding the uvs since the face will be removed anyway.
|
|
|
|
if DO_UV:
|
|
# UV COORDS
|
|
uvk= cfa.orig_uv[face_vert_index]
|
|
try:
|
|
tex_index= edge_my_uvs.index(uvk)
|
|
except:
|
|
tex_index= None
|
|
""" # DEBUG!
|
|
if DEBUG:
|
|
print 'not found', uvk, 'in', edge_my_uvs, 'ed index', ii, '\nwhat about', edge_other_uvs
|
|
"""
|
|
if tex_index != None: # This face uses a uv in the collapsing face. - do a merge
|
|
other_uv= edge_other_uvs[tex_index]
|
|
uv_vec= cfa.uv[face_vert_index]
|
|
uv_vec.x, uv_vec.y= uvs_mixed[tex_index]
|
|
|
|
# TEXFACE COLORS
|
|
if DO_VCOL:
|
|
colk= cfa.orig_col[face_vert_index]
|
|
try: tex_index= edge_my_cols.index(colk)
|
|
except: pass
|
|
if tex_index != None:
|
|
other_col= edge_other_cols[tex_index]
|
|
col_ob= cfa.col[face_vert_index]
|
|
col_ob.r, col_ob.g, col_ob.b= cols_mixed[tex_index]
|
|
|
|
# DEBUG! if DEBUG: rd()
|
|
|
|
# Execute the collapse
|
|
ced.v1.sel= ced.v2.sel= True
|
|
ced.v1.co= ced.v2.co= ced.collapse_loc
|
|
|
|
# DEBUG! if DEBUG: rd()
|
|
if current_face_count <= target_face_count:
|
|
ced.collapse_loc= None
|
|
break
|
|
|
|
'''
|
|
# Execute the collapse
|
|
for ced in collapse_edges:
|
|
# Since the list is ordered we can stop once the first non collapsed edge if sound.
|
|
if not ced.collapse_loc:
|
|
break
|
|
ced.v1.sel= ced.v2.sel= True
|
|
ced.v1.co= ced.v2.co= ced.collapse_loc
|
|
'''
|
|
|
|
# Copy weights back to the mesh before we remove doubles.
|
|
if DO_WEIGHTS:
|
|
BPyMesh.dict2MeshWeight(me, groupNames, vWeightDict)
|
|
|
|
doubles= me.remDoubles(0.0001)
|
|
me= ob.getData(mesh=1)
|
|
current_face_count= len(me.faces)
|
|
#break
|
|
|
|
if doubles==0: # should never happen.
|
|
break
|
|
|
|
if current_face_count <= target_face_count:
|
|
ced.collapse_loc= None
|
|
break
|
|
|
|
# Cleanup. BUGGY?
|
|
|
|
|
|
me.update()
|
|
Blender.Mesh.Mode(OLD_MESH_MODE)
|
|
|
|
|
|
# Example usage
|
|
def main():
|
|
Blender.Window.EditMode(0)
|
|
scn= Blender.Scene.GetCurrent()
|
|
active_ob= scn.getActiveObject()
|
|
t= Blender.sys.time()
|
|
redux(active_ob, 0.4)
|
|
print '%.4f' % (Blender.sys.time()-t)
|
|
|
|
if __name__=='__main__':
|
|
main()
|