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/diagnostics/observations/Atm/climatology'
param.reference_data_path = '/pscratch/sd/x/xie7/ls4p/zppy/output/ls4p-custom8_CTL-mem/ls4p-custom8_CTL-mem1/ncremap'
param.ref_name = 'ls4p-custom8_CTL-mem1'

param.test_data_path = '/pscratch/sd/x/xie7/ls4p/zppy/output/ls4p-custom8_TOFD-mem/ls4p-custom8_TOFD-mem1/ncremap'
param.test_name = 'ls4p-custom8_TOFD-mem1'
param.run_type = 'model_vs_model'
#param.run_type = 'mvm'
param.seasons = ["05","06","07","08","JJA"]

prefix = '/global/cfs/cdirs/e3sm/www/xie7/diags_ls4p'
param.results_dir = os.path.join(prefix, 'ls4p_mvm_TOFD_1_CTL_1')

runner.sets_to_run = ['lat_lon',"zonal_mean_xy","zonal_mean_2d","polar","cosp_histogram","meridional_mean_2d","annual_cycle_zonal_mean","zonal_mean_2d_stratosphere","area_mean_time_series"]
runner.run_diags([param])