From 45a58a1eee3b669b83bd2a862dfb5151db675f0b Mon Sep 17 00:00:00 2001
From: Benjamin Collins <kion@dashgl.com>
Date: Fri, 21 Aug 2020 07:57:55 +0900
Subject: [PATCH] read key frames

---
 NinjaModel.py  | 39 ++++++++++++++++++++++++---
 NinjaMotion.py | 71 ++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 107 insertions(+), 3 deletions(-)
 create mode 100644 NinjaMotion.py

diff --git a/NinjaModel.py b/NinjaModel.py
index e3bca1b..1a75095 100644
--- a/NinjaModel.py
+++ b/NinjaModel.py
@@ -32,8 +32,9 @@ from NinjaMaterial import NinjaMaterial
 from NinjaVertex import NinjaVertex
 from NinjaFace import NinjaFace
 from NinjaBone import NinjaBone
+from NinjaMotion import NinjaMotion
 
-class NinjaModel: # {
+class NinjaModel:
 
 	def __init__(self, model_name, tex_name):
 
@@ -436,6 +437,12 @@ class NinjaModel: # {
 		spline_interpolation = p[3] & 0x40
 		user_function_interpolation = p[3] & 0x80
 
+		anim = NinjaMotion()
+		anim.setName("anim_%03d" % len(self.anim_list))
+		anim.setFrames(frame_count)
+		anim.setSkeleton(self.bones)
+		self.anim_list.append(anim)
+
 		print("READING ANIMTION!!!!")
 		print(motion_type)
 
@@ -491,9 +498,35 @@ class NinjaModel: # {
 			entry['scl_num'] = p[0]
 			
 			anim_vals.append(entry)
-			print(entry)
 
-				
+		c = 2 * 3.141592 / 0x10000;
+		for entry in anim_vals:
+			
+			boneIndex = entry['bone_index']
+
+			self.file.seek(entry['pos_ofs'], 0)
+			for i in range(entry['pos_num']):
+				bytes = self.file.read(16)
+				p = struct.unpack('Ifff', bytes)
+				frame = p[0]
+				vec3 = [ p[1], p[2], p[3] ]
+				anim.appendKeyFrame(boneIndex, frame, vec3, 'pos')
+
+			self.file.seek(entry['rot_ofs'], 0)
+			for i in range(entry['rot_num']):
+				bytes = self.file.read(16)
+				p = struct.unpack('Ifff', bytes)
+				frame = p[0]
+				vec3 = [ p[1]*c, p[2]*c, p[3]*c ]
+				anim.appendKeyFrame(boneIndex, frame, vec3, 'rot')
+
+			self.file.seek(entry['scl_ofs'], 0)
+			for i in range(entry['scl_num']):
+				bytes = self.file.read(16)
+				p = struct.unpack('Ifff', bytes)
+				frame = p[0]
+				vec3 = [ p[1], p[2], p[3] ]
+				anim.appendKeyFrame(boneIndex, frame, vec3, 'scl')
 
 		return None
 
diff --git a/NinjaMotion.py b/NinjaMotion.py
new file mode 100644
index 0000000..5c29afe
--- /dev/null
+++ b/NinjaMotion.py
@@ -0,0 +1,71 @@
+"""
+
+ MIT License
+
+ Copyright (c) 2020 Benjamin Collins (kion @ dashgl.com)
+
+ Permission is hereby granted, free of charge, to any person obtaining a copy
+ of this software and associated documentation files (the "Software"), to deal
+ in the Software without restriction, including without limitation the rights
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ copies of the Software, and to permit persons to whom the Software is
+ furnished to do so, subject to the following conditions:
+
+ The above copyright notice and this permission notice shall be included in all
+ copies or substantial portions of the Software.
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+ SOFTWARE.
+
+"""
+
+import math
+import struct
+from Mat4 import Mat4 
+
+class NinjaMotion:
+
+	def __init__(self):
+		self.name = "";
+		self.skeleton = []
+		self.frames = 0
+		self.fps = 30
+		self.keyFrames = []
+		return None
+
+	def setName(self, name):
+		self.name = name
+		return None
+	
+	def setSkeleton(self, bones):
+		self.seleton = bones
+		for i in range(len(bones)):
+			self.keyFrames.append([])
+		return None
+
+	def setFrames(self, frames):
+		self.frames = frames
+		return None
+
+	def appendKeyFrame(self, boneIndex, frame, vec3, type):
+		keyFrames = self.keyFrames[boneIndex]
+
+		for keys in keyFrames:
+			if keys['frame'] != frame:
+				continue
+			keys[type] = vec3
+			return None
+		
+		keyFrame = {
+			'frame' : frame,
+			type : vec3
+		}
+
+		keyFrames.append(keyFrame)
+		return None
+		
+