Adding refactored classes for ble sensing with working data collection again

kano_wand/wand_sensors.py

@@ -0,0 +1,72 @@
+import pandas as pd
+
+
+class ThreeAxisSensor(object):
+    def __init__(self, x, y, z):
+        self.x = x
+        self.y = y
+        self.z = z
+
+    def __str__(self):
+        print(self.x, self.y, self.z)
+
+
+class WandSensors(object):
+    def __init__(self, wand_update_callback):
+        self.gyro = None
+        self.gyro_updates = 0
+
+        self.accel = None
+        self.accel_updates = 0
+
+        self.magneto = 0
+        self.button = 0
+        self.temperature = 0
+
+        self.dataframe = pd.DataFrame(columns=["gyro_x", "gyro_y", "gyro_z", "accel_x", "accel_y", "accel_z"])
+
+        self.wand_update_callback = wand_update_callback
+
+    def __str__(self):
+        return super().__str__()
+
+    def append_to_dataframe(self):
+        if self.gyro_updates == self.accel_updates:
+            self.dataframe = self.dataframe.append(
+                pd.DataFrame({
+                    "gyro_x": self.gyro.x,
+                    "gyro_y": self.gyro.y,
+                    "gyro_z": self.gyro.z,
+                    "accel_x": self.accel.x,
+                    "accel_y": self.accel.y,
+                    "accel_z": self.accel.z,
+                    "magneto": self.magneto,
+                    "button": self.button
+                }, index=[0])
+            )
+            self.wand_update_callback(self.dataframe)
+
+    def save_df_to_file(self, filename):
+        self.dataframe.to_csv(filename, index=False)
+
+    def set_gyro(self, x, y, z):
+        self.gyro = ThreeAxisSensor(x, y, z)
+        self.gyro_updates += 1
+        self.append_to_dataframe()
+
+    def set_accel(self, x, y, z):
+        self.accel = ThreeAxisSensor(x, y, z)
+        self.accel_updates += 1
+        self.append_to_dataframe()
+
+    def set_temp(self, temp):
+        self.temp = temp
+
+    def set_button(self, button):
+        self.button = button
+
+    def set_magneto(self, magneto):
+        self.magneto = magneto
+
+    def get_dataframe(self):
+        return self.dataframe