import xarray as xr
import sys



def merge_nc(gcm):
    top_dir = "/home/disk/rocinante/DATA/temp/kcp3/forcings/2.raw_nc/"
    bot_dir = "/home/disk/rocinante/DATA/temp/kcp3/forcings/xtra/2.raw_nc/"
    out_dir = "/home/disk/rocinante/DATA/temp/kcp3/forcings/xtra/2.big_raw/"


    ifile_top = "{}/{}".format(top_dir, gcm)
    ifile_bot = "{}/{}".format(bot_dir, gcm)
    ofile = "{}/{}".format(out_dir, gcm)
    

    top = xr.open_dataset(ifile_top)
    print(ifile_top)
    bot = xr.open_dataset(ifile_bot)
    print(ifile_bot)
    top = top.sel(y=slice(2,12))
    
    print(ofile)
    out = xr.concat([top, bot], dim='y')
    out.to_netcdf(ofile)


## manual input
#-----------------------
#args = sys.argv
#gcm = args[1]

## procedural input

gcms = {
    "access1.0_RCP85.nc",
    "access1.3_RCP85.nc",
    "bcc-csm1.1_RCP85.nc",
    "canesm2_RCP85.nc",
    "ccsm4_RCP85.nc",
    "csiro-mk3.6.0_RCP85.nc",
    "fgoals-g2_RCP85.nc",
    "gfdl-cm3_RCP85.nc",
    "giss-e2-h_RCP85.nc",
    "miroc5_RCP85.nc",
    "mri-cgcm3_RCP85.nc",
    #"noresm1-m_RCP85.nc",
}

for g in gcms:
    merge_nc(g)