something something i broke it meh

econ/business/Price_Believe_Business.py

@@ -2,29 +2,32 @@ from .base_business import Business
 from ..agents.price_believe_aquire import Price_Believe_Aquire_Agent
 from ..agents.price_believe_distribute import Price_Believe_Distribiute_Agent
 from ..agents.autoproduction import AutoProductionAgent
+import log
 
 
 class Price_Believe_Business(Business):
     def __init__(self, id, production, balance, exchange, simulation, location) -> None:
         super().__init__(id, production, balance, location)
 
+        self.max_storage = 10
         self.expense_per_unit = -1
         self.income_per_unit = -1
         self.distribute = Price_Believe_Distribiute_Agent(
-            simulation, self, production["name"], exchange, 0.2, 10)
+            simulation, self, production["name"], exchange, 0.2, 20)
         self.craft = AutoProductionAgent(simulation, self)
         self.aquire = {}
-      
+
         for k, v in production["prod"].items():
             a = Price_Believe_Aquire_Agent(
                 simulation, self, k, exchange, 0.2, 10)
             a.set_target(v*2)
             a.set_price_max(10)
             self.aquire[k] = a
-        self.distribute.set_price_min(10)
+        self.distribute.set_price_min(0)
         self.distribute.set_target(0)
 
     def tick_business_decisions(self, step):
+        amount= self.production["amount"]
         if step == 0:
 
             max_amount = self.production["amount"]*5
@@ -33,30 +36,39 @@ class Price_Believe_Business(Business):
         self.update_ex_per_unit()
         self.update_income_per_unit(step)
 
-        # set target for aquire
+        # calc descision
         orderForNewProds = 1
-        prod = self.production["prod"]
-        
-        for k, v in prod.items():
+        if self.income_per_unit<=0 or self.expense_per_unit<=0:
+            # dont have data
+            retain=0
+        else:
+          ie = (self.income_per_unit/self.expense_per_unit)-1
+          if ie < 0:
+              ie = 0
+          if ie > 1:
+              ie = 1
+          #retain=(self.max_storage-ie*self.max_storage)*amount
+          retain=0
 
+
+        # set target for aquire
+
+        prod = self.production["prod"]
+
+        for k, v in prod.items():
             self.aquire[k].set_target(v*orderForNewProds)
 
         # update production
-        targetUnit = self.production["amount"]*10
+        targetUnit = self.production["amount"]*self.max_storage
         self.craft.set_target(targetUnit)
 
         # set min distribute
         self.distribute.set_price_min(self.expense_per_unit)
-        self.distribute.set_target(0)
+        self.distribute.set_target(retain)
 
     def tick_business_episode(self, episode_count):
         start_balance = self.balance_history[-1]
         diff = self.balance-start_balance
-        if diff > 0:
-            # we have some profit -> increase production staff
-            self.craft.set_worker(self.craft.worker+1)
-        else:
-            self.craft.set_worker(self.craft.worker-1)
 
     def resource_in_possesion(self):
         return self.distribute.open_qty+self.inventory[self.production["name"]]
@@ -70,7 +82,7 @@ class Price_Believe_Business(Business):
             agent.update_trades()
             if agent.qty == 0:
                 return
-            if agent.qty<0:
+            if agent.qty < 0:
                 print("pla")
             req = prod[key]
 
@@ -89,7 +101,7 @@ class Price_Believe_Business(Business):
             for key, agent in self.aquire.items():
                 qtyToConfirm = prod[key]*qtyNewProds
                 agent.confirm_purchase(qtyToConfirm)
-                a=1+2
+                a = 1+2
 
         qtyNewUnits = qtyNewProds*amount
 
@@ -105,16 +117,31 @@ class Price_Believe_Business(Business):
         new_ex = ex_total/estimatedQTYUnits
         self.expense_per_unit = new_ex
 
-    def update_income_per_unit(self,step):
+    def update_income_per_unit(self, step):
         qty_sold = self.distribute.qty
         income = self.distribute.income
-        if income==0:
+        if income == 0:
             return
         self.income_per_unit = qty_sold/income
-        self.distribute.confirm_distribution(qty_sold,step)
+        self.distribute.confirm_distribution(qty_sold, step)
 
     def close_business(self):
         for k, a in self.aquire.items():
             a.unregister()
         self.distribute.unregister()
         self.craft.unregister()
+
+    def log(self, episode, episode_length, step):
+        data = {}
+        data["id"] = self.id
+        data["episode"] = episode
+        data["step"] = step
+        data["tstep"] = step+episode*episode_length
+        data["production"] = self.production["name"]
+        data["expense"] = self.expense_per_unit
+        data["income"] = self.income_per_unit
+        data["inventory"] = self.inventory[self.production["name"]]
+        data["retain"] = self.distribute.target
+        data["balance"] = self.balance
+        data["location"] = self.location
+        log.BUSINESSData.append(data)

econ/business/base_business.py

@@ -52,15 +52,3 @@ class Business(ABC):
     def close_business(self):
         assert "close_business has not been provided"
         
-    def log(self,episode,episode_length,step):
-        data={}
-        data["id"]=self.id
-        data["episode"]=episode
-        data["step"]=step
-        data["tstep"]=step+episode*episode_length
-        data["production"]=self.production["name"]
-        data["prod_inventory"]=self.inventory[self.production["name"]]
-        data["raw_inventory"]=self.inventory[self.production["name"]]
-        data["balance"]=self.balance
-        data["location"]=self.location
-        log.BUSINESSData.append(data)