Source code for tardis.montecarlo.estimators.continuum_radfield_properties

from dataclasses import dataclass

import numpy as np
import pandas as pd
from astropy import units as u

import tardis.constants as const
from tardis.io.atom_data import AtomData
from tardis.model.radiation_field_state import (
    DiluteBlackBodyRadiationFieldState,
)
from tardis.montecarlo.estimators.radfield_mc_estimators import (
    RadiationFieldMCEstimators,
)
from tardis.montecarlo.estimators.util import (
    bound_free_estimator_array2frame,
    integrate_array_by_blocks,
)
from tardis.plasma.properties.continuum_processes import PhotoIonBoltzmannFactor

H = const.h.cgs.value


[docs]class MCContinuumPropertiesSolver: def __init__( self, atom_data: AtomData, ): self.atom_data = atom_data
[docs] def solve( self, radfield_mc_estimators: RadiationFieldMCEstimators, time_simulation, volume, ): """ Solve for the continuum properties. Parameters ---------- radfield_mc_estimators : RadiationFieldMCEstimators The Monte Carlo estimators for the radiation field. time_simulation : float The simulation time. volume : float The volume of the cells. Returns ------- ContinuumProperties The calculated continuum properties. """ photo_ion_norm_factor = (time_simulation * volume * H) ** -1 photo_ionization_rate_coefficient = bound_free_estimator_array2frame( radfield_mc_estimators.photo_ion_estimator, self.atom_data.level2continuum_edge_idx, ) photo_ionization_rate_coefficient *= photo_ion_norm_factor stimulated_recomb_rate_factor = bound_free_estimator_array2frame( radfield_mc_estimators.stim_recomb_estimator, self.atom_data.level2continuum_edge_idx, ) stimulated_recomb_rate_factor *= photo_ion_norm_factor return ContinuumProperties( stimulated_recomb_rate_factor, photo_ionization_rate_coefficient )
[docs]class DiluteBlackBodyContinuumPropertiesSolver: def __init__(self, atom_data: AtomData) -> None: self.atom_data = atom_data
[docs] def solve( self, dilute_blackbody_radiationfield_state: DiluteBlackBodyRadiationFieldState, t_electrons: u.Quantity, ): """ Solve for the continuum properties. Parameters ---------- dilute_blackbody_radiationfield_state : DiluteBlackBodyRadiationFieldState The state of the dilute blackbody radiation field. t_electrons : u.Quantity The temperature of the electrons. Returns ------- ContinuumProperties The calculated continuum properties. """ photo_ion_boltzmann_factor = PhotoIonBoltzmannFactor.calculate( self.atom_data.photoionization_data, t_electrons ) mean_intensity_photo_ion_df = ( self.calculate_mean_intensity_photo_ion_table( dilute_blackbody_radiationfield_state ) ) photo_ion_rate_coeff = self.calculate_photo_ionization_rate_coefficient( mean_intensity_photo_ion_df ) stimulated_recomb_rate_coeff = ( self.calculate_stimulated_recomb_rate_factor( mean_intensity_photo_ion_df, photo_ion_boltzmann_factor, ) ) return ContinuumProperties( stimulated_recomb_rate_coeff, photo_ion_rate_coeff )
[docs] def calculate_photo_ionization_rate_coefficient( self, mean_intensity_photo_ion_df: pd.DataFrame, ): """ Calculate the photoionization rate coefficient. Parameters ---------- mean_intensity_photo_ion_df : pd.DataFrame The mean intensity of the radiation field at the frequencies of the photoionization table. Returns ------- pd.DataFrame The calculated photoionization rate coefficient. Notes ----- Equation 16 in Lucy 2003. """ gamma = mean_intensity_photo_ion_df.multiply( 4.0 * np.pi * self.atom_data.photoionization_data.x_sect / (self.atom_data.photoionization_data.nu * H), axis=0, ) gamma = integrate_array_by_blocks( gamma.values, self.atom_data.photoionization_data.nu.values, self.atom_data.photo_ion_block_references, ) gamma = pd.DataFrame(gamma, index=self.atom_data.photo_ion_unique_index) return gamma
[docs] def calculate_stimulated_recomb_rate_factor( self, mean_intensity_photo_ion_df: pd.DataFrame, photo_ion_boltzmann_factor: np.ndarray, ): """ Calculate the stimulated recombination rate factor (the phi_ik are multiplied in the plasma). Parameters ---------- mean_intensity_photo_ion_df : pd.DataFrame The mean intensity of the radiation field at the frequencies of the photoionization table. photo_ion_boltzmann_factor: np.ndarray The Boltzmann factor for the photoionization table. Returns ------- pd.DataFrame The calculated stimulated recombination rate factor. Notes ----- Equation 15 in Lucy 2003. """ alpha_stim_factor = ( mean_intensity_photo_ion_df * photo_ion_boltzmann_factor ) alpha_stim_factor = alpha_stim_factor.multiply( 4.0 * np.pi * self.atom_data.photoionization_data.x_sect / (self.atom_data.photoionization_data.nu * H), axis=0, ) alpha_stim_factor = integrate_array_by_blocks( alpha_stim_factor.values, self.atom_data.photoionization_data.nu.values, self.atom_data.photo_ion_block_references, ) alpha_stim_factor = pd.DataFrame( alpha_stim_factor, index=self.atom_data.photo_ion_unique_index ) return alpha_stim_factor
[docs] def calculate_mean_intensity_photo_ion_table( self, dilute_blackbody_radiationfield_state: DiluteBlackBodyRadiationFieldState, ): mean_intensity = ( dilute_blackbody_radiationfield_state.calculate_mean_intensity( self.atom_data.photoionization_data.nu.values ) ) mean_intensity_df = pd.DataFrame( mean_intensity, index=self.atom_data.photoionization_data.index, columns=np.arange( len(dilute_blackbody_radiationfield_state.t_radiative) ), ) return mean_intensity_df
[docs]@dataclass class ContinuumProperties: stimulated_recomb_rate_factor: pd.DataFrame photo_ionization_rate_coefficient: pd.DataFrame