import matplotlib.pyplot as plt
from specreduce.calibration_data import AtmosphericTransmission, SUPPORTED_EXTINCTION_MODELS

fig, ax = plt.subplots()
ext_default = AtmosphericTransmission()
ext_custom = AtmosphericTransmission(data_file="atm_transmission_secz1.5_1.6mm.dat")
ax.plot(ext_default.spectral_axis, ext_default.transmission, label=r"sec $z$ = 1; 1 mm H$_{2}$O", linewidth=1)
ax.plot(ext_custom.spectral_axis, ext_custom.transmission, label=r"sec $z$ = 1.5; 1.6 mm H$_{2}$O", linewidth=1)
ax.legend(loc="upper center", bbox_to_anchor=(0.5, 1.12), ncol=2, fancybox=True, shadow=True)
ax.set_xlabel("Wavelength (microns)")
ax.set_ylabel("Transmission")
fig.show()