From b89a4d6154b11fbbdd16ca0e4c8d4657ba9fe383 Mon Sep 17 00:00:00 2001
From: Jarrod Doyle <thejarroddoyle@gmail.com>
Date: Thu, 5 Sep 2024 17:32:29 +0100
Subject: [PATCH] Refactor model viewer to use new Context and support custom
 models

---
 project/code/TMV/Model.cs             | 98 ++-------------------------
 project/code/TMV/UI/ModelSelector.cs  | 73 ++++++++++++--------
 project/scenes/model_viewer.tscn      | 10 +--
 project/scenes/ui/model_selector.tscn |  6 ++
 4 files changed, 60 insertions(+), 127 deletions(-)

diff --git a/project/code/TMV/Model.cs b/project/code/TMV/Model.cs
index 1b4894c..7df45e0 100644
--- a/project/code/TMV/Model.cs
+++ b/project/code/TMV/Model.cs
@@ -1,7 +1,4 @@
-using System.Collections.Generic;
-using System.IO;
 using Godot;
-using KeepersCompound.LGS;
 using KeepersCompound.TMV.UI;
 
 namespace KeepersCompound.TMV;
@@ -11,104 +8,17 @@ public partial class Model : Node3D
     public override void _Ready()
     {
         var modelSelector = GetNode<Control>("%ModelSelector") as ModelSelector;
-        modelSelector.LoadModel += BuildModel;
+        modelSelector.ModelSelected += BuildModel;
     }
 
-    public void BuildModel(string rootPath, string modelPath)
+    public void BuildModel(string campaignName, string modelPath)
     {
         foreach (var node in GetChildren())
         {
             node.QueueFree();
         }
 
-        var modelFile = new ModelFile(modelPath);
-        if (modelFile == null)
-        {
-            GD.Print($"Failed to load model file: {modelPath}");
-            return;
-        }
-
-        // TODO: Remove this disgusting hack. Not only is it a hack, it doesn't support custom models
-        var baseDir = Path.GetDirectoryName(modelPath);
-        var options = new EnumerationOptions
-        {
-            MatchCasing = MatchCasing.CaseInsensitive,
-            RecurseSubdirectories = true,
-        };
-        var materials = new List<StandardMaterial3D>();
-        foreach (var material in modelFile.Materials)
-        {
-            if (material.Type == 0)
-            {
-                var paths = Directory.GetFiles(baseDir, material.Name, options);
-                if (paths.IsEmpty()) continue;
-
-                materials.Add(new StandardMaterial3D
-                {
-                    AlbedoTexture = TextureLoader.LoadTexture(paths[0])
-                });
-            }
-            else
-            {
-                var b = (material.Handle) & 0xff;
-                var g = (material.Handle >> 8) & 0xff;
-                var r = (material.Handle >> 16) & 0xff;
-                var colour = new Color(r / 255.0f, g / 255.0f, b / 255.0f, 1.0f);
-                materials.Add(new StandardMaterial3D
-                {
-                    AlbedoColor = colour
-                });
-            }
-        }
-
-        var surfaceDataMap = new Dictionary<int, MeshSurfaceData>();
-        foreach (var poly in modelFile.Polygons)
-        {
-            var vertices = new List<Vector3>();
-            var normal = modelFile.Normals[poly.Normal].ToGodotVec3();
-            var uvs = new List<Vector2>();
-            for (var i = 0; i < poly.VertexCount; i++)
-            {
-                var vertex = modelFile.Vertices[poly.VertexIndices[i]];
-                vertices.Add(vertex.ToGodotVec3());
-                if (i < poly.UvIndices.Length)
-                {
-                    var uv = modelFile.Uvs[poly.UvIndices[i]];
-                    uvs.Add(new Vector2(uv.X, uv.Y));
-                }
-                else
-                {
-                    uvs.Add(Vector2.Zero);
-                }
-            }
-
-            if (!surfaceDataMap.ContainsKey(poly.Data))
-            {
-                surfaceDataMap.Add(poly.Data, new MeshSurfaceData());
-            }
-
-            surfaceDataMap[poly.Data].AddPolygon(vertices, normal, uvs, uvs);
-        }
-
-        var mesh = new ArrayMesh();
-        foreach (var (materialId, surfaceData) in surfaceDataMap)
-        {
-            var array = surfaceData.BuildSurfaceArray();
-            var surfaceIdx = mesh.GetSurfaceCount();
-            mesh.AddSurfaceFromArrays(Mesh.PrimitiveType.Triangles, array);
-            for (var i = 0; i < materials.Count; i++)
-            {
-                var m = modelFile.Materials[i];
-                if (m.Slot == materialId)
-                {
-                    mesh.SurfaceSetMaterial(surfaceIdx, materials[i]);
-                    break;
-                }
-            }
-        }
-
-        var pos = -modelFile.Header.Center.ToGodotVec3();
-        var meshInstance = new MeshInstance3D { Mesh = mesh, Position = pos };
-        AddChild(meshInstance);
+        var model = Context.Instance.ModelLoader.Load(campaignName, modelPath);
+        AddChild(model);
     }
 }
