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()
../_images/quickstart_quickstart_4_0.png