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

param = CoreParameter()

param.reference_data_path = '/compyfs/e3sm_diags_data/obs_for_e3sm_diags/climatology/'
param.test_data_path = '/compyfs/zhou014/E3SM_simulations_AMIP_phs_irrig/tw_irr_phs_wm_inund.v1like.amip.ne30pg2_r05_oECv3.compy/post/atm/180x360_aave/clim/10yr/'
param.test_name = 'tw_irr_phs_wm_inund.v1like.amip.ne30pg2_r05_oECv3.compy'
#param.seasons = ["ANN"]   #all seasons ["ANN","DJF", "MAM", "JJA", "SON"] will run,if comment out"

prefix = '/compyfs/www/zhou014/E3SM/V2/'
param.results_dir = os.path.join(prefix, param.test_name)
# 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])