CiscoTheProot/rpi/roy-test.py

73 lines
1.7 KiB
Python

from rgbmatrix import RGBMatrix, RGBMatrixOptions
import paho.mqtt.client as mqtt
import time
from PIL import Image
# Configuration for the matrix
options = RGBMatrixOptions()
options.rows = 32
options.cols = 64
options.chain_length = 2
options.parallel = 1
options.hardware_mapping = 'regular' # If you have an Adafruit HAT: 'adafruit-hat'
matrix = RGBMatrix(options = options)
# offscreen canvas that can be written to which can then be set to the matrix async
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, matrix, offscreen_canvas
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
color += 1
color %= 3
updateScreen()
# 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))
toggleColor()
client = mqtt.Client()
client.on_connect = on_connect
client.on_message = on_message
client.connect("10.1.13.173", 1883, 60) # Replace with your MQTT broker's address
client.loop_start()
while True:
time.sleep(0.005)