Particle-Based-COVID19-Simulator/plot_current_states_lecco.m at main · IS2AI/Particle-Based-COVID19-Simulator · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
function plot_current_states_lecco(delta_t, cur_iter, tot_exposed, tot_infected, tot_recovered, tot_dead, ...
    tot_quarantined, tot_isolated, tot_severe_inf, tot_cases, actual_cases, actual_death, actual_date, save_figures)

% construct the time vector
time_vec = (0:cur_iter - 1) * delta_t;

f1 = figure(1);
set(f1,'Position',[60 60 800 500]);
subplot(3,1,1:2)
hold on;

legend_list = {};
p2 = stairs(time_vec, tot_exposed(1:cur_iter), 'm-', 'LineWidth', 1);
legend_list = [legend_list, 'Exposed'];

p3 = stairs(time_vec, tot_quarantined(1:cur_iter), 'm--', 'LineWidth', 1);
legend_list = [legend_list, 'Quarantined'];

p4 = stairs(time_vec, tot_infected(1:cur_iter), 'r-', 'LineWidth', 1);
legend_list = [legend_list, 'Infected'];

p5 = stairs(time_vec, tot_severe_inf(1:cur_iter), 'r--','LineWidth', 1);
legend_list = [legend_list, 'Severe Infected'];

p6 = stairs(time_vec, tot_recovered(1:cur_iter), 'g-', 'LineWidth', 1);
legend_list = [legend_list, 'Immunized'];

p7 = stairs(time_vec, tot_isolated(1:cur_iter), 'c-','LineWidth', 1);
legend_list = [legend_list, 'Isolated'];

p8 = stairs(time_vec, tot_dead(1:cur_iter), 'k-', 'LineWidth', 1);
legend_list = [legend_list, 'Dead'];
p8_ = stairs(actual_date, actual_death, 'k-o', 'LineWidth', 1);
legend_list = [legend_list, 'Dead (Actual)'];

p9 = plot(time_vec, tot_cases(1:cur_iter), 'b-', 'LineWidth', 1);
legend_list = [legend_list, 'Total cases'];
p9_ = plot(actual_date, actual_cases, 'b-o', 'LineWidth', 1);
legend_list = [legend_list, 'Total cases (Actual)'];

legend( legend_list, 'FontName','Arial', 'FontSize', 10','FontWeight','Demi','Location','northwest','Orientation','Vertical');

ylabel('Number of individuals','FontName','Arial', 'FontSize', 12, 'FontWeight', 'Demi');
xlabel('', 'FontName','Arial', 'FontSize', 12, 'FontWeight', 'Demi');
xlim([0 cur_iter * delta_t]);
%ylim([0 110000]);
set(gca,'YTickLabel',num2str(get(gca,'YTick').'))
grid on;  box on;

subplot 313
ha = plot(actual_date, actual_death, 'k-o', 'LineWidth', 1);
ax = gca;
ax.YAxis.Exponent = 0;
xlabel('Time (days)', 'FontName', 'Arial', 'FontSize', 12, 'FontWeight', 'Demi');
ylabel('Num. of Individuals', 'FontName', 'Arial', 'FontSize', 12, 'FontWeight', 'Demi');
xlim([0 cur_iter * delta_t]);
grid on; box on;
hold on
hb = plot(time_vec, tot_dead(1:cur_iter), 'k-', 'LineWidth', 1);
hc = plot(time_vec, tot_severe_inf(1:cur_iter), 'r--', 'LineWidth', 1);
legend([ha hb hc], 'Dead (Actual)', 'Dead', 'Severe Infected', 'Location','northwest','Orientation','Vertical');
hold off;

if save_figures == 1
    filename = sprintf('ind_%d.png', cur_iter);
    saveas(f1, filename);
end

end