Refactoring code from original POC phase - integers have been decoded roughly successfully so its time to start building a huge library for this bad boy

2019-11-08 12:58:37 -06:00
parent 48866f6af2
commit 564eed46d5
14 changed files with 223 additions and 246 deletions
@@ -0,0 +1,54 @@
+import pandas as pd
+
+class Gyro(object):
+    def __init__(self):
+        self.x = 0
+        self.y = 0
+        self.z = 0
+
+    def __str__(self):
+        print(self.x, self.y, self.z)
+
+
+class Accel(object):
+    def __init__(self):
+        self.x = 0
+        self.y = 0
+        self.z = 0
+
+    def __str__(self):
+        print(self.x, self.y, self.z)
+
+
+class SensorTracker(object):
+    def __init__(self):
+        self.gyro = None
+        self.gyro_updates = 0
+
+        self.accel = None
+        self.accel_updates = 0
+        self.dataframe = pd.DataFrame(columns=["gyro_x", "gyro_y", "gyro_z", "accel_x", "accel_y", "accel_z"])
+
+    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,
+                }, index=[0])
+            )
+
+    def save_df_to_file(self, filename):
+        self.dataframe.to_csv(filename, index=False)
+
+    def set_gyro(self, gyro):
+        self.gyro = gyro
+        self.gyro_updates += 1
+
+    def set_accel(self, accel):
+        self.accel = accel
+        self.accel_updates += 1