Create main.py

rainbow/main.py

@@ -0,0 +1,87 @@
+from sense_hat import SenseHat
+from time import sleep
+import time
+sense = SenseHat()
+
+# Define the colors for the rainbow
+colors = [
+    (255, 0, 0),    # Red
+    (255, 32, 0),   # Orange-Red
+    (255, 64, 0),   # Dark Orange
+    (255, 96, 0),   # Orange
+    (255, 128, 0),  # Orange
+    (255, 160, 0),  # Orange
+    (255, 192, 0),  # Orange
+    (255, 224, 0),  # Yellow-Orange
+    (255, 255, 0),  # Yellow
+    (224, 255, 0),  # Yellow-Green
+    (192, 255, 0),  # Green-Yellow
+    (160, 255, 0),  # Green-Yellow
+    (128, 255, 0),  # Green-Yellow
+    (96, 255, 0),   # Green
+    (64, 255, 0),   # Green
+    (32, 255, 0),   # Green
+    (0, 255, 0),    # Green
+    (0, 255, 32),   # Green-Blue
+    (0, 255, 64),   # Blue-Green
+    (0, 255, 96),   # Blue-Green
+    (0, 255, 128),  # Cyan
+    (0, 255, 160),  # Cyan
+    (0, 255, 192),  # Cyan
+    (0, 224, 255),  # Sky Blue
+    (0, 192, 255),  # Sky Blue
+    (0, 160, 255),  # Blue
+    (0, 128, 255),  # Blue
+    (0, 96, 255),   # Navy Blue
+    (0, 64, 255),   # Navy Blue
+    (0, 32, 255),   # Navy Blue
+    (0, 0, 255),    # Blue
+    (32, 0, 255),   # Indigo
+    (64, 0, 255),   # Indigo
+    (96, 0, 255),   # Purple
+    (128, 0, 255),  # Purple
+    (160, 0, 255),  # Violet
+    (192, 0, 255),  # Violet
+    (224, 0, 255),  # Magenta
+    (255, 0, 224),  # Red-Violet
+    (255, 0, 192),  # Red-Violet
+    (255, 0, 160),  # Red-Violet
+    (255, 0, 128),  # Red-Violet
+    (255, 0, 96),   # Red-Violet
+    (255, 0, 64),   # Red-Violet
+    (255, 0, 32)    # Red-Violet
+]
+
+
+# Function to draw the snake on the Sense HAT
+def draw_snake(x, y, color_index):
+    sense.set_pixel(x, y, colors[color_index])
+
+# Function to clear the Sense HAT display
+def clear_display():
+    sense.clear()
+
+# Main loop
+def main(shape, pixel_order):
+    # Clear the display
+    clear_display()
+
+    # Iterate through the pixel order list and the color list
+    color_index = 0
+    active_pixels = []  # List to store active pixels and their color indices
+
+    for x, y in pixel_order:
+        if shape[y][x] == 1:
+            draw_snake(x, y, color_index)
+            active_pixels.append((x, y, color_index))  # Add the pixel and its color index
+            color_index = (color_index + 1) % len(colors)
+            sleep(0.1)  # Adjust the speed of the snake
+
+    while True:
+        # Update the active pixels with the next color in the sequence
+        for i, (x, y, color_index) in enumerate(active_pixels):
+            new_color_index = (color_index + 1) % len(colors)
+            draw_snake(x, y, new_color_index)
+            active_pixels[i] = (x, y, new_color_index)  # Update the color index
+
+        sleep(0.1)  # Small delay to reduce CPU usage