import pandas as pd
import sys
import os
import numpy as np

args = sys.argv
ifile = args[1]
fn = os.path.basename(ifile).replace('delta', 'data').replace('.csv', '')
#opath = "/home/disk/rocinante/DATA/temp/WRF/scripts/regrid_c2c/longwave/output/rawWRF_ptmin_delta/"
#opath = "/home/disk/rocinante/DATA/temp/WRF/scripts/regrid_c2c/longwave/output/prism_delta_pyl/"
#opath = "/home/disk/rocinante/DATA/temp/WRF/scripts/regrid_c2c/longwave/output/prism_delta_kndy_5km/"
opath = "/home/disk/rocinante/DATA/temp/WRF/scripts/regrid_c2c/longwave/metfile/green/prism_delta_green_v2/"

ddir = "/home/disk/rocinante/DATA/temp/WRF/validation/forc3/"
#deltaf = "/home/disk/tsuga2/jswon11/workdir/2021_09_KingCounty-Phase3/scripts/lelv/delta_livneh_15km/low_elev_delta.csv"
#deltaf = "/home/disk/tsuga2/jswon11/workdir/2021_09_KingCounty-Phase3/scripts/lelv/delta/prism_delta_pyl/low_elev_delta.csv"
#eltaf = "/home/disk/tsuga2/jswon11/workdir/2021_09_KingCounty-Phase3/scripts/lelv/delta/prism_delta_kndy_tzfix_5km/low_elev_delta.csv"
#deltaf = "/home/disk/tsuga2/jswon11/workdir/2021_09_KingCounty-Phase3/scripts/lelv/delta/prism_delta_green_tzfix/low_elev_delta.csv"
deltaf = "/home/disk/tsuga2/jswon11/workdir/2021_09_KingCounty-Phase3/scripts/lelv/delta/green_v2/low_elev_delta.csv"
dtfmt = "%m/%d/%Y-%H:%M:%S"
os.makedirs(opath, exist_ok=True)

ifile = "{}/{}".format(ddir, fn)

df = pd.read_csv(ifile, header=None, sep=' ')
df.columns = ['date', 'temp', 'wind', 'rh', 'sw', 'lw', 'prec']
df.index = pd.to_datetime(df.date, format=dtfmt)

dlt = pd.read_csv(deltaf)
mons = df.index.month.values
delta = [dlt.Tmin[x-1] for x in mons]
prec = np.array([dlt.Prec[x-1] for x in mons])
df.temp += delta
df.prec *= prec
df.index = df.index.shift(-8, freq='H')
df.date = df.index.strftime(date_format=dtfmt)

opath = "{}/{}".format(opath, fn)
df.to_csv(opath, header=False, index=False, sep=' ')


#def read_rf(ifile):
#    dtfmt = "%m/%d/%Y-%H:%M:%S"
#    df = pd.read_csv(ifile, header=None, sep=' ')
#    df.columns = ['date', 'temp', 'wind', 'rh', 'sw', 'lw', 'prec']
#    df.drop['date', axis=1]
#    df.index = pd.to_datetime(df.date, format=dtfmt)
#    df.index = df.index.shift(-8, freq='H')
#    return df


#/home/disk/rocinante/DATA/temp/pnnl/westWAOR/4_output/forcs_dhsvm/