move model instances to model_scenarios.py

This commit is contained in:
markus 2021-02-23 11:46:04 +01:00
parent be6e5250dc
commit 805fa173c0
2 changed files with 361 additions and 357 deletions

View file

@ -0,0 +1,360 @@
from montecarlo import *
from cara import models
fixed_vl_exposure_models = [MCExposureModel(
concentration_model=MCConcentrationModel(
room=models.Room(volume=45),
ventilation=models.SlidingWindow(
active=models.PeriodicInterval(period=120, duration=10),
inside_temp=models.PiecewiseConstant((0, 24), (293,)),
outside_temp=models.PiecewiseConstant((0, 24), (283,)),
window_height=1.6, opening_length=0.6,
),
infected=MCInfectedPopulation(
number=1,
presence=models.SpecificInterval(((0, 4), (5, 9))),
masked=True,
virus=MCVirus(halflife=1.1, qID=100),
expiratory_activity=1,
samples=2000000,
breathing_category=1,
viral_load=float(vl)
)
),
exposed=models.Population(
number=2,
presence=models.SpecificInterval(((0, 4), (5, 9))),
activity=models.Activity.types['Seated'],
mask=models.Mask.types['Type I']
)
) for vl in range(6, 11)]
large_population_baselines = [MCExposureModel(
concentration_model=MCConcentrationModel(
room=models.Room(volume=800),
ventilation=models.HVACMechanical(
active=models.PeriodicInterval(period=120, duration=120),
q_air_mech=2000.
),
infected=MCInfectedPopulation(
number=1,
presence=models.SpecificInterval(((0, 4), (5, 9))),
masked=False,
virus=MCVirus(halflife=1.1, qID=qid),
expiratory_activity=1,
samples=200000,
breathing_category=2,
expiratory_activity_weights=(0.7, 0.3, 0)
)
),
exposed=models.Population(
number=60,
presence=models.SpecificInterval(((0, 4), (5, 9))),
activity=models.Activity.types['Standing'],
mask=models.Mask.types['No mask']
)
) for qid in (100, 60)]
exposure_models = [MCExposureModel(
concentration_model=MCConcentrationModel(
room=models.Room(volume=45),
ventilation=models.SlidingWindow(
active=models.PeriodicInterval(period=120, duration=120),
inside_temp=models.PiecewiseConstant((0, 24), (293,)),
outside_temp=models.PiecewiseConstant((0, 24), (283,)),
window_height=1.6, opening_length=0.2,
),
infected=MCInfectedPopulation(
number=1,
presence=models.SpecificInterval(((0, 4), (5, 9))),
masked=False,
virus=MCVirus(halflife=1.1, qID=qid),
expiratory_activity=3,
samples=2000000,
breathing_category=3,
expiratory_activity_weights=(0.7, 0.3, 0)
)
),
exposed=models.Population(
number=2,
presence=models.SpecificInterval(((0, 4), (5, 9))),
activity=models.Activity.types['Seated'],
mask=models.Mask.types['Type I']
)
) for qid in (100, 60)]
exposure_models_2 = [MCExposureModel(
concentration_model=MCConcentrationModel(
room=models.Room(volume=33),
ventilation=models.SlidingWindow(
active=models.PeriodicInterval(period=120, duration=120),
inside_temp=models.PiecewiseConstant((0, 24), (293,)),
outside_temp=models.PiecewiseConstant((0, 24), (283,)),
window_height=1.6, opening_length=0.6,
),
infected=MCInfectedPopulation(
number=1,
presence=models.SpecificInterval(((0, 4), (5, 9))),
masked=True,
virus=MCVirus(halflife=1.1, qID=qid),
expiratory_activity=4,
samples=2000000,
breathing_category=1,
expiratory_activity_weights=(0.7, 0.1, 0)
)
),
exposed=models.Population(
number=2,
presence=models.SpecificInterval(((0, 4), (5, 9))),
activity=models.Activity.types['Seated'],
mask=models.Mask.types['Type I']
)
) for qid in (100, 60)]
classroom_model = MCExposureModel(
concentration_model=MCConcentrationModel(
room=models.Room(volume=160),
ventilation=models.SlidingWindow(
active=models.PeriodicInterval(period=120, duration=10),
inside_temp=models.PiecewiseConstant((0, 24), (293,)),
outside_temp=models.PiecewiseConstant((0, 24), (283,)),
window_height=1.6, opening_length=0.6,
),
infected=MCInfectedPopulation(
number=1,
presence=models.SpecificInterval(((0, 2), (2.5, 4), (5, 7), (7.5, 9))),
masked=False,
virus=MCVirus(halflife=1.1, qID=60),
expiratory_activity=2,
samples=200000,
breathing_category=3,
)
),
exposed=models.Population(
number=19,
presence=models.SpecificInterval(((0, 2), (2.5, 4), (5, 7), (7.5, 9))),
activity=models.Activity.types['Seated'],
mask=models.Mask.types['No mask']
)
)
classroom_model_with_hepa = MCExposureModel(
concentration_model=MCConcentrationModel(
room=models.Room(volume=160),
ventilation=models.MultipleVentilation(
ventilations=(
models.SlidingWindow(
active=models.PeriodicInterval(period=120, duration=10),
inside_temp=models.PiecewiseConstant((0, 24), (293,)),
outside_temp=models.PiecewiseConstant((0, 24), (283,)),
window_height=1.6, opening_length=0.6,
),
models.AirChange(active=models.PeriodicInterval(period=120, duration=120),
air_exch=5)
)
),
infected=MCInfectedPopulation(
number=1,
presence=models.SpecificInterval(((0, 2), (2.5, 4), (5, 7), (7.5, 9))),
masked=False,
virus=MCVirus(halflife=1.1, qID=60),
expiratory_activity=2,
samples=200000,
breathing_category=3,
)
),
exposed=models.Population(
number=19,
presence=models.SpecificInterval(((0, 2), (2.5, 4), (5, 7), (7.5, 9))),
activity=models.Activity.types['Seated'],
mask=models.Mask.types['No mask']
)
)
shared_office_model = [MCExposureModel(
concentration_model=MCConcentrationModel(
room=models.Room(volume=50),
ventilation=models.SlidingWindow(
active=models.PeriodicInterval(period=120, duration=10),
inside_temp=models.PiecewiseConstant((0, 24), (293,)),
outside_temp=models.PiecewiseConstant((0, 24), (283,)),
window_height=1.6, opening_length=0.6,
),
infected=MCInfectedPopulation(
number=1,
presence=models.SpecificInterval(((0, 2), (2.1, 4), (5, 7), (7.1, 9))),
masked=True,
virus=MCVirus(halflife=1.35, qID=qid),
expiratory_activity=4,
samples=200000,
breathing_category=1,
expiratory_activity_weights=(0.7, 0.3, 0)
)
),
exposed=models.Population(
number=3,
presence=models.SpecificInterval(((0, 2), (2.1, 4), (5, 7), (7.1, 9))),
activity=models.Activity.types['Seated'],
mask=models.Mask.types['Type I']
)
) for qid in (100, 60)]
shared_office_worst_model = [MCExposureModel(
concentration_model=MCConcentrationModel(
room=models.Room(volume=50),
ventilation=models.AirChange(
active=models.PeriodicInterval(period=120, duration=120),
air_exch=0.,
),
infected=MCInfectedPopulation(
number=1,
presence=models.SpecificInterval(((0, 2), (2.1, 4), (5, 7), (7.1, 9))),
masked=False,
virus=MCVirus(halflife=1.35, qID=qid),
expiratory_activity=4,
samples=200000,
breathing_category=1,
expiratory_activity_weights=(0.7, 0.3, 0)
)
),
exposed=models.Population(
number=3,
presence=models.SpecificInterval(((0, 2), (2.1, 4), (5, 7), (7.1, 9))),
activity=models.Activity.types['Seated'],
mask=models.Mask.types['No mask']
)
) for qid in (100, 60)]
shared_office_better_model = [MCExposureModel(
concentration_model=MCConcentrationModel(
room=models.Room(volume=50),
ventilation=models.AirChange(
active=models.PeriodicInterval(period=120, duration=120),
air_exch=5.,
),
infected=MCInfectedPopulation(
number=1,
presence=models.SpecificInterval(((0, 2), (2.1, 4), (5, 7), (7.1, 9))),
masked=True,
virus=MCVirus(halflife=1.35, qID=qid),
expiratory_activity=4,
samples=200000,
breathing_category=1,
expiratory_activity_weights=(0.7, 0.3, 0)
)
),
exposed=models.Population(
number=3,
presence=models.SpecificInterval(((0, 2), (2.1, 4), (5, 7), (7.1, 9))),
activity=models.Activity.types['Seated'],
mask=models.Mask.types['Type I']
)
) for qid in (100, 60)]
ski_cabin_model = MCExposureModel(
concentration_model=MCConcentrationModel(
room=models.Room(volume=10),
ventilation=models.HVACMechanical(
active=models.PeriodicInterval(period=120, duration=120),
q_air_mech=0.,
),
infected=MCInfectedPopulation(
number=1,
presence=models.SpecificInterval(((0, 0.33),)),
masked=True,
virus=MCVirus(halflife=1.1, qID=60),
expiratory_activity=2,
samples=200000,
breathing_category=4,
expiratory_activity_weights=(0.7, 0.3, 0)
)
),
exposed=models.Population(
number=3,
presence=models.SpecificInterval(((0, 0.33),)),
activity=models.Activity.types['Moderate activity'],
mask=models.Mask.types['Type I']
)
)
gym_model = MCExposureModel(
concentration_model=MCConcentrationModel(
room=models.Room(volume=300),
ventilation=models.AirChange(
active=models.PeriodicInterval(period=120, duration=120),
air_exch=6.,
),
infected=MCInfectedPopulation(
number=2,
presence=models.SpecificInterval(((0, 1),)),
masked=False,
virus=MCVirus(halflife=1.1, qID=60),
expiratory_activity=1,
samples=200000,
breathing_category=5,
expiratory_activity_weights=(0.7, 0.3, 0)
)
),
exposed=models.Population(
number=28,
presence=models.SpecificInterval(((0, 1),)),
activity=models.Activity.types['Heavy exercise'],
mask=models.Mask.types['No mask']
)
)
waiting_room_model = MCExposureModel(
concentration_model=MCConcentrationModel(
room=models.Room(volume=100),
ventilation=models.AirChange(
active=models.PeriodicInterval(period=120, duration=120),
air_exch=0.,
),
infected=MCInfectedPopulation(
number=1,
presence=models.SpecificInterval(((0, 2),)),
masked=False,
virus=MCVirus(halflife=1.35, qID=60),
expiratory_activity=4,
samples=200000,
breathing_category=1,
expiratory_activity_weights=(0.7, 0.3, 0)
)
),
exposed=models.Population(
number=14,
presence=models.SpecificInterval(((0, 2),)),
activity=models.Activity.types['Seated'],
mask=models.Mask.types['No mask']
)
)
chorale_model = MCExposureModel(
concentration_model=MCConcentrationModel(
room=models.Room(volume=810),
ventilation=models.AirChange(
active=models.PeriodicInterval(period=120, duration=120),
air_exch=0.7,
),
infected=MCInfectedPopulation(
number=1,
presence=models.SpecificInterval(((0, 2.5),)),
masked=False,
virus=MCVirus(halflife=1.1, qID=100),
expiratory_activity=3,
samples=200000,
breathing_category=3,
expiratory_activity_weights=(0.7, 0.3, 0)
)
),
exposed=models.Population(
number=60,
presence=models.SpecificInterval(((0, 2.5),)),
activity=models.Activity.types['Moderate activity'],
mask=models.Mask.types['No mask']
)
)

