Improve CLI code dumping

- No longer exports sub-function code entries
- Improved performance of UMT_DUMP_ALL, and parallelized it

Author: colinator27

UndertaleModCli/Program.cs

@@ -12,8 +12,10 @@
 using System.Text;
 using System.Threading;
 using System.Threading.Tasks;
+using Underanalyzer.Decompiler;
 using UndertaleModLib;
 using UndertaleModLib.Compiler;
+using UndertaleModLib.Decompiler;
 using UndertaleModLib.Models;
 using UndertaleModLib.Project;
 using UndertaleModLib.Scripting;
@@ -499,15 +501,30 @@ private static int Dump(DumpOptions options)
         // If user provided code to dump, dump code
         if ((options.Code?.Length > 0) && (program.Data.Code?.Count > 0))
         {
+            GlobalDecompileContext globalDecompileContext = new(program.Data);
+            IDecompileSettings decompilerSettings = program.Data.ToolInfo.DecompilerSettings;
+
             // If user wanted to dump everything, do that, otherwise only dump what user provided
-            string[] codeArray;
             if (options.Code.Contains(UMT_DUMP_ALL))
-                codeArray = program.Data.Code.Select(c => c.Name.Content).ToArray();
+            {
+                int totalCores = Environment.ProcessorCount;
+                int outerLimit = Math.Max(1, totalCores / 4);
+                Parallel.ForEach(program.Data.Code, new ParallelOptions { MaxDegreeOfParallelism = outerLimit }, code =>
+                {
+                    if (code.ParentEntry is not null)
+                    {
+                        return;
+                    }
+                    program.DumpCodeEntry(code, globalDecompileContext, decompilerSettings);
+                });
+            }
             else
-                codeArray = options.Code;
-
-            foreach (string code in codeArray)
-                program.DumpCodeEntry(code);
+            {
+                foreach (string code in options.Code)
+                {
+                    program.DumpCodeEntry(code, globalDecompileContext, decompilerSettings);
+                }
+            }
         }
 
         // If user wanted to dump strings, dump all of them in a text file
@@ -817,32 +834,44 @@ private void CliQuickInfo()
     /// <summary>
     /// Dumps a code entry from a data file.
     /// </summary>
-    /// <param name="codeEntry">The code entry that should get dumped</param>
-    private void DumpCodeEntry(string codeEntry)
+    /// <param name="codeEntryName">The code entry name that should get dumped</param>
+    /// <param name="context">Decompile context to use when dumping</param>
+    /// <param name="settings">Settings to use for the decompiler</param>
+    private void DumpCodeEntry(string codeEntryName, GlobalDecompileContext context, IDecompileSettings settings)
     {
-        UndertaleCode code = Data.Code.ByName(codeEntry);
-
+        UndertaleCode code = Data.Code.ByName(codeEntryName);
 
         if (code == null)
         {
-            Console.Error.WriteLine($"Data file does not contain a code entry named {codeEntry}!");
+            Console.Error.WriteLine($"Data file does not contain a code entry named {codeEntryName}!");
             return;
         }
 
+        DumpCodeEntry(code, context, settings);
+    }
+
+    /// <summary>
+    /// Dumps a code entry from a data file.
+    /// </summary>
+    /// <param name="code">The code entry that should get dumped</param>
+    /// <param name="context">Decompile context to use when dumping</param>
+    /// <param name="settings">Settings to use for the decompiler</param>
+    private void DumpCodeEntry(UndertaleCode code, GlobalDecompileContext context, IDecompileSettings settings)
+    {
         string directory = Path.Join(Output.FullName, "CodeEntries");
 
         Directory.CreateDirectory(directory);
 
         if (Verbose)
-            Console.WriteLine($"Dumping {codeEntry}");
+            Console.WriteLine($"Dumping {code.Name?.Content}");
 
-        string dest = Paths.TryJoinVerifyWithinDirectory(directory, $"{codeEntry}.gml");
+        string dest = Paths.TryJoinVerifyWithinDirectory(directory, $"{code.Name?.Content}.gml");
         if (dest is null)
         {
-            Console.Error.WriteLine($"Failed to export code entry with name {codeEntry}");
+            Console.Error.WriteLine($"Failed to export code entry with name {code.Name?.Content}");
             return;
         }
-        File.WriteAllText(dest, GetDecompiledText(code));
+        File.WriteAllText(dest, GetDecompiledText(code, context, settings));
     }
 
     /// <summary>