indication_driver

asyncio --> threading

indication_driver.py

@@ -1,7 +1,7 @@
 import neopixel
 from time import sleep
-import asyncio
 from typing import List
+from threading import Thread
 
 import sys
 sys.path.append('.')
@@ -16,14 +16,15 @@ class IndicationDriver:
             auto_write=False, 
             pixel_order=neopixel.GRB
         )
-        self.clear()
-        self.indication_stop_event = asyncio.Event()
+        self._run_loop = True
+        self._indication_loop = None
+        self.stop()
 
 
-    def stop_loop(self):
-        self.indication_stop_event.set()
-        
-    def clear(self):
+    def stop(self):
+        if self._indication_loop:
+            self._run_loop = False
+            self._indication_loop.join()
         self.pixels.fill((0,0,0))
         self.pixels.show()
         
@@ -35,17 +36,15 @@ class IndicationDriver:
             raise ValueError
 
     def indicate(self, trash_type: str):
-        self.clear()
-        for idx in self._get_trash_location(trash_type):
+        self.stop()
+        location = self._get_trash_location(trash_type)
+
+        def _indication_loop(self, location: List[int]):
+            for idx in location:
                 self.pixels[idx] = (0, 255, 0)
             self.pixels.show()
-
-    async def indicate_async(self, trash_type: str):
-        self.clear()
-        self.indication_stop_event.clear()
-        self.indicate(trash_type)
-        location = self._get_trash_location(trash_type)
-        while not self.indication_stop_event.is_set():
+            self._run_loop = True
+            while self._run_loop:
                 for step in range (4):
                     for idx in range(location[0]):
                         if idx % 4 == step:
@@ -53,5 +52,8 @@ class IndicationDriver:
                         else:
                             self.pixels[idx] = (0, 0, 0)
                     self.pixels.show()
-                await asyncio.sleep(.1)
-        self.clear()
+                    sleep(.1)
+
+        self._indication_loop = Thread(target = _indication_loop, args = (self, location, ))
+        self._indication_loop.start()
+        sleep(1)

test_scripts/led_test.py

@@ -1,20 +1,12 @@
 from time import sleep
-import asyncio
 
 import sys
 sys.path.append('..')
 from TREx.indication_driver import IndicationDriver
 
-async def __main__():
-    task = asyncio.create_task(indication_driver.indicate_async("BOTTLE"))
-    await asyncio.sleep(5)
-    indication_driver.stop_loop()
-    await task
-    indication_driver.clear()
-
 indication_driver = IndicationDriver()
-asyncio.run(__main__())
-sleep(2)
 indication_driver.indicate("CAN")
-sleep(2)
-indication_driver.clear()
+sleep(5)
+indication_driver.indicate("INCOMBUSTIBLE")
+sleep(5)
+indication_driver.stop()

test_scripts/rfid_test.py

@@ -1,7 +1,6 @@
 from time import sleep
 import RPi.GPIO as GPIO
 from mfrc522 import SimpleMFRC522
-import asyncio
 
 import sys
 sys.path.append('..')
@@ -9,33 +8,15 @@ import TREx.config as config
 from TREx.indication_driver import IndicationDriver
 from TREx.rfid_driver import RFIDDriver
 
+reader = SimpleMFRC522()
+indication_driver = IndicationDriver()
 
-async def __main__():
-    rfid_driver = RFIDDriver()
-    indication_driver = IndicationDriver()
-    indication_task = None
-    while True:
-        id = await rfid_driver.get_next_id()
-        trash_category = None
-        try:
-            if indication_task is not None:
-                indication_driver.stop_loop()
-                await indication_task
+while True:
+    id, _ = reader.read()
+    if id in config.RFID_LOOKUP_TABLE.keys():
         trash_category = config.RFID_LOOKUP_TABLE[id]
-            indication_task = asyncio.create_task(indication_driver.indicate_async(trash_category))
-            await asyncio.sleep(1)
-        except KeyError:
-            indication_driver.stop_loop()
-
-asyncio.run(__main__())
-
-# while True:
-#     id, _ = reader.read()
-#     if id in config.RFID_LOOKUP_TABLE.keys():
-#         trash_category = config.RFID_LOOKUP_TABLE[id]
-#         print(f"{id}: {trash_category}") 
-#         indication_driver.indicate(trash_category)
-#     else:
-#         print(id)
-#     sleep(1)
-# GPIO.cleanup()
+        print(f"{id}: {trash_category}") 
+        indication_driver.indicate(trash_category)
+    else:
+        indication_driver.stop()
+        print(id)