Commit d7e2c4d6 authored by Jan Kiene's avatar Jan Kiene
Browse files

fix for trajectory inversion

parent 474a33b2
Loading
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -38,7 +38,7 @@ from ivas_processing_scripts.audiotools.rotation import Euler2Quat, Quat2Euler


def read_trajectory(trj_file: Path, return_quat=True):
    trj = np.genfromtext(trj_file, delimiter=",")
    trj = np.genfromtxt(trj_file, delimiter=",")

    if np.all(trj[:, 0] == -3):
        # Euler
+1 −1
Original line number Diff line number Diff line
@@ -121,4 +121,4 @@ def main():
        pad = np.array([[1, 0, 0, 0]])
        trj = np.vstack([np.repeat(pad, int(args.delay / 5), axis=0), trj])

    write_trajectory(trj, args.out_trj, quat=(args.output_format == "q"))
    write_trajectory(trj, args.out_trj, write_quat=(args.output_format == "q"))