From 1a40018aa21ba022cf2a8c306f616c54e3e63315 Mon Sep 17 00:00:00 2001
From: Alberto Nidasio <alberto.nidasio@skywarder.eu>
Date: Mon, 10 Oct 2022 21:47:00 +0100
Subject: [PATCH] [OrientationVisualizer] Now subscribing to both Main and
 Payload

---
 .../OrientationVisualizer/OrientationVisualizer.cpp      | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/src/shared/Modules/OrientationVisualizer/OrientationVisualizer.cpp b/src/shared/Modules/OrientationVisualizer/OrientationVisualizer.cpp
index 7b130298..a9196fd9 100644
--- a/src/shared/Modules/OrientationVisualizer/OrientationVisualizer.cpp
+++ b/src/shared/Modules/OrientationVisualizer/OrientationVisualizer.cpp
@@ -27,6 +27,15 @@ OrientationVisualizer::OrientationVisualizer(QWidget *parent)
 
     updateOrientation(0, 0, 0, 1);
 
+    getCore()->getMessageBroker()->subscribe(
+        Filter::fromString("Mav/ROCKET_FLIGHT_TM"), this,
+        [this](const Message &message, const Filter &filter)
+        {
+            updateOrientation(message.getField("nas_qx").getDouble(),
+                              message.getField("nas_qy").getDouble(),
+                              message.getField("nas_qz").getDouble(),
+                              message.getField("nas_qw").getDouble());
+        });
     getCore()->getMessageBroker()->subscribe(
         Filter::fromString("Mav/PAYLOAD_FLIGHT_TM"), this,
         [this](const Message &message, const Filter &filter)
-- 
GitLab