From e251ddb17dcfd097237ce9a6deb855ca782fdee3 Mon Sep 17 00:00:00 2001
From: Misha <106913236+MishaProductions@users.noreply.github.com>
Date: Sun, 14 Apr 2024 14:25:39 -0400
Subject: [PATCH] implement LegoAnimNodeData::FindKeys (#802)

* implement LegoAnimNodeData::FindKeys

* Update legoanim.cpp

* fix flickering

* Match

* Fix cond

* clang

---------

Co-authored-by: Christian Semmler <mail@csemmler.com>
---
 LEGO1/lego/sources/anim/legoanim.cpp | 47 ++++++++++++++++++++++++++--
 LEGO1/lego/sources/anim/legoanim.h   |  1 +
 2 files changed, 45 insertions(+), 3 deletions(-)

diff --git a/LEGO1/lego/sources/anim/legoanim.cpp b/LEGO1/lego/sources/anim/legoanim.cpp
index 1a0a5e9a..202f0236 100644
--- a/LEGO1/lego/sources/anim/legoanim.cpp
+++ b/LEGO1/lego/sources/anim/legoanim.cpp
@@ -635,7 +635,7 @@ LegoBool LegoAnimNodeData::FUN_100a0990(LegoFloat p_time)
 	return result;
 }
 
-// STUB: LEGO1 0x100a0a00
+// FUNCTION: LEGO1 0x100a0a00
 LegoU32 LegoAnimNodeData::FindKeys(
 	LegoFloat p_time,
 	LegoU32 p_numKeys,
@@ -645,8 +645,44 @@ LegoU32 LegoAnimNodeData::FindKeys(
 	LegoU32& p_old_index
 )
 {
-	// TODO
-	return 0;
+	LegoU32 numKeys;
+	if (p_numKeys == 0) {
+		numKeys = 0;
+	}
+	else if (p_time < GetKey(0, p_keys, p_size).GetTime()) {
+		numKeys = 0;
+	}
+	else if (p_time > GetKey(p_numKeys - 1, p_keys, p_size).GetTime()) {
+		p_new_index = p_numKeys - 1;
+		numKeys = 1;
+	}
+	else {
+		if (GetKey(p_old_index, p_keys, p_size).GetTime() <= p_time) {
+			for (p_new_index = p_old_index;
+				 p_new_index < p_numKeys - 1 && p_time >= GetKey(p_new_index + 1, p_keys, p_size).GetTime();
+				 p_new_index++) {
+			}
+		}
+		else {
+			for (p_new_index = 0;
+				 p_new_index < p_numKeys - 1 && p_time >= GetKey(p_new_index + 1, p_keys, p_size).GetTime();
+				 p_new_index++) {
+			}
+		}
+
+		p_old_index = p_new_index;
+		if (p_time == GetKey(p_new_index, p_keys, p_size).GetTime()) {
+			numKeys = 1;
+		}
+		else if (p_new_index < p_numKeys - 1) {
+			numKeys = 2;
+		}
+		else {
+			numKeys = 0;
+		}
+	}
+
+	return numKeys;
 }
 
 // FUNCTION: LEGO1 0x100a0b00
@@ -661,6 +697,11 @@ inline LegoFloat LegoAnimNodeData::Interpolate(
 	return p_value1 + (p_value2 - p_value1) * (p_time - p_key1.GetTime()) / (p_key2.GetTime() - p_key1.GetTime());
 }
 
+inline LegoAnimKey& LegoAnimNodeData::GetKey(LegoU32 p_i, LegoAnimKey* p_keys, LegoU32 p_size)
+{
+	return *((LegoAnimKey*) (((LegoU8*) p_keys) + (p_i * p_size)));
+}
+
 // FUNCTION: LEGO1 0x100a0b30
 LegoAnim::LegoAnim()
 {
diff --git a/LEGO1/lego/sources/anim/legoanim.h b/LEGO1/lego/sources/anim/legoanim.h
index 3527747c..7ba9cc04 100644
--- a/LEGO1/lego/sources/anim/legoanim.h
+++ b/LEGO1/lego/sources/anim/legoanim.h
@@ -165,6 +165,7 @@ class LegoAnimNodeData : public LegoTreeNodeData {
 		LegoAnimKey& p_key2,
 		LegoFloat p_value2
 	);
+	inline static LegoAnimKey& GetKey(LegoU32 p_i, LegoAnimKey* p_keys, LegoU32 p_size);
 
 	static LegoU32 FindKeys(
 		LegoFloat p_time,