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
|
# 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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
import theme
import kinematic
class KeyFrame:
def __init__(self):
pass
def empty(self):
return self.joints == None
def assign(self, x):
self.middle = x.middle
self.parts = x.parts.copy()
self.sticks = x.sticks.copy()
self.joints = x.joints.copy()
def clear(self):
self.parts = None
self.sticks = None
self.scaled_sticks = None
self.joints = None
self.scaled_joints = None
class StoredKeyFrame(KeyFrame):
def __init__(self):
KeyFrame.__init__(self)
self.clear()
def assign(self, x):
KeyFrame.assign(self, x)
self.scaled_sticks = x.sticks.copy()
self.scaled_joints = _initjoints()
_scalesticks(self.scaled_sticks, .2)
_setjoints(self.scaled_joints, self.scaled_sticks,
(self.x, int(theme.KEYFRAMEHEIGHT/2.0)))
class CurrentKeyFrame(KeyFrame):
def __init__(self):
KeyFrame.__init__(self)
self.parts = theme.PARTS.copy()
self.sticks = theme.STICKS.copy()
self.joints = _initjoints()
self.middle = (int(theme.DRAWWIDTH/2.0), int(theme.DRAWHEIGHT/2.0))
def assign(self, x):
KeyFrame.assign(self, x)
_setjoints(self.joints, self.sticks, self.middle)
keys = []
for i in range(5):
key = StoredKeyFrame()
keyframe_width = theme.KEYFRAMEWIDTH/5
key.x = keyframe_width/2 + i*keyframe_width
keys.append(key)
def _scalesticks(stickdict, i):
for key in stickdict:
(angle,len) = stickdict[key]
newlen = int(len * i)
stickdict[key] = (angle,newlen)
def _setjoints(joints, sticks, middle):
# have to traverse in order because
# parent joints must be set right
for stickname in theme.STICKLIST:
(angle,len) = sticks[stickname]
jname = theme.JOINTS[stickname]
(x,y) = kinematic.getparentjoint(jname, joints, middle)
parents = kinematic.getparentsticks(stickname)
panglesum = 0
for parentname in parents:
(pangle,plen) = sticks[parentname]
panglesum += pangle
(nx,ny) = kinematic.getpoints(x,y,angle+panglesum,len)
joints[jname] = (nx,ny)
def _initjoints():
joints = {}
for stickname in theme.JOINTS:
jname = theme.JOINTS[stickname]
joints[jname] = (0,0)
return joints
|
