From a016f34f9afb458fc8e708585d2c3faa02dd547d Mon Sep 17 00:00:00 2001
From: Luca Conterio <luca.conterio@skywarder.eu>
Date: Fri, 20 Aug 2021 17:31:35 +0200
Subject: [PATCH] [AirBrakes] Added scripts for drag coeffs and trajectories

---
 scripts/airbrakes/coeffs.py       | 20 ++++++++++++++++++++
 scripts/airbrakes/trajectories.py | 22 ++++++++++++++++++++++
 2 files changed, 42 insertions(+)
 create mode 100644 scripts/airbrakes/coeffs.py
 create mode 100644 scripts/airbrakes/trajectories.py

diff --git a/scripts/airbrakes/coeffs.py b/scripts/airbrakes/coeffs.py
new file mode 100644
index 000000000..fa2de47fa
--- /dev/null
+++ b/scripts/airbrakes/coeffs.py
@@ -0,0 +1,20 @@
+import scipy.io as sio
+import sys
+
+#filename = "coeffs.mat"
+fieldname = "coeffs"
+
+if len(sys.argv) < 2:
+    print("\nError, missing path to file \nUsage : python3 coeffs.py <path_to_mat_file>\n")
+    
+mat = sio.loadmat(sys.argv[1])
+
+data = mat[fieldname][0][0]
+names = data.dtype.fields.keys()
+
+coeffs = {}
+
+for coeff, name in zip(data, names):
+    coeffs[name] = coeff.item()
+
+print(coeffs)
diff --git a/scripts/airbrakes/trajectories.py b/scripts/airbrakes/trajectories.py
new file mode 100644
index 000000000..3202be92c
--- /dev/null
+++ b/scripts/airbrakes/trajectories.py
@@ -0,0 +1,22 @@
+import scipy.io as sio
+import sys
+
+#filename = "Trajectories.mat"
+fieldname = "trajectories_saving"
+
+if len(sys.argv) < 2:
+    print("\nError, missing path to file \nUsage : python3 coeffs.py <path_to_mat_file>\n")
+
+mat = sio.loadmat(sys.argv[1])
+trajectories = mat[fieldname][0]
+
+with open("Trajectories_data.h", "w") as f:
+    f.write("{\n")
+    for trajectory in trajectories:
+        zs, vzs, xs, vxs, ys, vys, sbar = trajectory
+        f.write("\t{\n")
+        f.write("\t\t%d, %d,\n\t\t{\n" % (len(zs), sbar))
+        for z, vz in zip(zs, vzs):
+            f.write("\t\t\t{%f, %f},\n" % (z, vz))
+        f.write("\t\t}\n\t},\n")
+    f.write("}")
-- 
GitLab