RE CV: add PRS cache

biorand · Mar 24, 2024 · 54b7add · 54b7add
1 parent 98808ee
commit 54b7add
Showing 1 changed file with 33 additions and 3 deletions.
diff --git a/IntelOrca.Biohazard.BioRand/RECV/ReCvRandomiser.cs b/IntelOrca.Biohazard.BioRand/RECV/ReCvRandomiser.cs
@@ -247,7 +247,7 @@ public override void Generate(RandoConfig config, IRandoProgress progress, FileR
 
                 using (progress.BeginTask(null, "Compressing room files"))
                 {
-                    _rdxAfs = WriteRdxAfs();
+                    _rdxAfs = WriteRdxAfs(fileRepository);
                 }
 
                 using (progress.BeginTask(null, "Creating ISO file"))
@@ -466,7 +466,7 @@ private byte[] ReadFile(UdfEditor editor, FileIdentifier fileId)
             return data;
         }
 
-        private AfsFile WriteRdxAfs()
+        private AfsFile WriteRdxAfs(FileRepository fileRepository)
         {
             if (_rdxAfs == null)
                 throw new Exception();
@@ -482,13 +482,43 @@ private AfsFile WriteRdxAfs()
                 var rrdt = _rdts[i];
                 if (rrdt != null)
                 {
-                    var prs = PrsFile.Compress(rrdt.RdtFile.Data);
+                    var prs = CachePrs(fileRepository, rrdt.RdtId, rrdt.RdtFile.Data);
                     builder.Replace(i, prs.Data);
                 }
             });
             return builder.ToAfsFile();
         }
 
+        private PrsFile CachePrs(FileRepository fileRepository, RdtId rtdId, ReadOnlyMemory<byte> uncompressed)
+        {
+            var cacheDir = Path.Combine(Path.GetDirectoryName(fileRepository.DataPath), "prs_cache");
+            var cacheFile = Path.Combine(cacheDir, rtdId + "_" + uncompressed.CalculateFnv1a().ToString("X8"));
+            if (File.Exists(cacheFile))
+            {
+                try
+                {
+                    return new PrsFile(File.ReadAllBytes(cacheFile));
+                }
+                catch
+                {
+                }
+            }
+
+            var compressedData = PrsFile.Compress(uncompressed);
+            try
+            {
+                if (Directory.Exists(cacheDir))
+                {
+                    Directory.CreateDirectory(cacheDir);
+                    compressedData.Data.WriteToFile(cacheFile);
+                }
+            }
+            catch
+            {
+            }
+            return compressedData;
+        }
+
         protected override string[] GetDefaultNPCs()
         {
             return new[] { "claire", "steve", "chris", "rodrigo", "alfred", "alexia" };