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


param = CoreParameter()

param.run_type = 'model_vs_model'
param.reference_data_path = '/pscratch/sd/x/xsongsio/E3SM_simulation/20230721.F2010New.v3alpha02.pm-cpu_intel_O0/diag/clim_rgr/'
param.reference_name = '20230721.F2010New.v3alpha02.pm-cpu_intel_O0'
param.test_data_path = '/pscratch/sd/x/xsongsio/E3SM_simulation/20230721.F2010New.v3alpha02.pm-cpu_intel/diag/clim_rgr/'
param.test_name = '20230721.F2010New.v3alpha02.pm-cpu_intel'
#param.seasons = ["ANN"]   #all seasons ["ANN","DJF", "MAM", "JJA", "SON"] will run,if comment out"

prefix = '/global/cfs/cdirs/e3sm/www/xsongsio/v3alpha02.pm-cpu_intel_O2-O0/'
param.results_dir = os.path.join(prefix, '2-11')
param.test_title = ' '
param.reference_title = ' '
param.diff_title = 'pm-cpu_intel_O2 - pm-cpu_intel_O0'
# 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']
#runner.sets_to_run = ['lat_lon','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])