IMU Norden kann kalibriert werden

scripts/exomy_admin_api.py

@@ -29,6 +29,9 @@ ROUTE_LIBRARY_FILE = os.path.join(
     os.path.dirname(__file__), '..', 'config', 'route_library.json'
 )
 CAMERA_RUNTIME_SETTINGS_FILE = '/tmp/exomy_camera_settings.json'
+IMU_HEADING_CALIBRATION_FILE = os.path.join(
+    os.path.dirname(__file__), '..', 'config', 'imu_heading_calibration.json'
+)
 MOTOR_TEST_LOCK = threading.Lock()
 GPS_PORT_PATTERNS = [
     '/dev/serial/by-id/*',
@@ -89,6 +92,7 @@ IMU_ROS_CACHE = {
     'quaternion': 'unbekannt',
     'roll_pitch_yaw': 'unbekannt',
     'error': 'Noch keine IMU-Daten aus ROS',
+    'heading_deg': None,
     'updated_at': 0.0,
 }
 
@@ -153,6 +157,13 @@ def _quaternion_to_euler_deg(values):
     )
 
 
+def _normalize_heading_deg(value):
+    heading = float(value) % 360.0
+    if heading < 0.0:
+        heading += 360.0
+    return heading
+
+
 def _imu_ros_cache_update(**kwargs):
     with IMU_ROS_LOCK:
         IMU_ROS_CACHE.update(kwargs)
@@ -169,6 +180,7 @@ def _on_imu_data_message(message):
         float(orientation.get('z', 0.0)),
         float(orientation.get('w', 1.0)),
     )
+    euler_deg = _quaternion_to_euler_deg(quaternion)
     _imu_ros_cache_update(
         state='active',
         acceleration=_format_vector((
@@ -182,7 +194,8 @@ def _on_imu_data_message(message):
             float(angular_velocity.get('z', 0.0)),
         ), 'rad/s', 3),
         quaternion=_format_quaternion(quaternion),
-        roll_pitch_yaw=_format_euler_deg(_quaternion_to_euler_deg(quaternion)),
+        roll_pitch_yaw=_format_euler_deg(euler_deg),
+        heading_deg=None if not euler_deg else _normalize_heading_deg(euler_deg[2]),
         error='',
     )
 
@@ -842,6 +855,7 @@ def collect_status():
     time_details = get_system_time_details()
     container_details = get_container_details(EXOMY_CONTAINER)
     imu_details = read_imu_ros_status()
+    imu_heading = get_imu_heading_status()
     return {
         'status': 'Bereit',
         'system': {
@@ -884,6 +898,7 @@ def collect_status():
             'exomy_restarts': container_details['restarts'],
         },
         'imu': imu_details,
+        'imu_heading': imu_heading,
     }
 
 
@@ -1021,6 +1036,52 @@ def read_camera_settings_version():
         return 0
 
 
+def get_default_imu_heading_calibration():
+    return {
+        'offset_deg': 0.0,
+        'updated_at': None,
+        'source': 'default',
+    }
+
+
+def read_imu_heading_calibration():
+    data = get_default_imu_heading_calibration()
+    try:
+        with open(IMU_HEADING_CALIBRATION_FILE, 'r', encoding='utf-8') as handle:
+            payload = json.load(handle)
+    except (FileNotFoundError, json.JSONDecodeError, OSError, ValueError, TypeError):
+        return data
+
+    try:
+        offset_deg = float(payload.get('offset_deg', 0.0))
+    except (TypeError, ValueError):
+        return data
+
+    data['offset_deg'] = offset_deg
+    data['updated_at'] = payload.get('updated_at')
+    data['source'] = 'saved'
+    return data
+
+
+def write_imu_heading_calibration(offset_deg):
+    payload = {
+        'offset_deg': float(offset_deg),
+        'updated_at': time.strftime('%Y-%m-%d %H:%M:%S'),
+    }
+    os.makedirs(os.path.dirname(IMU_HEADING_CALIBRATION_FILE), exist_ok=True)
+    with open(IMU_HEADING_CALIBRATION_FILE, 'w', encoding='utf-8') as handle:
+        json.dump(payload, handle, ensure_ascii=False, indent=2)
+        handle.write('\n')
+    payload['source'] = 'saved'
+    return payload
+
+
+def get_corrected_heading_deg(raw_heading_deg, offset_deg):
+    if raw_heading_deg is None:
+        return None
+    return _normalize_heading_deg(float(raw_heading_deg) + float(offset_deg))
+
+
 def get_camera_config():
     profile = read_camera_settings()
     return {
@@ -1036,6 +1097,51 @@ def get_camera_config():
     }
 
 
