mpl.rcParams["legend.fontsize"] = 20
# Hydro Risk Curves and current storage situation
path = """/home/dw/comit/data"""
os.chdir(path)
connection = ea.DW_connect(linux=True)
print "Getting COMIT and HRC data"
inflow_file = os.path.join("/home", "dw", "comit", "data", "inflows.pickle")
storage_file = os.path.join("/home", "dw", "comit", "data", "storage.pickle")
ea_hrc = os.path.join("/home", "dw", "comit", "data", "hydroriskcurves", "HRC_History_20130618.csv")
so_hrc = """http://www.systemoperator.co.nz/f3933,83729137/HRC_data_2013_2014_-_September_update.xlsx"""
inflows, storage = ea.get_comit_data(inflow_file, storage_file)
NZ = ea.panel_beater(storage.sum(axis=1), inflows.sum(axis=1), 365)
Taupo = ea.panel_beater(storage["Taupo"], inflows["Taupo"], 365)
Tekapo = ea.panel_beater(storage["Tekapo"], inflows["Tekapo"], 365)
Pukaki = ea.panel_beater(storage["Pukaki"], inflows["Pukaki"], 365)
Hawea = ea.panel_beater(storage["Hawea"], inflows["Hawea"], 365)
TeAnauManapouri = ea.panel_beater(
storage["TeAnau"] + storage["Manapouri"], inflows["TeAnau"] + inflows["Manapouri"], 365
)
NZ_HRC_SO, SI_HRC_SO = ea.get_SO_HRC(so_hrc)
NZ_HRC, SI_HRC = ea.get_EA_HRC(ea_hrc)
SI_HRC2 = SI_HRC_SO["6/2013":].append(SI_HRC["12/2012":"5/2013"])
NZ_HRC2 = NZ_HRC_SO["6/2013":].append(NZ_HRC["12/2012":"5/2013"])
