X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=blender%2Fio_mspgl%2Fanimation.py;fp=blender%2Fio_mspgl%2Fanimation.py;h=5ad937df196060b934bd1269718beb6833ac76d4;hb=f2cab83d5704f3f4a91f551f402187637e5063d2;hp=0000000000000000000000000000000000000000;hpb=553f3ec4fbe28a37978ee53b9b6e22fedb691e1d;p=libs%2Fgl.git

diff --git a/blender/io_mspgl/animation.py b/blender/io_mspgl/animation.py
new file mode 100644
index 00000000..5ad937df
--- /dev/null
+++ b/blender/io_mspgl/animation.py
@@ -0,0 +1,88 @@
+import mathutils
+
+class Curve:
+	def __init__(self, curve):
+		self._curve = curve
+		self.knots = []
+
+		for i in range(len(self.keyframe_points)-1):
+			kf1 = self.keyframe_points[i]
+			kf2 = self.keyframe_points[i+1]
+			dx = (kf2.co[0]-kf1.co[0])/3
+			slope1 = (kf1.handle_right[1]-kf1.co[1])/(kf1.handle_right[0]-kf1.co[0])
+			slope2 = (kf2.co[1]-kf2.handle_left[1])/(kf2.co[0]-kf2.handle_left[0])
+
+			if i==0:
+				self.knots.append(mathutils.Vector(kf1.co))
+			self.knots.append(kf1.co+mathutils.Vector((dx, slope1*dx)))
+			self.knots.append(kf2.co-mathutils.Vector((dx, slope2*dx)))
+			self.knots.append(mathutils.Vector(kf2.co))
+
+	def __getattr__(self, attr):
+		return getattr(self._curve, attr)
+
+
+class Keyframe:
+	def __init__(self, time):
+		self.time = time
+		self.control = False
+		self.curves = []
+
+
+class Animation:
+	def __init__(self, action):
+		self._action = action
+		self.curves = [Curve(c) for c in action.fcurves]
+		self.fps = 1
+
+		for c in self.curves:
+			for i in range(0, len(c.knots)-1, 3):
+				p0 = c.knots[i]
+				p1 = c.knots[i+1]
+				p2 = c.knots[i+2]
+				p3 = c.knots[i+3]
+				for j in range(50):
+					t = (p0[0]*(50-j)+p3[0]*j)/50
+					x = (t-p0[0])/(p3[0]-p0[0])
+					v1 = c._curve.evaluate(t)
+					v2 = p0[1]*(1-x)**3+3*p1[1]*x*(1-x)**2+3*p2[1]*x**2*(1-x)+p3[1]*x**3
+
+		keyframes_by_time = {}
+		controls_by_time = {}
+		self.keyframes = []
+		for c in self.curves:
+			for i, k in enumerate(c.knots):
+				x = k[0]
+				control = i%3!=0
+				kf_map = controls_by_time if control else keyframes_by_time
+				if x in kf_map:
+					kf = kf_map[x]
+				else:
+					kf = Keyframe(x)
+					kf.control = control
+					self.keyframes.append(kf)
+					kf_map[x] = kf
+				kf.curves.append((c, i))
+
+		self.keyframes.sort(key=lambda k: k.time)
+
+	def __getattr__(self, attr):
+		return getattr(self._curve, attr)
+
+	def change_fps(self, fps):
+		scale = self.fps/fps
+		self.fps = fps
+
+		for c in self.curves:
+			for k in c.knots:
+				k[0] *= scale
+
+		for k in self.keyframes:
+			k.time *= scale
+
+
+def create_animation_from_action(context, action):
+	anim = Animation(action)
+	render = context.scene.render
+	anim.change_fps(render.fps/render.fps_base)
+	return anim