openmv/usr/tests/test_drawing_2.py
Kwabena W. Agyeman bcd5151412 New drawing functions.
All the drawing functions have been updated to handle automatic clipping
when drawing offscren and work with both grayscale and RGB565.
Additionally, all functions now accept color arguments.

I've also updated the example scripts with the new functions and tested
them out to make sure they work.

Additionally, I wrote a test suite for the drawing functions to make
sure they work.
2016-02-18 22:40:12 -05:00

64 lines
2.4 KiB
Python

import sensor, image, time
sensor.reset()
sensor.set_framesize(sensor.QVGA)
# All drawing functions use the same code to pass color.
# So we just need to test one function.
# TODO: Why does MP need int(c) when c is computed using "//"?
while(True):
# Test Draw Line (GRAYSCALE)
sensor.set_pixformat(sensor.GRAYSCALE)
for i in range(10):
img = sensor.snapshot()
for i in range(img.width()):
c = ((i * 255) + (img.width()/2)) // img.width()
img.draw_line([i, 0, i, img.height()-1], int(c))
sensor.snapshot()
time.sleep(1000)
for i in range(img.width()):
c = (((img.width() - i) * 255) + (img.width()/2)) // img.width()
img.draw_line([i, 0, i, img.height()-1], color = int(c))
sensor.snapshot()
time.sleep(1000)
# Test Draw Line (RGB565)
sensor.set_pixformat(sensor.RGB565)
for i in range(10):
img = sensor.snapshot()
for i in range(img.width()):
c = ((i * 255) + (img.width()/2)) // img.width()
img.draw_line([i, 0, i, img.height()-1], [int(c), 0, 0])
sensor.snapshot()
time.sleep(1000)
for i in range(img.width()):
c = (((img.width() - i) * 255) + (img.width()/2)) // img.width()
img.draw_line([i, 0, i, img.height()-1], color = [int(c), 0, 0])
sensor.snapshot()
time.sleep(1000)
# Test Draw Line (RGB565)
sensor.set_pixformat(sensor.RGB565)
for i in range(10):
img = sensor.snapshot()
for i in range(img.width()):
c = ((i * 255) + (img.width()/2)) // img.width()
img.draw_line([i, 0, i, img.height()-1], [0, int(c), 0])
sensor.snapshot()
time.sleep(1000)
for i in range(img.width()):
c = (((img.width() - i) * 255) + (img.width()/2)) // img.width()
img.draw_line([i, 0, i, img.height()-1], color = [0, int(c), 0])
sensor.snapshot()
time.sleep(1000)
# Test Draw Line (RGB565)
sensor.set_pixformat(sensor.RGB565)
for i in range(10):
img = sensor.snapshot()
for i in range(img.width()):
c = ((i * 255) + (img.width()/2)) // img.width()
img.draw_line([i, 0, i, img.height()-1], [0, 0, int(c)])
sensor.snapshot()
time.sleep(1000)
for i in range(img.width()):
c = (((img.width() - i) * 255) + (img.width()/2)) // img.width()
img.draw_line([i, 0, i, img.height()-1], color = [0, 0, int(c)])
sensor.snapshot()
time.sleep(1000)