mirror of
https://github.com/openmv/openmv.git
synced 2025-11-04 14:49:50 +08:00
94 lines
2.5 KiB
Python
94 lines
2.5 KiB
Python
# Draw Image Testing script with bounce
|
|
#
|
|
# Exercise draw image with many different values for testing
|
|
|
|
import sensor, image, time, pyb
|
|
|
|
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.IMAGE_HINT_BILINEAR | image.IMAGE_HINT_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.IMAGE_HINT_BILINEAR | image.IMAGE_HINT_CENTER)
|
|
status += 'alpha:128 '
|
|
|
|
img.draw_string(8, 0, status, mono_space = False)
|
|
|
|
print(clock.fps())
|