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

param = CoreParameter()

#param.test_data_path = '/Users/zhang40/Documents/ACME/e3sm_arm_diags/data/post-processed/20210122.F2010.armsites'
#param.test_name = '20210122.F2010.armsites' 
#param.test_start_yr = '0001'
#param.test_end_yr = '0001'

param.test_data_path = '/Users/zhang40/Documents/e3sm_diags_data/e3sm_diags_test_data/E3SM_v2'
param.test_name = 'E3SM_v2' 
param.test_start_yr = '2000'
param.test_end_yr = '2000'
param.reference_data_path = '/Users/zhang40/Documents/ACME/ARMDiags/data/arm_diags_data_v3.0_09232022/observation'
#param.ref_name = 'armdiags'
param.run_type = 'model_vs_obs'
prefix = '/Users/zhang40/Downloads'
#param.multiprocessing = True
#param.num_workers = 4
param.results_dir = os.path.join(prefix, 'mixed_phase_partition')
#param.results_dir = os.path.join(prefix, 'arm_diags_v3_m2o_diurnal_cycle')
#param.test_timeseries_input = True
#param.ref_timeseries_input = True
#param.ref_start_yr = '0001'
#param.ref_end_yr = '0001'

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