View file

@ -11,6 +11,7 @@ import matplotlib.pyplot as plt
import matplotlib.patches as patches
import matplotlib.lines as mlines
from sklearn.neighbors import KernelDensity
from model_scenarios import *
TIME_STEP = 0.05
USE_SCOEH = False
@ -945,363 +946,6 @@ def compare_concentration_curves(exp_models: typing.List[MCExposureModel], label
plt.show()
fixed_vl_exposure_models = [MCExposureModel(
concentration_model=MCConcentrationModel(
room=models.Room(volume=45),
ventilation=models.SlidingWindow(
active=models.PeriodicInterval(period=120, duration=10),
inside_temp=models.PiecewiseConstant((0, 24), (293,)),
outside_temp=models.PiecewiseConstant((0, 24), (283,)),
window_height=1.6, opening_length=0.6,
),
infected=MCInfectedPopulation(
number=1,
presence=models.SpecificInterval(((0, 4), (5, 9))),
masked=True,
virus=MCVirus(halflife=1.1, qID=100),
expiratory_activity=1,
samples=2000000,
breathing_category=1,
viral_load=float(vl)
)
),
exposed=models.Population(
number=2,
presence=models.SpecificInterval(((0, 4), (5, 9))),
activity=models.Activity.types['Seated'],
mask=models.Mask.types['Type I']
)
) for vl in range(6, 11)]
large_population_baselines = [MCExposureModel(
concentration_model=MCConcentrationModel(
room=models.Room(volume=800),
ventilation=models.HVACMechanical(
active=models.PeriodicInterval(period=120, duration=120),
q_air_mech=2000.
),
infected=MCInfectedPopulation(
number=1,
presence=models.SpecificInterval(((0, 4), (5, 9))),
masked=False,
virus=MCVirus(halflife=1.1, qID=qid),
expiratory_activity=1,
samples=200000,
breathing_category=2,
expiratory_activity_weights=(0.7, 0.3, 0)
)
),
exposed=models.Population(
number=60,
presence=models.SpecificInterval(((0, 4), (5, 9))),
activity=models.Activity.types['Standing'],
mask=models.Mask.types['No mask']
)
) for qid in (100, 60)]
exposure_models = [MCExposureModel(
concentration_model=MCConcentrationModel(
room=models.Room(volume=45),
ventilation=models.SlidingWindow(
active=models.PeriodicInterval(period=120, duration=120),
inside_temp=models.PiecewiseConstant((0, 24), (293,)),
outside_temp=models.PiecewiseConstant((0, 24), (283,)),
window_height=1.6, opening_length=0.2,
),
infected=MCInfectedPopulation(
number=1,
presence=models.SpecificInterval(((0, 4), (5, 9))),
masked=False,
virus=MCVirus(halflife=1.1, qID=qid),
expiratory_activity=3,
samples=2000000,
breathing_category=3,
expiratory_activity_weights=(0.7, 0.3, 0)
)
),
exposed=models.Population(
number=2,
presence=models.SpecificInterval(((0, 4), (5, 9))),
activity=models.Activity.types['Seated'],
mask=models.Mask.types['Type I']
)
) for qid in (100, 60)]
exposure_models_2 = [MCExposureModel(
concentration_model=MCConcentrationModel(
room=models.Room(volume=33),
ventilation=models.SlidingWindow(
active=models.PeriodicInterval(period=120, duration=120),
inside_temp=models.PiecewiseConstant((0, 24), (293,)),
outside_temp=models.PiecewiseConstant((0, 24), (283,)),
window_height=1.6, opening_length=0.6,
),
infected=MCInfectedPopulation(
number=1,
presence=models.SpecificInterval(((0, 4), (5, 9))),
masked=True,
virus=MCVirus(halflife=1.1, qID=qid),
expiratory_activity=4,
samples=2000000,
breathing_category=1,
expiratory_activity_weights=(0.7, 0.1, 0)
)
),
exposed=models.Population(
number=2,
presence=models.SpecificInterval(((0, 4), (5, 9))),
activity=models.Activity.types['Seated'],
mask=models.Mask.types['Type I']
)
) for qid in (100, 60)]
classroom_model = MCExposureModel(
concentration_model=MCConcentrationModel(
room=models.Room(volume=160),
ventilation=models.SlidingWindow(
active=models.PeriodicInterval(period=120, duration=10),
inside_temp=models.PiecewiseConstant((0, 24), (293,)),
outside_temp=models.PiecewiseConstant((0, 24), (283,)),
window_height=1.6, opening_length=0.6,
),
infected=MCInfectedPopulation(
number=1,
presence=models.SpecificInterval(((0, 2), (2.5, 4), (5, 7), (7.5, 9))),
masked=False,
virus=MCVirus(halflife=1.1, qID=60),
expiratory_activity=2,
samples=200000,
breathing_category=3,
)
),
exposed=models.Population(
number=19,
presence=models.SpecificInterval(((0, 2), (2.5, 4), (5, 7), (7.5, 9))),
activity=models.Activity.types['Seated'],
mask=models.Mask.types['No mask']
)
)
classroom_model_with_hepa = MCExposureModel(
concentration_model=MCConcentrationModel(
room=models.Room(volume=160),
ventilation=models.MultipleVentilation(
ventilations=(
models.SlidingWindow(
active=models.PeriodicInterval(period=120, duration=10),
inside_temp=models.PiecewiseConstant((0, 24), (293,)),
outside_temp=models.PiecewiseConstant((0, 24), (283,)),
window_height=1.6, opening_length=0.6,
),
models.AirChange(active=models.PeriodicInterval(period=120, duration=120),
air_exch=5)
)
),
infected=MCInfectedPopulation(
number=1,
presence=models.SpecificInterval(((0, 2), (2.5, 4), (5, 7), (7.5, 9))),
masked=False,
virus=MCVirus(halflife=1.1, qID=60),
expiratory_activity=2,
samples=200000,
breathing_category=3,
)
),
exposed=models.Population(
number=19,
presence=models.SpecificInterval(((0, 2), (2.5, 4), (5, 7), (7.5, 9))),
activity=models.Activity.types['Seated'],
mask=models.Mask.types['No mask']
)
)
shared_office_model = [MCExposureModel(
concentration_model=MCConcentrationModel(
room=models.Room(volume=50),
ventilation=models.SlidingWindow(
active=models.PeriodicInterval(period=120, duration=10),
inside_temp=models.PiecewiseConstant((0, 24), (293,)),
outside_temp=models.PiecewiseConstant((0, 24), (283,)),
window_height=1.6, opening_length=0.6,
),
infected=MCInfectedPopulation(
number=1,
presence=models.SpecificInterval(((0, 2), (2.1, 4), (5, 7), (7.1, 9))),
masked=True,
virus=MCVirus(halflife=1.35, qID=qid),
expiratory_activity=4,
samples=200000,
breathing_category=1,
expiratory_activity_weights=(0.7, 0.3, 0)
)
),
exposed=models.Population(
number=3,
presence=models.SpecificInterval(((0, 2), (2.1, 4), (5, 7), (7.1, 9))),
activity=models.Activity.types['Seated'],
mask=models.Mask.types['Type I']
)
) for qid in (100, 60)]
shared_office_worst_model = [MCExposureModel(
concentration_model=MCConcentrationModel(
room=models.Room(volume=50),
ventilation=models.AirChange(
active=models.PeriodicInterval(period=120, duration=120),
air_exch=0.,
),
infected=MCInfectedPopulation(
number=1,
presence=models.SpecificInterval(((0, 2), (2.1, 4), (5, 7), (7.1, 9))),
masked=False,
virus=MCVirus(halflife=1.35, qID=qid),
expiratory_activity=4,
samples=200000,
breathing_category=1,
expiratory_activity_weights=(0.7, 0.3, 0)
)
),
exposed=models.Population(
number=3,
presence=models.SpecificInterval(((0, 2), (2.1, 4), (5, 7), (7.1, 9))),
activity=models.Activity.types['Seated'],
mask=models.Mask.types['No mask']
)
) for qid in (100, 60)]
shared_office_better_model = [MCExposureModel(
concentration_model=MCConcentrationModel(
room=models.Room(volume=50),
ventilation=models.AirChange(
active=models.PeriodicInterval(period=120, duration=120),
air_exch=5.,
),
infected=MCInfectedPopulation(
number=1,
presence=models.SpecificInterval(((0, 2), (2.1, 4), (5, 7), (7.1, 9))),
masked=True,
virus=MCVirus(halflife=1.35, qID=qid),
expiratory_activity=4,
samples=200000,
breathing_category=1,
expiratory_activity_weights=(0.7, 0.3, 0)
)
),
exposed=models.Population(
number=3,
presence=models.SpecificInterval(((0, 2), (2.1, 4), (5, 7), (7.1, 9))),
activity=models.Activity.types['Seated'],
mask=models.Mask.types['Type I']
)
) for qid in (100, 60)]
ski_cabin_model = MCExposureModel(
concentration_model=MCConcentrationModel(
room=models.Room(volume=10),
ventilation=models.HVACMechanical(
active=models.PeriodicInterval(period=120, duration=120),
q_air_mech=0.,
),
infected=MCInfectedPopulation(
number=1,
presence=models.SpecificInterval(((0, 0.33),)),
masked=True,
virus=MCVirus(halflife=1.1, qID=60),
expiratory_activity=2,
samples=200000,
breathing_category=4,
expiratory_activity_weights=(0.7, 0.3, 0)
)
),
exposed=models.Population(
number=3,
presence=models.SpecificInterval(((0, 0.33),)),
activity=models.Activity.types['Moderate activity'],
mask=models.Mask.types['Type I']
)
)
gym_model = MCExposureModel(
concentration_model=MCConcentrationModel(
room=models.Room(volume=300),
ventilation=models.AirChange(
active=models.PeriodicInterval(period=120, duration=120),
air_exch=6.,
),
infected=MCInfectedPopulation(
number=2,
presence=models.SpecificInterval(((0, 1),)),
masked=False,
virus=MCVirus(halflife=1.1, qID=60),
expiratory_activity=1,
samples=200000,
breathing_category=5,
expiratory_activity_weights=(0.7, 0.3, 0)
)
),
exposed=models.Population(
number=28,
presence=models.SpecificInterval(((0, 1),)),
activity=models.Activity.types['Heavy exercise'],
mask=models.Mask.types['No mask']
)
)
waiting_room_model = MCExposureModel(
concentration_model=MCConcentrationModel(
room=models.Room(volume=100),
ventilation=models.AirChange(
active=models.PeriodicInterval(period=120, duration=120),
air_exch=0.,
),
infected=MCInfectedPopulation(
number=1,
presence=models.SpecificInterval(((0, 2),)),
masked=False,
virus=MCVirus(halflife=1.35, qID=60),
expiratory_activity=4,
samples=200000,
breathing_category=1,
expiratory_activity_weights=(0.7, 0.3, 0)
)
),
exposed=models.Population(
number=14,
presence=models.SpecificInterval(((0, 2),)),
activity=models.Activity.types['Seated'],
mask=models.Mask.types['No mask']
)
)
chorale_model = MCExposureModel(
concentration_model=MCConcentrationModel(
room=models.Room(volume=810),
ventilation=models.AirChange(
active=models.PeriodicInterval(period=120, duration=120),
air_exch=0.7,
),
infected=MCInfectedPopulation(
number=1,
presence=models.SpecificInterval(((0, 2.5),)),
masked=False,
virus=MCVirus(halflife=1.1, qID=100),
expiratory_activity=3,
samples=200000,
breathing_category=3,
expiratory_activity_weights=(0.7, 0.3, 0)
)
),
exposed=models.Population(
number=60,
presence=models.SpecificInterval(((0, 2.5),)),
activity=models.Activity.types['Moderate activity'],
mask=models.Mask.types['No mask']
)
)
compare_concentration_curves([classroom_model, classroom_model_with_hepa], ['Just window', 'Window and HEPA'])
#print(np.mean(chorale_model.infection_probability()))
#print(np.mean(chorale_model.infection_probability())+np.std(chorale_model.infection_probability()))