From f8d91df9543ee387df206d882820fb8cb0bbb4e3 Mon Sep 17 00:00:00 2001
From: HaoYizhi <haoyizhi@agentlab.cn>
Date: Sun, 26 Feb 2023 21:58:05 +0800
Subject: [PATCH] model + firm

---
 __pycache__/firm.cpython-38.pyc | Bin 726 -> 1016 bytes
 firm.py                         |   5 ++++
 model.py                        |  48 +++++++++++++++++++++++---------
 3 files changed, 40 insertions(+), 13 deletions(-)

diff --git a/__pycache__/firm.cpython-38.pyc b/__pycache__/firm.cpython-38.pyc
index 9b6f664427ad03f8aeede32ccae33d42b3ff4233..cb39303bfda47991b36d5c889200d24b3f01bc14 100644
GIT binary patch
delta 348
zcmZvXze>bF5XSdU_RJlZD2U*JjaEj)XHW#e<{GuRW)0cBm`Ji=cOxi~E3~k=;z;9b
zSokUy`y^tc6Ym<qZ<udj=HmzR!9Lf*s~{L4&};COFuDk@@y<eR4TblNCZ1pyDC)hv
zyyICyy3|S(CWAhOHv3==%2AGU@;hmRMo3^NGN+HkdJTEP7zqS0I=+mE^KRw6ah`~6
z95L+yz_C=v)s5?`+HxsUX<WZjCbO9;jbqhJXQhp(4k0@UwJg+~<ds&UPOQULufu+6
z8y#(LmFP&^DxTEFszPd>P5JU9HX&z*U<mtofCwVrJ@RmPmg(Xom8CuUgV_F;F^*V!
UOb+m<oswO@`@_e6*v`rM4^nzup8x;=

delta 59
zcmeytevOqcl$V!_0SM-Q`k8FOG?6cXQDkDiC`&MdCd<YNTN#=CG+8D`Giy!mVwPoN
N0;=O+oV=dd69C(?4x#`6

diff --git a/firm.py b/firm.py
index 8c933bf..0225ed1 100644
--- a/firm.py
+++ b/firm.py
@@ -15,3 +15,8 @@ class FirmAgent(ap.Agent):
 
         self.dct_product_is_disrupted = dict.fromkeys(list_product, False)
         self.dct_product_is_removed = dict.fromkeys(list_product, False)
+
+    def remove_edge_to_customer_if_removed(self, remove_product):
+        t = self.firm_network.graph.out_edges(
+            self.firm_network.positions[self], keys=True, data=True)
+        print(t)
diff --git a/model.py b/model.py
index ff8ca56..9523a06 100644
--- a/model.py
+++ b/model.py
@@ -86,8 +86,10 @@ class Model(ap.Model):
 
         self.firm_network = ap.Network(self, G_Firm)
         # print([node.label for node in self.firm_network.nodes])
-        # print([list(self.firm_network.graph.predecessors(node)) for node in self.firm_network.nodes])
-        # print([self.firm_network.graph.nodes[node.label]['Name'] for node in self.firm_network.nodes])
+        # print([list(self.firm_network.graph.predecessors(node))
+        #        for node in self.firm_network.nodes])
+        # print([self.firm_network.graph.nodes[node.label]['Name']
+        #        for node in self.firm_network.nodes])
         # print([v for v in self.firm_network.graph.nodes(data=True)])
 
         # init firm
@@ -103,10 +105,9 @@ class Model(ap.Model):
                 capacity=self.firm_network.graph.out_degree(ag_node))
             self.firm_network.add_agents([firm_agent], [ag_node])
         self.a_list_total_firms = ap.AgentList(self, self.firm_network.agents)
-        # print(
-        #     list(
-        #         zip(self.a_list_total_firms.code, self.a_list_total_firms.name,
-        #             self.a_list_total_firms.capacity)))
+        # print(list(zip(self.a_list_total_firms.code,
+        #                self.a_list_total_firms.name,
+        #                self.a_list_total_firms.capacity)))
 
         # set the initial firm product that are removed
         for firm_code, list_product in self.dct_list_remove_firm_prod.items():
@@ -118,17 +119,36 @@ class Model(ap.Model):
                 firm.dct_product_is_removed[product] = True
 
     def update(self):
-        # Update list of unhappy people
-        # self.agents.update_happiness()
-        # self.unhappy = self.agents.select(self.agents.happy == False)
+        # Update the firm that is removed
+        self.dct_list_remove_firm_prod = {}
+        for firm in self.a_list_total_firms:
+            for product, flag in firm.dct_product_is_removed.items():
+                if flag is True:
+                    if firm.code in self.dct_list_remove_firm_prod.keys():
+                        self.dct_list_remove_firm_prod[firm.code].append(
+                            product)
+                    else:
+                        self.dct_list_remove_firm_prod[firm.code] = [product]
+        # print(self.dct_list_remove_firm_prod)
 
         # Stop simulation if reached terminal number of iteration
-        if self.t == self.int_n_iter:
+        if self.t == self.int_n_iter or len(
+                self.dct_list_remove_firm_prod) == 0:
             self.stop()
 
     def step(self):
-        # Move unhappy people to new location
-        self.unhappy.find_new_home()
+        # shuffle self.dct_list_remove_firm_prod
+        dct_key_list = list(self.dct_list_remove_firm_prod.keys())
+        self.nprandom.shuffle(dct_key_list)
+        self.dct_list_remove_firm_prod = {
+            key: self.dct_list_remove_firm_prod[key]
+            for key in dct_key_list
+        }
+        for firm_code, list_product in self.dct_list_remove_firm_prod.items():
+            firm = self.a_list_total_firms.select(
+                self.a_list_total_firms.code == firm_code)[0]
+            for product in list_product:
+                firm.remove_edge_to_customer_if_removed(product)
 
     def end(self):
         pass
@@ -169,4 +189,6 @@ class Model(ap.Model):
 
 model = Model(dct_sample_para)
 model.setup()
-model.draw_network()
+model.update()
+model.step()
+# model.draw_network()