import matplotlib.pyplot as plt
from roentgen.lines import get_lines
import astropy.units as u
import padre_meddea
from padre_meddea.io.amptek import read_mca
mca_file = padre_meddea._test_files_directory / "minix_20kV_15uA_sdd.mca"
spec = read_mca(mca_file)
au_lines = get_lines(1 * u.keV, 20 * u.keV, "au")
f = spec.meta['calib']
energy_ax = f(spec.spectral_axis.value)
fig, ax = plt.subplots(layout="constrained")
ax.plot(energy_ax, spec.flux)  # doctest: +SKIP
for this_line in au_lines:
    this_label = f"{this_line['energy'].to('keV'):0.2f} Au {this_line['transition']} {this_line['intensity']}"
    ax.axvline(this_line['energy'].to('keV').value, label=this_label, color='red')  # doctest: +SKIP
plt.legend()  # doctest: +SKIP