Box2x2.py

from netCDF4 import Dataset
import numpy as np

# - grid file
gridfile = '/store/CT1/hmg2840/lbrodeau/eNATL60/eNATL60-I/coordinates_eNATL60.nc'

# - Define read data function
def read_datagrid(gridfile,latmin=None,latmax=None,lonmin=None,lonmax=None):
    """Return navlon,navlat."""
    ncfile = Dataset(gridfile,'r')
    # load navlon and navlat
    _navlon = ncfile.variables['glamt'][:,:]
    _navlat = ncfile.variables['gphit'][:,:]
    #-Define domain
    domain = (lonmin<_navlon) * (_navlon<lonmax) * (latmin<_navlat) * (_navlat<latmax)
    where = np.where(domain)
    vlats = _navlat[where]
    vlons = _navlon[where]
    #get indice
    jmin = where[0][vlats.argmin()]
    jmax = where[0][vlats.argmax()]
    imin = where[1][vlons.argmin()]
    imax = where[1][vlons.argmax()]
    #load arrays
    navlon = _navlon[jmin:jmax+1,imin:imax+1]
    navlat = _navlat[jmin:jmax+1,imin:imax+1]
    return navlon,navlat,jmin,jmax,imin,imax

# - Define box dimensions
Box_01 = ['32.0','30.0','-70.0','-68.0','Box_1','1']
Box_02 = ['32.0','30.0','-68.0','-66.0','Box_2','2']
Box_03 = ['32.0','30.0','-66.0','-64.0','Box_3','3']
Box_04 = ['32.0','30.0','-64.0','-62.0','Box_4','4']
Box_05 = ['32.0','30.0','-62.0','-60.0','Box_5','5']
Box_06 = ['32.0','30.0','-60.0','-58.0','Box_6','6']
Box_07 = ['32.0','30.0','-58.0','-56.0','Box_7','7']
Box_08 = ['32.0','30.0','-56.0','-54.0','Box_8','8']
Box_09 = ['32.0','30.0','-54.0','-52.0','Box_9','9']
Box_10 = ['32.0','30.0','-52.0','-50.0','Box_10','10']
Box_11 = ['32.0','30.0','-50.0','-48.0','Box_11','11']
Box_12 = ['32.0','30.0','-48.0','-46.0','Box_12','12']
Box_13 = ['32.0','30.0','-46.0','-44.0','Box_13','13']
Box_14 = ['32.0','30.0','-44.0','-42.0','Box_14','14']

# - Generate box array
box_arr = []
for ii in np.arange(1,15,1):
    name = eval('Box_'+str(ii).zfill(2))
    box_arr.append(name)

#- defining dictionaries for the boxes
class box: # empty container.
    def __init__(self,name=None):
        self.name = name
        return

dictboxes = {}

for ibox in box_arr:
    
    y2 = eval(ibox[0]) ;y1 = eval(ibox[1]);
    x2 = eval(ibox[2]) ;x1 = eval(ibox[3]);
    box_name = ibox[4]
    print(box_name)

    # - Obtain navlon and Navlat
    navlon,navlat,jmin,jmax,imin,imax = read_datagrid(gridfile,latmin=y1,latmax=y2,lonmin=x2,lonmax=x1)

    # - save box parameter
    abox = box(box_name)
    abox.lonmin = navlon.min()
    abox.lonmax = navlon.max()
    abox.latmin = navlat.min()
    abox.latmax = navlat.max()
    abox.navlon = navlon
    abox.navlat = navlat
    abox.imin = imin
    abox.imax = imax
    abox.jmin = jmin
    abox.jmax = jmax
    abox.nb = ibox[5]
    dictboxes[box_name] = abox

boxes = dictboxes.values()