diff --git a/pipeline.py b/pipeline.py index 5a66b46..4dd26a1 100644 --- a/pipeline.py +++ b/pipeline.py @@ -3,15 +3,13 @@ import argparse import glob import os -import time -from functools import reduce -from shutil import copy import numpy as np -from plotter import * -from postprocessor import * -from pp_params import * +from plotter import Plotter, plt +from postprocessor import PostProcessor, get_time +from comparator import Comparator +from pp_params import default_params, load_params fake_pp = PostProcessor() @@ -173,7 +171,7 @@ pp_args.add_argument( plot_args = parser.add_argument_group("plot", "Plot configuration") plot_args.add_argument( - "--colormap", help="Colormap used", choices=P.colormaps(), default="plasma" + "--colormap", help="Colormap used", choices=plt.colormaps(), default="plasma" ) plot_args.add_argument( "--format", @@ -213,20 +211,17 @@ pp_params.disk.nb_bin = args.nb_bin pp_params.pdf.nb_bin = args.pdf_nb_bin # extension for out files -P.style.use("seaborn-deep") +plt.style.use("seaborn-deep") if args.format == "pdf": - P.style.use("~/.config/matplotlib/pdf.mplstyle") + plt.style.use("~/.config/matplotlib/pdf.mplstyle") if args.beamer: - P.rcParams["font.family"] = "sans-serif" - P.rcParams["figure.figsize"] = (7, 4.5) + plt.rcParams["font.family"] = "sans-serif" + plt.rcParams["figure.figsize"] = (7, 4.5) # Plot properties -P.rcParams["image.cmap"] = args.colormap -P.rcParams["savefig.dpi"] = args.dpi -P.rcParams["lines.linewidth"] = 2 -P.rcParams["lines.markersize"] = 10 -P.rcParams["errorbar.capsize"] = 4 +plt.rcParams["image.cmap"] = args.colormap +plt.rcParams["savefig.dpi"] = args.dpi # List of id that were successfully computed nums_success = {} @@ -244,11 +239,6 @@ for run in runs: if not os.path.exists(path_out): os.makedirs(path_out) - try: - copy(path_in + "/disk.nml", path_out) - copy(path_in + "/output_00001/compilation.txt", path_out) - except: - pass nums_success[run] = []