HaoYizhi/IIabm
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

remove firm that denied request from candidate list

Browse Source
This commit is contained in:
HaoYizhi
2023-05-15 13:37:05 +08:00
parent aac2607284
commit 7e8980d0c6
15 changed files with 82 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

79
firm.py
View File

@@ -18,7 +18,8 @@ class FirmAgent(ap.Agent):
self.a_lst_product_disrupted = ap.AgentList(self.model, [])
self.a_lst_product_removed = ap.AgentList(self.model, [])
self.dct_n_trial_up_product_removed = {}
self.dct_n_trial_up_prod_removed = {}
self.dct_cand_alt_supply_up_prod_removed = {}
self.dct_request_prod_from_firm = {}
def remove_edge_to_cus_remove_cus_up_prod(self, remove_product):
@@ -50,32 +51,39 @@ class FirmAgent(ap.Agent):
customer.a_lst_up_product_removed:
customer.a_lst_up_product_removed.append(
remove_product)
customer.dct_n_trial_up_product_removed[
customer.dct_n_trial_up_prod_removed[
remove_product] = 0
def seek_alt_supply(self):
for product in self.a_lst_up_product_removed:
# print(f"{self.name} seek alt supply for {product.code}")
if self.dct_n_trial_up_product_removed[
print(f"{self.name} seek alt supply for {product.code}")
if self.dct_n_trial_up_prod_removed[
product] <= self.model.int_n_max_trial:
# select a list of candidate firm that has the product
candidate_alt_supply = self.model.a_lst_total_firms.select([
product in firm.a_lst_product
and product not in firm.a_lst_product_removed
for firm in self.model.a_lst_total_firms
])
if not candidate_alt_supply:
if self.dct_n_trial_up_prod_removed[product] == 0:
# select a list of candidate firm that has the product
self.dct_cand_alt_supply_up_prod_removed[product] = \
self.model.a_lst_total_firms.select([
product in firm.a_lst_product
and product not in firm.a_lst_product_removed
for firm in self.model.a_lst_total_firms
])
if not self.dct_cand_alt_supply_up_prod_removed[product]:
continue
# select based on size
lst_prob = [
size / sum(candidate_alt_supply.revenue_log)
for size in candidate_alt_supply.revenue_log
size /
sum(self.dct_cand_alt_supply_up_prod_removed[
product].revenue_log)
for size in self.dct_cand_alt_supply_up_prod_removed[
product].revenue_log
]
select_alt_supply = self.model.nprandom.choice(
candidate_alt_supply, p=lst_prob)
# print(
# f"{self.name} selct alt supply for {product.code} from {select_alt_supply.name}"
# )
self.dct_cand_alt_supply_up_prod_removed[product],
p=lst_prob)
print(
f"{self.name} selct alt supply for {product.code} "
f"from {select_alt_supply.name}"
)
assert product in select_alt_supply.a_lst_product, \
f"{select_alt_supply} \
does not produce requested product {product}"
@@ -95,10 +103,10 @@ class FirmAgent(ap.Agent):
# select_alt_supply.dct_request_prod_from_firm.items()
# })
self.dct_n_trial_up_product_removed[product] += 1
self.dct_n_trial_up_prod_removed[product] += 1
def handle_request(self):
# print(self.name, 'handle_request')
print(self.name, 'handle_request')
for product, lst_firm in self.dct_request_prod_from_firm.items():
if self.dct_prod_capacity[product] > 0:
if len(lst_firm) == 0:
@@ -114,8 +122,12 @@ class FirmAgent(ap.Agent):
in_edges = self.model.firm_network.graph.in_edges(
self.model.firm_network.positions[firm], keys=True)
lst_adj_firm = []
lst_adj_firm += [ap.AgentIter(self.model, edge[1]).to_list()[0].code for edge in out_edges]
lst_adj_firm += [ap.AgentIter(self.model, edge[0]).to_list()[0].code for edge in in_edges]
lst_adj_firm += \
[ap.AgentIter(self.model, edge[1]).to_list()[
0].code for edge in out_edges]
lst_adj_firm += \
[ap.AgentIter(self.model, edge[0]).to_list()[
0].code for edge in in_edges]
if self.code in lst_adj_firm:
lst_firm_connect.append(firm)
if len(lst_firm_connect) == 0:
@@ -124,19 +136,21 @@ class FirmAgent(ap.Agent):
lst_prob = [
size / sum(lst_firm_size) for size in lst_firm_size
]
select_customer = self.model.nprandom.choice(lst_firm,
p=lst_prob)
select_customer = \
self.model.nprandom.choice(lst_firm, p=lst_prob)
self.accept_request(select_customer, product)
elif len(lst_firm_connect) == 1:
self.accept_request(lst_firm_connect[0], product)
elif len(lst_firm_connect) > 1:
# handling based on size of firm that has connection
lst_firm_size = [firm.revenue_log for firm in lst_firm_connect]
lst_firm_size = [
firm.revenue_log for firm in lst_firm_connect]
lst_prob = [
size / sum(lst_firm_size) for size in lst_firm_size
]
select_customer = self.model.nprandom.choice(lst_firm_connect,
p=lst_prob)
select_customer = \
self.model.nprandom.choice(lst_firm_connect,
p=lst_prob)
self.accept_request(select_customer, product)
def accept_request(self, down_firm, product):
@@ -151,18 +165,21 @@ class FirmAgent(ap.Agent):
(self.firm_network.positions[self],
self.firm_network.positions[down_firm], {
'Product': product.code
})
})
])
self.dct_prod_capacity[product] -= 1
self.dct_request_prod_from_firm[product].remove(down_firm)
down_firm.a_lst_up_product_removed.remove(product)
# print(
# f"{self.name} accept {product.code} request from {down_firm.name}"
# )
print(
f"{self.name} accept {product.code} request "
f"from {down_firm.name}"
)
else:
down_firm.dct_cand_alt_supply_up_prod_removed[product].remove(self)
def clean_before_trial(self):
self.dct_request_prod_from_firm = {}
def clean_before_time_step(self):
self.dct_n_trial_up_product_removed = {}
self.dct_n_trial_up_prod_removed = {}
self.a_lst_up_product_removed = ap.AgentList(self.model, [])