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


param = CoreParameter()


# 'mpl' and 'vcs' are for matplotlib or vcs plots respectively.
#run_type= 'model_vs_obs'



multiprocessing = True
num_workers= 32

# Name of folder where all results will be stored.



param.reference_data_path = '/global/cscratch1/sd/terai/E3SM_NGD-ATMPhys/Convection_diagnostic/v2P3/ne30/climo_rgr/'
param.reference_name = 'E3SM_alpha5_59_v2candidate_NGD_Conv.wP3_ngdConv_Output.ne30pg2_r05_oECv3.compy'
param.short_reference_name='v2+P3'
#param.reference_data_path = '/global/cfs/cdirs/e3sm/acme_diags/obs_for_e3sm_diags/climatology/'
param.test_data_path = '/global/cscratch1/sd/terai/E3SM_NGD-ATMPhys/Convection_diagnostic/MCSPv1/ne30/climo_rgr/' #to modify
param.test_name = 'E3SM_alpha5_59_v2candidate_NGD_Conv.F2010SC5-P3_wP3_MCSPv1.ne30pg2_r05_oECv3.cori-knl'         #to modify
param.short_test_name = 'MCSPv1'                                                                                  #to modify

param.seasons = ["ANN","DJF","JJA"]

param.run_type = 'model_vs_model'
param.backend = 'mpl'


prefix = '/global/cfs/cdirs/e3sm/www/terai/v3_ConvAssessment/MCSPv1/'
param.results_dir = os.path.join(prefix, 'model')
#prefix = '/var/www/acme/acme-diags/zhang40/tests/'
#param.results_dir = os.path.join(prefix, 'lat_lon_demo')
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_xy', 'zonal_mean_2d', 'polar', 'meridional_mean_2d']
runner.run_diags([param])