tardis.visualization.widgets.grotrian module

Grotrian Diagram Widget for TARDIS simulation models.

This widget displays a Grotrian Diagram of the last line interactions of the simulation packets

class tardis.visualization.widgets.grotrian.GrotrianPlot(atom_data, level_energy_data, level_population_data, line_interaction_analysis)

Bases: object

Class for the Grotrian Diagram

Parameters

atom_datapandas.DataFrame

Mapping from atomic number to symbol and name

level_energy_datapandas.Series

Level energies (in eV) indexed by (atomic_number, ion_number, level_number)

level_population_datapandas.DataFrame

Level populations indexed by (atomic_number, ion_number, level_number) and each column representing the supernova shell

line_interaction_analysistardis.analysis.LastLineInteraction

LastLineInteraction object with the appropriate filters

FILTER_MODES = ('packet_out_nu', 'packet_in_nu')

FILTER_MODES_DESC = ('Emitted Wavelength', 'Absorbed Wavelength')

Y_SCALE_OPTION = {'Linear': <function GrotrianPlot.<lambda>>, 'Log': <ufunc 'log'>}

property atomic_name

property atomic_number

property atomic_symbol

display()

Function to draw the plot and the reference scales (calls other draw methods independently)

property filter_mode

classmethod from_simulation(sim, **kwargs)

Creates a GrotrianPlot object from a Simulation object

Parameters

simtardis.simulation.Simulation

TARDIS simulation object

Returns

tardis.visualization.widgets.grotrian.GrotrianPlot

GrotrianPlot object

property ion_number

property level_diff_threshold

property max_levels

property max_wavelength

property min_wavelength

reset_selected_plot_wavelength_range()

Resets the wavelength range of the selected plot

set_ion(atomic_number, ion_number)

Sets the atomic number and ion number

property shell

property y_scale

class tardis.visualization.widgets.grotrian.GrotrianWidget(plot, num_shells, **kwargs)

Bases: object

A wrapper class for the Grotrian Diagram, containing the Grotrian Plot and the IpyWidgets

Parameters

plottardis.visualization.widgets.grotrian.GrotrianPlot

GrotrianPlot object

num_shellsint

Number of shells in the sim.simulation_state.v_inner

display()

Function to render the Grotrian Widget containing the plot and IpyWidgets together

classmethod from_simulation(sim, **kwargs)

Creates a GrotrianWidget object from a Simulation object

Parameters

simtardis.simulation.Simulation

TARDIS simulation object

Returns

tardis.visualization.widgets.grotrian.GrotrianWidget

GrotrianWidget object

tardis.visualization.widgets.grotrian.is_zero_defined(transform)

Utility function to decide if a certain transform is defined at zero

Parameters

transformfunction

Returns

bool

True if transform is defined at 0 else False

tardis.visualization.widgets.grotrian.standardize(values, transform=<function <lambda>>, min_value=None, max_value=None, zero_undefined_offset=0)

Utility function to standardize displayed values like wavelengths, num_packets, levels populations to the range [0, 1] This helps in computing visual elements like widths, colors, etc.

Parameters

valuespandas.Series

The data to standardize

transformfunction, optional

Transformations like np.log, np.exp, etc. to apply on the data. Defaults to identity

min_valuefloat, optional

The lower bound of the range

max_valuefloat, optional

The upper bound of the range

zero_undefined_offsetint, optional

This is useful for log transformation because log(0) is -inf. Hence, value=0 gives y=0 while the output for other values start at zero_undefined_offset (y = log(value) + zero_undefined_offset) Default value is 0

Returns

pandas.Series

Values after standardization