pingu_98/cara
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Extending the time range for the plot in the report_generator

Browse source
This commit is contained in:
Nicolas Mounet 2020-11-12 18:50:14 +01:00
parent 306e7fc31d
commit a9e07f77f6
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
cara/apps/calculator/report_generator.py
View file

@ -24,8 +24,10 @@ class RepeatEvents:
def calculate_report_data(model: models.ExposureModel):
resolution = 600
t_start = model.exposed.presence.boundaries()[0][0]
t_end = model.exposed.presence.boundaries()[-1][1]
t_start = min(model.exposed.presence.boundaries()[0][0],
model.concentration_model.infected.presence.boundaries()[0][0])
t_end = max(model.exposed.presence.boundaries()[-1][1],
model.concentration_model.infected.presence.boundaries()[-1][1])
times = list(np.linspace(t_start, t_end, resolution))
concentrations = [model.concentration_model.concentration(time) for time in times]