import xarray as xr
from glob import glob
import os

gcms = [
    "access1.0_RCP45",
    "access1.0_RCP85",
    "access1.3_RCP85",
    "bcc-csm1.1_RCP85",
    "canesm2_RCP85",
    "ccsm4_RCP85",
    "csiro-mk3.6.0_RCP85",
    "fgoals-g2_RCP85",
    "gfdl-cm3_RCP85",
    "giss-e2-h_RCP85",
    "miroc5_RCP85",
    "mri-cgcm3_RCP85",
    "noresm1-m_RCP85"
]
gcms = ["pnnl"]

locs = ['eugene', 'omak', 'salem', 'vancouver']
idir = "/home/disk/rocinante/DATA/temp/crystal_fire/slp/wrf_data/"
odir = "/home/disk/rocinante/DATA/temp/crystal_fire/slp/data_daily/"
for g in gcms:
    gdir = "{}/{}".format(odir, g)
    os.makedirs(gdir, exist_ok=True)
    for l in locs:
        print(g, l)
        data = glob('{}/{}/*{}*.nc'.format(idir, g, l))
        for d in data:
            ds = xr.open_dataset(d)
            ds = ds.resample(times='d').mean()
            fn = os.path.basename(d).replace('.nc_', '_')
            ds.to_netcdf("{}/{}".format(gdir, fn))