Merge pull request #132 from MichaelRelaxen/dl-missions

Deadlocked missions data files support

Author: MilchRatchet

LibReplanetizer/Headers/MissionHeader.cs

@@ -50,6 +50,21 @@ public static List<string> FindMissionFiles(GameType game, string enginePath)
 
             return files;
         }
+        public static List<string> FindMissionDataFiles(GameType game, string enginePath)
+        {
+            string? dir = Path.GetDirectoryName(enginePath);
+
+            List<string> files = new List<string>();
+
+            if (dir == null) return files;
+
+            if (game == GameType.DL)
+            {
+                files.AddRange(Directory.GetFiles(dir, "gameplay_mission_data*.dat"));
+            }
+
+            return files;
+        }
 
         public byte[] Serialize()
         {

LibReplanetizer/Level.cs

@@ -14,6 +14,7 @@
 using System.Collections.Generic;
 using System.IO;
 using LibReplanetizer.Serializers;
+using System.Linq;
 
 namespace LibReplanetizer
 {
@@ -348,6 +349,7 @@ public Level(string enginePath)
             }
 
             List<string> missionPaths = MissionHeader.FindMissionFiles(game, enginePath);
+            List<string> missionDataPaths = MissionHeader.FindMissionDataFiles(game, enginePath);
             missions = new List<Mission>();
 
             for (int i = 0; i < missionPaths.Count; i++)
@@ -376,6 +378,15 @@ public Level(string enginePath)
                     parser.GetTextures(mission.textures);
                 }
 
+                string? datPath = missionDataPaths.FirstOrDefault(p =>
+                    Path.GetFileNameWithoutExtension(p).Contains($"[{i}]"));
+
+                if (datPath != null)
+                {
+                    using var datParser = new MissionDataParser(datPath, game);
+                    mission.mobies = datParser.GetMobies(mission.models, mobyModels);
+                }
+
                 missions.Add(mission);
             }
 

LibReplanetizer/Parsers/MissionParser.cs

@@ -64,4 +64,57 @@ public void Dispose()
             fileStream.Close();
         }
     }
+
+    public class MissionDataParser : IDisposable
+    {
+        private FileStream fileStream;
+        private GameType game;
+
+        public MissionDataParser(string datFilePath, GameType game)
+        {
+            this.game = game;
+            this.fileStream = File.OpenRead(datFilePath);
+        }
+
+        public List<Moby> GetMobies(List<Model> missionModels, List<Model> levelModels)
+        {
+            byte[] fileHeader = ReadBlock(fileStream, 0x00, 0x20);
+            int mobyInstancesOffset = ReadInt(fileHeader, 0x04);
+
+            byte[] mobySecHeader = ReadBlock(fileStream, mobyInstancesOffset, 0x10);
+            int mobyCount = ReadInt(mobySecHeader, 0x00);
+
+            if (mobyCount <= 0 || mobyCount > 10000)
+                return new List<Moby>();
+
+            int mobyDataStart = mobyInstancesOffset + 0x10;
+            byte[] mobyBlock = ReadBlock(fileStream, mobyDataStart, mobyCount * 0x70);
+
+            List<Model> allModels = new List<Model>(missionModels);
+            foreach (var m in levelModels)
+                if (!allModels.Exists(x => x.id == m.id))
+                    allModels.Add(m);
+
+            var mobies = new List<Moby>();
+            for (int i = 0; i < mobyCount; i++)
+            {
+                int offset = i * 0x70;
+
+                // Null so the constructor doesnt try to index into an empty pvar list
+                mobyBlock[offset + 0x50] = 0xFF;
+                mobyBlock[offset + 0x51] = 0xFF;
+                mobyBlock[offset + 0x52] = 0xFF;
+                mobyBlock[offset + 0x53] = 0xFF;
+                Moby moby = new Moby(GameType.DL, mobyBlock, i, allModels, new List<byte[]>());
+                mobies.Add(moby);
+            }
+
+            return mobies;
+        }
+
+        public void Dispose()
+        {
+            fileStream?.Close();
+        }
+    }
 }

Replanetizer/Frames/LevelFrame.cs

@@ -36,6 +36,7 @@ public class LevelFrame : Frame
         private FramebufferRenderer? renderer;
         public LevelRenderer? levelRenderer;
         private RendererPayload rendererPayload;
+        private HashSet<Mission> selectedMissions = new HashSet<Mission>();
         public Level level { get; set; }
         private bool enableCameraInfo = true;
         public ShaderTable shaderTable;
@@ -336,6 +337,24 @@ private void RenderMenuBar()
                     ImGui.EndMenu();
                 }
 
