Source code for pycif.plugins.datastreams.fluxes.point_sources.get_domain

import numpy as np
import itertools
import os
import pandas as pd
from .....utils.classes.domains import Domain




def get_domain(ref_dir, ref_file, input_interval, target_dir, tracer=None):
    """

    """

    # Looking for a reference file to read lon/lat in
    ref_file = list(tracer.input_files.values())[0][0]
    if not os.path.isfile(ref_file):
        raise Exception(
            "TNO domain could not be initialized as no file was found"
        )

    # Read the emissions file
    ds = pd.read_csv(ref_file, sep=";", parse_dates=["datei", "datef"])

    # Put it to a domain Plugin
    zlon, zlat = np.meshgrid(ds["lon"], ds["lat"])
    domain = Domain(nlon=len(ds), nlat=len(ds),
                    zlon=zlon, zlat=zlat,
                    zlonc=zlon, zlatc=zlat,
                    nlev=len(ds),
                    heights=ds["alt"], height_unit="m")

    return domain