"""
=================================
Sub-hourly ET from Jena, Germany
=================================
"""

if __name__ == '__main__':
    import os
    import site
    # add parent directory to path
    et_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
    print(et_dir)
    site.addsitedir(et_dir)

# %%

import pandas as pd
from ETUtil import PenmanMonteith

_df = pd.read_csv(os.path.join(et_dir, 'data/jen_obj.txt'))
# https://www.bgc-jena.mpg.de/wetter/Weatherstation.pdf
# TODO input df contains 'SWDR (W/m**2)', can that be used?
df = pd.DataFrame()
df[['rel_hum', 'wind_speed', 'temp', 'tdew',
    'rn']] = _df[['rh (%)', 'wv (m/s)', 'T (degC)', 'Tdew (degC)', 'Rn (W/m**2)']]

# rn = df['rn']
# rn = np.where(rn < 0.0, 0.0, rn)
# df['rn'] = rn

df.index = pd.to_datetime(_df['index'])
df.index.freq = pd.infer_freq(df.index)

units = {'temp': 'Centigrade',
         'rel_hum': 'percent',
         'tdew': 'Fahrenheit',
         'wind_speed': 'MilesPerHour'}

constants = dict()
constants['lat_dec_deg'] = 50.9
constants['altitude'] = 155
# These values are not accurate
constants['a_s'] = 0.23
constants['albedo'] = 0.23
constants['b_s'] = 0.5
constants['wind_z'] = 2

eto_model = PenmanMonteith(df, units=units, constants=constants, verbosity=2)

# %%

pet = eto_model()

# %%

pet.head()

# %%

eto_model.plot_inputs()

# %%

eto_model.plot_outputs()