import xarray as xr
import pandas as pd
import os
import sys

#idir = ""
#d1 = xr.open_dataset("/home/disk/rocinante/DATA/temp/WRF/var/pnnl/2d/T2_year/pnnl_T2_1988.nc")
#d2 = xr.open_dataset("/home/disk/rocinante/xiaodong/PNW_WRF.T2.1988.nc")
#d3 = d2.drop(['x','y']).rename({"time":"times", "x":"y", "y":"x"}).transpose("times", "x", "y")
#ddiff = d1.T2 - d3.T2
#d2 = d2.rename({"time":"times", "x":"y", "y":"x"})
# fix time values
# fix x,y,times label
# add lat/lon
# fix transpose
# drop x,y


odir = "/home/disk/rocinante/DATA/temp/WRF/var/pnnl/2d/new/"
geomap = "/home/disk/rocinante/DATA/temp/WRF/scripts/pnnl_geo.nc"

ifile = sys.argv[1]
ds = xr.open_dataset(ifile)
geo = xr.open_dataset(geomap)

s = os.path.basename(ifile).split('.')
var = s[1]
yr = s[2]

#print(ds)
ds = ds.drop_vars(ds.coords)
ds = ds.rename({"time":"times", "x":"y", "y":"x"})
ds = ds.transpose("times", "x", "y")
dates = pd.date_range("01/01/{}-00:00:00".format(yr), "12/31/{}-23:00:00".format(yr), freq='H')
ds = ds.assign_coords({"times":dates, "lat":geo.lat, "lon":geo.lon})

odir = "{}/{}_year/".format(odir, var)
os.makedirs(odir, exist_ok=True)
out = "{}/pnnl_{}_{}.nc".format(odir, var, yr)
print(out)
ds.to_netcdf(out, unlimited_dims='times')