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

param = CoreParameter()

# Location of the ref data.
param.reference_data_path = '/global/homes/m/meng/scratch/camdiagout/climo/c7_F_tau_F.f2000/yrs1999_2000/'
# Name of the ref model data, used to find the climo files.
param.ref_name = 'c7_F_tau_F.f2000'
# An optional, shorter name to be used instead of the ref_name.
param.short_ref_name = 'Ref: noTAU'

# Location of the test data.
param.test_data_path = '/global/homes/m/meng/scratch/camdiagout/climo/c7_F_tau_T.f2000/yrs1999_2000/'
# Name of the test model data, used to find the climo files.
param.test_name = 'c7_F_tau_T.f2000'
# An optional, shorter name to be used instead of the test_name.
param.short_test_name = 'Test: TAU'

# What plotsets to run the diags on.
param.sets = ['lat_lon']
# What seasons to run the diags on.
# If not defined, diags are run on ['ANN', 'DJF', 'MAM', 'JJA', 'SON'].
param.seasons = ['ANN']
# Name of the folder where the results are stored.
# Change `prefix` to use your directory.
prefix = '/global/project/projectdirs/e3sm/www/zhang40/tests'
param.results_dir = os.path.join(prefix, 'tau_test_ANN')
# This parameter modifies the software to accommodate model vs model runs.
# The default setting for run_type is 'model_vs_obs'.
param.run_type = 'model_vs_model'

# Below are more optional arguments.

# 'mpl' is to create matplotlib plots, 'vcs' is for vcs plots.
param.backend = 'mpl'
# Title of the difference plots.
param.diff_title = 'Test Model - Ref Model'
# For running with multiprocessing.
multiprocessing = True
num_workers = 2

runner.sets_to_run = ['lat_lon']
runner.run_diags([param])