Source code for g3d.flow.solver.step.crank_nicolson

from typing import Any
from g3d.flow.solver.step import DualTimeStepper



[docs]
class CrankNicolson(DualTimeStepper):
    def __init__(self, *args: Any, **kwargs: Any) -> None:
        super().__init__(*args, **kwargs)

    def _flux_inc(self, RK_stage: int = 1) -> None:
        self._cons.WAXPY(0.5, self._cons_incr, self._cons_curr)
        self._flux_inc_common(RK_stage)