def launch_run(params, env_dic, display_progress=True):
stats = get_zero_stats(params)
for r in range(params[nrun_key]):
params[store_preference_key] = 0.95
params[remote_work_key] = 0.98
params[innoculation_number_key] = 5
available_beds = params[icu_bed_per_thousand_individual_key] * params[nindividual_key] / 1000
virus_dic = get_virus_simulation_t0(params)
for day in range(params[nday_key]):
if display_progress:
print_progress_bar(r * params[nday_key] + day + 1, params[nrun_key] * params[nday_key],
prefix='Progress:', suffix='Complete', length=50)
propagate_to_houses(env_dic, virus_dic, params[house_infect_key])
if not is_weekend(day):
propagate_to_transportation(env_dic, virus_dic, params[transport_infection_key],
params[remote_work_key], params[transport_contact_cap_key])
propagate_to_workplaces(env_dic, virus_dic, params[work_infection_key], params[remote_work_key])
if is_weekend(day):
propagate_to_stores(env_dic, virus_dic, params[store_infection_key], params[store_preference_key])
increment_pandemic_1_day(env_dic, virus_dic, available_beds)
update_stats(virus_dic, stats, r, day)
stats["loc"][r][day] = measure_lockdown_strength(params)
return stats
def launch_run(params, env_dic, display_progress=True):
stats = get_zero_stats(params)
loosening_day = np.zeros((params[nrun_key]))
for r in range(params[nrun_key]):
params[store_preference_key] = 0.95
params[remote_work_key] = 0.98
params[house_infect_key] = 0.5
params[work_infection_key] = 0.01
params[store_infection_key] = 0.001
params[transport_infection_key] = 0.01
params[innoculation_number_key] = 50
available_beds = params[icu_bed_per_thousand_individual_key] * params[nindividual_key] / 1000
if len(params[additional_scenario_params_key]) < 1:
raise AssertionError("Need an additional_scenario parameter")
days_to_lockdown_loosening = params[additional_scenario_params_key][0]
days_with_no_cases = 0
first_day_lockdown_loosening = -1
virus_dic = get_virus_simulation_t0(params)
for day in range(params[nday_key]):
if display_progress:
print_progress_bar(r * params[nday_key] + day + 1, params[nrun_key] * params[nday_key],
prefix='Progress:', suffix='Complete', length=50)
propagate_to_houses(env_dic, virus_dic, params[house_infect_key])
if not is_weekend(day):
propagate_to_transportation(env_dic, virus_dic, params[transport_infection_key],
params[remote_work_key], params[transport_contact_cap_key])
propagate_to_workplaces(env_dic, virus_dic, params[work_infection_key], params[remote_work_key])
if is_weekend(day):
propagate_to_stores(env_dic, virus_dic, params[store_infection_key], params[store_preference_key])
increment_pandemic_1_day(env_dic, virus_dic, available_beds)
update_stats(virus_dic, stats, r, day)
stats["loc"][r][day] = measure_lockdown_strength(params)
if stats['new'][r][day] == 0:
days_with_no_cases += 1
else:
days_with_no_cases = 0
if days_with_no_cases > days_to_lockdown_loosening:
if first_day_lockdown_loosening == -1:
first_day_lockdown_loosening = day
days_with_no_cases = 0
soften_full_lockdown(params)
loosening_day[r] = first_day_lockdown_loosening
loosening_day_ok = [l_v for l_v in loosening_day if l_v != -1]
if display_progress:
print(loosening_day)
print("Lockdown removal occured %d times in average after %.2f days"
% (len(loosening_day_ok), sum(loosening_day_ok)/(1+len(loosening_day_ok))))
return stats
def launch_run(params, env_dic, display_progress=True):
if len(params[additional_scenario_params_key]) < 2:
raise AssertionError("Need more additional_scenario parameter")
nb_rogue = int(params[additional_scenario_params_key][0])
rogue_factor = params[additional_scenario_params_key][1]
stats = get_zero_stats(params)
for r in range(params[nrun_key]):
params[store_preference_key] = 0.95
params[remote_work_key] = 0.98
params[house_infect_key] = 0.5
params[work_infection_key] = 0.01
params[store_infection_key] = 0.001
params[transport_infection_key] = 0.01
params[innoculation_number_key] = 50
available_beds = params[icu_bed_per_thousand_individual_key] * params[
nindividual_key] / 1000
virus_dic = get_virus_simulation_t0(params)
for day in range(params[nday_key]):
if display_progress:
print_progress_bar(r * params[nday_key] + day + 1,
params[nrun_key] * params[nday_key],
prefix='Progress:',
suffix='Complete',
length=50)
propagate_to_houses(env_dic, virus_dic, params[house_infect_key])
if not is_weekend(day):
propagate_to_transportation(env_dic, virus_dic,
params[transport_infection_key],
params[remote_work_key],
params[transport_contact_cap_key])
propagate_to_workplaces(env_dic, virus_dic,
params[work_infection_key],
params[remote_work_key])
if is_weekend(day):
propagate_to_stores(env_dic, virus_dic,
params[store_infection_key],
params[store_preference_key])
increment_pandemic_1_day(env_dic, virus_dic, available_beds)
contagious_people = get_contagious_people(virus_dic)
if len(contagious_people) > 0:
rogues = get_random_sample(contagious_people, nb_rogue)
for rogue in rogues:
env_dic[IBE_K][rogue] *= rogue_factor
update_stats(virus_dic, stats, r, day)
stats["loc"][r][day] = measure_lockdown_strength(params)
return stats
def launch_run(params, env_dic, display_progress=True):
if len(params[additional_scenario_params_key]) < 1:
raise AssertionError("Need more additional_scenario parameter")
percent_increase = params[additional_scenario_params_key][0]
stats = get_zero_stats(params)
for r in range(params[nrun_key]):
params[store_preference_key] = 0.3
params[remote_work_key] = 0.58
params[house_infect_key] = 0.5
params[work_infection_key] = 0.01
params[store_infection_key] = 0.001
params[transport_infection_key] = 0.01
params[innoculation_number_key] = 50
params[icu_bed_per_thousand_individual_key] = 100
available_beds = params[icu_bed_per_thousand_individual_key] * params[
nindividual_key] / 1000
virus_dic = get_virus_simulation_t0(params)
for day in range(params[nday_key]):
if display_progress:
print_progress_bar(r * params[nday_key] + day + 1,
params[nrun_key] * params[nday_key],
prefix='Progress:',
suffix='Complete',
length=50)
propagate_to_houses(env_dic, virus_dic, params[house_infect_key])
if not is_weekend(day):
propagate_to_transportation(env_dic, virus_dic,
params[transport_infection_key],
params[remote_work_key],
params[transport_contact_cap_key])
propagate_to_workplaces(env_dic, virus_dic,
params[work_infection_key],
params[remote_work_key])
if is_weekend(day):
propagate_to_stores(env_dic, virus_dic,
params[store_infection_key],
params[store_preference_key])
increment_pandemic_1_day(env_dic, virus_dic, available_beds)
if day > 0 and day % 7 == 0:
params[work_infection_key] *= (1 + percent_increase)
params[store_infection_key] *= (1 + percent_increase)
params[transport_infection_key] *= (1 + percent_increase)
params[innoculation_number_key] *= (1 + percent_increase)
update_stats(virus_dic, stats, r, day)
stats["loc"][r][day] = measure_lockdown_strength(params)
return stats
def launch_run(params, env_dic, fun, display_progress=True):
stats = get_zero_stats(params)
stats_l = []
for run_id in range(params[nrun_key]):
if display_progress:
print_progress_bar(run_id,
params[nrun_key],
prefix='Progress:',
suffix='Complete',
length=50)
stats_l.append(fun(env_dic, params, run_id))
for run_id, run_stats in stats_l:
merge_run_stat(stats, run_stats, run_id)
return stats