2021-07-01 19:14:36 +00:00
import paho . mqtt . client as mqtt
import time
import board
import busio
2021-07-04 15:23:35 +00:00
2021-07-01 19:14:36 +00:00
from adafruit_ht16k33 import segments
2021-07-03 19:14:56 +00:00
from enum import Enum
2021-07-01 19:14:36 +00:00
2021-07-04 15:23:35 +00:00
from gpiozero import Button
import socket #to get local ip
2021-07-01 19:14:36 +00:00
__TOPIC_BRIGHTNESS__ = " brightness "
2021-07-03 19:14:56 +00:00
__TOPIC_SCROLL_INTERVAL__ = " scrollinterval "
__TOPIC_TEXT_STATIC__ = " textstatic "
__TOPIC_TEXT_ONCE__ = " textonce "
__TOPIC_TEXT_SCROLL__ = " textscroll "
2021-07-03 21:11:45 +00:00
__TOPIC_TEXT_MARQUEE__ = " textmarquee "
__TOPIC_TEXT_PAN__ = " textpan "
2021-07-03 19:14:56 +00:00
2021-07-04 15:23:35 +00:00
__TOPIC_BUTTON__ = " button "
2021-07-03 19:14:56 +00:00
NUMDIGITS = 8
class Mode ( Enum ) :
STATIC = 1 #show text constantly. may cut digits
ONCE = 3 #show once for some time
SCROLL = 2 #scroll thorugh once
2021-07-03 21:11:45 +00:00
PAN = 4 #scroll back and forth
MARQUEE = 5
2021-07-01 19:14:36 +00:00
class Controller :
2021-07-01 19:50:35 +00:00
connected = False
2021-07-03 19:14:56 +00:00
displayL = None
displayR = None
2023-08-19 09:33:06 +00:00
basetopic = None
mqtthost = None
mqttport = None
2021-07-03 21:11:45 +00:00
2021-07-01 19:50:35 +00:00
def __init__ ( self , base_topic , mqtt_host , mqtt_port = 1883 ) :
self . scroll_interval = 0.25
2021-07-03 21:11:45 +00:00
self . text = " "
self . text_last = " "
2021-07-01 19:50:35 +00:00
self . poscount = 0
2021-07-03 21:11:45 +00:00
self . last_scrollupdate = 0
self . mode = Mode . STATIC
2021-07-04 15:23:35 +00:00
self . buttonLeft = Button ( 23 , pull_up = False , hold_time = 1 ) # https://gpiozero.readthedocs.io/en/stable/api_input.html
self . buttonRight = Button ( 24 , pull_up = False , hold_time = 1 ) # with deactivated pullup input is active high
self . buttonLeft . when_released = self . buttonLeft_released
self . buttonLeft . when_held = self . buttonLeft_held
self . buttonLeft_heldFlag = False #if true, next released will be ignored
self . buttonRight . when_released = self . buttonRight_released
self . buttonRight . when_held = self . buttonRight_held
self . buttonRight_heldFlag = False #if true, next released will be ignored
2021-07-01 19:50:35 +00:00
# Create the I2C interface.
self . i2c = busio . I2C ( board . SCL , board . SDA )
# Create the LED segment class.
# This creates a 14 segment 4 character display:
2021-07-03 19:14:56 +00:00
self . displayL = segments . Seg14x4 ( self . i2c , address = 0x70 ) # find address with "sudo i2cdetect -y 1" . you need to install: sudo apt-get install -y python-smbus i2c-tools
self . displayR = segments . Seg14x4 ( self . i2c , address = 0x71 )
2021-07-01 19:50:35 +00:00
# Clear the display.
2021-07-03 19:14:56 +00:00
self . displayL . fill ( 0 )
self . displayR . fill ( 0 )
2021-07-01 19:50:35 +00:00
# set brightness, range 0-1.0, 1.0 max brightness
2021-07-03 19:14:56 +00:00
self . displayL . brightness = 1.0
self . displayR . brightness = 1.0
2023-08-19 09:33:06 +00:00
self . basetopic = basetopic
self . mqtthost = mqtt_host
self . mqttport = mqtt_port
2023-08-19 09:29:56 +00:00
2023-08-19 09:33:06 +00:00
connectToMQTT ( )
2023-08-19 09:29:56 +00:00
2023-08-19 09:33:06 +00:00
def connectToMQTT ( self ) :
print ( " trying to connect to mqtt " )
2021-07-01 19:50:35 +00:00
self . mqtt = mqtt . Client ( )
self . mqtt . on_connect = self . on_connect
self . mqtt . on_message = self . on_message
self . mqtt . on_disconnect = self . on_disconnect
2023-08-19 09:29:56 +00:00
self . mqtt . connect ( self . mqtthost , self . mqttport )
self . topic = self . basetopic
2021-07-01 19:50:35 +00:00
2021-07-04 15:23:35 +00:00
2021-07-01 19:50:35 +00:00
def on_disconnect ( self , client , userdata , rc ) :
print ( " MQTT disconnected " )
2023-08-19 09:33:06 +00:00
self . mode = Mode . STATIC
2023-08-19 09:29:56 +00:00
self . text = " - - - - "
2021-07-01 19:50:35 +00:00
self . connected = False
def on_message ( self , client , userdata , message ) :
msg = str ( message . payload . decode ( " utf-8 " ) )
print ( " msg = " + msg )
if message . topic . endswith ( __TOPIC_BRIGHTNESS__ + " /set " ) :
val = float ( msg )
if val > = 0.0 and val < = 1.0 :
2021-07-03 19:14:56 +00:00
self . displayL . brightness = val
self . displayR . brightness = val
print ( " changed brightness to " , self . displayL . brightness )
self . mqtt . publish ( self . topic + " / " + __TOPIC_BRIGHTNESS__ , self . displayL . brightness , 1 )
2021-07-01 19:50:35 +00:00
else :
print ( " invalid brightness " , val )
if message . topic . endswith ( __TOPIC_SCROLL_INTERVAL__ + " /set " ) :
val = float ( msg )
if val > 0.0 :
self . scroll_interval = val
print ( " changed scroll_interval to " , self . scroll_interval )
self . mqtt . publish ( self . topic + " / " + __TOPIC_SCROLL_INTERVAL__ , self . scroll_interval , 1 )
else :
print ( " error not >0.0: " , val )
2021-07-03 21:11:45 +00:00
if message . topic . endswith ( __TOPIC_TEXT_STATIC__ + " /set " ) :
self . mode = Mode . STATIC
self . text = msg . ljust ( NUMDIGITS )
print ( " changed text to " , self . text )
self . mqtt . publish ( self . topic + " / " + __TOPIC_TEXT_STATIC__ , self . text , 1 )
2021-07-03 21:28:47 +00:00
if message . topic . endswith ( __TOPIC_TEXT_ONCE__ + " /set " ) :
self . mode = Mode . ONCE
self . text = msg . ljust ( NUMDIGITS )
self . poscount = 0 #use for timing
print ( " changed text to " , self . text )
self . mqtt . publish ( self . topic + " / " + __TOPIC_TEXT_ONCE__ , self . text , 1 )
2021-07-01 19:50:35 +00:00
2021-07-03 21:11:45 +00:00
if message . topic . endswith ( __TOPIC_TEXT_MARQUEE__ + " /set " ) :
self . mode = Mode . MARQUEE
2021-07-01 20:06:18 +00:00
self . poscount = 0 #start at beginning of new text
2021-07-01 19:50:35 +00:00
self . text = msg
2021-07-03 21:11:45 +00:00
for i in range ( NUMDIGITS ) :
self . text = " " + self . text + " " #add spaces left and right for marquee to enter and leave the full frame
2021-07-01 19:50:35 +00:00
print ( " changed text to " , self . text )
2021-07-03 21:11:45 +00:00
self . mqtt . publish ( self . topic + " / " + __TOPIC_TEXT_MARQUEE__ , self . text , 1 )
2021-07-03 19:14:56 +00:00
2021-07-03 21:11:45 +00:00
if message . topic . endswith ( __TOPIC_TEXT_PAN__ + " /set " ) :
self . mode = Mode . PAN
2021-07-03 19:14:56 +00:00
self . poscount = 0 #start at beginning of new text
self . text = msg
2021-07-03 21:11:45 +00:00
for i in range ( int ( NUMDIGITS / 4 ) ) :
self . text = " " + self . text + " " #add spaces left and right for marquee to enter and leave the full frame
2021-07-03 19:14:56 +00:00
print ( " changed text to " , self . text )
2021-07-03 21:11:45 +00:00
self . mqtt . publish ( self . topic + " / " + __TOPIC_TEXT_PAN__ , self . text , 1 )
2021-07-03 21:17:56 +00:00
if message . topic . endswith ( __TOPIC_TEXT_SCROLL__ + " /set " ) :
self . mode = Mode . SCROLL
self . poscount = 0 #start at beginning of new text
self . text = msg
for i in range ( NUMDIGITS ) :
self . text = " " + self . text + " " #add spaces left and right for marquee to enter and leave the full frame
print ( " changed text to " , self . text )
self . mqtt . publish ( self . topic + " / " + __TOPIC_TEXT_SCROLL__ , self . text , 1 )
2021-07-01 19:50:35 +00:00
def on_connect ( self , client , userdata , flags , rc ) :
print ( " Connected to MQTT Broker " )
self . mqtt . subscribe ( self . topic + " /# " )
2021-07-04 15:23:35 +00:00
2021-07-01 19:50:35 +00:00
self . connected = True
2021-07-03 21:11:45 +00:00
2021-07-04 15:23:35 +00:00
self . mqtt . publish ( self . topic + " / " + " $state " , " ready " , 1 )
self . mqtt . publish ( self . topic + " / " + " $hostname " , socket . gethostname ( ) , 1 )
self . mqtt . publish ( self . topic + " / " + " $localip " , self . get_ip ( ) , 1 )
2023-08-19 09:33:06 +00:00
#clear display
self . mode = Mode . STATIC
2023-08-19 09:29:56 +00:00
self . text = " "
2021-07-04 15:23:35 +00:00
def get_ip ( self ) :
s = socket . socket ( socket . AF_INET , socket . SOCK_DGRAM )
try :
# doesn't even have to be reachable
s . connect ( ( ' 10.255.255.255 ' , 1 ) )
IP = s . getsockname ( ) [ 0 ]
except Exception :
IP = ' 127.0.0.1 '
finally :
s . close ( )
return IP
2021-07-03 21:11:45 +00:00
def writeSegment ( self , _digit , _char , clear = False ) :
if _digit > = 0 and _digit < 4 :
if clear :
self . displayL [ _digit ] = ' '
self . displayL [ _digit ] = _char
elif _digit < 8 :
if clear :
self . displayR [ _digit - 4 ] = ' '
self . displayR [ _digit - 4 ] = _char
def displayTextOffset ( self , _text , _offset = 0 ) :
# This function is needed because library print functions only work for a single display module. This function handles the '.' character with its special cases.
_text + = " " #append NUMDIGITS spaces at end for when scrolling past end
2021-07-03 19:14:56 +00:00
2021-07-03 21:11:45 +00:00
_text = _text [ _offset : ] #move by offset
i = 0 #counter for reading pos
d = 0 #counter for writing segment pos
while d < NUMDIGITS and i < len ( _text ) :
if _text [ i ] == ' . ' and _text [ max ( 0 , i - 1 ) ] == ' . ' : #is dot and was before (or is at start)
self . writeSegment ( d , _text [ i ] , True )
d + = 1
i + = 1
else :
if _text [ i ] != ' . ' : #normal char
self . writeSegment ( d , _text [ i ] )
i + = 1
if i < len ( _text ) :
if _text [ i ] == ' . ' : #dot, but can be appended to last char
self . writeSegment ( d , _text [ i ] )
i + = 1
d + = 1
def seglen ( self , _text ) : #len function but optimized for 14-segment display
i = 0 #counter for reading pos
d = 0 #counter for writing segment pos
while i < len ( _text ) :
if _text [ i ] == ' . ' and _text [ max ( 0 , i - 1 ) ] == ' . ' : #is dot and was before (or is at start)
d + = 1
i + = 1
else :
if _text [ i ] != ' . ' : #normal char
i + = 1
if i < len ( _text ) :
if _text [ i ] == ' . ' : #dot, but can be appended to last char
i + = 1
d + = 1
return d
2021-07-01 19:50:35 +00:00
def loop_forever ( self ) :
run = True
while run :
2023-08-19 09:33:06 +00:00
2023-08-19 09:33:55 +00:00
if not self . connected :
2023-08-19 09:33:06 +00:00
if ( time . time ( ) > self . last_reconnectTry + 30 ) :
connectToMQTT ( )
self . last_reconnectTry = time . time ( )
2021-07-01 20:06:18 +00:00
self . mqtt . loop ( 0.1 ) #with block timeout
2023-08-19 09:33:06 +00:00
2021-07-04 15:23:35 +00:00
2021-07-01 19:50:35 +00:00
2021-07-03 19:14:56 +00:00
if self . displayL is not None and self . displayR is not None : #displays initialized
2021-07-03 21:11:45 +00:00
2021-07-03 21:28:47 +00:00
if self . mode == Mode . STATIC or self . mode == Mode . ONCE :
2021-07-03 21:11:45 +00:00
if self . text != self . text_last : #time to update animation
self . displayTextOffset ( self . text , 0 )
2021-07-03 21:28:47 +00:00
if self . mode == Mode . ONCE :
if time . time ( ) > self . last_scrollupdate + self . scroll_interval :
if self . poscount > = NUMDIGITS : #time it stays depends on scrollspeed (how long it would take to pass over the frame)
self . clearAndStop ( )
self . poscount + = 1 #used for timing
self . last_scrollupdate = time . time ( )
2021-07-03 21:11:45 +00:00
2021-07-03 21:17:56 +00:00
if self . mode == Mode . MARQUEE or self . mode == Mode . SCROLL :
2021-07-03 21:11:45 +00:00
if time . time ( ) > self . last_scrollupdate + self . scroll_interval or self . text != self . text_last : #time to update animation
self . displayTextOffset ( self . text , self . poscount )
2021-07-03 19:14:56 +00:00
self . poscount + = 1
2021-07-03 21:17:56 +00:00
if self . mode == Mode . MARQUEE :
self . poscount % = max ( 1 , int ( self . seglen ( self . text ) - NUMDIGITS + 1 ) )
elif self . mode == Mode . SCROLL :
if self . poscount > = max ( 1 , int ( self . seglen ( self . text ) - NUMDIGITS + 1 ) ) : #reached end for scroll once
2021-07-03 21:28:47 +00:00
self . clearAndStop ( )
2021-07-03 21:17:56 +00:00
2021-07-03 19:14:56 +00:00
self . last_scrollupdate = time . time ( )
2021-07-03 21:17:56 +00:00
2021-07-03 21:11:45 +00:00
if self . mode == Mode . PAN :
if time . time ( ) > self . last_scrollupdate + self . scroll_interval or self . text != self . text_last : #time to update animation
2021-07-03 19:14:56 +00:00
print ( " pos= " , self . poscount )
scrolloffset = self . poscount
2021-07-03 21:11:45 +00:00
if scrolloffset > = ( self . seglen ( self . text ) - NUMDIGITS ) :
scrolloffset = ( self . seglen ( self . text ) - NUMDIGITS ) - ( scrolloffset - ( self . seglen ( self . text ) - NUMDIGITS ) ) #reverse
2021-07-03 19:14:56 +00:00
scrolloffset = max ( 0 , scrolloffset ) #cap at zero
print ( " scrolloffset= " , scrolloffset )
2021-07-03 21:11:45 +00:00
self . displayTextOffset ( self . text , scrolloffset )
2021-07-03 19:14:56 +00:00
self . poscount + = 1
2021-07-03 21:11:45 +00:00
self . poscount % = ( self . seglen ( self . text ) - NUMDIGITS ) * 2 #if poscount over scrollable width means scroll backwards
2021-07-03 19:14:56 +00:00
self . last_scrollupdate = time . time ( )
2021-07-03 21:11:45 +00:00
self . text_last = self . text
2021-07-03 21:28:47 +00:00
def clearAndStop ( self ) :
self . mode == Mode . STATIC #Stop animation
self . text = " " . ljust ( NUMDIGITS ) #empty
2021-07-04 15:23:35 +00:00
self . displayTextOffset ( self . text , 0 ) #update immediately
def buttonLeft_released ( self ) :
if self . buttonLeft_heldFlag :
self . buttonLeft_heldFlag = False
else :
print ( " button left pressed " )
if self . connected :
self . mqtt . publish ( self . topic + " / " + __TOPIC_BUTTON__ , " lpress " , 1 )
def buttonLeft_held ( self ) :
print ( " button left held " )
self . buttonLeft_heldFlag = True
if self . connected :
self . mqtt . publish ( self . topic + " / " + __TOPIC_BUTTON__ , " lhold " , 1 )
def buttonRight_released ( self ) :
if self . buttonRight_heldFlag :
self . buttonRight_heldFlag = False
else :
print ( " button right pressed " )
if self . connected :
self . mqtt . publish ( self . topic + " / " + __TOPIC_BUTTON__ , " rpress " , 1 )
def buttonRight_held ( self ) :
print ( " button right held " )
self . buttonRight_heldFlag = True
if self . connected :
self . mqtt . publish ( self . topic + " / " + __TOPIC_BUTTON__ , " rhold " , 1 )