skyscapes.physical_model.exojax.components.absorption#

Absorption opacity components.

Absorption iterates over the atmosphere’s MolecularSpecies tuple and sums per-molecule line-list / cross-section contributions. Each species owns its own opa engine and altitude-resolved mmr profile; this component is a thin orchestrator with no per-molecule state of its own.

Classes#

`Absorption`	Sum of per-species line-list / cross-section absorption.
`PrecomputedAbsorption`	Absorption with per-species `xsmatrix` precomputed at a fixed TP.

Module Contents#

class skyscapes.physical_model.exojax.components.absorption.Absorption[source]#

Bases: skyscapes.physical_model.exojax.components.base.AbstractAbsorption

Sum of per-species line-list / cross-section absorption.

Iterates over the species tuple, skipping any with opa is None (e.g. a species included purely for its Rayleigh contribution).

compute(species, Tarr, pressure, gravity, rt_engine)[source]#

Sum per-species absorption optical depth.

Parameters:

species (tuple[skyscapes.physical_model.exojax.components.species.MolecularSpecies, Ellipsis])
Tarr (jaxtyping.Array)
pressure (jaxtyping.Array)
gravity (jaxtyping.Array)

Return type:

skyscapes.physical_model.exojax.components.base.Contribution

class skyscapes.physical_model.exojax.components.absorption.PrecomputedAbsorption[source]#

Bases: skyscapes.physical_model.exojax.components.base.AbstractAbsorption

Absorption with per-species xsmatrix precomputed at a fixed TP.

Drop-in for Absorption for retrievals: holds each absorbing species’ cross-section matrix (computed once via opa.xsmatrix at the fixed temperature-pressure structure) and applies only the live mmr scaling. The spectrum is therefore differentiable in abundance but INERT IN TEMPERATURE – the stored xsmatrix bakes Tarr, so tp_profile leaves no longer change the output. To fit temperature, use the full recompute path (from_default_setup) instead. Build via ExoJaxPhysicalModel.for_retrieval.

xsmatrix_per_species corresponds, in order, to the species with opa is not None (the same ones Absorption would sum over).

xsmatrix_per_species: tuple[jaxtyping.Array, Ellipsis]#

compute(species, Tarr, pressure, gravity, rt_engine)[source]#

Sum per-species absorption from the stored cross-sections.

Parameters:

species (tuple[skyscapes.physical_model.exojax.components.species.MolecularSpecies, Ellipsis])
Tarr (jaxtyping.Array)
pressure (jaxtyping.Array)
gravity (jaxtyping.Array)

Return type:

skyscapes.physical_model.exojax.components.base.Contribution