KeyframeMesh/KeyframeMesh.py at main · dbkdavid/KeyframeMesh · GitHub

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
import maya.cmds as cmds

def selectMesh( isChecked ):

	global obj
	global verts

	obj = cmds.ls( selection=True, long=True )[0]
	verts = cmds.polyListComponentConversion( obj, toVertex=True )
	verts = cmds.ls( verts, flatten=True, long=True )

	updateTextLabel( True )

	if not cmds.attributeQuery( 'KeyframeMesh', node=obj, exists=True ):
		cmds.addAttr( obj, ln="KeyframeMesh", at="bool", keyable=True )

def initializeMesh( isChecked ):

	global obj

	cmds.addAttr( obj, ln="KeyframeMesh", at="bool", keyable=True )

def storeIniVertPos( isChecked ):

	global obj
	global vert_pos

	for v in verts:
		pos = cmds.xform( v, query=True, translation=True, worldSpace=False )
		vert_pos.append( pos )

def keyMesh( isChecked ):

	global obj
	global verts
	global vert_pos

	initial_vert_pos = vert_pos

	for i in range( len( verts ) ):

		new_vert_pos = cmds.xform( verts[i], query=True, translation=True, worldSpace=False )

		result_vert_pos = list()

		for j in range( len( new_vert_pos ) ):
			result_vert_pos.append( new_vert_pos[j] - initial_vert_pos[i][j] )

		cmds.setKeyframe( verts[i], attribute="pntx", value=result_vert_pos[0] )
		cmds.setKeyframe( verts[i], attribute="pnty", value=result_vert_pos[1] )
		cmds.setKeyframe( verts[i], attribute="pntz", value=result_vert_pos[2] )

	cmds.setKeyframe( obj + ".KeyframeMesh", value=True )

def selectObjAndVerts( isChecked ):

	global obj
	global verts

	selection = list()

	selection.append( obj )
	for v in verts:
		selection.append( v )

	cmds.select( selection )

def addKeyframeFromMesh ( isChecked ):

	global obj
	global verts
	global vert_pos

	initial_vert_pos = vert_pos

	secondary_obj = cmds.ls( selection=True, long=True )[0]
	secondary_verts = cmds.polyListComponentConversion( secondary_obj, toVertex=True )
	secondary_verts = cmds.ls( secondary_verts, flatten=True, long=True )

	if len( secondary_verts ) == len( verts ):

		for i in range( len( secondary_verts ) ):

			new_vert_pos = cmds.xform( secondary_verts[i], query=True, translation=True, worldSpace=False )

			result_vert_pos = list()

			for j in range( len( new_vert_pos ) ):
				result_vert_pos.append( new_vert_pos[j] - initial_vert_pos[i][j] )

			cmds.setKeyframe( verts[i], attribute="pntx", value=result_vert_pos[0] )
			cmds.setKeyframe( verts[i], attribute="pnty", value=result_vert_pos[1] )
			cmds.setKeyframe( verts[i], attribute="pntz", value=result_vert_pos[2] )

		cmds.setKeyframe( obj + ".KeyframeMesh", value=True )

		cmds.select( obj )
		cmds.currentTime( ( cmds.currentTime( query=True ) ) )

	else:
		print("Error: Second mesh does not match.")

def updateTextLabel( isChecked ):

	try:
		mesh_name = cmds.ls( obj, long=False )[0]
	except:
		mesh_name = "Undefined"

	cmds.text( 'text_sel_mesh', label='Selected Mesh:          ' + mesh_name, edit=True )

def makeUI():

	# UI settings
	win_width = 400
	win_padding = 5
	win_row_spacing = 2
	win_name = 'window_ui'

	# Delete the window ui if it already exists
	if cmds.window( win_name, exists = True ):
		cmds.deleteUI( win_name )

	# Window
	main_window = cmds.window( win_name, title="Keyframe Mesh", iconName='KM', width=win_width, height=10, rtf=True, ret=False )
	main_layout = cmds.columnLayout( adjustableColumn=True, columnAttach=('both', win_padding), columnOffset=('both', 0), rowSpacing=win_row_spacing )
	cmds.text( label='', height=win_padding )

	# Buttons
	cmds.text( 'text_sel_mesh', label='Selected Mesh:          Undefined', align="left")
	cmds.text( label='', height=win_padding )
	cmds.button( label='Select Mesh', command=selectMesh, annotation="Defines the mesh to apply keyframes." )
	#cmds.button( label='Update Label', command=updateTextLabel, annotation="Defines the mesh to apply keyframes." )
	#cmds.button( label='Add Attribute', command=initializeMesh, annotation="Adds a custom KeyframeMesh attribute to the selected mesh transform." )
	cmds.button( label='Store Initial Pose', command=storeIniVertPos, annotation="Stores the initial vertex positions of the mesh. This is necessary for edits made when in sculpt mode." )
	cmds.button( label='Add Keyframe', command=keyMesh, bgc=[0.5, 0.03, 0.03], annotation="Adds a keyframe to all vertices of the selected mesh at the current time." )
	cmds.button( label='Add Keyframe From Second Mesh', bgc=[0.4, 0.2, 0.2], command=addKeyframeFromMesh, annotation="Adds a keyframe to all vertices of the selected mesh at the current time based on the vertex positions of a second selected mesh." )
	cmds.button( label='Edit Keyframes', command=selectObjAndVerts, annotation="Selects all keyframed components for easy editing on the timeline." )

	# Close Button
	cmds.separator( height=( win_padding * 4 ), style='in' )
	cmds.button( label='Close', command=('cmds.deleteUI(\"' + main_window + '\", window=True)') )
	cmds.text( label='', height=win_padding )

	# Show Window
	cmds.showWindow( main_window )

# global variables
obj = None
verts = list()
vert_pos = list()

# make UI
makeUI()