STARDIS
[1]:
# Import necessary code
import numpy as np
import matplotlib.pyplot as plt
from tardis.io.atom_data.util import download_atom_data
from stardis.base import run_stardis
from astropy import units as u, constants as const
[2]:
# Download atomic data
download_atom_data('kurucz_cd23_chianti_H_He')
[3]:
# Create array of wavelengths and run STARDIS
tracing_lambdas = np.arange(6540, 6590, .01) * u.Angstrom
sim = run_stardis('stardis_example.yml', tracing_lambdas)
/home/smithis7/anaconda3/envs/stardis/lib/python3.8/site-packages/tardis/plasma/properties/radiative_properties.py:93: RuntimeWarning: invalid value encountered in true_divide
(g_lower * n_upper) / (g_upper * n_lower)
[4]:
# Plot spectra
#%matplotlib notebook
plt.figure(figsize=(10,6))
plt.plot(sim.lambdas, sim.spectrum_lambda)
plt.xlim((6540,6590))
plt.title("STARDIS Solar Spectrum")
plt.xlabel("Wavelength [$\AA$]")
plt.ylabel("Flux density [erg/s/cm$^2$/$\AA$]")
plt.show()