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_data_path = '/Users/zhang40/Documents/ACME_simulations'
#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/O3'
#param.reference_name = '20210603.v2rc3c.piControl.northamericax4v1pg2_WC14to60E2r3.chrysalis'
param.ref_name = '20210603.v2rc3c.piControl.northamericax4v1pg2_WC14to60E2r3.chrysalis'
#param.reference_data_path = '/Users/zhang40/Documents/ACME_simulations'
#param.refer_name = '20180215.DECKv1b_H1.ne30_oEC.edison'
param.run_type = 'model_vs_model'
prefix = '/Users/zhang40/Downloads'
#param.seasons = ["ANN"]
param.multiprocessing = True
#param.num_workers = 16
param.results_dir = os.path.join(prefix, 'O3_test_m2m')
#param.no_viewer = True


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