def f(x, p, pf): return fnc.f(test.fn, x, p, pf)
def f(x, p, pf): return fnc.f(cubic_spline_fit.fn, x, p, pf)
from f_fnc import fnc import numpy p = numpy.asarray([5.0]) pf = numpy.asarray([0.0]) y = fnc.f("fs_embedding",2.0, p, pf) print(y)
def run(): print("Plot") potfiles = [] for file_name in os.listdir("pots"): if (".pot" in file_name): potfiles.append(file_name) for file_name in potfiles: print(file_name) x = numpy.linspace(0.0, 6.5, 1001) y = numpy.zeros((1001, ), ) p_fixed = numpy.asarray([0.0]) fh = open("pots" + "/" + file_name, 'r') for line in fh: line = plot.one_space(line.strip()) f = line.split(" ") print(f) if (f[0] == "#TYPE"): f_name = f[1] elif (f[0] == "#P"): p = numpy.asarray(f[1:]) elif (f[0] == "#PF"): p_fixed = numpy.asarray(f[1:]) elif (f[0] == "#TITLE"): title = line[6:].strip() elif (f[0] == "#X"): xlabel = line[2:].strip() elif (f[0] == "#Y"): ylabel = line[2:].strip() elif (f[0] == "#PLOT"): plot_name = line[5:].strip() fh.close() # for n in range(len(x)): y[n] = fnc.f(f_name, x[n], p, p_fixed) print(title) plt.figure(figsize=(8, 6)) plt.title(title) plt.xlabel(xlabel) plt.ylabel(ylabel) plt.rc('font', family='serif') plt.rc('xtick', labelsize='x-small') plt.rc('ytick', labelsize='x-small') plt.plot(x, y, color='k', ls='solid') if (max(y) > 1.0e4): plt.ylim(-5, 20) if (min(y) < -1.0e4): plt.ylim(-50, 50) plt.grid(True) plt.savefig("pots/" + plot_name, type='efs') exit() plot_dir = "" plot_file = sys.argv[2] pot_files = [] fh = open(plot_dir + "/" + plot_file, 'r') for line in fh: line = plot.one_space(line.strip()) f = line.split(" ") if (f[0] == "FILE"): pot_files.append(f[1]) fh.close() for file_name in pot_files: x = numpy.linspace(0.0, 6.5, 1001) y = numpy.zeros((1001, ), ) fh = open(plot_dir + "/" + file_name, 'r') for line in fh: line = plot.one_space(line.strip()) f = line.split(" ") if (f[0] == "#TYPE"): f_name = f[1] elif (f[0] == "#P"): p = numpy.asarray(f[1:]) elif (f[0] == "#PF"): p_fixed = numpy.asarray(f[1:]) fh.close() for n in range(len(x)): y[n] = fnc.f(f_name, x[n], p, p_fixed) plot_name = file_name.split(".") plot_name = plot_name[0] + '.eps' plt.plot(x, y, 'b') plt.savefig(plot_dir + "/" + plot_name, type='efs') plt.show() """