Source code for pycif.plugins.datastreams.fields.lmdz_prodloss3d.write

from .....utils.classes.fluxes import Flux
import numpy as np




def write(self, prescr_file, prodloss, mode="a", **kwargs):
    """Write prescribed species files for LMDZ

    Args:
        self (Fluxes): the Fluxes plugin
        prescr_file (str): the file where to write fluxes
        prodloss (xarray.DataArray): prescribed species data to write
        mode (str): 'w' to overwrite, 'a' to append
        """

    prodloss_fwd = prodloss["fwd"].values
    prodloss_tl = prodloss["tl"].values
    np.transpose([prodloss_fwd, prodloss_tl], axes=(0, 4, 3, 2, 1)).T.tofile(prescr_file)