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

args = sys.argv
ifile = args[1]
fn = os.path.basename(ifile)
#gcm = os.path.basename(os.path.dirname(ifile))
#opath = "/home/disk/rocinante/DATA/temp/WRF/scripts/regrid_c2c/longwave/output/prism_grid_snoho_tlapse/"
#opath = "/home/disk/rocinante/DATA/temp/WRF/scripts/regrid_c2c/longwave/output/delta_skkm_tlapse/"
#opath = "/home/disk/rocinante/DATA/temp/kcp3/forcings/7.final/snoho/"
#opath = "./{}/{}_tlapse/".format(os.path.dirname(os.path.dirname(ifile)), gcm)
#opath = "./{}_tlapse/".format(os.path.dirname(os.path.dirname(ifile)))
opath = "{}_tlapseHON-LS/".format(os.path.dirname(ifile))
print(fn)
print(opath)
os.makedirs(opath, exist_ok=True)


#lapse = np.array([5.5, 5.5, 5.5, 5.0, 4.5, 4.0, 4.0, 4.0, 4.0, 4.5, 5.5, 5.5 ])

lapse = np.array([5.5, 5.5, 5.5, 5.0, 4.5, 4.0, 4.0, 4.0, 2.0, 6.5, 7.5, 5.5 ])
#lapse = np.array([4.5, 5.5, 6.5, 7.0, 7.5, 8.0, 7.0, 6.5, 6.0, 5.0, 4.5, 4.5])
lapse /= -1000

df = pd.read_csv(ifile, header=None, sep=' ')
dates = pd.to_datetime(df[0], format='%m/%d/%Y-%H:%M:%S')
lr = [ lapse[x-1] for x in dates.dt.month]
df['lp'] = lr

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

