Polish

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")