From 4741c3476fe2cdd06e97fadef80eb814cad3fd02 Mon Sep 17 00:00:00 2001 From: Lionel GUEZ <guez@lmd.ens.fr> Date: Fri, 24 Jul 2020 17:16:06 +0200 Subject: [PATCH] Polish --- Analysis/plot_traj.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/Analysis/plot_traj.py b/Analysis/plot_traj.py index b3e37649..cd6f8154 100755 --- a/Analysis/plot_traj.py +++ b/Analysis/plot_traj.py @@ -19,8 +19,8 @@ def compute_ishape(n, k1, ishape_last): else: return ishape -def get_coord(k, i, k1, ishape_last, sr): - ishape = compute_ishape((k, i), k1, ishape_last) +def get_coord(n, k1, ishape_last, sr): + ishape = compute_ishape(n, k1, ishape_last) return sr.shape(ishape).points[0] if __name__ == "__main__": @@ -48,10 +48,10 @@ if __name__ == "__main__": for row_in in cr: k = int(row_in[0]) i = int(row_in[1]) - extr_1 = get_coord(k, i, k1, ishape_last, sr) + extr_1 = get_coord((k, i), k1, ishape_last, sr) k = int(row_in[2]) i = int(row_in[3]) - extr_2 = get_coord(k, i, k1, ishape_last, sr) + extr_2 = get_coord((k, i), k1, ishape_last, sr) plt.plot((extr_1[0], extr_2[0]), (extr_1[1], extr_2[1]), "-+", color = "black") -- GitLab