From 555c397d766a2f0b1ae1da0320776740fe82d7eb Mon Sep 17 00:00:00 2001
From: Cory <cory@balaton.dev>
Date: Thu, 19 Oct 2023 13:15:12 +0200
Subject: [PATCH] Fix directories list

---
 src/scripts/plot_relative_error.py | 21 +++++++++++----------
 1 file changed, 11 insertions(+), 10 deletions(-)

diff --git a/src/scripts/plot_relative_error.py b/src/scripts/plot_relative_error.py
index 54214c8..de8b692 100644
--- a/src/scripts/plot_relative_error.py
+++ b/src/scripts/plot_relative_error.py
@@ -2,10 +2,10 @@ import matplotlib.pyplot as plt
 import numpy as np
 
 def main():
-    directories = {
+    directories = [
         "output/relative_error/RK4/",
         "output/relative_error/euler/",
-    }
+    ]
     files = [
         "4000_steps.txt",
         "8000_steps.txt",
@@ -22,8 +22,12 @@ def main():
         "Relative error for the RK4 method",
         "Relative error for the forward Euler method"
     ]
+    methods = [
+        "rk4",
+        "euler"
+    ]
     fig1, axs1 = plt.subplots(2,1)
-    for i, (dir, title) in enumerate(zip(directories, titles)):
+    for i, (dir, title) in enumerate(list(zip(directories, titles))):
         max_err = []
         for label, file in zip(labels, files):
             with open(dir+file) as f:
@@ -33,17 +37,14 @@ def main():
                 max_err.append(max(r))
                 axs1[i].plot(t, r, label=label)
         
-            axs1[i].set(xlabel=r"t $(\mu s)$", ylabel = r"relative_error $(\mu m)$")
-
-        
+        axs1[i].set(xlabel=r"t $(\mu s)$", ylabel = r"relative_error $(\mu m)$")
         axs1[i].legend()
         axs1[i].set_title(title)
 
-        conv_rate = 1/3 * sum([np.log10(max_err[i+1]/max_err[i])/np.log10(.5) for i in range(3)])
-        print(conv_rate)
+        conv_rate = 1/3 * sum([np.log2(max_err[i+1]/max_err[i])/np.log2(.5) for i in range(3)])
+        print(f"{methods[i]}: {conv_rate}")
 
-    plt.show()
-    # fig1.savefig("../latex/images/phase_space_2_particles_x.pdf")
+    fig1.savefig("../latex/images/phase_space_2_particles_x.pdf")
 
 
 if __name__ == "__main__":