import os
from e3sm_diags.parameter.core_parameter import CoreParameter
from e3sm_diags.parameter.arm_diags_parameter import ARMDiagsParameter
from e3sm_diags.run import runner

param = CoreParameter()

model_config_shortname='v2P3'
param.test_name = 'v2+P3'
#infilename='ARMdiags.eam.h4.1996-01-01_2010-12-31.nc'
#param.short_test_name='v2+P3'
param.test_data_path = ''.join(['/global/cscratch1/sd/terai/E3SM_NGD-ATMPhys/Integration/',model_config_shortname,'/F20TR/arm_output/'])
param.reference_data_path = '/global/cfs/cdirs/e3sm/e3sm_diags/obs_for_e3sm_diags/arm-diags-data/'
param.ref_name = 'armdiags'
param.run_type = 'model_vs_obs'
prefix = ''.join(['/global/cfs/cdirs/e3sm/www/terai/Integration/F20TR/'])
#param.multiprocessing = True
#param.num_workers = 4
param.results_dir = os.path.join(prefix,'arm_diags_v2P3')
#param.test_timeseries_input = True
#param.ref_timeseries_input = True
param.test_start_yr = '1996'
param.test_end_yr = '2009'
param.ref_start_yr = '0001'
param.ref_end_yr = '0001'

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