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

formatting

Browse Source
This commit is contained in:
HaoYizhi
2023-07-28 18:10:25 +08:00
parent bf82aa62ff
commit 9c26d04d76
2 changed files with 42 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
model.py
View File

@@ -222,14 +222,14 @@ class Model(ap.Model):
self.dct_lst_init_disrupt_firm_prod = t_dct
# set the initial firm product that are disrupted
print('\n', '=' * 20, 'step', self.t, '=' * 20)
# print('\n', '=' * 20, 'step', self.t, '=' * 20)
for firm, a_lst_product in self.dct_lst_init_disrupt_firm_prod.items():
for product in a_lst_product:
assert product in firm.dct_prod_up_prod_stat.keys(), \
f"product {product.code} not in firm {firm.code}"
firm.dct_prod_up_prod_stat[
product]['p_stat'].append(('D', self.t))
print(f"initial disruption {firm.name} {product.code}")
# print(f"initial disruption {firm.name} {product.code}")
# proactive strategy
# get all the firm prod affected
@@ -331,10 +331,10 @@ class Model(ap.Model):
'Product': di_supp_prod.code
})
])
print(f"proactive add {select_cand.name} to "
f"{pro_firm.name} "
f"for {di_supp_node['Firm_Code']} "
f"{di_supp_node['Product_Code']}")
# print(f"proactive add {select_cand.name} to "
# f"{pro_firm.name} "
# f"for {di_supp_node['Firm_Code']} "
# f"{di_supp_node['Product_Code']}")
# change capacity
select_cand.dct_prod_capacity[di_supp_prod] -= 1
break
@@ -356,8 +356,8 @@ class Model(ap.Model):
/ len(firm.dct_prod_up_prod_stat.keys()) \
/ self.remove_t
firm.size_stat.append((size, self.t))
print(f'in ts {self.t}, reduce {firm.name} size '
f'to {firm.size_stat[-1][0]} due to {prod.code}')
# print(f'in ts {self.t}, reduce {firm.name} size '
# f'to {firm.size_stat[-1][0]} due to {prod.code}')
lst_is_disrupt = \
[stat == 'D' for stat, _ in
firm.dct_prod_up_prod_stat[prod]['p_stat']
@@ -377,7 +377,7 @@ class Model(ap.Model):
firm in self.dct_lst_init_disrupt_firm_prod.keys() \
and prod in self.dct_lst_init_disrupt_firm_prod[firm]
if status == 'D' and not is_init:
print("not stop because", firm.name, prod.code)
# print("not stop because", firm.name, prod.code)
break
else:
continue
@@ -390,7 +390,7 @@ class Model(ap.Model):
self.stop()
def step(self):
print('\n', '=' * 20, 'step', self.t, '=' * 20)
# print('\n', '=' * 20, 'step', self.t, '=' * 20)
# remove edge to customer and disrupt customer up product
for firm in self.a_lst_total_firms:
@@ -410,7 +410,7 @@ class Model(ap.Model):
firm.disrupt_cus_prod(prod, up_prod)
for n_trial in range(self.int_n_max_trial):
print('=' * 10, 'trial', n_trial, '=' * 10)
# print('=' * 10, 'trial', n_trial, '=' * 10)
# seek_alt_supply
# shuffle self.a_lst_total_firms
self.a_lst_total_firms = self.a_lst_total_firms.shuffle()
@@ -447,7 +447,7 @@ class Model(ap.Model):
# self.a_lst_total_firms.dct_request_prod_from_firm = {} why?
def end(self):
print('/' * 20, 'output', '/' * 20)
# print('/' * 20, 'output', '/' * 20)
qry_result = db_session.query(Result).filter_by(s_id=self.sample.id)
if qry_result.count() == 0:
@@ -458,8 +458,8 @@ class Model(ap.Model):
lst_is_normal = [stat == 'N' for stat, _
in dct_status_supply['p_stat']]
if not all(lst_is_normal):
print(f"{firm.name} {prod.code}:")
print(dct_status_supply['p_stat'])
# print(f"{firm.name} {prod.code}:")
# print(dct_status_supply['p_stat'])
for status, ts in dct_status_supply['p_stat']:
db_r = Result(s_id=self.sample.id,
id_firm=firm.code,