import os
from acme_diags.parameter.core_parameter import CoreParameter
from acme_diags.parameter.area_mean_time_series_parameter import AreaMeanTimeSeriesParameter

from acme_diags.run import runner

param = CoreParameter()

#For compy
machine_path_prefix = '/compyfs/e3sm_diags_data/'
#For cori
#machine_path_prefix = '/global/project/projectdirs/acme/acme_diags'

param.reference_data_path = os.path.join(machine_path_prefix, 'obs_for_e3sm_diags/time-series/')
param.test_data_path = os.path.join(machine_path_prefix, 'test_model_data_for_acme_diags/time-series/E3SM_v1/')
param.test_name = 'e3sm_v1'

param.output_format = ['png', 'pdf', 'svg']
param.output_format_subplot = ['pdf']
param.dpi = 300

prefix = '/compyfs/www/gola749/examples/'
param.results_dir = os.path.join(prefix, 'area_mean_with_obs')
#param.multiprocessing = True
#param.num_workers =  40

# We're passing in this new object as well, in
# addtion to the CoreParameter object.

ts_param = AreaMeanTimeSeriesParameter()
#ts_param.ref_names = ['none']   #This setting plot model data only
ts_param.start_yr = '2002'
ts_param.end_yr = '2008'

runner.sets_to_run = ['area_mean_time_series']
runner.run_diags([param, ts_param])