diff --git a/scripts/airbrakes/coeffs.py b/scripts/airbrakes/coeffs.py new file mode 100644 index 0000000000000000000000000000000000000000..fa2de47fa3f969fa180181e1edc46ef6c1b527ea --- /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 0000000000000000000000000000000000000000..3202be92c24d966bc8a6e81b0d5cb8d251e801e2 --- /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("}")