import xarray as xr


prism_dir = '/home/disk/rocinante/DATA/temp/TNC_stormwater/prism-daily/maps/'
wrf_dir = '/home/disk/rocinante/DATA/temp/TNC_stormwater/bias_correction/maps/'
out_dir = "/home/disk/rocinante/DATA/temp/TNC_stormwater/report/check/"

varlist = [
    'Quantile_OCT-MAR',
    'SEASONAL_TTL_APR-JUN',
    'SEASONAL_TTL_JAN-MAR',
    'SEASONAL_TTL_JUL-SEP',
    'SEASONAL_TTL_OCT-DEC',
]


for v in varlist:
    print(v)
    prism_file = "{}/PRISM-daily_{}_interp.nc".format(prism_dir, v)
    rawWRF_file = "{}/WRF-NARR_bc1.0-time_{}.nc".format(wrf_dir, v)
    bcWRF_file = "{}/PRISM-bc1.0_{}.nc".format(wrf_dir, v)

    prism = xr.open_dataset(prism_file)
    rawWRF = xr.open_dataset(rawWRF_file)
    bcWRF = xr.open_dataset(bcWRF_file)
    
    ratio_raw = rawWRF / prism
    ratio_bc = bcWRF / prism

    #print(ratio_raw)
    #print(ratio_bc)
    #prism.to_netcdf('{}/prism_')

    if 'Quantile' in v:
        for q in prism['quantile'].values:        
            qrawr = ratio_raw.sel(quantile=q)
            qbcr = ratio_bc.sel(quantile=q)
            qprism = prism.sel(quantile=q)
            qraw = rawWRF.sel(quantile=q)
            qbc = bcWRF.sel(quantile=q)
            
            
            
            qrawr.to_netcdf('{}/rawWRF_ratio_{}_q{}.nc'.format(out_dir, v, q))
            qbcr.to_netcdf('{}/bcWRF_ratio_{}_q{}.nc'.format(out_dir, v, q))
            qprism.to_netcdf('{}/prism_{}_q{}.nc'.format(out_dir, v, q))
            qraw.to_netcdf('{}/rawWRF_{}_q{}.nc'.format(out_dir, v, q))
            qbc.to_netcdf('{}/bcWRF_{}_q{}.nc'.format(out_dir, v, q))

        
    else:
        ratio_raw.to_netcdf('{}/rawWRF_ratio_{}.nc'.format(out_dir, v))
        ratio_bc.to_netcdf('{}/bcWRF_ratio_{}.nc'.format(out_dir, v))
        prism.to_netcdf('{}/prism_{}.nc'.format(out_dir, v))
        rawWRF.to_netcdf('{}/rawWRF_{}.nc'.format(out_dir, v))
        bcWRF.to_netcdf('{}/bcWRF_{}.nc'.format(out_dir, v))
        



    
#PRISM-bc1.0
#WRR-NARR_bc1.0-time
#PRISM

#PRISM-bc1.0_Quantile_OCT-MAR.nc
#PRISM-bc1.0_SEASONAL_TTL_APR-JUN.nc
#PRISM-bc1.0_SEASONAL_TTL_JAN-MAR.nc
#PRISM-bc1.0_SEASONAL_TTL_JUL-SEP.nc
#PRISM-bc1.0_SEASONAL_TTL_OCT-DEC.nc

#WRF-NARR_bc1.0-time_Quantile_OCT-MAR.nc
#WRF-NARR_bc1.0-time_SEASONAL_TTL_APR-JUN.nc
#WRF-NARR_bc1.0-time_SEASONAL_TTL_JAN-MAR.nc
#WRF-NARR_bc1.0-time_SEASONAL_TTL_JUL-SEP.nc
#WRF-NARR_bc1.0-time_SEASONAL_TTL_OCT-DEC.nc

#PRISM_Quantile_OCT-MAR_interp.nc
#PRISM_SEASONAL_TTL_APR-JUN_interp.nc
#PRISM_SEASONAL_TTL_JAN-MAR_interp.nc
#PRISM_SEASONAL_TTL_JUL-SEP_interp.nc
#PRISM_SEASONAL_TTL_OCT-DEC_interp.nc