#!/usr/bin/env python
import pytools as pyt
import numpy as np


def main():
    srcPath = '../data/nnans/ssv_2011_sum.nc'
    desPath = '../data/land_sea_mask.nc'

    varName = 'ssv'
    ntPerYear = 1460
    cutOffRatio = 0.95

    nnans, dims = pyt.nct.ncreadByDimRange(srcPath, varName, [[None]*2]*3)
    nnans = np.squeeze(nnans, axis=0)
    # print(f'{nnans.shape =}')

    isLand = np.zeros(nnans.shape)
    isLand[(nnans > ntPerYear * cutOffRatio)] = 1

    pyt.nct.save(
        desPath, {
            'isLand': isLand,
            'lat': dims[-2],
            'lon': dims[-1],
        }, overwrite=True
    )


if __name__ == '__main__':
    main()