pimoroni-pico/micropython/examples/tufty2040/pride_badge.py

111 lines
4.1 KiB
Python

# A name badge with customisable Pride flag background.
from picographics import PicoGraphics, DISPLAY_TUFTY_2040
import math
display = PicoGraphics(display=DISPLAY_TUFTY_2040)
WIDTH, HEIGHT = display.get_bounds()
# List of available pen colours, add more if necessary
RED = display.create_pen(209, 34, 41)
ORANGE = display.create_pen(246, 138, 30)
YELLOW = display.create_pen(255, 216, 0)
GREEN = display.create_pen(0, 121, 64)
INDIGO = display.create_pen(36, 64, 142)
VIOLET = display.create_pen(115, 41, 130)
WHITE = display.create_pen(255, 255, 255)
PINK = display.create_pen(255, 175, 200)
BLUE = display.create_pen(116, 215, 238)
BROWN = display.create_pen(97, 57, 21)
BLACK = display.create_pen(0, 0, 0)
MAGENTA = display.create_pen(255, 33, 140)
CYAN = display.create_pen(33, 177, 255)
AMETHYST = display.create_pen(156, 89, 209)
# Uncomment one of these to change flag
# If adding your own, colour order is left to right (or top to bottom)
COLOUR_ORDER = [RED, ORANGE, YELLOW, GREEN, INDIGO, VIOLET] # traditional pride flag
# COLOUR_ORDER = [BLACK, BROWN, RED, ORANGE, YELLOW, GREEN, INDIGO, VIOLET] # Philadelphia pride flag
# COLOUR_ORDER = [BLUE, PINK, WHITE, PINK, BLUE] # trans flag
# COLOUR_ORDER = [MAGENTA, YELLOW, CYAN] # pan flag
# COLOUR_ORDER = [MAGENTA, VIOLET, INDIGO] # bi flag
# COLOUR_ORDER = [YELLOW, WHITE, AMETHYST, BLACK] # non-binary flag
# Add chevrons to the left
# CHEVRONS = [] # No chevrons
CHEVRONS = [WHITE, PINK, BLUE, BROWN, BLACK] # Progress Pride Flag
# Initial chevron height compared to screen height
FIRST_CHEVRON_HEIGHT = 0.4
# Change this for vertical stripes
STRIPES_DIRECTION = "horizontal"
# Change details here! Works best with a short, one word name
NAME = "Hel"
PRONOUNS = "She/they"
# Change the colour of the text (swapping these works better on a light background)
TEXT_COLOUR = WHITE
DROP_SHADOW_COLOUR = BLACK
# Draw the flag
if STRIPES_DIRECTION == "horizontal":
stripe_width = round(HEIGHT / len(COLOUR_ORDER))
for x in range(len(COLOUR_ORDER)):
display.set_pen(COLOUR_ORDER[x])
display.rectangle(0, stripe_width * x, WIDTH, stripe_width)
if STRIPES_DIRECTION == "vertical":
stripe_width = round(WIDTH / len(COLOUR_ORDER))
for x in range(len(COLOUR_ORDER)):
display.set_pen(COLOUR_ORDER[x])
display.rectangle(stripe_width * x, 0, stripe_width, HEIGHT)
if len(CHEVRONS) > 0:
stripe_width = round((HEIGHT * (1 - FIRST_CHEVRON_HEIGHT)) / len(CHEVRONS))
offset = -stripe_width * math.floor((len(CHEVRONS) + 1) / 2)
middle = round(HEIGHT / 2)
for x in range(len(CHEVRONS) - 1, -1, -1):
display.set_pen(CHEVRONS[x])
display.triangle(
x * stripe_width + offset, -stripe_width,
(x + 1) * stripe_width + offset + middle, middle,
x * stripe_width + offset, HEIGHT + stripe_width)
# Set a starting scale for text size.
# This is intentionally bigger than will fit on the screen, we'll shrink it to fit.
name_size = 20
pronouns_size = 20
# These loops adjust the scale of the text until it fits on the screen
while True:
display.set_font("bitmap8")
name_length = display.measure_text(NAME, name_size)
if name_length >= WIDTH - 20:
name_size -= 1
else:
# comment out this section if you hate drop shadow
DROP_SHADOW_OFFSET = 5
display.set_pen(DROP_SHADOW_COLOUR)
display.text(NAME, int((WIDTH - name_length) / 2 + 10) - DROP_SHADOW_OFFSET, 10 + DROP_SHADOW_OFFSET, WIDTH, name_size)
# draw name and stop looping
display.set_pen(TEXT_COLOUR)
display.text(NAME, int((WIDTH - name_length) / 2 + 10), 10, WIDTH, name_size)
break
while True:
display.set_font("bitmap8")
pronouns_length = display.measure_text(PRONOUNS, pronouns_size)
if pronouns_length >= WIDTH - 60:
pronouns_size -= 1
else:
# draw pronouns and stop looping
display.set_pen(TEXT_COLOUR)
display.text(PRONOUNS, int((WIDTH - pronouns_length) / 2), 175, WIDTH, pronouns_size)
break
# Once all the adjusting and drawing is done, update the display.
display.update()