# -*- coding: utf-8 -*-0
"""
Created on Fri Mar 15 15:02:02 2019

Run all AoB models
    
@author: theot
"""

main_fn="C:/Users/theot/Dropbox (Sense.)/2018-19 EA TPM CBA (Shared)/Step 3/AoB model/Models/"
model_fn="C:/Users/theot/Dropbox (Sense.)/2018-19 EA TPM CBA (Shared)/Step 3/AoB model/Models/"
output_fn="C:/Users/theot/Dropbox (Sense.)/2018-19 EA TPM CBA (Shared)/Step 3/AoB model/Output/"

model_names = ['AoB_All',
                'AoB_All_Major_Capex',
               'MWH_demand_Major_Capex',
               'AoB_demand',
               'AoB_demand_and_DG_investment',
             'AoB_demand_and_gen_investment',
             'AoB_WUNI',
             'AoB_No_AoB_on_existing',
             'AoB_All_Major_Capex_Tiwai_Off',
             'AoB_demand_Major_Capex',
             'AoB_demand_No_AoB_on_existing',
             'AoB_All_Major_Capex_2024',
             'AoB_All_Major_Capex_alternative',
             'AoB_All_Major_Capex_Gen_Benefits',
             'AoB_All_Major_Capex_30_70']

output_names = ['All',
                    'All_major_capex',
                    'Demand',
                    'Demand_and_DG_investment',
                 'Demand_and_gen_investment',
                 'All_major_capex_alternative',
                 'WUNI',
                 'No_AoB_on_existing',
                 'Demand_no_aob_on_existing',
                 'All_major_capex_tiwai_off',
                 'Demand_major_capex',
                 'MWh_Demand_major_capex',
                 'All_major_capex_2024',
                 'All_major_capex_gen_benefits',
                 'All_Major_Capex_30_70']


#Run all models - except sensitivity models 
for i, s in enumerate(model_names):
    print(s)
    runfile(model_fn+s+'.py')
    matplotlib.pyplot.close("all")
    

#Run Welfare and costs
for i, s in enumerate(output_names):
    print(s)
    runfile(output_fn+s+'/Aggregates'+'.py')    
    runfile(output_fn+s+'/Welfare_and_costs'+'.py')
    print("npv", npv/1000000)
    print("npv_p0", npv_p0/1000000)
    
#Run Welfare and costs
for i, s in enumerate(output_names):
    print(s)
    runfile(output_fn+s+'/Welfare_and_costs_CS'+'.py')
    print("npv", npv/1000000)
    print("npv_p0", npv_p0/1000000)