import os
from e3sm_diags.parameter.core_parameter import CoreParameter
from e3sm_diags.parameter.area_mean_time_series_parameter import AreaMeanTimeSeriesParameter
from e3sm_diags.run import runner

param = CoreParameter()

param.reference_data_path = '/global/cfs/cdirs/e3sm/e3sm_diags/obs_for_e3sm_diags/climatology/'
param.test_data_path = '/global/cfs/cdirs/e3sm/feng809/e3sm_test/e3sm_diag/data/postprocessed_e3sm_v2_data_for_e3sm_diags/v2.LR.BGC-LNDATM.SPINUP/climatology/rgr/'
param.test_name = 'v2.LR.BGC-LNDATM.SPINUP'
param.multiprocessing = True
param.num_workers = 50
prefix = '/global/cfs/cdirs/e3sm/www/sfeng/doc_examples/v2.LR.BGC-LNDATM.SPINUP/001_050/'
param.results_dir = os.path.join(prefix, 'all_sets')

#
##Set specific parameters for new sets
ts_param = AreaMeanTimeSeriesParameter()
ts_param.reference_data_path = '/global/cfs/cdirs/e3sm/e3sm_diags/obs_for_e3sm_diags/time-series/'
ts_param.test_data_path = ''
ts_param.test_name = 'v2.LR.BGC-LNDATM.SPINUP'
ts_param.start_yr = '0001'
ts_param.end_yr = '0050'

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