#!/home/CWA_MDPS/.conda/envs/rd/bin/python
from pytools.caltools import interp_1d
import pytools.nctools as nct
import pytools.timetools as tt
import numpy as np
import os
# !!! run this file on betago !!!


def main():
    for year in [2012, 2013, 2014]:
        run(year=year)


def run(year):
    lonw, lone, lats, latn = 150, 151, 30, 31
    lonw, lone, lats, latn = 10, 11, 70, 71
    lonw, lone, lats, latn = 10, 11, -62, -61
    times, timee = None, None

    desPath = getDesPath(year)
    data, dims = nct.ncreadByDimRange(
        desPath, 'ssh', [[times, timee], [lats, latn], [lonw, lone]]
    )

    data = np.mean(data, axis=(-1, -2))
    print(f'{year = }, area={lonw, lone, lats, latn}')

    print(f'  nnans, min, max, std =' \
        + f' {np.sum(np.isnan(data)):5d}' \
        + f', {np.nanmin(data):5.2f}' \
        + f', {np.nanmax(data):5.2f}' \
        + f', {np.nanstd(data):5.2f}')


def getDesPath(year):
    return f'../data/merged/ssh_{year}.nc'



if __name__ == '__main__':
    main()