openmv/scripts/examples/02-Image-Processing/00-Drawing/image_drawing_advanced.py
2023-07-05 19:03:37 +02:00

119 lines
2.7 KiB
Python

# Draw Image Testing script with bounce
#
# Exercise draw image with many different values for testing
import sensor
import image
import time
sensor.reset()
sensor.set_pixformat(sensor.RGB565) # or GRAYSCALE...
sensor.set_framesize(sensor.QVGA) # or QQVGA...
sensor.skip_frames(time=2000)
clock = time.clock()
BOUNCE = True
RESCALE = True
SMALL_IMAGE_SCALE = 3
CYCLE_FORMATS = True
CYCLE_MASK = True
# Used when CYCLE_FORMATS or CYCLE_MASK is true
value_mixer = 0
# Location of small image
x = 100
y = 50
# Bounce direction
xd = 1
yd = 1
# Small image scaling
rescale = 1.0
rd = 0.1
max_rescale = 5
min_rescale = rd * 2
# Boundary to bounce within
xmin = -sensor.width() / SMALL_IMAGE_SCALE - 8
ymin = -sensor.height() / SMALL_IMAGE_SCALE - 8
xmax = sensor.width() + 8
ymax = sensor.height() + 8
while True:
clock.tick()
status = ""
value_mixer = value_mixer + 1
img = sensor.snapshot()
# Makes a scaled copy of the sensor
small_img = img.mean_pooled(SMALL_IMAGE_SCALE, SMALL_IMAGE_SCALE)
status = "rgb565 "
if CYCLE_FORMATS:
image_format = (value_mixer >> 8) & 3
# To test combining different formats
if image_format == 1:
small_img = small_img.to_bitmap(copy=True)
status = "bitmap "
if image_format == 2:
small_img = small_img.to_grayscale(copy=True)
status = "grayscale "
if image_format == 3:
small_img = small_img.to_rgb565(copy=True)
status = "rgb565 "
# update small image location
if BOUNCE:
x = x + xd
if x < xmin or x > xmax:
xd = -xd
y = y + yd
if y < ymin or y > ymax:
yd = -yd
# Update small image scale
if RESCALE:
rescale = rescale + rd
if rescale < min_rescale or rescale > max_rescale:
rd = -rd
# Find the center of the image
scaled_width = int(small_img.width() * abs(rescale))
scaled_height = int(small_img.height() * abs(rescale))
apply_mask = CYCLE_MASK and ((value_mixer >> 9) & 1)
if apply_mask:
img.draw_image(
small_img,
int(x),
int(y),
mask=small_img.to_bitmap(copy=True),
x_scale=rescale,
y_scale=rescale,
alpha=240,
hint=image.BILINEAR | image.CENTER,
)
status += "alpha:240 "
status += "+mask "
else:
img.draw_image(
small_img,
int(x),
int(y),
x_scale=rescale,
y_scale=rescale,
alpha=128,
hint=image.BILINEAR | image.CENTER,
)
status += "alpha:128 "
img.draw_string(8, 0, status, mono_space=False)
print(clock.fps())