mindstorms-python/hub_button_lights.py at main · nateswartz/mindstorms-python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
from mindstorms import MSHub, Motor, MotorPair, ColorSensor, DistanceSensor, App
from mindstorms.control import wait_for_seconds, wait_until, Timer
from mindstorms.operator import greater_than, greater_than_or_equal_to, less_than, less_than_or_equal_to, equal_to, not_equal_to
import math
import random

def getRandomColor():
    colors = ["azure","black","blue","cyan","green","orange","pink","red","violet","yellow","white"]
    random_index = random.randint(0, len(colors)-1)
    return colors[random_index]

def getRandomNumber():
    return random.randint(0, 9)

def turnLightMatrixOn(isRandomPattern):
    for x in range(5):
        for y in range(5):
            hub.light_matrix.set_pixel(x, y)

def setLightMatrixToRandomEvenDistribution():
    hub.light_matrix.off()
    led_state = [ [0]*5 for _ in range(5) ]
    lit_count = 0
    hub.speaker.beep()
    while (lit_count < 12):
        x = random.randint(0,4)
        y = random.randint(0,4)
        if (led_state[x][y] != 1):
            led_state[x][y] = 1
            lit_count += 1
    for x in range(5):
        for y in range(5):
            hub.light_matrix.set_pixel(x, y, led_state[x][y]*100)


# Create your objects here.
hub = MSHub()


# Write your program here.
while True:
    hub.left_button.wait_until_pressed()
    hub.status_light.off()
    hub.light_matrix.off()
    hub.left_button.wait_until_released()
    hub.status_light.on(getRandomColor())
    setLightMatrixToRandomEvenDistribution()