from constants import *
from mesh import Mesh
import numpy as np
import weak
import vacuum
import akhmedov
weak_mesh = Mesh()
vacuum_mesh = Mesh()
akhmedov_mesh = Mesh()
length = 70 # km
for x in np.arange(0, length * 1e3 * 1e2, 0.5e4):
weak_mesh.add_on_mesh(x, abs(weak.rho_flavours(x)[0][1]))
vacuum_mesh.add_on_mesh(x, abs(vacuum.rho_flavours(x)[0][1]))
akhmedov_mesh.add_on_mesh(x, abs(akhmedov.rho_flavours(x)[0][1]))
fig = Figure()
FigureCanvas(fig)
axs1 = fig.add_subplot(2, 1, 1)
axs2 = fig.add_subplot(2, 1, 2)
l1, = axs1.plot(weak_mesh.x(1e-5),
weak_mesh.y(1e3),
'-',
markersize=1,
color='#000000')
axs1.set_title("Oscillations in vacuum and normal matter")
