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

param = CoreParameter()

param.run_type = 'model_vs_model'
param.reference_data_path = '/global/cscratch1/sd/gxlin/MMF_regrid_data/E3SM_MMF_ne30pg2_overview/climo_outdir/'
param.test_data_path = '/global/cscratch1/sd/gxlin/MMF_regrid_data/E3SM_MMF_MAML_ne30pg2_overview/climo_outdir/'
param.test_name = 'E3SM_MMF_MAML_ne30pg2_overview'
param.ref_name = 'E3SM_MMF_ne30pg2_overview'
param.short_test_name = 'MAML'
param.short_ref_name = 'MMF'
param.diff_title = 'MAML - MMF'

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

prefix = '/global/project/projectdirs/acme/www/gxlin/doc_examples/'
param.results_dir = os.path.join(prefix, 'model_to_model')
# 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','zonal_mean_xy','zonal_mean_2d']
runner.run_diags([param])