oasis-grids/oasisgrids.py at master · nichannah/oasis-grids · 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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
#!/usr/bin/env python

import sys, os
import argparse
import netCDF4 as nc
import numpy as np

sys.path.append('./esmgrids')
from esmgrids.oasis_grid import OasisGrid
from grid_factory import factory

def check_args(args):

    err = None

    if args.model_name in ['MOM', 'NEMO']:
        if args.model_hgrid is None or args.model_mask is None:
            err = 'Please provide MOM or NEMO grid definition and mask files.'

    return err

def check_file_exist(files):

    err = None

    for f in files:
        if f is not None and not os.path.exists(f):
            err = "Can't find input file {}.".format(f)

    return err

def main():

    parser = argparse.ArgumentParser()
    parser.add_argument("model_name", help="""
        The the model name. Supported names are:
            MOM   (1, 0.25 and 0.1 degree MOM ocean),
            NEMO  (ocean),
            SPE   (T42 spectral atmos),
            FVO   (2 degree atmos),
            CORE2 (CORE2 atmosphere),
            JRA55 (JRA55 atmosphere)
            """)
    parser.add_argument("--grid_name", default=None, help="""
        The OASIS name for the grid being created.
        """)
    parser.add_argument("--model_hgrid", default=None, help="""
        The model horizonatal grid definition file.
        Only needed for MOM and NEMO grids""")
    parser.add_argument("--model_mask", default=None, help="""
        The model mask file.
        Only needed for MOM and NEMO grids""")
    parser.add_argument("--model_cols", type=int, default=128, help="""
        Number of model columns
        Only needed for atmospheric grids""")
    parser.add_argument("--model_rows", type=int, default=64, help="""
        Number of model rows
        Only needed for atmospheric grids""")
    parser.add_argument("--grids", default="grids.nc",
                        help="The path to output OASIS grids.nc file")
    parser.add_argument("--areas", default="areas.nc",
                        help="The path to output OASIS areas.nc file")
    parser.add_argument("--masks", default="masks.nc",
                        help="The path to output OASIS masks.nc file")

    args = parser.parse_args()

    # The model name needs to be a certain length because OASIS requires that
    # grid names are exactly 4 characters long plus postfix.
    assert len(args.model_name) >= 3

    if args.grid_name is None:
        args.grid_name = args.model_name.lower()
    args.model_name = args.model_name.upper()

    err = check_args(args)
    if err is not None:
        print(err, file=sys.stderr)
        parser.print_help()
        return 1

    err = check_file_exist([args.model_hgrid, args.model_mask])
    if err is not None:
        print(err, file=sys.stderr)
        parser.print_help()
        return 1

    model_grid = factory(args.model_name, args.model_hgrid, args.model_mask,
                         args.model_rows, args.model_cols)
    coupling_grid = OasisGrid(args.grid_name, model_grid)

    coupling_grid.write_grids(args.grids)
    coupling_grid.write_areas(args.areas)
    coupling_grid.write_masks(args.masks)

    return 0

if __name__ == "__main__":
    sys.exit(main())