forked from kattkieru/medium_def_layers
/
skin_merge.py
161 lines (126 loc) · 5.18 KB
/
skin_merge.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
from maya.api import OpenMaya as om2
from maya.api import OpenMayaAnim as oma2
import pymel.core as pm
## ======================================================================
def get_deform_shape( ob ):
"""
Gets the visible geometry shape regardless of whether or not
the object is deformed or not.
:param ob: The object to check.
:returns: The object's deform shape.
"""
ob = pm.PyNode(ob)
if ob.type() in ['nurbsSurface', 'mesh', 'nurbsCurve']:
ob = ob.getParent()
shapes = pm.PyNode(ob).getShapes()
if len(shapes) == 1:
return( shapes[0] )
else:
real_shapes = [ x for x in shapes if not x.intermediateObject.get() ]
return( real_shapes[0] if len(real_shapes) else None )
## ---------------------------------------------------------------------
def get_skin_cluster(ob):
"""
Find the first skinCluster on the object that actually effects it.
When multiple deformers are layered, this means that the latest skin
in the deform chain will be returned.
In situations where multiple skins are layered through blendshapes,
this function will only return the skin on the current mesh and not
any of the blendshape sources.
:param ob: The object whose skin you wish to return.
:returns: A skinCluster object, or None.
"""
shape = get_deform_shape( ob )
if shape is None:
return(None)
skins = pm.ls( pm.listHistory( shape ), type='skinCluster' )
if len( skins ):
for skin in skins:
# log( '\t+ Processing %s...' % skin )
groupId = skin.input[0].groupId.inputs( plugs=True )[0]
outputs = groupId.outputs(plugs=True)
for outp in outputs:
node = outp.node()
if node == shape or node == ob:
## force neighbors!
skin.weightDistribution.set(1) ## neighbors
return( skin )
return( None )
## ---------------------------------------------------------------------
def log( msg, warn=False, error=False ):
if warn and error:
raise ValueError('Please specify only one of warn / error.')
log_func = om.MGlobal.displayInfo
if warn:
log_func = om.MGlobal.displayWarning
elif error:
log_func = om.MGlobal.displayError
log_func( msg )
## ======================================================================
## om2 utilities
def get_mobject( name ):
sel = om2.MGlobal.getSelectionListByName( name )
return sel.getDependNode(0)
## ---------------------------------------------------------------------
def get_dag_path( name ):
sel = om2.MGlobal.getSelectionListByName( name )
return sel.getDagPath(0)
## ---------------------------------------------------------------------
def get_mfn_skin( skin_ob ):
if isinstance( skin_ob, pm.PyNode ):
skin_ob = get_mobject( skin_ob.longName() )
return oma2.MFnSkinCluster( skin_ob )
## ---------------------------------------------------------------------
def get_mfn_mesh( mesh_ob ):
if isinstance( mesh_ob, pm.PyNode ):
mesh_ob = get_mobject( mesh_ob.longName() )
return om2.MFnMesh( mesh_ob )
## ---------------------------------------------------------------------
def get_complete_components( mesh_ob ):
assert( isinstance(mesh_ob, om2.MFnMesh) )
comp = om2.MFnSingleIndexedComponent()
ob = comp.create( om2.MFn.kMeshVertComponent )
comp.setCompleteData( mesh_ob.numVertices )
return( ob )
## ======================================================================
def move_skin( source, target ):
source_shape = get_deform_shape( source )
source_dp = get_dag_path( source_shape.longName() )
source_skin = get_skin_cluster( source )
source_mfn = get_mfn_skin( source_skin )
source_mesh = get_mfn_mesh( get_deform_shape(source) )
components = get_complete_components( source_mesh )
weights, influence_count = source_mfn.getWeights( source_dp, components )
pm.select( cl=True )
target_skin = pm.deformer( target, type='skinCluster', n='MERGED__' + source_skin.name() )[0]
## copy over input values / connections
bind_inputs = [ (x.inputs(plugs=True)[0] if x.isConnected() else None) for x in source_skin.bindPreMatrix ]
bind_values = [ x.get() for x in source_skin.bindPreMatrix ]
mat_inputs = [ (x.inputs(plugs=True)[0] if x.isConnected() else None) for x in source_skin.matrix ]
mat_values = [ x.get() for x in source_skin.matrix ]
for index, bind_value, mat_value in zip( xrange(influence_count), bind_values, mat_values ):
target_skin.bindPreMatrix[index].set( bind_value )
target_skin.matrix[index].set( mat_value )
for index, bind_input, mat_input in zip( xrange(influence_count), bind_inputs, mat_inputs ):
if bind_input:
bind_input >> target_skin.bindPreMatrix[index]
if mat_input:
mat_input >> target_skin.matrix[index]
## copy over weights
target_mfn = get_mfn_skin( target_skin )
target_mesh = get_mfn_mesh( get_deform_shape(target) )
target_dp = get_dag_path( get_deform_shape(target).longName() )
components = get_complete_components( target_mesh )
all_indices = om2.MIntArray( range(influence_count) )
target_mfn.setWeights( target_dp, components, all_indices, weights )
## ======================================================================
## main
items = pm.selected()
if len(items) == 2:
move_skin( items[0], items[1] )
log(
'+ SkinMerge: Complete. Merged skin from {} onto {}.'
.format( *items )
)
else:
log( "-- Please select a skinned mesh and a target mesh.", error=True )