class Strip(object):
def __init__(self,length,port):
self._disp_length = length
self._s = Serial(port, 115200*4, timeout=0.5)
self._bar = [[0, 0, 0]]*self._disp_length
self._commit()
def blink(self, index, duration=0.1, dutycycle=0.5, color=None):
assert duration >= 0
self.set_bit(index, color)
time.sleep(duration*dutycycle)
self.clear_bit(index)
time.sleep(duration*(1 - dutycycle))
def wink(self, index, duration=0.1, dutycycle=0.5):
assert duration >= 0
color = self._bar[index - 1]
self.clear_bit(index)
time.sleep(duration*dutycycle)
self.set_bit(index, color)
time.sleep(duration*(1 - dutycycle))
def set_bit(self, index, color):
"""Color is a tuple (R,G,B) where R, G, and B are numbers within [0,1]. Index starts from zero."""
assert index >= 0 and index <= self._disp_length - 1, 'index should be within [0,length-1]'
self._bar[index] = color
self._commit()
def clear_bit(self, index):
assert index >= 0 and index <= self._disp_length - 1, 'index should be within [0,length-1]'
self.set_bit(index, [0,0,0])
def set_all(self, pixels):
assert len(pixels) >= len(self._bar)
assert len(pixels[0]) == 3
assert pixels[0][0] >= 0 and pixels[0][0] <= 1
self._bar = list(pixels[0:self._disp_length])
self._commit()
def _commit(self):
write_led_strip(self._s, self._bar)
def __enter__(self):
return self
def __exit__(self, exc_type, exc_value, traceback):
self._s.__exit__()
def __del__(self):
self._s.__del__()
