Add optional estimation without magnetometer.

Mayitzin · Jul 12, 2022 · 0368775 · 0368775
1 parent 18be81d
commit 0368775
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/ahrs/filters/complementary.py b/ahrs/filters/complementary.py
@@ -109,6 +109,7 @@
 
 import numpy as np
 from ..common.orientation import ecompass
+from ..common.orientation import acc2q
 from ..utils.core import _assert_numerical_iterable
 
 class Complementary:
@@ -268,6 +269,8 @@ def am_estimation(self, acc: np.ndarray, mag: np.ndarray = None) -> np.ndarray:
         q_am : numpy.ndarray
             Estimated attitude.
         """
+        if mag is None:
+            return acc2q(acc)
         return ecompass(acc, mag, frame='NED', representation='quaternion')
 
     def update(self, q: np.ndarray, gyr: np.ndarray, acc: np.ndarray, mag: np.ndarray = None, dt: float = None) -> np.ndarray: