diff --git a/Barotrauma/BarotraumaShared/Source/GameSession/CargoManager.cs b/Barotrauma/BarotraumaShared/Source/GameSession/CargoManager.cs index 24b1c9c03..31aa45682 100644 --- a/Barotrauma/BarotraumaShared/Source/GameSession/CargoManager.cs +++ b/Barotrauma/BarotraumaShared/Source/GameSession/CargoManager.cs @@ -110,6 +110,7 @@ namespace Barotrauma } Dictionary availableContainers = new Dictionary(); + ItemPrefab containerPrefab = null; foreach (PurchasedItem Pi in itemsToSpawn) { Vector2 position = new Vector2( @@ -125,7 +126,7 @@ namespace Barotrauma if (itemContainer == null) { - var containerPrefab = MapEntityPrefab.List.Find(ep => + containerPrefab = MapEntityPrefab.List.Find(ep => ep.NameMatches(Pi.itemPrefab.CargoContainerName) || (ep.Tags != null && ep.Tags.Contains(Pi.itemPrefab.CargoContainerName.ToLowerInvariant()))) as ItemPrefab; @@ -151,6 +152,13 @@ namespace Barotrauma } for (int i = 0; i < Pi.quantity; i++) { + if (!availableContainers.ContainsKey(itemContainer)) + { + Item containerItemOverFlow = new Item(containerPrefab, position, wp.Submarine); + itemContainer = containerItemOverFlow.GetComponent(); + availableContainers.Add(itemContainer, itemContainer.Capacity); + } + if (itemContainer == null) { //no container, place at the waypoint @@ -177,8 +185,12 @@ namespace Barotrauma } //reduce the number of available slots in the container - availableContainers[itemContainer]--; - if (availableContainers[itemContainer] <= 0) + //if there is a container + if (availableContainers.ContainsKey(itemContainer)) + { + availableContainers[itemContainer]--; + } + if (availableContainers.ContainsKey(itemContainer) && availableContainers[itemContainer] <= 0) { availableContainers.Remove(itemContainer); }