57 lines
1.7 KiB
Python
57 lines
1.7 KiB
Python
import neopixel
|
|
from time import sleep
|
|
import asyncio
|
|
from typing import List
|
|
|
|
import sys
|
|
sys.path.append('.')
|
|
import config
|
|
|
|
class IndicationDriver:
|
|
def __init__(self):
|
|
self.pixels = neopixel.NeoPixel(
|
|
config.PIXEL_PIN,
|
|
config.NUM_PIXELS,
|
|
brightness= config.NEOPIXEL_BRIGHTNESS,
|
|
auto_write=False,
|
|
pixel_order=neopixel.GRB
|
|
)
|
|
self.clear()
|
|
self.indication_stop_event = asyncio.Event()
|
|
|
|
|
|
def stop_loop(self):
|
|
self.indication_stop_event.set()
|
|
|
|
def clear(self):
|
|
self.pixels.fill((0,0,0))
|
|
self.pixels.show()
|
|
|
|
def _get_trash_location(self, trash_type: str) -> List[int]:
|
|
if trash_type in config.TRASH_LOCATION:
|
|
center_location = config.TRASH_LOCATION[trash_type]
|
|
return range(center_location-2, center_location+3)
|
|
else:
|
|
raise ValueError
|
|
|
|
def indicate(self, trash_type: str):
|
|
self.clear()
|
|
for idx in self._get_trash_location(trash_type):
|
|
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():
|
|
for step in range (4):
|
|
for idx in range(location[0]):
|
|
if idx % 4 == step:
|
|
self.pixels[idx] = (0, 0, 255)
|
|
else:
|
|
self.pixels[idx] = (0, 0, 0)
|
|
self.pixels.show()
|
|
await asyncio.sleep(.1)
|
|
self.clear() |