+                if (level.game == GameType.DL && level.missions.Count > 0 && ImGui.BeginMenu("Missions"))
+                {
+                    foreach (var mission in level.missions)
+                    {
+                        bool selected = selectedMissions.Contains(mission);
+                        if (ImGui.Checkbox($"Mission {mission.missionID}", ref selected))
+                        {
+                            if (selected)
+                                selectedMissions.Add(mission);
+                            else
+                                selectedMissions.Remove(mission);
+
+                            RefreshMissionMobies();
+                        }
+                    }
+                    ImGui.EndMenu();
+                }
+
                 ImGui.Separator();
 
                 if (interactiveSession)
@@ -611,7 +630,7 @@ private void LoadLevel(Level level)
                 camera.SetRotation(0, 0);
             }
             selectedObjects.Clear();
-
+            activeMobies = level.mobs.ToList();
             InvalidateView();
         }
 
@@ -1012,8 +1031,17 @@ protected void OnResize()
                     return level.shrubs.Find(x => x.globalID == hitId);
                 case RenderedObjectType.Tie:
                     return level.ties.Find(x => x.globalID == hitId);
+
                 case RenderedObjectType.Moby:
-                    return level.mobs.Find(x => x.globalID == hitId);
+                    Moby? foundMoby = level.mobs.Find(x => x.globalID == hitId);
+                    if (foundMoby != null) return foundMoby;
+                    foreach (var mission in selectedMissions)
+                    {
+                        Moby? missionMoby = mission.mobies.Find(x => x.globalID == hitId);
+                        if (missionMoby != null) return missionMoby;
+                    }
+                    return null;
+
                 case RenderedObjectType.Spline:
                     // Both "Splines" and "GrindPaths" use Spline objects.
                     // Search for a hit in the Splines first then look
@@ -1102,6 +1130,36 @@ private void ExportListOfProperties<T>(string path, IEnumerable<T> list, Func<T,
             }
         }
 
+        private List<Moby> activeMobies = new List<Moby>();
+        private void RefreshMissionMobies()
+        {
+            if (levelRenderer == null) return;
+
+            foreach (var moby in activeMobies.ToList())
+                levelRenderer.Remove(moby, level);
+
+            var mobies = new List<Moby>(level.mobs);
+            foreach (var mission in selectedMissions)
+                mobies.AddRange(mission.mobies);
+
+            foreach (var moby in mobies)
+            {
+                Mission? sourceMission = selectedMissions.FirstOrDefault(m => m.mobies.Contains(moby));
+                if (sourceMission != null)
+                {
+                    bool isMissionModel = sourceMission.models.Exists(m => m.id == moby.modelID);
+                    levelRenderer.Include(moby, isMissionModel ? sourceMission.textures : level.textures);
+                }
+                else
+                {
+                    levelRenderer.Include(moby);
+                }
+            }
+
+            activeMobies = mobies;
+            InvalidateView();
+        }
+
         protected void OnPaint()
         {
             rendererPayload.visibility.hideCameraMoby = interactiveSession && hookLiveUpdate && hookUpdateCamera;

Replanetizer/Frames/ModelFrame.cs

@@ -695,6 +695,18 @@ private void PrepareForArrowInput()
 
         private void SelectModel(Model? model)
         {
+            if (model == null)
+                return;
+
+            foreach (var mission in level.missions)
+            {
+                if (mission.models.Contains(model))
+                {
+                    SelectModel(model, mission.textures);
+                    return;
+                }
+            }
+
             SelectModel(model, level.textures);
         }
 

Replanetizer/Renderer/LevelRenderer.cs

@@ -183,14 +183,14 @@ private void Add(Tie tie)
             tieRenderers.Add(tieRenderer);
         }
 
-        private void Add(Moby mob)
+        private void Add(Moby mob, List<Texture>? textureOverride = null)
         {
             if (textures == null)
             {
                 throw new NullReferenceException();
             }
 
-            MeshRenderer mobRenderer = new MeshRenderer(shaderTable, textures, textureIDs);
+            MeshRenderer mobRenderer = new MeshRenderer(shaderTable, textureOverride ?? textures, textureIDs);
             mobRenderer.Include(mob);
             mobyRenderers.Add(mobRenderer);
         }
@@ -215,6 +215,10 @@ public override void Include<T>(T obj)
 
             throw new NotImplementedException();
         }
+        public void Include(Moby mob, List<Texture>? textureOverride = null)
+        {
+            Add(mob, textureOverride);
+        }
         public override void Include<T>(List<T> list) => throw new NotImplementedException();
 
         public void Remove<T>(T obj, Level level)