MCH2022 badge?
go to mch2022.badge.team

Content import buttons # TODO import display import leds from color import Color import utime import urandom # TODO: https://www.rdocumentation.org/packages/base/versions/3.6.1/topics/seq def seq(start, stop, step=1): n = int(round((stop - start)/float(step))) if n > 1: return([start + step*i for i in range(n+1)]) elif n == 1: return([start]) else: return([]) # TODO: https://en.wikipedia.org/wiki/Fisher%E2%80%93Yates_shuffle def randomize(arr, n): # Start from the last element and swap one by one. We don't # need to run for the first element that's why i > 0 for i in range(n-1, 0, -1): # Pick a random index from 0 to i j = urandom.randint(0, i+1) # Swap arr[i] with the element at random index arr[i], arr[j] = arr[j], arr[i] return arr class stroboscope: # TODO: add full RGB spectrum, check against previously selected color purple15 = [Color(230, 230, 250), Color(216, 191, 216), Color(221, 160, 221), Color(238, 130, 238), Color(218, 112, 214), Color(255, 0, 255), Color(255, 0, 255), Color(186, 85, 211), Color(147, 112, 219), Color(138, 43, 226), Color(148, 0, 211), Color(153, 50, 204), Color(139, 0, 139), Color(128, 0, 128), Color(75, 0, 130)] yellow11 = [Color(255, 228, 181), Color(255, 218, 185), Color(238, 232, 170), Color(240, 230, 140), Color(189, 183, 107), Color(255, 255, 0), Color(128, 128, 0), Color(173, 255, 47), Color(154, 205, 50), Color(255, 255, 102), Color(255, 255, 51)] def __init__(self, delay, matrix, width, random=False, rocket=False, top_bar=False, top_width=1): # TODO: implement LED brightness if len(self.purple15) % width != 0: raise ValueError if len(self.yellow11) % top_width != 0: raise ValueError if len(matrix) != 4: raise ValueError self.color_random = random self.color_width = width self.led_delay = delay self.led_matrix = matrix self.rocket = rocket self.top_bar = top_bar self.top_width = top_width def begin(self): # TODO: process button events self.color_range = seq(0, len(self.purple15)-1, self.color_width) self.top_range = seq(0, len(self.yellow11)-1, self.top_width) self.rocket_on = False while True: r = urandom.randint(0, 255) g = urandom.randint(0, 255) b = urandom.randint(0, 255) with display.open() as disp: disp.clear([r, g, b]).update() if self.color_random is True: try: self.color_range = randomize( self.color_range, len(self.purple15)) self.top_range = randomize( self.top_range, len(self.yellow11)) except IndexError: # FIXME: index out of range errors in index swap (L26) print("bug: sequence shuffle failed (out of range)") # TODO: set amount of used top leds "width" for k, i in enumerate(self.top_range): # TODO: check self.top_bar leds.prep(k, self.yellow11[i]) for i in self.color_range: if self.led_matrix[0] == 1: leds.prep(leds.TOP_LEFT, self.purple15[i]) if self.led_matrix[1] == 1: leds.prep(leds.TOP_RIGHT, self.purple15[i]) if self.led_matrix[2] == 1: leds.prep(leds.BOTTOM_LEFT, self.purple15[i]) if self.led_matrix[3] == 1: leds.prep(leds.BOTTOM_RIGHT, self.purple15[i]) leds.update() if self.rocket is False: continue if self.rocket_on is True: leds.set_rocket(0, 31) # TODO: define brightness leds.set_rocket(1, 31) leds.set_rocket(2, 31) else: leds.set_rocket(0, 0) leds.set_rocket(1, 0) leds.set_rocket(2, 0) self.rocket_on = not self.rocket_on # main LED delay utime.sleep_ms(self.led_delay) def main(): Strobo = stroboscope(100, [1, 1, 1, 1], 1, True, True, 1) Strobo.begin() if __name__ == "__main__": main()