Faction Test v1.0.1.0

2023-02-16 15:01:28 +02:00
parent caa5a2f762
commit 2c5a7923b0
309 changed files with 7502 additions and 4335 deletions
@@ -1,13 +1,134 @@
-using System.Collections.Generic;
+using System;
+using System.Collections.Generic;
+using System.Xml.Linq;
+using System.Linq;
+using Barotrauma.IO;
+using XmlWriterSettings = System.Xml.XmlWriterSettings;
+#nullable enable

 namespace Barotrauma
 {
-    public class CreatureMetrics
+    public static class CreatureMetrics
    {
-        public readonly HashSet<Identifier> RecentlyEncountered = new HashSet<Identifier>();
-        public readonly HashSet<Identifier> Encountered = new HashSet<Identifier>();
-        public readonly HashSet<Identifier> Killed = new HashSet<Identifier>();
+        private const string path = "creature_metrics.xml";

-        public readonly static CreatureMetrics Instance = new CreatureMetrics();
+        /// <summary>
+        /// Resets every round.
+        /// </summary>
+        public static HashSet<Identifier> RecentlyEncountered { get; private set; } = new HashSet<Identifier>();
+        public static HashSet<Identifier> Encountered { get; private set; } = new HashSet<Identifier>();
+        public static HashSet<Identifier> Unlocked { get; private set; } = new HashSet<Identifier>();
+        public static HashSet<Identifier> Killed { get; private set; } = new HashSet<Identifier>();
+        public static bool IsInitialized { get; private set; }
+        public static bool UnlockAll { get; set; }
+
+        public static void Init()
+        {
+            IsInitialized = true;
+            if (File.Exists(path))
+            {
+                Load();
+            }
+            Save();
+        }
+
+        private static void Load()
+        {
+            XDocument doc = XMLExtensions.TryLoadXml(path);
+            XElement? root = doc?.Root;
+            if (root == null)
+            {
+                DebugConsole.AddWarning($"Failed to load creature metrics from {path}!");
+                return;
+            }
+            UnlockAll = root.GetAttributeBool(nameof(UnlockAll), UnlockAll);
+            Unlocked = new HashSet<Identifier>(root.GetAttributeIdentifierArray(nameof(Unlocked), Array.Empty<Identifier>()));
+            Encountered = new HashSet<Identifier>(root.GetAttributeIdentifierArray(nameof(Encountered), Array.Empty<Identifier>()));
+            Killed = new HashSet<Identifier>(root.GetAttributeIdentifierArray(nameof(Killed), Array.Empty<Identifier>()));
+            SyncSets();
+        }
+
+        public static void Save()
+        {
+            if (!IsInitialized)
+            {
+                throw new Exception("Creature Metrics not yet initialized!");
+            }
+            SyncSets();
+            XDocument configDoc = new XDocument();
+            XElement root = new XElement("CreatureMetrics");
+            configDoc.Add(root);
+            root.SetAttributeValue(nameof(UnlockAll), UnlockAll);
+            root.SetAttributeValue(nameof(Unlocked), string.Join(",", Unlocked).Trim().ToLowerInvariant());
+            root.SetAttributeValue(nameof(Encountered), string.Join(",", Encountered).Trim().ToLowerInvariant());
+            root.SetAttributeValue(nameof(Killed), string.Join(",", Killed).Trim().ToLowerInvariant());
+            configDoc.SaveSafe(path);
+            XmlWriterSettings settings = new XmlWriterSettings
+            {
+                Indent = true,
+                OmitXmlDeclaration = true,
+                NewLineOnAttributes = true
+            };
+            try
+            {
+                using var writer = XmlWriter.Create(path, settings);
+                configDoc.WriteTo(writer);
+                writer.Flush();
+            }
+            catch (Exception e)
+            {
+                DebugConsole.ThrowError("Saving creature metrics failed.", e);
+                GameAnalyticsManager.AddErrorEventOnce("CreatureMetrics.Save:SaveFailed", GameAnalyticsManager.ErrorSeverity.Error,
+                    "Saving creature metrics failed.\n" + e.Message + "\n" + e.StackTrace.CleanupStackTrace());
+            }
+        }
+
+        public static void RecordKill(Identifier species)
+        {
+            AddEncounter(species);
+            if (!Killed.Contains(species))
+            {
+                Killed.Add(species);
+            }
+        }
+
+        public static void AddEncounter(Identifier species)
+        {
+            if (species == CharacterPrefab.HumanSpeciesName) { return; }
+            if (Encountered.Contains(species)) { return; }
+            Encountered.Add(species);
+            RecentlyEncountered.Add(species);
+            UnlockInEditor(species);
+        }
+
+        private static IEnumerable<CharacterFile>? vanillaCharacters;
+        public static void UnlockInEditor(Identifier species)
+        {
+            if (species == CharacterPrefab.HumanSpeciesName) { return; }
+            if (Unlocked.Contains(species)) { return; }
+            vanillaCharacters ??= GameMain.VanillaContent.GetFiles<CharacterFile>();
+            var contentFile = CharacterPrefab.FindBySpeciesName(species);
+            if (contentFile == null) { return; }
+            if (!vanillaCharacters.Contains(contentFile.ContentFile))
+            {
+                // Don't try to unlock custom characters. They are always unlocked.
+                return;
+            }
+            Unlocked.Add(species);
+        }
+
+        private static void SyncSets()
+        {
+            // Ensure that all killed are also encountered and both unlocked.
+            // Otherwise we could permanently hide some creatures by manually adding them to the encountered or by removing from unlocked in the xml file.
+            foreach (var species in Killed)
+            {
+                Encountered.Add(species);
+            }
+            foreach (var species in Encountered)
+            {
+                Unlocked.Add(species);
+            }
+        }
    }
 }