import matplotlib.pyplot as plt
from shone.opacity import Opacity, generate_synthetic_opacity

opacity = generate_synthetic_opacity()
opacity_sample = opacity.grid.sel(
    dict(
        pressure=10,  # [bar]
        temperature=200  # [K]
    )
)

plt.semilogy(
    opacity_sample.wavelength, opacity_sample,
    label=f"T={opacity_sample.temperature} K"
)
plt.gca().set(
    xlabel='Wavelength [µm]',
    ylabel='Opacity, $\kappa$ [cm$^2$ g$^{-1}$]'
)