diff --git a/project/code/TMV/UI/ModelSelector.cs b/project/code/TMV/UI/ModelSelector.cs
index b8bdaba..f713faf 100644
--- a/project/code/TMV/UI/ModelSelector.cs
+++ b/project/code/TMV/UI/ModelSelector.cs
@@ -6,15 +6,15 @@ using Godot;
 namespace KeepersCompound.TMV.UI;
 
 public partial class ModelSelector : Control
-{
-	[Signal]
-	public delegate void LoadModelEventHandler(string rootPath, string modelPath);
 
-	private InstallPaths _installPaths;
+{
+	public event ModelSelectedEventHandler ModelSelected;
+	public delegate void ModelSelectedEventHandler(string campaign, string mission);
 
 	private FileDialog _FolderSelect;
 	private LineEdit _FolderPath;
 	private Button _BrowseButton;
+	private ItemList _Campaigns;
 	private ItemList _Models;
 	private Button _LoadButton;
 	private Button _CancelButton;
@@ -28,13 +28,15 @@ public partial class ModelSelector : Control
 		_FolderSelect = GetNode<FileDialog>("%FolderSelect");
 		_FolderPath = GetNode<LineEdit>("%FolderPath");
 		_BrowseButton = GetNode<Button>("%BrowseButton");
+		_Campaigns = GetNode<ItemList>("%Campaigns");
 		_Models = GetNode<ItemList>("%Models");
 		_LoadButton = GetNode<Button>("%LoadButton");
 		_CancelButton = GetNode<Button>("%CancelButton");
 
 		_BrowseButton.Pressed += () => _FolderSelect.Visible = true;
-		_FolderSelect.DirSelected += (string dir) => { _FolderPath.Text = dir; BuildModelList(dir); };
-		_FolderPath.TextSubmitted += BuildModelList;
+		_FolderSelect.DirSelected += SetInstallPath;
+		_FolderPath.TextSubmitted += SetInstallPath;
+		_Campaigns.ItemSelected += BuildModelList;
 		_Models.ItemSelected += (long _) => _LoadButton.Disabled = false;
 		_LoadButton.Pressed += EmitLoadModel;
 		_CancelButton.Pressed += () => Visible = false;
@@ -51,44 +53,57 @@ public partial class ModelSelector : Control
 		}
 	}
 
-	private void BuildModelList(string path)
+	private void SetInstallPath(string path)
 	{
-		_installPaths = new InstallPaths(path);
-		ExtractObjFiles();
-
-		_Models.Clear();
-		_LoadButton.Disabled = true;
-
-		var paths = Directory.GetFiles(_extractedObjectsPath, "*.bin", SearchOption.AllDirectories);
-		foreach (var m in paths.OrderBy(s => s))
+		_FolderPath.Text = path;
+		if (Context.Instance.PathManager.Init(path))
 		{
-			_Models.AddItem(m.TrimPrefix(_extractedObjectsPath));
+			BuildCampaignList();
 		}
 	}
 
