jaxrts

A Python package for X-ray Thomson Scattering from dense plasmas, using jax.

Modules

absorption_edges_ionized_ions_data

Ionization Energy Data Generated by the flexible atomic code [Gu, 2008] (https://github.com/flexible-atomic-code/fac).

analysis

Some analysis functions, mainly used for benchmarking and testing.

bound_free

This submodule is dedicated to calculate the contribution of tightly bound electrons to the dynamic structure factor.

ee_localfieldcorrections

This submodule is dedicated to the calculation of static and dynamic local field corrections.

ee_localfieldcorrections_dornheim_2021

This file is the taken from https://github.com/ToDor90/LFC/blob/master/ESA_PARAMETRIZATION/ collecting both scripts in one, replacing numpy by jax.numpy and performing some qol changes.

elements

This submodule contains data for different chemical elements.

experimental

Contains functionalities that go beyond the jaxrts base implementation.

form_factors

This submodule is dedicated to form factors.

free_bound

This submodule is dedicated to calculate the contribution of the free bound contributions to the dynamic structure factor.

free_free

This submodule is dedicated to the calculation of the free electron dynamic structure.

helpers

Miscellaneous helper functions.

hnc_potentials

HNC Potentials

hypernetted_chain

This submodule is dedicated to the using the hypernetted chain approach to calculate static structure factors.

instrument_function

This submodule is dedicated to the modelling and handling of instrument functions.

ion_feature

This submodule is dedicated to the calculation of the ion-feature.

ipd

Module to account for Ionization Potential Depression (IPD)

literature

Submodule handeling the citation of references

math

This submodule is dedicated to math functions which occur frequently in the calculations.

models

This submodule contains high-level wrappers for the different Models implemented.

plasma_physics

This submodule contains basic formulas used in plasma physics.

plasmastate

Definition of the PlasmaState class, is describing the experiment's target.

saha

Module containing functions to solve the Saha-equation, linking the temperature of a plasma to it's ionization.

saving

This module allows for saving and loading jaxrts.models.Model, jaxrts.plasmastate.PlasmaState and other objects introduced in jaxrts.

setup

This submodule is dedicated to handling the setup configuration for the calculation of the synthetic spectra.

static_structure_factors

Module handeling the calculation of static structure factors.

units

This submodule defines the unit registry for calculations involving quantities, consisting of a magtitude and a unit.