Simulation_Modeling_Code/chapter_3/same_python_file/chemical_reaction_animation.py at master · Rakibul73/Simulation_Modeling_Code · 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
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation

c1 = [50.0]
c2 = [25.0]
c3 = [0.0]
kl = 0.025
k2 = 0.01
time_difference = 0.1
t = 0.0
last_time = 15.0
i = 0

fig, ax = plt.subplots()
line_c1, = ax.plot([], [], label='C1')
line_c2, = ax.plot([], [], label='C2')
line_c3, = ax.plot([], [], label='C3')
ax.set_xlim(0, int(last_time / time_difference))
ax.set_ylim(0, max(max(c1), max(c2), max(c3)) + 5)
ax.set_xlabel('Time')
ax.set_ylabel('Concentration')
ax.set_title('Chemical Reaction Simulation')
ax.legend()

def init():
    line_c1.set_data([], [])
    line_c2.set_data([], [])
    line_c3.set_data([], [])
    return line_c1, line_c2, line_c3

def animate(frame):
    global t, i, time_difference

    if t <= last_time:
        c1_next = c1[i] + (k2 * c3[i] - kl * c1[i] * c2[i]) * time_difference
        c2_next = c2[i] + (k2 * c3[i] - kl * c1[i] * c2[i]) * time_difference
        c3_next = c3[i] + 2.0 * (kl * c1[i] * c2[i] - k2 * c3[i]) * time_difference

        c1.append(c1_next)
        c2.append(c2_next)
        c3.append(c3_next)

        i += 1
        t += time_difference

        if t >= 2.0:
            time_difference = 0.2
        if t == 6.0:
            time_difference = 0.4

    line_c1.set_data(range(len(c1)), c1)
    line_c2.set_data(range(len(c2)), c2)
    line_c3.set_data(range(len(c3)), c3)
    return line_c1, line_c2, line_c3

ani = FuncAnimation(fig, animate, frames=int(last_time / time_difference),
                    init_func=init, blit=True, interval=200)

plt.show()