-	// TODO: Move this to a resource manager
-	private void ExtractObjFiles()
+	private void BuildCampaignList()
 	{
-		var dir = new DirectoryInfo(_extractedObjectsPath);
-		if (dir.Exists)
-		{
-			dir.Delete(true);
-		}
+		_Campaigns.Clear();
+		_Models.Clear();
+		_LoadButton.Disabled = true;
 
-		var zip = ZipFile.OpenRead(_installPaths.objPath);
-		zip.ExtractToDirectory(_extractedObjectsPath);
+		var pathManager = Context.Instance.PathManager;
+		_Campaigns.AddItem("Original Missions");
+		foreach (var campaign in pathManager.GetCampaignNames())
+		{
+			_Campaigns.AddItem(campaign);
+		}
+	}
+
+	private void BuildModelList(long idx)
+	{
+		_Models.Clear();
+		_LoadButton.Disabled = true;
+
+		var pathManager = Context.Instance.PathManager;
+		var campaignName = idx == 0 ? null : _Campaigns.GetItemText((int)idx);
+		var modelNames = pathManager.GetModelNames(campaignName);
+		foreach (var model in modelNames)
+		{
+			_Models.AddItem(model);
+		}
 	}
 
 	private void EmitLoadModel()
 	{
-		var selected = _Models.GetSelectedItems();
-		if (selected.IsEmpty())
+		var campaignIdxs = _Campaigns.GetSelectedItems();
+		var modelIdxs = _Models.GetSelectedItems();
+		if (campaignIdxs.IsEmpty() || modelIdxs.IsEmpty())
 		{
 			return;
 		}
 
-		var path = _extractedObjectsPath + _Models.GetItemText(selected[0]);
-		EmitSignal(SignalName.LoadModel, _installPaths.rootPath, path);
+		var campaignIdx = campaignIdxs[0];
+		var modelIdx = modelIdxs[0];
+		var campaignName = campaignIdx == 0 ? null : _Campaigns.GetItemText(campaignIdx);
+		var modelName = _Models.GetItemText(modelIdx);
+		ModelSelected(campaignName, modelName);
 
 		Visible = false;
 	}
diff --git a/project/scenes/model_viewer.tscn b/project/scenes/model_viewer.tscn
index da3c757..830298b 100644
--- a/project/scenes/model_viewer.tscn
+++ b/project/scenes/model_viewer.tscn
@@ -15,13 +15,15 @@ ssao_enabled = true
 transform = Transform3D(1, 0, 0, 0, 1, 0, 0, 0, 1, 0.110309, 0.187101, -0.461656)
 script = ExtResource("1_dax7s")
 
-[node name="Camera3D" type="Camera3D" parent="."]
-script = ExtResource("2_ov7rc")
-
 [node name="UI" type="CanvasLayer" parent="."]
 
 [node name="ModelSelector" parent="UI" instance=ExtResource("3_ovrmo")]
 unique_name_in_owner = true
 
-[node name="WorldEnvironment" type="WorldEnvironment" parent="."]
+[node name="Environment" type="Node3D" parent="."]
+
+[node name="WorldEnvironment" type="WorldEnvironment" parent="Environment"]
 environment = SubResource("Environment_e4172")
+
+[node name="Camera3D" type="Camera3D" parent="Environment"]
+script = ExtResource("2_ov7rc")
diff --git a/project/scenes/ui/model_selector.tscn b/project/scenes/ui/model_selector.tscn
index 5e623dc..8a83c22 100644
--- a/project/scenes/ui/model_selector.tscn
+++ b/project/scenes/ui/model_selector.tscn
@@ -77,6 +77,12 @@ text = "Browse"
 [node name="HBoxContainer" type="HBoxContainer" parent="PanelContainer/MarginContainer/VBoxContainer"]
 layout_mode = 2
 
+[node name="Campaigns" type="ItemList" parent="PanelContainer/MarginContainer/VBoxContainer/HBoxContainer"]
+unique_name_in_owner = true
+custom_minimum_size = Vector2(0, 480)
+layout_mode = 2
+size_flags_horizontal = 3
+
 [node name="Models" type="ItemList" parent="PanelContainer/MarginContainer/VBoxContainer/HBoxContainer"]
 unique_name_in_owner = true
 custom_minimum_size = Vector2(0, 480)