Skip to content
Snippets Groups Projects
make_figures.py 2.26 KiB
Newer Older
Pierre Augier's avatar
Pierre Augier committed
import sys
from runpy import run_path
Pierre Augier's avatar
Pierre Augier committed
from pprint import pprint

import subprocess
Pierre Augier's avatar
Pierre Augier committed

import matplotlib.pyplot as plt

Pierre Augier's avatar
Pierre Augier committed
from util import (
    has_to_be_made,
    has_to_save,
    here,
    paths_simuls_regimes,
    paths_1couple,
)
Pierre Augier's avatar
Pierre Augier committed
def make_fig(name, source=None, letter=None):
Pierre Augier's avatar
Pierre Augier committed
    if source is None:
        source = f"save_{name}.py"
Pierre Augier's avatar
Pierre Augier committed
    name = f"fig_{name}"
    if has_to_be_made(name, source):
        print(f"Running {source} to produce {name}")
        if letter is not None:
            command = f"{sys.executable} {here / source} SAVE"
            command += f" {letter}"
            subprocess.run(command.split(), check=True)
        else:
            run_path(here / source)
Pierre Augier's avatar
Pierre Augier committed


def make_table(name, source=None):
    if source is None:
        source = f"save_table_{name}.py"
    if has_to_be_made(f"table_{name}.tex", source):
Pierre Augier's avatar
Pierre Augier committed
        print(f"Running {source} to produce {name}")
        run_path(str(here / source))
Pierre Augier's avatar
Pierre Augier committed


make_table("methods_1couple")
make_table("better_simuls")
make_table("simuls_regimes")

make_fig("E_vs_time_N40_Ri20")
make_fig("means_vs_kmaxeta_N40_Ri20")
make_fig("spectra_1couple")

make_fig("kmaxeta_vs_FhR")
make_fig("epsK2overepsK_vs_FhR")

make_fig("isotropy_coef_vs_FhR")
make_fig("isotropy_diss_vs_R")
make_fig("isotropy_diss_vs_Fh")
make_fig("isotropy_velo_vs_R")
make_fig("isotropy_velo_vs_Fh")

vlabarre's avatar
vlabarre committed
make_fig("epsK_vs_FhR")
Pierre Augier's avatar
Pierre Augier committed
make_fig("mixing_coef_vs_FhR")
make_fig("mixing_coef_vs_Fh")

Pierre Augier's avatar
Pierre Augier committed
# make_fig("ratio_EA_EK_vs_FhR")
# make_fig("ratio_EA_EK_vs_Fh")

Pierre Augier's avatar
Pierre Augier committed
make_fig("ratio_velo_vs_Fh")
Pierre Augier's avatar
Pierre Augier committed
make_fig("ratio_EA_E_vs_FhR", "save_ratio_Ekeys_E_vs_FhR.py")
make_fig("ratio_EA_E_vs_Fh", "save_ratio_Ekeys_E_vs_Fh.py")
Pierre Augier's avatar
Pierre Augier committed
make_fig("ratio_length_scales_vs_Fh")

Pierre Augier's avatar
Pierre Augier committed
pprint(paths_1couple)
print("paths_simuls_regimes")
Pierre Augier's avatar
Pierre Augier committed
pprint(paths_simuls_regimes)
Pierre Augier's avatar
Pierre Augier committed

for letter in paths_simuls_regimes.keys():
Pierre Augier's avatar
Pierre Augier committed
    make_fig(f"spectra_regime_{letter}", "save_spectra_regimes.py", letter=letter)
vlabarre's avatar
vlabarre committed
    make_fig(f"seb_regime_{letter}", "save_seb_regimes.py", letter=letter)
Pierre Augier's avatar
Pierre Augier committed
    make_fig(
vlabarre's avatar
vlabarre committed
        f"ratio_spectra_regime_{letter}",
        "save_ratio_spectra_regimes.py",
Pierre Augier's avatar
Pierre Augier committed
        letter=letter,
    )
Pierre Augier's avatar
Pierre Augier committed

make_fig("spectra_1strat")
make_fig("spectra_1R")

vlabarre's avatar
vlabarre committed
make_fig("old_simuls_vs_FhR")

vlabarre's avatar
vlabarre committed
make_fig(f"spectra_khkz_regime_L", "save_spectra_khkz_regimes.py", letter="L")


Pierre Augier's avatar
Pierre Augier committed
if not has_to_save:
    plt.show()