import numpy as np
import matplotlib.pyplot as plt
from literature_attenuation import *
import astropy.units as u


fig = plt.figure()
ax = fig.add_subplot(111)

wav = np.linspace(3,4,1000) #log angstrom
wav = 10.**(wav)*u.angstrom
x = (1./wav.to(u.micron)).value

tau_conroy = conroy(wav.value)
tau_calzetti = calzetti(wav.value)
tau_smc = smc(wav.value)
tau_lmc = lmc(wav.value)
tau_mw = cardelli(wav.value)

ax.plot(x,tau_mw,color='black',lw=3,label='Milky Way Average')
ax.plot(x,tau_lmc,color='firebrick',lw=3,label='LMC Average')
ax.plot(x,tau_smc,color='dodgerblue',lw=3,label='SMC Average')

plt.legend(fontsize=14)


ax.set_xlabel(r'1/$\mu$m',fontsize=14)
ax.set_ylabel(r'$\tau$',fontsize=14)

fig.savefig('local_group.png',dpi=300)