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

param = CoreParameter()

param.test_data_path = '/Users/zhang40/Documents/ACME_simulations/O3'
param.test_name = '20210603.v2rc3c.piControl.northamericax4v1pg2_WC14to60E2r3.chrysalis' 
#param.test_name = '20180215.DECKv1b_H1.ne30_oEC.edison'
#param.test_data_path = '/Users/zhang40/Documents/ACME_simulations/RRM_test'
#param.test_name = '20200404.alpha20_ne30_phydt.amip.northamericax4v1pg2_r0125_northamericax4v1pg2.cori-knl'
#param.short_test_name = 'alpha20_rrm_test'
param.reference_data_path = '/Users/zhang40/Documents/ACME_simulations/O3/climatology'
#param.reference_data_path = '/Users/zhang40/Documents/ACME_simulations/'
#param.refer_name = '20160520.A_WCYCL1850.ne30_oEC.edison.alpha6_01'
param.run_type = 'model_vs_obs'
prefix = '/Users/zhang40/Downloads'
#param.seasons = ["ANN"]
param.multiprocessing = True
#param.num_workers = 16
param.results_dir = os.path.join(prefix, 'O3_test1')
#param.no_viewer = True


runner.sets_to_run = ['lat_lon','annual_cycle_zonal_mean']
runner.run_diags([param])