From 57c126a7bcb467bb678146298008b011a3dc591a Mon Sep 17 00:00:00 2001
From: Rune Harlyk <rune@harlyk.dk>
Date: Thu, 14 Nov 2024 11:19:35 +0100
Subject: [PATCH] =?UTF-8?q?=F0=9F=8E=AE=20Makes=20controller=20sub=20for?=
 =?UTF-8?q?=20imu=20updates?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/src/lib/stores/imu.ts              | 36 +++++++++++++-------------
 app/src/lib/types/models.ts            |  2 +-
 app/src/routes/controller/+page.svelte | 33 +++++++++++++++++------
 3 files changed, 44 insertions(+), 27 deletions(-)

diff --git a/app/src/lib/stores/imu.ts b/app/src/lib/stores/imu.ts
index d303d0f..55fcc1c 100644
--- a/app/src/lib/stores/imu.ts
+++ b/app/src/lib/stores/imu.ts
@@ -4,24 +4,24 @@ import type { IMU } from '$lib/types/models';
 const maxIMUData = 100;
 
 export const imu = (() => {
-	const { subscribe, update } = writable({
-		x: [] as number[],
-		y: [] as number[],
-		z: [] as number[],
-		imu_temp: [] as number[],
-		altitude: [] as number[],
-		pressure: [] as number[],
-		bmp_temp: [] as number[]
-	});
+    const { subscribe, update } = writable({
+        x: [] as number[],
+        y: [] as number[],
+        z: [] as number[],
+        heading: [] as number[],
+        altitude: [] as number[],
+        pressure: [] as number[],
+        bmp_temp: [] as number[]
+    });
 
-	const addData = (content: IMU) => {
-		update((data) => {
-			(Object.keys(content) as (keyof IMU)[]).forEach((key) => {
-				data[key] = [...data[key], content[key]].slice(-maxIMUData);
-			});
-			return data;
-		});
-	};
+    const addData = (content: IMU) => {
+        update(data => {
+            (Object.keys(content) as (keyof IMU)[]).forEach(key => {
+                data[key] = [...data[key], content[key]].slice(-maxIMUData);
+            });
+            return data;
+        });
+    };
 
-	return { subscribe, addData };
+    return { subscribe, addData };
 })();
diff --git a/app/src/lib/types/models.ts b/app/src/lib/types/models.ts
index de4bf4e..b91e290 100644
--- a/app/src/lib/types/models.ts
+++ b/app/src/lib/types/models.ts
@@ -150,7 +150,7 @@ export type IMU = {
     x: number;
     y: number;
     z: number;
-    imu_temp: number;
+    heading: number;
     altitude: number;
     bmp_temp: number;
     pressure: number;
diff --git a/app/src/routes/controller/+page.svelte b/app/src/routes/controller/+page.svelte
index 542342b..7b4a186 100644
--- a/app/src/routes/controller/+page.svelte
+++ b/app/src/routes/controller/+page.svelte
@@ -1,14 +1,31 @@
 <script lang="ts">
-	import Controls from './Controls.svelte';
-	import WidgetContainer from '$lib/components/layout/WidgetContainer.svelte';
-	import { selectedView, views } from '$lib/stores/application';
+    import Controls from './Controls.svelte';
+    import WidgetContainer from '$lib/components/layout/WidgetContainer.svelte';
+    import { selectedView, views } from '$lib/stores/application';
+    import { onMount } from 'svelte';
+    import { mpu, socket } from '$lib/stores';
+    import { imu } from '$lib/stores/imu';
+    import type { IMU } from '$lib/types/models';
 
-    $: layout = $views.find(v => v.name === $selectedView)!
+    $: layout = $views.find(v => v.name === $selectedView)!;
+
+    onMount(() => {
+        socket.on('imu', (data: IMU) => {
+            imu.addData(data);
+            if (data.heading)
+                mpu.update(mpuData => {
+                    mpuData.heading = data.heading;
+                    console.log(data.heading);
+
+                    return mpuData;
+                });
+        });
+    });
 </script>
 
 <div class="absolute top-0 select-none w-screen h-screen">
-	<Controls />
-	<div class="absolute w-full h-screen top-0 overflow-hidden lg:pt-16 pt-12">
-		<WidgetContainer container={layout.content} />
-	</div>
+    <Controls />
+    <div class="absolute w-full h-screen top-0 overflow-hidden lg:pt-16 pt-12">
+        <WidgetContainer container={layout.content} />
+    </div>
 </div>