98 lines
2.5 KiB
Python
98 lines
2.5 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)
|
|
|
|
|
|
# Load image and make sure it fits our screen.
|
|
def pure_pil_alpha_to_color_v2(image, color=(0, 0, 0)):
|
|
"""Alpha composite an RGBA Image with a specified color.
|
|
|
|
Simpler, faster version than the solutions above.
|
|
|
|
Source: http://stackoverflow.com/a/9459208/284318
|
|
|
|
Keyword Arguments:
|
|
image -- PIL RGBA Image object
|
|
color -- Tuple r, g, b (default 255, 255, 255)
|
|
|
|
"""
|
|
image.load() # needed for split()
|
|
background = Image.new('RGB', image.size, color)
|
|
background.paste(image, mask=image.split()[3]) # 3 is the alpha channel
|
|
return background
|
|
|
|
image = Image.open("../testImg.bmp")
|
|
image.thumbnail((128,62), Image.ANTIALIAS)
|
|
RGBImage = pure_pil_alpha_to_color_v2(image)
|
|
|
|
|
|
|
|
|
|
# offscreen canvas that can be written to which can then be set to the matrix async
|
|
offscreen_canvas = matrix.CreateFrameCanvas()
|
|
offscreen_canvas.brightness = 50
|
|
offscreen__text_canvas = matrix.CreateFrameCanvas()
|
|
offscreen__text_canvas.brightness = 50
|
|
offscreen__text_canvas.SetImage(RGBImage)
|
|
|
|
|
|
def updateScreen():
|
|
global color, matrix, offscreen_canvas
|
|
nextCanvas = offscreen_canvas
|
|
if(color == 1):
|
|
offscreen_canvas.Fill(255,255,255)
|
|
nextCanvas = offscreen_canvas
|
|
elif(color == 0):
|
|
offscreen_canvas.Fill(255,0,0)
|
|
nextCanvas = offscreen_canvas
|
|
elif(color == 2):
|
|
nextCanvas = offscreen__text_canvas
|
|
|
|
nextCanvas = matrix.SwapOnVSync(nextCanvas)
|
|
|
|
|
|
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)
|
|
|