Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
https://davidsword.ca/learning-to-count-to-5-with-a-raspberry-pi/ Raspberry Pi - Learn to count to 5 game. Pressing button, count up to number on OLED screen, then celebrate once reached.
# https://davidsword.ca/learning-to-count-to-5-with-a-raspberry-pi/
import time
import busio
import adafruit_ssd1306
import RPi.GPIO as GPIO
from board import SCL, SDA
from PIL import Image, ImageDraw, ImageFont
# settings
countTo = 5 # probably don't change as there's 5 LEDs to count along with.
onCompleteMsg = "YAYY !!"
animateLoop = 20 # how many LED chases to do
animateSpeed = 0.04 # how fast to chase
screenPadding = 3
fontSize = 40
fontTTF = 'whatever-font.ttf' # probably need to change this, see http://www.dafont.com/bitmap.php
# @see https://pinout.xyz/
# @see https://davidsword.ca/wp-content/uploads/2020/06/learning-to-count-to-5-with-a-raspberry-pi.png
GPIO.setmode(GPIO.BCM)
led1 = 5
led2 = 6
led3 = 13
led4 = 19
led5 = 26
btn = 22
# Display Setup
# @see https://learn.adafruit.com/adafruit-pioled-128x32-mini-oled-for-raspberry-pi/usage
i2c = busio.I2C(SCL, SDA)
disp = adafruit_ssd1306.SSD1306_I2C(128, 32, i2c)
disp.fill(0)
disp.show()
width = disp.width
height = disp.height
image = Image.new("1", (width, height))
draw = ImageDraw.Draw(image)
font = ImageFont.truetype(fontTTF, fontSize)
def displayClear():
# Draw a black filled box to clear the image.
draw.rectangle((0, 0, width, height), outline=0, fill=0)
disp.fill(0)
disp.show()
def display(val):
displayClear()
draw.text((screenPadding, screenPadding), str(val), font=font, fill=255)
disp.image(image)
disp.show()
def turnOn(pin):
GPIO.output(pin, GPIO.HIGH)
def turnOff(pin):
GPIO.output(pin, GPIO.LOW)
# @TODO loop this dynmically
def animation():
turnOff(led1)
turnOff(led2)
turnOff(led3)
turnOff(led4)
turnOff(led5)
turnOn(led1)
time.sleep(animateSpeed)
turnOff(led1)
turnOn(led2)
time.sleep(animateSpeed)
turnOff(led2)
turnOn(led3)
time.sleep(animateSpeed)
turnOff(led3)
turnOn(led4)
time.sleep(animateSpeed)
turnOff(led4)
turnOn(led5)
time.sleep(animateSpeed)
turnOff(led5)
GPIO.setup(led1, GPIO.OUT)
GPIO.setup(led2, GPIO.OUT)
GPIO.setup(led3, GPIO.OUT)
GPIO.setup(led4, GPIO.OUT)
GPIO.setup(led5, GPIO.OUT)
GPIO.setup(btn, GPIO.IN, pull_up_down=GPIO.PUD_DOWN)
displayClear()
count = 0
while True:
if GPIO.input(btn) == GPIO.HIGH:
if count < ( countTo - 1 ):
count += 1
display(count)
if count == 1:
turnOn(led1)
if count == 2:
turnOn(led2)
if count == 3:
turnOn(led3)
if count == 4:
turnOn(led4)
# no long-pressing the button
time.sleep(0.75)
else:
# we hit the limit, do some fun stuff.
turnOn(led5)
count = countTo
display(count)
time.sleep(0.75)
display(onCompleteMsg)
for z in range(animateLoop):
animation()
count = 0
displayClear()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.