+def get_imu_heading_status():
+    calibration = read_imu_heading_calibration()
+    with IMU_ROS_LOCK:
+        raw_heading_deg = IMU_ROS_CACHE.get('heading_deg')
+
+    corrected_heading_deg = get_corrected_heading_deg(raw_heading_deg, calibration['offset_deg'])
+    return {
+        'offset_deg': calibration['offset_deg'],
+        'raw_heading_deg': raw_heading_deg,
+        'corrected_heading_deg': corrected_heading_deg,
+        'updated_at': calibration['updated_at'],
+        'source': calibration['source'],
+    }
+
+
+def set_imu_heading_offset(offset_deg):
+    payload = write_imu_heading_calibration(offset_deg)
+    corrected_heading_deg = None
+    with IMU_ROS_LOCK:
+        raw_heading_deg = IMU_ROS_CACHE.get('heading_deg')
+        if raw_heading_deg is not None:
+            corrected_heading_deg = get_corrected_heading_deg(raw_heading_deg, payload['offset_deg'])
+    return {
+        'offset_deg': payload['offset_deg'],
+        'raw_heading_deg': raw_heading_deg,
+        'corrected_heading_deg': corrected_heading_deg,
+        'updated_at': payload['updated_at'],
+        'source': payload['source'],
+    }
+
+
+def calibrate_imu_heading_to_north():
+    with IMU_ROS_LOCK:
+        raw_heading_deg = IMU_ROS_CACHE.get('heading_deg')
+
+    if raw_heading_deg is None:
+        return 409, {'status': 'Noch keine frischen IMU-Daten für Nord-Kalibrierung'}
+
+    calibration = set_imu_heading_offset(-float(raw_heading_deg))
+    return 200, {
+        'status': 'Nordrichtung gespeichert',
+        'imu_heading': calibration,
+    }
+
+
 def set_camera_config(profile_id=None, fps=None):
     current = read_camera_settings()
     if profile_id is None:
@@ -1805,7 +1911,7 @@ class Handler(http.server.BaseHTTPRequestHandler):
 
     def do_POST(self):
         raw_body = b'{}'
-        if self.path in ('/api/routes/save', '/api/routes/delete'):
+        if self.path in ('/api/routes/save', '/api/routes/delete', '/api/imu/heading-offset'):
             try:
                 content_length = int(self.headers.get('Content-Length', '0'))
             except ValueError:
@@ -1918,6 +2024,25 @@ class Handler(http.server.BaseHTTPRequestHandler):
             self.write_json(status_code, response)
             return
 
+        if self.path == '/api/imu/heading-offset':
+            try:
+                payload = json.loads(raw_body.decode('utf-8'))
+            except (UnicodeDecodeError, json.JSONDecodeError):
+                self.write_json(400, {'status': 'Ungültige Anfrage'})
+                return
+
+            try:
+                offset_deg = float(payload.get('offset_deg', 0.0))
+            except (TypeError, ValueError):
+                self.write_json(400, {'status': 'Ungültiger Heading-Offset'})
+                return
+
+            self.write_json(200, {
+                'status': 'Heading-Offset gespeichert',
+                'imu_heading': set_imu_heading_offset(offset_deg),
+            })
+            return
+
         if self.path == '/api/motor-test/drive-neutral/preview':
             try:
                 content_length = int(self.headers.get('Content-Length', '0'))
@@ -2067,6 +2192,11 @@ class Handler(http.server.BaseHTTPRequestHandler):
                 self.write_json(500, {'status': str(exc)})
             return
 
+        if self.path == '/api/imu/set-north':
+            status_code, response = calibrate_imu_heading_to_north()
+            self.write_json(status_code, response)
+            return
+
         actions = {
             '/api/cold-start-gps': {
                 'handler': cold_start_gps_receiver,