diff --git a/Assets/Scripts/ProductionObjects/IOPort.cs b/Assets/Scripts/ProductionObjects/IOPort.cs index 2700216..176dbaf 100644 --- a/Assets/Scripts/ProductionObjects/IOPort.cs +++ b/Assets/Scripts/ProductionObjects/IOPort.cs @@ -8,15 +8,35 @@ public class IOPort : MonoBehaviour public int itemCount = 0; private int maxItemCount = 100; - public void IncreaseItemCount(int value) + public int IncreaseItemCount(int value) // Returns rest of value { + int restOfValue = 0; + if (itemCount + value > maxItemCount) + { + restOfValue = value - (maxItemCount - itemCount); + itemCount = maxItemCount; + } + else + { + itemCount += value; + } + return restOfValue; } - public void DecreaseItemCount(int value) + public int DecreaseItemCount(int value) // Returns rest of value { + int restOfValue = 0; + if (itemCount - value < 0) + { + restOfValue = value - itemCount; + itemCount = 0; + } + else + { + itemCount -= value; + } + return restOfValue; } - - } \ No newline at end of file