Matrix/API/main.py

229 lines
5.1 KiB
Python
Raw Normal View History

2023-11-22 15:21:15 +00:00
#!/usr/bin/env python
2023-11-22 15:44:26 +00:00
from deps.samplebase import SampleBase
2023-11-22 15:21:15 +00:00
from rgbmatrix import graphics
2023-11-23 07:15:02 +00:00
2023-11-23 09:07:17 +00:00
from flask import Flask, request, jsonify
2023-11-23 07:49:47 +00:00
from waitress import serve
import time, threading
2023-11-22 14:41:13 +00:00
2023-11-23 07:15:02 +00:00
2023-11-23 09:07:17 +00:00
api = Flask(__name__)
2023-11-23 07:15:02 +00:00
2023-11-23 09:07:41 +00:00
@api.route('/text', methods=['POST'])
def display_text():
2023-11-23 10:45:56 +00:00
# prepare response data
2023-11-23 09:07:41 +00:00
response = { 'success': True }
try:
global matrix
2023-11-23 10:45:56 +00:00
# receive client data
2023-11-23 09:07:41 +00:00
data = request.get_json()
2023-11-23 10:45:56 +00:00
# try unpacking text and x,y coordinates
2023-11-23 09:07:41 +00:00
text = data.get('text')
x_coord = data.get('x')
y_coord = data.get('y')
2023-11-23 10:45:56 +00:00
# call matrix method with data
2023-11-23 09:07:41 +00:00
matrix.text(x_coord, y_coord, text)
2023-11-23 09:44:57 +00:00
except Exception as e:
2023-11-23 10:45:56 +00:00
# error handling
2023-11-23 09:44:57 +00:00
print(e)
2023-11-23 09:07:41 +00:00
response['success'] = False
2023-11-23 10:45:56 +00:00
# respond to client
2023-11-23 09:07:41 +00:00
return jsonify(response)
2023-11-23 09:46:15 +00:00
@api.route('/pixel', methods=['POST'])
def set_pixel():
2023-11-23 10:45:56 +00:00
# prepare response data
2023-11-23 09:46:15 +00:00
response = { 'success': True }
try:
global matrix
2023-11-23 10:45:56 +00:00
# receive client data
2023-11-23 09:46:15 +00:00
data = request.get_json()
2023-11-23 10:45:56 +00:00
# try unpacking x,y coordinates
2023-11-23 09:46:15 +00:00
x_coord = data.get('x')
y_coord = data.get('y')
2023-11-23 10:45:56 +00:00
# call matrix method with data
2023-11-23 09:46:15 +00:00
matrix.pixel(x_coord, y_coord)
except Exception as e:
2023-11-23 10:45:56 +00:00
# error handling
2023-11-23 09:46:15 +00:00
print(e)
response['success'] = False
2023-11-23 10:45:56 +00:00
# respond to client
2023-11-23 09:46:15 +00:00
return jsonify(response)
2023-11-23 09:47:40 +00:00
@api.route('/circle', methods=['POST'])
def draw_circle():
2023-11-23 10:45:56 +00:00
# prepare response data
2023-11-23 09:47:40 +00:00
response = { 'success': True }
try:
global matrix
2023-11-23 10:45:56 +00:00
# receive client data
2023-11-23 09:47:40 +00:00
data = request.get_json()
2023-11-23 10:45:56 +00:00
# try unpacking radius and x,y coordinates
2023-11-23 09:47:40 +00:00
radius = data.get('r')
x_coord = data.get('x')
y_coord = data.get('y')
2023-11-23 10:45:56 +00:00
# call matrix method with data
2023-11-23 09:47:40 +00:00
matrix.circle(x_coord, y_coord, radius)
except Exception as e:
2023-11-23 10:45:56 +00:00
# error handling
2023-11-23 09:47:40 +00:00
print(e)
response['success'] = False
2023-11-23 10:45:56 +00:00
# respond to client
2023-11-23 09:47:40 +00:00
return jsonify(response)
2023-11-23 09:48:19 +00:00
@api.route('/rectangle', methods=['POST'])
def draw_rectangle():
2023-11-23 10:45:56 +00:00
# prepare response data
2023-11-23 09:48:19 +00:00
response = { 'success': True }
try:
global matrix
2023-11-23 10:45:56 +00:00
# receive client data
2023-11-23 09:48:19 +00:00
data = request.get_json()
2023-11-23 10:45:56 +00:00
# try unpacking rectangle width and height
2023-11-23 09:48:19 +00:00
width = data.get('w')
height = data.get('h')
2023-11-23 10:45:56 +00:00
# try unpacking x,y coordinates
2023-11-23 09:48:19 +00:00
x_coord = data.get('x')
y_coord = data.get('y')
2023-11-23 10:45:56 +00:00
# call matrix method with data
2023-11-23 09:48:19 +00:00
matrix.rectangle(x_coord, y_coord, width, height)
except Exception as e:
2023-11-23 10:45:56 +00:00
# error handling
2023-11-23 09:48:19 +00:00
print(e)
response['success'] = False
2023-11-23 10:45:56 +00:00
# respond to client
2023-11-23 09:48:19 +00:00
return jsonify(response)
2023-11-23 09:46:43 +00:00
@api.route('/color', methods=['POST'])
def set_color():
2023-11-23 10:45:56 +00:00
# prepare response data
2023-11-23 09:46:43 +00:00
response = { 'success': True }
try:
global matrix
2023-11-23 10:45:56 +00:00
# receive client data
2023-11-23 09:46:43 +00:00
data = request.get_json()
2023-11-23 10:45:56 +00:00
# try unpacking red, green and blue color values
2023-11-23 09:46:43 +00:00
red = data.get('r')
green = data.get('g')
blue = data.get('b')
2023-11-23 10:45:56 +00:00
# call matrix method with data
2023-11-23 09:46:43 +00:00
matrix.set_color(red, green, blue)
except Exception as e:
2023-11-23 10:45:56 +00:00
# error handling
2023-11-23 09:46:43 +00:00
print(e)
response['success'] = False
2023-11-23 10:45:56 +00:00
# respond to client
2023-11-23 09:46:43 +00:00
return jsonify(response)
2023-11-23 09:49:03 +00:00
@api.route('/clear', methods=['GET'])
def update_matrix():
2023-11-23 10:45:56 +00:00
# prepare response data
2023-11-23 09:49:03 +00:00
response = { 'success': True }
try:
global matrix
2023-11-23 10:45:56 +00:00
# receive client data
2023-11-23 09:49:03 +00:00
matrix.clear()
except Exception as e:
2023-11-23 10:45:56 +00:00
# error handling
2023-11-23 09:49:03 +00:00
print(e)
response['success'] = False
return jsonify(response)
2023-11-23 09:07:41 +00:00
@api.route('/update', methods=['GET'])
def update_matrix():
response = { 'success': True }
try:
global matrix
matrix.update()
2023-11-23 09:44:57 +00:00
except Exception as e:
print(e)
2023-11-23 09:07:41 +00:00
response['success'] = False
2023-11-23 10:45:56 +00:00
# respond to client
2023-11-23 09:07:41 +00:00
return jsonify(response)
2023-11-23 07:49:47 +00:00
2023-11-23 07:15:02 +00:00
2023-11-23 10:45:56 +00:00
# derived class to access rgb-led-matrix
2023-11-23 09:07:17 +00:00
class Matrix(SampleBase):
2023-11-22 15:22:19 +00:00
def __init__(self, *args, **kwargs):
2023-11-23 09:07:17 +00:00
super(Matrix, self).__init__(*args, **kwargs)
2023-11-22 14:41:13 +00:00
2023-11-23 10:45:56 +00:00
# run display program
2023-11-22 15:22:19 +00:00
def run(self):
2023-11-23 10:45:56 +00:00
# initialize font
2023-11-23 07:49:47 +00:00
self.font = graphics.Font()
self.font.LoadFont("deps/fonts/9x18B.bdf")
2023-11-23 10:45:56 +00:00
# initialize color
2023-11-23 07:49:47 +00:00
self.color = graphics.Color(255, 255, 255)
2023-11-23 09:07:17 +00:00
2023-11-23 10:45:56 +00:00
# initialize canvas
2023-11-23 07:49:47 +00:00
self.canvas = self.matrix.CreateFrameCanvas()
2023-11-23 09:07:17 +00:00
self.canvas.Clear()
2023-11-22 14:41:13 +00:00
2023-11-23 10:45:56 +00:00
# wait in idle
2023-11-22 15:21:15 +00:00
while True:
2023-11-23 07:49:47 +00:00
time.sleep(.05)
2023-11-23 07:15:02 +00:00
2023-11-23 10:45:56 +00:00
# clear canvas
2023-11-23 09:44:57 +00:00
def clear(self):
self.canvas.Clear()
2023-11-23 10:45:56 +00:00
# swap canvas with buffer
2023-11-23 07:49:47 +00:00
def update(self):
self.canvas = self.matrix.SwapOnVSync(self.canvas)
2023-11-23 09:44:57 +00:00
self.clear()
2023-11-23 09:07:17 +00:00
2023-11-23 10:45:56 +00:00
# set current color
2023-11-23 09:46:15 +00:00
def set_color(self, r, g, b):
self.color = graphics.Color(r, g, b)
2023-11-23 10:45:56 +00:00
# set specified pixel to current color
2023-11-23 09:46:15 +00:00
def pixel(self, x, y):
self.canvas.SetPixel(x, y, self.color.red, self.color.green, self.color.blue)
2023-11-23 10:45:56 +00:00
# display text at position with current color
2023-11-23 09:07:17 +00:00
def text(self, x, y, t):
graphics.DrawText(self.canvas, self.font, x*9, (y+1)*18, self.color, t)
2023-11-22 14:41:13 +00:00
2023-11-23 10:45:56 +00:00
# display circle at position with radius in current color
2023-11-23 09:47:40 +00:00
def circle(self, x, y, r):
graphics.DrawCircle(self.canvas, self.font, x, y, r, self.color)
2023-11-23 10:45:56 +00:00
# display rectangle at position with dimensions in current color
2023-11-23 09:48:19 +00:00
def rectangle(self, x, y, w, h):
2023-11-23 10:45:56 +00:00
# loop through each point in dimensions
2023-11-23 09:48:19 +00:00
for i in range(w):
for j in range(h):
2023-11-23 10:45:56 +00:00
# set next pixel in rectangle
2023-11-23 09:48:19 +00:00
self.canvas.SetPixel(x+i, y+j, self.color.red, self.color.green, self.color.blue)
2023-11-22 15:21:15 +00:00
2023-11-23 10:45:56 +00:00
# initialize global matrix variable
2023-11-23 09:07:17 +00:00
matrix = None
2023-11-22 15:21:15 +00:00
if __name__ == "__main__":
2023-11-23 10:45:56 +00:00
# set matrix to instance
2023-11-23 09:07:17 +00:00
matrix = Matrix()
2023-11-23 10:45:56 +00:00
# run rest api in thread, serving with 'waitress'
2023-11-23 09:07:17 +00:00
threading.Thread(target=lambda: serve(api, host="0.0.0.0", port=8080)).start()
2023-11-23 07:15:02 +00:00
2023-11-23 10:45:56 +00:00
# help menu on call
2023-11-23 09:07:17 +00:00
if not matrix.process():
matrix.print_help()