From f56407d64f60354f64066b4e83e3c8023cd2773b Mon Sep 17 00:00:00 2001
From: Yukai Li <yukaili.geek@gmail.com>
Date: Wed, 17 Mar 2021 14:04:06 -0600
Subject: [PATCH] Add mapanim conversion

---
 .gitignore               |  4 +++-
 DgfTxmConvert/Program.cs | 52 ++++++++++++++++++++++++++++++++++++++++
 README.md                |  2 ++
 3 files changed, 57 insertions(+), 1 deletion(-)

diff --git a/.gitignore b/.gitignore
index 4ce6fdd..f9d3d17 100644
--- a/.gitignore
+++ b/.gitignore
@@ -337,4 +337,6 @@ ASALocalRun/
 .localhistory/
 
 # BeatPulse healthcheck temp database
-healthchecksdb
\ No newline at end of file
+healthchecksdb
+
+/DgfTxmConvert/Properties/launchSettings.json
diff --git a/DgfTxmConvert/Program.cs b/DgfTxmConvert/Program.cs
index 5c82052..880a93f 100644
--- a/DgfTxmConvert/Program.cs
+++ b/DgfTxmConvert/Program.cs
@@ -185,6 +185,23 @@ namespace DgfTxmConvert
                 });
             });
 
+            app.Command("convert-mapanim", config =>
+            {
+                config.FullName = "Converts mapanim files";
+                config.Description = "Converts mapanim models.";
+
+                var mapAnimPathArg = config.Argument("mapAnimPathArg", "Path of the mapanim model to extract").IsRequired();
+                mapAnimPathArg.Accepts().ExistingFile();
+                var outBaseArg = config.Argument("outBase", "Directory to write converted files");
+                outBaseArg.Accepts().LegalFilePath();
+                config.HelpOption();
+
+                config.OnExecute(() =>
+                {
+                    ExtractMapAnim(mapAnimPathArg.Value, outBaseArg.Value);
+                });
+            });
+
             app.VersionOptionFromAssemblyAttributes(System.Reflection.Assembly.GetExecutingAssembly());
             app.HelpOption();
 
@@ -668,5 +685,40 @@ namespace DgfTxmConvert
                 converter.ExportTextures(sw, basePath + ".", true);
             }
         }
+
+        static void ExtractMapAnim(string mapAnimPath, string basePath = null)
+        {
+            if (basePath == null)
+                basePath = Path.ChangeExtension(mapAnimPath, null);
+            else
+                basePath = Path.Combine(basePath, Path.GetFileNameWithoutExtension(mapAnimPath));
+
+            using DatReader dat = new DatReader(Utils.CheckDecompress(File.OpenRead(mapAnimPath)));
+            // Second entry is texture DAT
+            using DatReader txmDat = new DatReader(new MemoryStream(dat.GetData(1)));
+            using ObjConverter converter = new ObjConverter(txmDat);
+            string mtlPath = basePath + ".mtl";
+            string mtlName = Path.GetFileName(mtlPath);
+            // First entry is a collection of weird PDBs
+            using DatReader pdbDat = new DatReader(new MemoryStream(dat.GetData(0)));
+            for (int i = 0; i < pdbDat.EntriesCount; ++i)
+            {
+                using MemoryStream ms = new MemoryStream(pdbDat.GetData(i));
+                // This is a DAT, but we're going to pretend it's a normal PDB
+                BinaryReader br = new BinaryReader(ms);
+                Pdb pdb = new Pdb();
+                pdb.Read(br);
+                using StreamWriter sw = File.CreateText($"{basePath}.{i}.obj");
+                sw.WriteLine($"mtllib {mtlName}");
+                sw.WriteLine();
+
+                converter.ConvertObj(pdb, sw);
+            }
+
+            using (StreamWriter sw = File.CreateText(mtlPath))
+            {
+                converter.ExportTextures(sw, basePath + ".");
+            }
+        }
     }
 }
diff --git a/README.md b/README.md
index d29aa4b..6480e89 100644
--- a/README.md
+++ b/README.md
@@ -26,6 +26,8 @@ option.
   corresponding texture DAT needs to be supplied as well.
 - `convert-bg`: Converts sky dome to Wavefront OBJ models. Note only the first
   far texture is extracted.
+- `convert-mapanim`: Converts files from the mapanim folder to Wavefront OBJ
+  models.
 
 Other
 -----