# -*- coding: utf-8 -*-
"""
.. _example_atomic_spectra:

========================
Atomic spectrum #0: Calculate an atomic spectrum
========================

In this example, the NIST and Kurucz databases are used to compute an atomic spectrum.
The same partition function is employed. The Lorentzian broadening is assumed equal for all lines.

"""


# %%
def broad_arbitrary(**kwargs):
    """An arbitrary broadening formula for the Lorentzian component"""
    HWHM = kwargs["pressure_atm"] * (296 / kwargs["Tgas"]) ** 0.7
    shift = None
    return HWHM, shift


from radis import SpectrumFactory, plot_diff

sf = SpectrumFactory(
    wavelength_min=498,
    wavelength_max=508,
    species="N_II",
    wstep=0.01,
    path_length=1,  # cm
    pressure=20,  # atm
    mole_fraction=0.1,
    verbose=0,  # to keep output quiet
    lbfunc=broad_arbitrary,
    pfsource="nist",
)

# NIST
sf.fetch_databank("nist")
s_NIST = sf.eq_spectrum(Tgas=10000, name="NIST")

# Kurucz
sf.fetch_databank("kurucz")
s_KURUCZ = sf.eq_spectrum(Tgas=10000, name="Kurucz")

plot_diff(
    s_NIST,
    s_KURUCZ,
    "radiance_noslit",
    wunit="nm",  # "nm" or "cm-1"
)