from src.surfEmis_driver import gen_surf_emis
from src.elevEmis_driver import gen_elev_emis

files = ['BC-em-anthro_input4MIPs_emissions_CMIP_CEDS-2021-04-21_gn_199901-201912.nc',\
         'OC-em-anthro_input4MIPs_emissions_CMIP_CEDS-2021-04-21_gn_199901-201912.nc',\
         'SO2-em-anthro_input4MIPs_emissions_CMIP_CEDS-2021-04-21_gn_199901-201912.nc',\
         'SO2-em-anthro_input4MIPs_emissions_CMIP_CEDS-2021-04-21_gn_199901-201912.nc',\
         'SO2-em-anthro_input4MIPs_emissions_CMIP_CEDS-2021-04-21_gn_199901-201912.nc']

#files = [None]*5

# files = ['BC-em-anthro_input4MIPs_emissions_CMIP_CEDS-2021-04-21_gn_1850Clim.nc',\
#          'OC-em-anthro_input4MIPs_emissions_CMIP_CEDS-2021-04-21_gn_1850Clim.nc',\
#          'SO2-em-anthro_input4MIPs_emissions_CMIP_CEDS-2021-04-21_gn_1850Clim.nc',\
#          'SO2-em-anthro_input4MIPs_emissions_CMIP_CEDS-2021-04-21_gn_1850Clim.nc',\
#          'SO2-em-anthro_input4MIPs_emissions_CMIP_CEDS-2021-04-21_gn_1850Clim.nc']

#files = ['BC-em-biomassburning_input4MIPs_emissions_CMIP_VUA-CMIP-BB4CMIP6-1-2_gn_1850Clim.nc',\
#         'OC-em-biomassburning_input4MIPs_emissions_CMIP_VUA-CMIP-BB4CMIP6-1-2_gn_1850Clim.nc',\
#         'SO2-em-biomassburning_input4MIPs_emissions_CMIP_VUA-CMIP-BB4CMIP6-1-2_gn_1850Clim.nc',\
#         'SO2-em-biomassburning_input4MIPs_emissions_CMIP_VUA-CMIP-BB4CMIP6-1-2_gn_1850Clim.nc',\
#         'SO2-em-biomassburning_input4MIPs_emissions_CMIP_VUA-CMIP-BB4CMIP6-1-2_gn_1850Clim.nc']
# CUS_ne32x32pg2_scrip.nc, CUS_ne32x32pg2_scrip.nc, CUS_ne32x32pg2_scrip.nc, CUS_ne32x32pg2_scrip.nc
syr = 1845
eyr = 1855
for spc,file in zip(['bc_a4','pom_a4','so2','so4_a1','so4_a2'],files):
    print('Start year:',syr)
    print('End year:',eyr)
    aa = gen_surf_emis(syr,eyr,file,variable=spc,output='E3SM_surf_emis_CUS.nc',\
                                 indir='/compyfs/hass877/inpData/HR_data',mean=True,mean_yr=1850,\
                                 outdir='/compyfs/hass877/inpData/HR_data',\
                                 grid='CUS_ne32x32pg2_scrip.nc',cgrid='SE_gridarea_CUS.nc')

    #aa = get_species_elev_params(syr,eyr,file,variable=spc,output='E3SM_elev_emis_CUS.nc',\
    #                             indir='/compyfs/hass877/inpData/HR_data',mean=True,mean_yr=1850,\
    #                             outdir='/compyfs/hass877/inpData/HR_data',checker=True,\
    #                             grid='CUS_ne32x32pg2_scrip.nc',cgrid='SE_gridarea_CUS.nc')
    aa.prod_emis()

aa.combine_numvars('num_a1')
aa.combine_numvars('num_a2')
aa.combine_numvars('num_a4')

print('\n======Processing Surface emissions Complete=======\n')

files = ['BC-em-biomassburning_input4MIPs_emissions_CMIP_VUA-CMIP-BB4CMIP6-1-2_gn_1850Clim.nc',\
         'OC-em-biomassburning_input4MIPs_emissions_CMIP_VUA-CMIP-BB4CMIP6-1-2_gn_1850Clim.nc',\
         'SO2-em-biomassburning_input4MIPs_emissions_CMIP_VUA-CMIP-BB4CMIP6-1-2_gn_1850Clim.nc',\
         'SO2-em-biomassburning_input4MIPs_emissions_CMIP_VUA-CMIP-BB4CMIP6-1-2_gn_1850Clim.nc',\
         'SO2-em-biomassburning_input4MIPs_emissions_CMIP_VUA-CMIP-BB4CMIP6-1-2_gn_1850Clim.nc']

files = [None]*5

files = ['BC-em-biomassburning_input4MIPs_emissions_CMIP_VUA-CMIP-BB4CMIP6-1-2_gn_1850Clim.nc',\
         'OC-em-biomassburning_input4MIPs_emissions_CMIP_VUA-CMIP-BB4CMIP6-1-2_gn_1850Clim.nc',\
         'SO2-em-biomassburning_input4MIPs_emissions_CMIP_VUA-CMIP-BB4CMIP6-1-2_gn_1850Clim.nc',\
         'SO2-em-biomassburning_input4MIPs_emissions_CMIP_VUA-CMIP-BB4CMIP6-1-2_gn_1850Clim.nc',\
         'SO2-em-biomassburning_input4MIPs_emissions_CMIP_VUA-CMIP-BB4CMIP6-1-2_gn_1850Clim.nc']

for spc,file in zip(['bc_a4','pom_a4','so2','so4_a1','so4_a2'],files):
    print('Start year:',syr)
    print('End year:',eyr)
    #aa = get_species_params(syr,eyr,file,variable=spc,output='E3SM_surf_emis_CUS.nc',\
    #                             indir='/compyfs/hass877/inpData/HR_data',mean=True,mean_yr=1850,\
    #                             outdir='/compyfs/hass877/inpData/HR_data',\
    #                             grid='CUS_ne32x32pg2_scrip.nc',cgrid='SE_gridarea_CUS.nc',checker=True)

    aa = gen_elev_emis(syr,eyr,file,variable=spc,output='E3SM_elev_emis_CUS.nc',\
                                 indir='/compyfs/hass877/inpData/HR_data',mean=True,mean_yr=1850,\
                                 outdir='/compyfs/hass877/inpData/HR_data',\
                                 grid='CUS_ne32x32pg2_scrip.nc',cgrid='SE_gridarea_CUS.nc')
    aa.prod_emis()

aa.combine_numvars('num_a1')
aa.combine_numvars('num_a2')
aa.combine_numvars('num_a4')
print('\n======Processing Elevated emissions Complete=======\n')