import os
from acme_diags.parameter.core_parameter import CoreParameter
from acme_diags.run import runner

param = CoreParameter()

param.reference_data_path = '/global/cscratch1/sd/beharrop/E3SM_simulations/v1DECK/HR_firstish20/reg_climo/yrs0007_0025/'
param.reference_name = 'theta.20180906.branch_noCNT.A_WCYCL1950S_CMIP6_HR.ne120_oRRS18v3_ICG'
param.short_reference_name = 'firstish_20'
param.test_data_path = '/global/cscratch1/sd/beharrop/E3SM_simulations/v1DECK/HR_last20/reg_climo/theta.20190910.branch_noCNT.n438b.unc03.A_WCYCL1950S_CMIP6_HR.ne120_oRRS18v3_ICG/yrs0057_0076/'
param.test_name = 'theta.20190910.branch_noCNT.n438b.unc03.A_WCYCL1950S_CMIP6_HR.ne120_oRRS18v3_ICG'
param.short_test_name = 'last_20'
param.seasons = ["ANN","DJF", "MAM", "JJA", "SON"]
param.run_type = 'model_vs_model'

param.diff_levels = [-2,-1,-.5,-.4,-.3,-.2,-.1,.1,.2,.3,.4,.5,1,2]

prefix = '/global/cscratch1/sd/beharrop/E3SM_simulations/v1DECK/HR_last20/'
param.results_dir = os.path.join(prefix, 'mod_v_mod_comp_diags')
param.multiprocessing = True
param.num_workers = 32

#use below to run all core sets of diags:
#runner.sets_to_run = ['lat_lon','zonal_mean_xy', 'zonal_mean_2d', 'polar', 'cosp_histogram', 'meridional_mean_2d']
#use below to run lat_lon map only
runner.sets_to_run = ['lat_lon', 'zonal_mean_xy', 'zonal_mean_2d']
runner.run_diags([param])