diff --git a/rpi/roy-test.py b/rpi/roy-test.py index a23dd7f..05b0d5e 100644 --- a/rpi/roy-test.py +++ b/rpi/roy-test.py @@ -1,6 +1,7 @@ from rgbmatrix import RGBMatrix, RGBMatrixOptions import paho.mqtt.client as mqtt import time +from PIL import Image # Configuration for the matrix @@ -16,19 +17,41 @@ matrix = RGBMatrix(options = options) offscreen_canvas = matrix.CreateFrameCanvas() +image = Image.open("../testImg.bmp") + +# Make image fit our screen. +image.thumbnail((matrix.width, matrix.height), Image.ANTIALIAS) + + + +def updateScreen(): + global color + if(color == 1): + offscreen_canvas.Fill(255,255,255) + elif(color == 0): + offscreen_canvas.Fill(255,0,0) + elif(color == 2): + matrix.SetImage(image.convert('RGB')) + + offscreen_canvas.brightness = 50 + offscreen_canvas = matrix.SwapOnVSync(offscreen_canvas) + + color = 0 def toggleColor(): global color - if(color == 1): - color = 0 - else: - color = 1 + color += 1 + color %= 3 + + updateScreen() -def on_connect(client, userdata, flags, rc): - print("Connected to MQTT broker with result code " + str(rc)) +# functions called by the MQTT listener +def on_connect(client, userdata, flags, responseCode): + print("Connected to MQTT broker with result code " + str(responseCode)) client.subscribe("test") + def on_message(client, userdata, message): print("Received message '" + str(message.payload) + "' on topic '" + message.topic + "' with QoS " + str(message.qos)) @@ -44,13 +67,7 @@ client.connect("10.1.13.173", 1883, 60) # Replace with your MQTT broker's addre client.loop_start() + while True: time.sleep(0.005) - - if(color == 1): - offscreen_canvas.Fill(255,255,255) - else: - offscreen_canvas.Fill(255,0,0) - - offscreen_canvas.brightness = 50 - offscreen_canvas = matrix.SwapOnVSync(offscreen_canvas) \ No newline at end of file + diff --git a/rpi/show-image.py b/rpi/show-image.py new file mode 100644 index 0000000..e69de29