mirror of
https://github.com/pybricks/pybricks-api.git
synced 2026-09-15 02:54:52 +00:00
Laurens Valk
adf8ed6b56
First pass including Image and ImageFile. Largely migrated from old ev3dev version. We can refine this for the embedded version in the next commits.
131 lines
3.1 KiB
Python
131 lines
3.1 KiB
Python
import math
|
|
|
|
from pybricks.hubs import EV3Brick
|
|
from pybricks.parameters import Color
|
|
from pybricks.tools import wait
|
|
from pybricks.media.ev3dev import Font, Image
|
|
|
|
|
|
# Initialize the EV3
|
|
ev3 = EV3Brick()
|
|
|
|
|
|
# SPLIT SCREEN ################################################################
|
|
|
|
# Make a sub-image for the left half of the screen
|
|
left = Image(
|
|
ev3.screen,
|
|
sub=True,
|
|
x1=0,
|
|
y1=0,
|
|
x2=ev3.screen.width // 2 - 1,
|
|
y2=ev3.screen.height - 1,
|
|
)
|
|
|
|
# Make a sub-image for the right half of the screen
|
|
right = Image(
|
|
ev3.screen,
|
|
sub=True,
|
|
x1=ev3.screen.width // 2,
|
|
y1=0,
|
|
x2=ev3.screen.width - 1,
|
|
y2=ev3.screen.height - 1,
|
|
)
|
|
|
|
# Use a monospaced font so that text is vertically aligned when we print
|
|
right.set_font(Font(size=8, monospace=True))
|
|
|
|
|
|
# Graphing y = sin(x)
|
|
def f(x):
|
|
return math.sin(x)
|
|
|
|
|
|
for t in range(200):
|
|
# Graph on left side
|
|
|
|
# Scale t to x-axis and compute y values
|
|
x0 = (t - 1) * 2 * math.pi / left.width
|
|
y0 = f(x0)
|
|
x1 = t * 2 * math.pi / left.width
|
|
y1 = f(x1)
|
|
|
|
# Scale y values to screen coordinates
|
|
sy0 = (-y0 + 1) * left.height / 2
|
|
sy1 = (-y1 + 1) * left.height / 2
|
|
|
|
# Shift the current graph to the left one pixel
|
|
left.draw_image(-1, 0, left)
|
|
# Fill the last column with white to erase the previous plot point
|
|
left.draw_line(left.width - 1, 0, left.width - 1, left.height - 1, 1, Color.WHITE)
|
|
# Draw the new value of the graph in the last column
|
|
left.draw_line(left.width - 2, int(sy0), left.width - 1, int(sy1), 3)
|
|
|
|
# Print every 10th value on right side
|
|
if t % 10 == 0:
|
|
right.print("{:10.2f}{:10.2f}".format(x1, y1))
|
|
|
|
wait(100)
|
|
|
|
|
|
# SPRITE ANIMATION ############################################################
|
|
|
|
# Copy of screen for double-buffering
|
|
buf = Image(ev3.screen)
|
|
|
|
# Load images from file
|
|
bg = Image("background.png")
|
|
sprite = Image("sprite.png")
|
|
|
|
# Number of cells in each sprite animation
|
|
NUM_CELLS = 8
|
|
|
|
# Each cell in the sprite is 75 x 100 pixels
|
|
CELL_WIDTH, CELL_HEIGHT = 75, 100
|
|
|
|
# Get sub-images for each individual cell
|
|
# This is more efficient that loading individual images
|
|
walk_right = [
|
|
Image(
|
|
sprite,
|
|
sub=True,
|
|
x1=x * CELL_WIDTH,
|
|
y1=0,
|
|
x2=(x + 1) * CELL_WIDTH - 1,
|
|
y2=CELL_HEIGHT - 1,
|
|
)
|
|
for x in range(NUM_CELLS)
|
|
]
|
|
walk_left = [
|
|
Image(
|
|
sprite,
|
|
sub=True,
|
|
x1=x * CELL_WIDTH,
|
|
y1=CELL_HEIGHT,
|
|
x2=(x + 1) * CELL_WIDTH - 1,
|
|
y2=2 * CELL_HEIGHT - 1,
|
|
)
|
|
for x in range(NUM_CELLS)
|
|
]
|
|
|
|
|
|
# Walk from left to right
|
|
for x in range(-100, 200, 2):
|
|
# Start with the background image
|
|
buf.draw_image(0, 0, bg)
|
|
# Draw the current sprite - purple is treated as transparent
|
|
buf.draw_image(x, 5, walk_right[x // 5 % NUM_CELLS], Color.PURPLE)
|
|
# Copy the double-buffer to the screen
|
|
ev3.screen.draw_image(0, 0, buf)
|
|
# 20 frames per second
|
|
wait(50)
|
|
|
|
# Walk from right to left
|
|
for x in range(200, -100, -2):
|
|
buf.draw_image(0, 0, bg)
|
|
buf.draw_image(x, 5, walk_left[x // 5 % NUM_CELLS], Color.PURPLE)
|
|
ev3.screen.draw_image(0, 0, buf)
|
|
wait(50)
|
|
|
|
wait(1000)
|