From bb27d4f1771ffe89066ee503f1407483ab0eb25f Mon Sep 17 00:00:00 2001
From: Sepehr Madani <ssepehrmadani@gmail.com>
Date: Tue, 1 Sep 2020 00:21:00 -0400
Subject: [PATCH] Adds wrapToPi function for angles
---
utils/converter.py | 9 +++++++++
1 file changed, 9 insertions(+)
diff --git a/utils/converter.py b/utils/converter.py
index 33bc2e1..f19d01f 100644
--- a/utils/converter.py
+++ b/utils/converter.py
@@ -1,5 +1,6 @@
from math import cos, acos, pi
from numpy import arange
+from cmath import exp, phase
def deg_to_u(degrees):
return [cos(x * pi / 180) for x in degrees]
@@ -9,3 +10,11 @@ def range_in_deg(res):
def u_to_deg(u):
return [acos(x) * 180 / pi for x in u]
+
+def vectorWrapToPi(vector):
+ """Wraps a list of angles in the [-π, π] range."""
+ return [wrapToPi(x) for x in vector]
+
+def wrapToPi(theta):
+ """Wraps an angle in the [-π, π] range."""
+ return phase(exp(1j * theta))
\ No newline at end of file
--
GitLab