Add tempo, live status

pyleds/lib/Litsimaja.py

@@ -3,17 +3,16 @@ from rpi_ws281x import PixelStrip
 # from lib.strip.TkinterStrip import TkinterStrip
 from lib.LoopSwitch import LoopSwitch
 from lib.Regions import Regions
+from lib.Tempo import Tempo
 
 
 class Litsimaja(object):
-    _loops: []
-    _regions: Regions
-
     def __init__(self):
         self._strip = PixelStrip(290, 18, 800000, 10, False, 255, 0, 4104)
         self._loops = []
         self._strip.begin()
         self._regions: Regions = Regions(self.count_pixels(), [46, 96, 191, 241])
+        self._tempo: Tempo = Tempo(60)
 
     def count_pixels(self) -> int:
         return self._strip.numPixels()
@@ -38,9 +37,26 @@ class Litsimaja(object):
         for loop in self._loops:
             loop.stop()
         self._loops.clear()
-        
+
     def switch_region(self, region_id: int):
         self._regions.switch_region(region_id)
 
     def get_region_ids(self):
         return self._regions.list_region_ids()
+
+    def build_status_array(self):
+        data = {'success': True}
+        features = {
+            'tempo': {
+                'bpm': self.get_tempo().get_bpm()
+            }
+        }
+        regions = []
+        for region_id in self._regions.list_region_ids():
+            regions.append(self._regions.is_region_enabled(region_id))
+        features['region'] = regions
+        data['features'] = features
+        return data
+
+    def get_tempo(self):
+        return self._tempo

pyleds/lib/Regions.py

@@ -21,9 +21,12 @@ class Regions(object):
             self._pixelsEnabled[i] = enabled
         self._regions[region_id][2] = enabled
 
-    def is_pixel_enabled(self, pixel_id: int):
+    def is_pixel_enabled(self, pixel_id: int) -> bool:
         return self._pixelsEnabled[pixel_id]
     
+    def is_region_enabled(self, region_id: int) -> bool:
+        return self._regions[region_id][2]
+    
     def switch_region(self, region_id: int):
         status: bool = self._regions[region_id][2]
         self.region_switch(region_id, not status)

pyleds/lib/Tempo.py

@@ -0,0 +1,45 @@
+import time
+from typing import Optional
+
+
+class Tempo(object):
+    _last_beat: float
+    _override_beat: Optional[float] = None
+    
+    def __init__(self, bpm: float = 60):
+        self._bpm: float = bpm
+        self.update_last_beat()
+    
+    def set_bpm(self, bpm: float) -> None:
+        self._bpm = bpm
+    
+    def get_bpm(self) -> float:
+        return self._bpm
+    
+    def update_last_beat(self) -> None:
+        self._last_beat = time.time()
+    
+    def get_beat_time(self) -> float:
+        return 1000 / (self._bpm / 60)
+    
+    def get_beat_delay(self) -> Optional[float]:
+        tick = self._override_beat
+        if tick is None:
+            return None
+        self._override_beat = None
+        return tick - self._last_beat
+    
+    def sync_beat(self):
+        self._override_beat = time.time()
+        
+    def wait(self, beat_divider: int = 1):
+        now = time.time()
+        next_beat = self._last_beat + (self.get_beat_time() / beat_divider / 1000)
+        wait_time = next_beat - now
+        beat_delay = self.get_beat_delay()
+        if beat_delay is not None:
+            wait_time += beat_delay
+        if wait_time > 0:
+            time.sleep(wait_time)
+        self.update_last_beat()
+        return True