import numpy as np
import h5py

from util.util import get_grid_values_all


def acspo_validate(oper_file, cspp_file):

    h5f_oper = h5py.File(oper_file, 'r')
    h5f_cspp = h5py.File(cspp_file, 'r')

    lon_cspp = get_grid_values_all(h5f_cspp, 'lon')
    lat_cspp = get_grid_values_all(h5f_cspp, 'lat')
    print('cspp shape: ', lat_cspp.shape)
    c_idx = lat_cspp.shape[1] // 2
    cntr_lon_cspp = lon_cspp[:, c_idx]
    cntr_lat_cspp = lat_cspp[:, c_idx]

    sst_cspp = get_grid_values_all(h5f_cspp, 'sea_surface_temperature')[0, ]
    l2p_flags_cspp = get_grid_values_all(h5f_cspp, 'l2p_flags')[0, ]

    lon_oper = get_grid_values_all(h5f_oper, 'lon')
    lat_oper = get_grid_values_all(h5f_oper, 'lat')
    cntr_lon_oper = lon_oper[:, c_idx]
    cntr_lat_oper = lat_oper[:, c_idx]
    print('oper shape: ', lat_oper.shape)
    sst_oper = get_grid_values_all(h5f_oper, 'sea_surface_temperature')[0, ]
    l2p_flags_oper = get_grid_values_all(h5f_oper, 'l2p_flags')[0, ]

    # generate a ndarray of boolean
    cspp_clear = (l2p_flags_cspp & (1 << 15)) != 0
    oper_clear = (l2p_flags_oper & (1 << 15)) != 0

    idxs_lat = np.isin(cntr_lat_oper, cntr_lat_cspp, assume_unique=True)
    idxs_lon = np.isin(cntr_lon_oper, cntr_lon_cspp, assume_unique=True)

    start_idx_oper = -1
    stop_idx_oper = -1
    start_idx_cspp = -1
    stop_idx_cspp = -1

    for k in range(len(cntr_lat_oper)):
        c_a = np.isclose(cntr_lat_oper[k], cntr_lat_cspp)
        if np.size(np.nonzero(c_a)[0]) == 1:
            if start_idx_oper == -1:
                start_idx_oper = k
                start_idx_cspp = np.nonzero(c_a)[0][0]
            else:
                stop_idx_oper = k
                stop_idx_cspp = np.nonzero(c_a)[0][0]

    print(start_idx_oper, stop_idx_oper)
    print(start_idx_cspp, stop_idx_cspp)