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

param = CoreParameter()

param.reference_data_path = '/global/cfs/cdirs/e3sm/e3sm_diags/obs_for_e3sm_diags/climatology/'
#param.test_data_path = '/global/cscratch1/sd/mengz/AMIP/cori-knl.E3SM_maint1_0_012022.e3smv1_v2like_freerun_10yrs_CTL.F2010C5-CMIP6-LR.ne30_ne30/run/e3sm_diag_input/'
#param.test_data_path = '/global/cscratch1/sd/mengz/E3SMv2_simulation/cori-knl.E3SM_maint2_0_012022.e3smv2_freerun_10yrs_CTL.F2010-CICE.ne30_ne30/run/e3sm_diag_input/' #'/global/cfs/cdirs/e3sm/e3sm_diags/test_model_data_for_acme_diags/climatology/'
param.test_data_path = '/global/cscratch1/sd/mengz/E3SMv2_simulation/cori-knl.E3SM_maint2_0_012022.e3smv2_freerun_10yrs_SENS_MINCDNC.F2010-CICE.ne30_ne30/run/e3sm_diag_input/'

#param.test_name = 'cori-knl.E3SM_maint1_0_012022.e3smv1_v2like_freerun_10yrs_CTL.F2010C5-CMIP6-LR.ne30_ne30'
#param.test_name = 'cori-knl.E3SM_maint2_0_012022.e3smv2_freerun_10yrs_CTL.F2010-CICE.ne30_ne30' #'20161118.beta0.FC5COSP.ne30_ne30.edison'
param.test_name = 'cori-knl.E3SM_maint2_0_012022.e3smv2_freerun_10yrs_SENS_MINCDNC.F2010-CICE.ne30_ne30'

param.seasons = ["ANN"]   #all seasons ["ANN","DJF", "MAM", "JJA", "SON"] will run,if comment out"

prefix = '/global/cfs/cdirs/e3sm/www/mengz/doc_examples/'
param.results_dir = os.path.join(prefix, 'lat_lon_v2_mincdnc')
# Use the following if running in parallel:
#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']
runner.run_diags([param])