Source code for solrat.atom_model.shared.common_api.multi_slab_atmosphere

from solrat.atom_model.shared.common_api.constant_property_slab import ConstantPropertySlabAtmosphere
from solrat.atom_model.shared.object.stokes import Stokes
from solrat.engine.functions.decorators import log_method




class MultiSlabAtmosphere:
    r"""
    Container that consecutively combines multiple slabs to create a stratified atmosphere.
    """

    def __init__(self, *slabs: ConstantPropertySlabAtmosphere):
        self.slabs = slabs



    @log_method
    def forward(self, initial_stokes: Stokes) -> Stokes:
        r"""
        Propagate radiation through slabs sequentially (one after another).
        Each slab uses the output of the previous slab as input.

        :param initial_stokes:  Initial Stokes vector that is entering the slab.
        """

        current_stokes = self.slabs[0].forward(initial_stokes=initial_stokes)

        for i in range(1, len(self.slabs)):
            current_stokes = self.slabs[i].forward(initial_stokes=current_stokes)

        return current_stokes