pyFMS/pyfms/py_horiz_interp/interp.py at eb22abc7978ef8a586f7893a65c7ea9e640c3d92 · NOAA-GFDL/pyFMS · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
from pyfms.py_horiz_interp import horiz_interp


class ConserveInterp:
    def __init__(self, interp_id: int = None, save_xgrid_area: bool = False):

        """
        Python counterpart to FmsHorizInterp_type
        for conservative interpolation
        """

        self.interp_id = interp_id
        self.xgrid_area = None
        if interp_id is not None:
            self.nxgrid = horiz_interp.get_nxgrid(interp_id)
            self.i_src = horiz_interp.get_i_src(interp_id)
            self.j_src = horiz_interp.get_j_src(interp_id)
            self.i_dst = horiz_interp.get_i_dst(interp_id)
            self.j_dst = horiz_interp.get_j_dst(interp_id)
            self.nlon_src = horiz_interp.get_nlon_src(interp_id)
            self.nlat_src = horiz_interp.get_nlat_src(interp_id)
            self.nlon_dst = horiz_interp.get_nlon_dst(interp_id)
            self.nlat_dst = horiz_interp.get_nlat_dst(interp_id)
            self.interp_method = horiz_interp.get_interp_method(interp_id)
            self.area_frac_dst = horiz_interp.get_area_frac_dst(interp_id)
            if save_xgrid_area:
                self.xgrid_area = horiz_interp.get_xgrid_area(interp_id)
        else:
            self.nxgrid = None
            self.i_src = None
            self.j_src = None
            self.i_dst = None
            self.j_dst = None
            self.nlon_src = None
            self.nlat_src = None
            self.nlon_dst = None
            self.nlat_dst = None
            self.interp_method = None
            self.area_frac_dst = None

    def __repr__(self):

        repr_str = f"""
            interp_id: {self.interp_id}
            nxgrid: {self.nxgrid}
            nlon_src: {self.nlon_src}
            nlat_src: {self.nlat_src}
            nlon_dst: {self.nlon_dst}
            nlat_dst: {self.nlat_dst}
            interp_method: {self.interp_method}
            i_src_minmax: [{self.i_src.min()}, {self.i_src.max()}]
            j_src_minmax [{self.j_src.min()}, {self.j_src.max()}]
            i_dst_minmax: [{self.i_dst.min()}, {self.i_dst.max()}]
            j_dst_minmax: [{self.j_dst.min()}, {self.j_dst.max()}]
            area_frac_dst_minmax: [{self.area_frac_dst.min()}, {self.area_frac_dst.max()}]
        """

        return repr_str