Source code for g3d.flow.solver.rhs.fluxmodifiers.scale

"""flux scale operator (flux modifier)"""
from g3d.flow.solver.rhs.fluxmodifiers import FluxModifier




class FluxScale(FluxModifier):
    """

    """

    def __init__(self) -> None:
        super().__init__()



    def evaluate(self) -> None:
        """

        """
        # divide net fluxes by cell volumes to obtain time derivatives in all cells
        idx1, idx2 = self._get_range()
        for i in range(self._n_var):
            self._pointers[i][idx1:idx2] = (
                    self._pointers[i][idx1:idx2] * self._voli[idx1:idx2] * self._dt[idx1:idx2])