from spectractor.simulation.atmosphere import AtmosphereGrid, plot_transmission_simple
from spectractor import parameters
import numpy as np
import matplotlib.pyplot as plt
a = AtmosphereGrid(atmgrid_filename='tests/data/reduc_20170530_134_atmsim.fits')
lambdas = np.arange(200, 1200)
fig = plt.figure()
for pwv in np.arange(5):
    transmission = a.simulate(ozone=400, pwv=pwv, aerosols=0.05)
    plot_transmission_simple(plt.gca(), lambdas, transmission(lambdas),
    title=a.title, label=a.label)
if parameters.DISPLAY: plt.show()