def f_decorated(*args, **kwargs): filename = kwargs.pop('filename', None) if filename: makedirs(os.path.dirname(filename)) ext = os.path.splitext(filename)[1][1:] if filename else None if ext in ('pdf', 'pgf'): # setup fonts using Latin Modern mpl.rcParams.update({ "font.family": ["serif"], "font.serif": ["Latin Modern Roman"], "font.sans-serif": ["Latin Modern Sans"], "font.monospace": ["Latin Modern Mono"] }) elif ext == 'png': mpl.rcParams.update({"savefig.dpi": 240}) f(*args, **kwargs) if filename: if ext == "tikz": from matplotlib2tikz import save as tikz_save tikz_save(filename, figureheight='\\figureheight', figurewidth='\\figurewidth') else: plt.savefig(filename, bbox_inches='tight')
def get_pseudospectra_wrap( df_weights, df_nauto, tmask, pmask, tnorm, pnorm, pixel_size, lmax, n_x, h5_weight_basedir, h5_signal_basedir, weight_name, processes, compress_level, outdir, mapcase, name, isim, nullsplit, redirect=False, mode='multithreading', ): makedirs(outdir) filename = os.path.join( outdir, name if nullsplit is None else '_'.join((name, nullsplit))) outpath = filename + '.hdf5' if os.path.isfile(outpath): print('{} existed. Skipping...'.format(outpath)) return df = get_pseudospectra_IO( df_weights, df_nauto, tmask, pmask, tnorm, pnorm, pixel_size, lmax, n_x, h5_weight_basedir, h5_signal_basedir, weight_name, processes, mapcase, name, isim, nullsplit, stdout=filename + '.out' if redirect else None, stderr=filename + '.err' if redirect else None, timeit_filename=filename + '.time', mode=mode, ) df.to_hdf(outpath, 'pseudospectra', format='table', complevel=compress_level, fletcher32=True) return
def main(args): makedirs(args.o) with h5py.File(args.first, "r") as f1: with h5py.File(args.second, "r") as f2: plot_h5diff(f1, f2, args.o, prefix=os.path.splitext(os.path.basename( args.first))[0], verbose=args.verbose)
def convert(basedir, output, in_path, protocol=2): out_path = in_path.replace(basedir, output, 1) if os.path.isfile(out_path): print('{} existed, skip.'.format(out_path)) return makedirs(os.path.dirname(out_path)) with open(in_path, 'rb') as f: data = pickle.load(f) if PY2 else pickle.load(f, encoding='latin1') with open(out_path, 'wb') as f: pickle.dump(data, f, protocol=2) return