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