Isca/run_script.py at main · jduffield65/Isca · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
from isca_tools.run import run_experiment
import os
# import os
# from functools import wraps
#
# import sh
#
# mkdir = sh.mkdir.bake('-p')
# cd = sh.cd
# git = sh.git.bake('--no-pager')
#
# P = os.path.join
# codedir_git = git.bake('-C', os.environ['GFDL_BASE'])
# git_test = codedir_git.log('-1', '--format="%H"').stdout

# When calling the script with no arguments, it just runs the experiment.
jobs_dir = os.path.join(os.environ['HOME'], 'Isca', 'jobs')  # all jobs saved here - CHANGE FOR EACH EXPERIMENT
# exp_dir = os.path.join(jobs_dir, 'tau_sweep/aquaplanet/depth=1/no_conv/k=1')  # specific experiment - CHANGE FOR EACH EXPERIMENT
# exp_dir = os.path.join(jobs_dir, 'tau_sweep/land/meridional_band/depth=1/bucket_evap/ras_conv/k=1_5')  # specific experiment - CHANGE FOR EACH EXPERIMENT
# exp_dir = os.path.join(jobs_dir, 'tau_sweep/land/meridional_band/evap/evap=0/k=0_6')
# exp_dir = os.path.join(jobs_dir, 'aquaplanet/vary_depth/depth=25')
# exp_dir = os.path.join(jobs_dir, 'aquaplanet/vary_obliquity/5')
# exp_dir = os.path.join(jobs_dir, 'aquaplanet/vary_depth/depth=1/k=0_3')
# exp_dir = os.path.join(jobs_dir, 'play/temp_tendency/k=3_5')
# exp_dir = os.path.join(jobs_dir, 'tau_sweep/land/meridional_band/evap/evap=0/k=1_5')
# exp_dir = os.path.join(jobs_dir, 'play/bucket/depth=5/k=1_5')
# exp_dir = os.path.join(jobs_dir, 'rossbypalooza/depth=20/lat_10/rough')
exp_dir = os.path.join(jobs_dir, 'aquaplanet/latent_heat_temp_only/base')
# exp_dir = os.path.join(jobs_dir, 'rossbypalooza/depth=20/all_nh/lat_10/rot_6x')
# exp_dir = os.path.join(jobs_dir, 'aquaplanet/latent_heat_temp_only/k=0_6')

namelist_file = os.path.join(exp_dir, 'namelist.nml')
diag_table_file = os.path.join(exp_dir, 'diag_table')
run_experiment(namelist_file, diag_table_file, slurm=True)

# For creating co2 timeseries .nc file
# from isca_tools.time_series.base import create_time_series_file
# def co2_func(days, pressure, lat, lon):
#     co2_val = np.ones((days.shape[0], pressure.shape[0], lat.shape[0], lon.shape[0])) * 500    # last 50 years have 500 ppmv
#     co2_val[days < 360 * 5] = 360      # First 40 years have 300ppmv
#     return co2_val
# create_time_series_file(os.path.join(exp_dir, 'co2_timeseries_t42.nc'), namelist_file, 'co2', co2_func, 360)

# # For creating land .nc file
# from isca_tools.land.base import write_land
# from isca_tools.plot import show_land
# # write_land('land.nc', namelist_file, 'square', [-90, 90, 180 - 30, 180 + 30])
# write_land('land.nc', namelist_file, 'square', [-90, 90, 180 - 15, 180 + 15])
# show_land(os.path.join(exp_dir, 'land.nc'))
#
# from isca_tools.utils.land import get_ocean_coords
# get_ocean_coords(namelist_file)