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

param = CoreParameter()

param.test_data_path = '/Users/zhang40/Documents/ACME/e3sm_arm_diags/data/post-processed/20210122.F2010.armsites'
#param.test_data_path = '/Users/zhang40/Documents/ACME/e3sm_arm_diags/data/post-processed/20200922.F2010SC5.ne30pg2_r05.armsites'
#param.test_data_path = '/Users/zhang40/Documents/ACME/e3sm_arm_diags/data/post-processed/DECKv1.A1.branch2000'
param.test_name = '20200922.F2010SC5.ne30pg2_r05.armsites' 
#param.test_name = 'DECKv1.A1.branch2000' 
#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/obs_for_e3sm_diags/climatology'
#param.reference_data_path = '/Users/zhang40/Documents/ARM_LLNL/repo/arm_diags_data_v2_release_fin_08072020/observation'
param.reference_data_path = '/Users/zhang40/Documents/ARM_LLNL/repo/arm_diags_data_v2.1_01272021/observation'
param.ref_name = 'armdiags'
param.run_type = 'model_vs_obs'
prefix = '/Users/zhang40/Downloads'
param.seasons = ["ANN"]
#param.multiprocessing = True
#param.num_workers = 4
param.results_dir = os.path.join(prefix, 'arm_diags_v2_test')
param.test_timeseries_input = True
param.ref_timeseries_input = True
param.test_start_yr = '0001'
param.test_end_yr = '0001'
param.ref_start_yr = '0001'
param.ref_end_yr = '0001'
#param.no_viewer = True


runner.sets_to_run = ['arm_diags']
runner.run_diags([param])