(61d00a474) v0.9.7.1

2020-03-04 13:04:10 +01:00
parent 3c50efa5c9
commit 3c09ebe02f
5086 changed files with 786063 additions and 295871 deletions
@@ -0,0 +1,174 @@
+using Barotrauma.Items.Components;
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using Barotrauma.Extensions;
+
+namespace Barotrauma
+{
+    static class AutoItemPlacer
+    {
+        private static readonly List<Item> spawnedItems = new List<Item>();
+
+        public static bool OutputDebugInfo = false;
+
+        public static void PlaceIfNeeded(GameMode gameMode)
+        {
+            if (GameMain.NetworkMember != null && !GameMain.NetworkMember.IsServer) { return; }
+            
+            CampaignMode campaign = gameMode as CampaignMode;
+            if (campaign == null || !campaign.InitialSuppliesSpawned)
+            {
+                for (int i = 0; i < Submarine.MainSubs.Length; i++)
+                {
+                    if (Submarine.MainSubs[i] == null) { continue; }
+                    List<Submarine> subs = new List<Submarine>() { Submarine.MainSubs[i] };
+                    subs.AddRange(Submarine.MainSubs[i].DockedTo.Where(d => !d.IsOutpost));
+                    Place(subs);
+                }
+                if (campaign != null) { campaign.InitialSuppliesSpawned = true; }
+            }            
+        }
+
+        private static void Place(IEnumerable<Submarine> subs)
+        {
+            if (GameMain.NetworkMember != null && GameMain.NetworkMember.IsClient)
+            {
+                DebugConsole.ThrowError("Clients are not allowed to use AutoItemPlacer.\n" + Environment.StackTrace);
+                return;
+            }
+
+            int sizeApprox = MapEntityPrefab.List.Count() / 3;
+            var containers = new List<ItemContainer>(100);
+            var prefabsWithContainer = new List<ItemPrefab>(sizeApprox / 3);
+            var prefabsWithoutContainer = new List<ItemPrefab>(sizeApprox);
+            var removals = new List<ItemPrefab>();
+
+            foreach (Item item in Item.ItemList)
+            {
+                if (!subs.Contains(item.Submarine)) { continue; }
+                containers.AddRange(item.GetComponents<ItemContainer>());
+            }
+
+            foreach (MapEntityPrefab prefab in MapEntityPrefab.List)
+            {
+                if (!(prefab is ItemPrefab ip)) { continue; }
+
+                if (ip.ConfigElement.Elements().Any(e => string.Equals(e.Name.ToString(), typeof(ItemContainer).Name.ToString(), StringComparison.OrdinalIgnoreCase)))
+                {
+                    prefabsWithContainer.Add(ip);
+                }
+                else
+                {
+                    prefabsWithoutContainer.Add(ip);
+                }
+            }
+
+            spawnedItems.Clear();
+            var validContainers = new Dictionary<ItemContainer, PreferredContainer>();
+            prefabsWithContainer.Shuffle();
+            // Spawn items that have an ItemContainer component first so we can fill them up with items if needed (oxygen tanks inside the spawned diving masks, etc)
+            for (int i = 0; i < prefabsWithContainer.Count; i++)
+            {
+                var itemPrefab = prefabsWithContainer[i];
+                if (itemPrefab == null) { continue; }
+                if (SpawnItems(itemPrefab))
+                {
+                    removals.Add(itemPrefab);
+                }
+            }
+            // Remove containers that we successfully spawned items into so that they are not counted in in the second pass.
+            removals.ForEach(i => prefabsWithContainer.Remove(i));
+            // Another pass for items with containers because also they can spawn inside other items (like smg magazine)
+            prefabsWithContainer.ForEach(i => SpawnItems(i));
+            // Spawn items that don't have containers last
+            prefabsWithoutContainer.Shuffle();
+            prefabsWithoutContainer.ForEach(i => SpawnItems(i));
+
+            if (OutputDebugInfo)
+            {
+                DebugConsole.NewMessage("Automatically placed items: ");
+                foreach (string itemName in spawnedItems.Select(it => it.Name).Distinct())
+                {
+                    DebugConsole.NewMessage(" - " + itemName + " x" + spawnedItems.Count(it => it.Name == itemName));
+                }
+            }
+
+            bool SpawnItems(ItemPrefab itemPrefab)
+            {
+                if (itemPrefab == null)
+                {
+                    string errorMsg = "Error in AutoItemPlacer.SpawnItems - itemPrefab was null.\n"+Environment.StackTrace;
+                    DebugConsole.ThrowError(errorMsg);
+                    GameAnalyticsManager.AddErrorEventOnce("AutoItemPlacer.SpawnItems:ItemNull", GameAnalyticsSDK.Net.EGAErrorSeverity.Error, errorMsg);
+                    return false;
+                }
+                bool success = false;
+                foreach (PreferredContainer preferredContainer in itemPrefab.PreferredContainers)
+                {
+                    if (preferredContainer.SpawnProbability <= 0.0f || preferredContainer.MaxAmount <= 0) { continue; }
+                    validContainers = GetValidContainers(preferredContainer, containers, validContainers, primary: true);
+                    if (validContainers.None())
+                    {
+                        validContainers = GetValidContainers(preferredContainer, containers, validContainers, primary: false);
+                    }
+                    foreach (var validContainer in validContainers)
+                    {
+                        if (SpawnItem(itemPrefab, containers, validContainer))
+                        {
+                            success = true;
+                        }
+                    }
+                }
+                return success;
+            }
+        }
+
+        private static Dictionary<ItemContainer, PreferredContainer> GetValidContainers(PreferredContainer preferredContainer, IEnumerable<ItemContainer> allContainers, Dictionary<ItemContainer, PreferredContainer> validContainers, bool primary)
+        {
+            validContainers.Clear();
+            foreach (ItemContainer container in allContainers)
+            {
+                if (!container.AutoFill) { continue; }
+                if (primary)
+                {
+                    if (!ItemPrefab.IsContainerPreferred(preferredContainer.Primary, container)) { continue; }
+                }
+                else
+                {
+                    if (!ItemPrefab.IsContainerPreferred(preferredContainer.Secondary, container)) { continue; }
+                }
+                if (!validContainers.ContainsKey(container))
+                {
+                    validContainers.Add(container, preferredContainer);
+                }
+            }
+            return validContainers;
+        }
+
+        private static bool SpawnItem(ItemPrefab itemPrefab, List<ItemContainer> containers, KeyValuePair<ItemContainer, PreferredContainer> validContainer)
+        {
+            bool success = false;
+            if (Rand.Value() > validContainer.Value.SpawnProbability) { return success; }
+            int amount = Rand.Range(validContainer.Value.MinAmount, validContainer.Value.MaxAmount + 1);
+            for (int i = 0; i < amount; i++)
+            {
+                if (validContainer.Key.Inventory.IsFull())
+                {
+                    containers.Remove(validContainer.Key);
+                    break;
+                }
+
+                var item = new Item(itemPrefab, validContainer.Key.Item.Position, validContainer.Key.Item.Submarine);
+                spawnedItems.Add(item);
+#if SERVER
+                Entity.Spawner.CreateNetworkEvent(item, remove: false);
+#endif
+                validContainer.Key.Inventory.TryPutItem(item, null);
+                containers.AddRange(item.GetComponents<ItemContainer>());
+                success = true;
+            }
+            return success;
+        }
+    }
+}