import netCDF4 as nc
import numpy as np

def get_MOM6grid(path:str,idm,jdm):
    """
    Convert MOM6 supergrid to regular array grid
    path     : string containing the path and name of the file to open
    idm, jdm : dimension for full domain
    """
    ## read supergrid
    dgrid=nc.Dataset(path)
    x=dgrid['x'][:]
    y=dgrid['y'][:]
    ## create plon and plat
    plon=np.zeros([jdm,idm])
    plat=np.zeros([jdm,idm])
    for j in np.arange(jdm):
       for i in np.arange(idm):
          plon[j,i] = x[2*j,2*i]  #HYCOM 1,1 is MOM6 2,2
          plat[j,i] = y[2*j,2*i]  #HYCOM 1,1 is MOM6 2,2

    return plon,plat