Skip to content

Commit 72eaf99

Browse files
refactor: increase font sizes of profiles plots
also relabel E_coal -> E_coll
1 parent 0bd4a5b commit 72eaf99

1 file changed

Lines changed: 25 additions & 11 deletions

File tree

scripts_for_plotting/compare_profiles.py

Lines changed: 25 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,18 @@
6262
assert args.cleo_path2build.is_dir(), f"cleo_path2build: {args.cleo_path2build}"
6363
assert args.path4figs.is_dir(), f"path4figs: {args.path4figs}"
6464

65+
# %% font sizes for beautifying plots
66+
SMALL_SIZE = 15
67+
MEDIUM_SIZE = 16
68+
BIG_SIZE = 18.5
69+
70+
plt.rc("font", size=MEDIUM_SIZE) # controls default text sizes
71+
plt.rc("axes", titlesize=BIG_SIZE) # fontsize of the axes title
72+
plt.rc("axes", labelsize=BIG_SIZE) # fontsize of the x and y labels
73+
plt.rc("xtick", labelsize=MEDIUM_SIZE) # fontsize of the tick labels
74+
plt.rc("ytick", labelsize=MEDIUM_SIZE) # fontsize of the tick labels
75+
plt.rc("legend", fontsize=BIG_SIZE) # legend fontsize
76+
plt.rc("figure", titlesize=BIG_SIZE) # fontsize of the figure title
6577

6678
# %%
6779
cleo_config, cleo_consts, cleo_gbxs, cleo_time = led.get_cleo_consts_gbxs_time(
@@ -126,7 +138,7 @@ def retrieve_different_profile_datasets(profiles_datasets, simtype):
126138
fullscheme_datasets, "fullscheme"
127139
)
128140
# %% Plot Hill figure 4 (top 2 rows only)
129-
fig, axes = plt.subplots(nrows=2, ncols=2, figsize=(9, 5), width_ratios=[3, 2])
141+
fig, axes = plt.subplots(nrows=2, ncols=2, figsize=(11, 7), width_ratios=[5, 3.7])
130142
gs = axes[0, -1].get_gridspec()
131143
axes[0, -1].remove()
132144
stdax = axes[1, -1]
@@ -172,7 +184,7 @@ def get_style(profile, simtype):
172184

173185
label_for_simtype = { # simtype: label for legend
174186
"condevap_only": "no precip",
175-
"fullscheme": "with $E_{coal}$",
187+
"fullscheme": "with $E_{coll}$",
176188
}
177189

178190
label_for_profile = { # profile: label for legend
@@ -272,6 +284,8 @@ def get_style(profile, simtype):
272284
stdax.set_xticklabels(labels)
273285
stdax.set_xlabel("")
274286

287+
fig.tight_layout()
288+
275289
plt.savefig(args.path4figs / "fig4_profiles.pdf", format="pdf", bbox_inches="tight")
276290
plt.show()
277291

@@ -357,10 +371,10 @@ def add_surfprecip_cumulative_to_plot(
357371

358372

359373
fullscheme_plot_kwargs = { # fullscheme_datasets : plot_kwargs
360-
"hydrostatic": {"color": "black", "linestyle": "-", "fmt": "+"},
361-
"approx_drhod_dz": {"color": "black", "linestyle": ":", "fmt": "+"},
362-
"exner_novapour": {"color": "black", "linestyle": "--", "fmt": "+"},
363-
"exner_novapour_constrho": {"color": "black", "linestyle": "-.", "fmt": "+"},
374+
# "hydrostatic": {"color": "purple", "linestyle": "-", "fmt": "+"},
375+
"approx_drhod_dz": {"color": "purple", "linestyle": ":", "fmt": "+"},
376+
"exner_novapour": {"color": "purple", "linestyle": "--", "fmt": "+"},
377+
"exner_novapour_constrho": {"color": "purple", "linestyle": "-.", "fmt": "+"},
364378
}
365379

366380
cleo_datasets_plot_kwargs = {
@@ -415,15 +429,15 @@ def add_surfprecip_cumulative_to_plot(
415429
}
416430

417431
fullscheme_note = { # profile: label for legend
418-
"hydrostatic": None,
432+
# "hydrostatic": None,
419433
"approx_drhod_dz": "H$_{no-vapour}$",
420434
"exner_novapour": "$\u03A0$",
421-
"exner_novapour_constrho": "$\u03A0_{constant-density}$",
435+
"exner_novapour_constrho": "CLEO with $E_{coll}$\n$\u03A0_{constant-density}$",
422436
}
423437

424438
cleo_datasets_note = {
425439
"is_precipTrue_numconc50p000_nsupers256_alpha0p000_fixedeffFalse": None,
426-
"is_precipTrue_numconc50p000_nsupers256_alpha0p500_fixedeffFalse": "CLEO, with $E_{coal}$, various \u03B1",
440+
"is_precipTrue_numconc50p000_nsupers256_alpha0p500_fixedeffFalse": "CLEO with $E_{coll}$,\nvarious \u03B1",
427441
"is_precipTrue_numconc50p000_nsupers256_alpha1p000_fixedeffFalse": None,
428442
"is_precipTrue_numconc50p000_nsupers256_alpha0p000_fixedeffTrue": None,
429443
"is_precipTrue_numconc50p000_nsupers256_alpha0p500_fixedeffTrue": "CLEO, various \u03B1",
@@ -436,7 +450,7 @@ def add_surfprecip_cumulative_to_plot(
436450
"is_precipTrue_numconc50p000_nsupers256_alpha1p000_fixedeffTrue": None,
437451
}
438452

439-
fig, axs = plt.subplots(nrows=1, ncols=2, width_ratios=[24, 1], figsize=(7, 4))
453+
fig, axs = plt.subplots(nrows=1, ncols=2, width_ratios=[24, 1], figsize=(10, 7))
440454

441455
for key, ds in fullscheme_datasets.items():
442456
if key in fullscheme_plot_kwargs:
@@ -540,7 +554,7 @@ def add_surfprecip_cumulative_to_plot(
540554
"is_precipTrue_numconc50p000_nsupers4096_alpha0p500_fixedeffFalse": 4096,
541555
}
542556

543-
fig, axs = plt.subplots(nrows=1, ncols=2, width_ratios=[6, 4])
557+
fig, axs = plt.subplots(nrows=1, ncols=2, width_ratios=[6, 4], figsize=(10, 7))
544558

545559
for key, ds in nsupers_datasets.items():
546560
if key in nsupers_datasets_plot_kwargs:

0 commit comments

Comments
 (0)