Add lookup function

07th-mod · Aug 24, 2024 · 98eacbd · 98eacbd
1 parent fe131ca
commit 98eacbd
Showing 1 changed file with 90 additions and 1 deletion.
diff --git a/MOD.Scripts.AssetManager/MODImageMapping.cs b/MOD.Scripts.AssetManager/MODImageMapping.cs
@@ -4,6 +4,7 @@
 using System.IO;
 using System.Linq;
 using System.Text;
+using BGICompiler.Compiler.Logger;
 
 namespace MOD.Scripts.AssetManager
 {
@@ -24,12 +25,17 @@ private class ModToOGMapping
 		{
 			// A dictionary mapping mod paths to a OG paths
 			// eg.modToOG could be { "background/hina_bus_03": "bg/hina/bus_03", "sprite/hyos1a_09_": "sprites/yos/yos_d3" }
-			public Dictionary<string, string> modToOG;
+			Dictionary<string, string> modToOG;
 
 			public ModToOGMapping(Dictionary<string, string> modToOG)
 			{
 				this.modToOG = modToOG;
 			}
+
+			public bool GetOGImage(string modImagePath, out string ogImagePath)
+			{
+				return modToOG.TryGetValue(modImagePath, out ogImagePath);
+			}
 		}
 
 		private class ScriptFallback
@@ -42,6 +48,17 @@ public ScriptFallback(Dictionary<string, Dictionary<string, string>> scriptFallb
 			{
 				scriptToFallback = scriptFallbackJSON.ToDictionary(kvp => kvp.Key, kvp => new ModToOGMapping(kvp.Value));
 			}
+
+			public bool GetModToOGMapping(string scriptFileName, string modImagePath, out string ogImagePath)
+			{
+				if(scriptToFallback.TryGetValue(scriptFileName, out ModToOGMapping modToOGMapping))
+				{
+					return modToOGMapping.GetOGImage(modImagePath, out ogImagePath);
+				}
+
+				ogImagePath = null;
+				return false;
+			}
 		}
 
 		private class VoiceMapping
@@ -54,6 +71,16 @@ public VoiceMapping(Dictionary<string, Dictionary<string, string>> voiceToMappin
 			{
 				voiceToMapping = voiceToMappingJSON.ToDictionary(kvp => kvp.Key, kvp => new ModToOGMapping(kvp.Value));
 			}
+			public bool GetModToOGMapping(string lastPlayedVoice, string modImagePath, out string ogImagePath)
+			{
+				if(voiceToMapping.TryGetValue(lastPlayedVoice, out ModToOGMapping mapping))
+				{
+					return mapping.GetOGImage(modImagePath, out ogImagePath);
+				}
+
+				ogImagePath = null;
+				return false;
+			}
 		}
 
 		private class VoiceDatabase
@@ -66,6 +93,16 @@ public VoiceDatabase(Dictionary<string, Dictionary<string, Dictionary<string, st
 			{
 				scriptToVoiceMapping = scriptToVoiceMappingJSON.ToDictionary(kvp => kvp.Key, kvp => new VoiceMapping(kvp.Value));
 			}
+			public bool GetVoiceMapping(string scriptFileName, string lastPlayedVoice, string modImagePath, out string ogImagePath)
+			{
+				if(scriptToVoiceMapping.TryGetValue(scriptFileName, out VoiceMapping mapping))
+				{
+					return mapping.GetModToOGMapping(lastPlayedVoice, modImagePath, out ogImagePath);
+				}
+
+				ogImagePath = null;
+				return false;
+			}
 		}
 
 		public class ImageMapping
@@ -90,6 +127,58 @@ public static ImageMapping GetVoiceBasedMapping(string path)
 				return new ImageMapping(DeserializeFromJSONFile<ImageMappingFromJSON>(path));
 			}
 
+			public bool GetOGImage(string scriptFileName, string lastPlayedVoice, string modImagePath, out string ogImagePath, out string debug_info)
+			{
+				// Since JSON doesn't support 'null' keys, lastPlayedVoice to the empty string ("") before continuing
+				if(lastPlayedVoice == null)
+				{
+					lastPlayedVoice = "";
+				}
+
+				// 1) Try to get from voice database
+				if (voiceDatabase.GetVoiceMapping(scriptFileName, lastPlayedVoice, modImagePath, out ogImagePath))
+				{
+					debug_info = "Source: Voice Mapping";
+					return true;
+				}
+
+				// 2) Try script fallback
+				if (scriptFallback.GetModToOGMapping(scriptFileName, modImagePath, out ogImagePath))
+				{
+					debug_info = "Source: Script Fallback";
+					return true;
+				}
+
+				// 3) Try global fallback
+				if(globalFallback.GetOGImage(modImagePath,out ogImagePath))
+				{
+					debug_info = "Source: Global Fallback";
+					return true;
+				}
+
+				debug_info = "No match found";
+				ogImagePath = null;
+				return false;
+			}
+
+			public static void TEST_Lookup(string JSONPath)
+			{
+				ImageMapping mapping = GetVoiceBasedMapping(JSONPath);
+
+				string scriptName = "busstop01";
+				string lastVoice = null;
+				string modImage = "background/hina_bus_01";
+				Debug.Log($"Looking up {scriptName} - {(lastVoice == null ? "[null]" : lastVoice)} - {modImage}");
+				if (mapping.GetOGImage(scriptName, lastVoice, modImage, out string ogImagePath, out string debug))
+				{
+					Debug.Log($"Got match {ogImagePath} - Debug: [{debug}]");
+				}
+				else
+				{
+					Debug.Log($"Match failed - Debug: [{debug}]");
+				}
+			}
+
 			ImageMapping(ImageMappingFromJSON mappingJSON)
 			{
 				globalFallback = new ModToOGMapping(mappingJSON.global_fallback);