From 44461ff9557eee11e060e5feacf6ba6fb25f07a7 Mon Sep 17 00:00:00 2001
From: Janis <janis.meister87@gmail.com>
Date: Tue, 7 Jun 2022 19:10:55 +0200
Subject: [PATCH] io

---
 Assets/Scripts/ProductionObjects/IOPort.cs | 28 ++++++++++++++++++----
 1 file changed, 24 insertions(+), 4 deletions(-)

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