adafruit-mirror/Adafruit_CircuitPython_turtle
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

allow changing colors, backward()

Browse source
This commit is contained in:
ladyada 2019-06-17 22:41:53 -04:00
parent 1d715956b9
commit 17b24ce364
2 changed files with 55 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

47
adafruit_turtle.py
View file

@ -27,12 +27,14 @@ import gc
import math import math
import time import time
class color: class Color:
WHITE = 0xFFFF WHITE = 0xFFFFFF
BLACK = 0x0000 BLACK = 0x0000
RED = 0xF800 RED = 0xFF0000
BLUE = 0x001F GREEN = 0x00FF00
BLUE = 0x0000FF
colors = (WHITE, BLACK, RED, GREEN, BLUE)
class Vec2D(tuple): class Vec2D(tuple):
"""A 2 dimensional vector class, used as a helper class """A 2 dimensional vector class, used as a helper class
@ -93,19 +95,17 @@ class turtle:
self._bg_bitmap = displayio.Bitmap(self._w, self._h, 1) self._bg_bitmap = displayio.Bitmap(self._w, self._h, 1)
self._bg_palette = displayio.Palette(1) self._bg_palette = displayio.Palette(1)
self._bg_palette[0] = color.BLACK self._bg_palette[0] = Color.BLACK
self._bg_sprite = displayio.TileGrid(self._bg_bitmap, self._bg_sprite = displayio.TileGrid(self._bg_bitmap,
pixel_shader=self._bg_palette, pixel_shader=self._bg_palette,
x=0, y=0) x=0, y=0)
self._splash.append(self._bg_sprite) self._splash.append(self._bg_sprite)
self._fg_bitmap = displayio.Bitmap(self._w, self._h, 5) self._fg_bitmap = displayio.Bitmap(self._w, self._h, 5)
self._fg_palette = displayio.Palette(5) self._fg_palette = displayio.Palette(len(Color.colors)+1)
self._fg_palette.make_transparent(0) self._fg_palette.make_transparent(0)
self._fg_palette[1] = color.WHITE for i,c in enumerate(Color.colors):
self._fg_palette[2] = color.BLACK self._fg_palette[i+1] = c
self._fg_palette[3] = color.RED
self._fg_palette[4] = color.BLUE
self._fg_sprite = displayio.TileGrid(self._fg_bitmap, self._fg_sprite = displayio.TileGrid(self._fg_bitmap,
pixel_shader=self._fg_palette, pixel_shader=self._fg_palette,
x=0, y=0) x=0, y=0)
@ -114,7 +114,7 @@ class turtle:
self._turtle_bitmap = displayio.Bitmap(9, 9, 2) self._turtle_bitmap = displayio.Bitmap(9, 9, 2)
self._turtle_palette = displayio.Palette(2) self._turtle_palette = displayio.Palette(2)
self._turtle_palette.make_transparent(0) self._turtle_palette.make_transparent(0)
self._turtle_palette[1] = color.WHITE self._turtle_palette[1] = Color.WHITE
for i in range(4): for i in range(4):
self._turtle_bitmap[4-i, i] = 1 self._turtle_bitmap[4-i, i] = 1
self._turtle_bitmap[i, 4+i] = 1 self._turtle_bitmap[i, 4+i] = 1
@ -128,7 +128,7 @@ class turtle:
self._penstate = False self._penstate = False
self._pencolor = None self._pencolor = None
self.pencolor(color.BLACK) self.pencolor(Color.WHITE)
self._display.show(self._splash) self._display.show(self._splash)
self._display.refresh_soon() self._display.refresh_soon()
@ -145,15 +145,22 @@ class turtle:
p = self.pos() p = self.pos()
x1 = p[0] + math.sin(math.radians(self._heading))*distance x1 = p[0] + math.sin(math.radians(self._heading))*distance
y1 = p[1] + math.cos(math.radians(self._heading))*distance y1 = p[1] + math.cos(math.radians(self._heading))*distance
print("* Forward to", x1, y1)
self.goto(x1, y1) self.goto(x1, y1)
def backward(self, distance):
self.forward(-distance)
def goto(self, x1, y1): def goto(self, x1, y1):
x1 += self._w//2 x1 += self._w//2
y1 = self._h//2 - y1 y1 = self._h//2 - y1
x0 = self._x x0 = self._x
y0 = self._y y0 = self._y
print("* GoTo from", x0, y0, "to", x1, y1) print("* GoTo from", x0, y0, "to", x1, y1)
if not self.isdown():
self._x = x1 # woot, we just skip ahead
self._y = y1
self._drawturtle()
return
steep = abs(y1 - y0) > abs(x1 - x0) steep = abs(y1 - y0) > abs(x1 - x0)
rev = False rev = False
dx = x1 - x0 dx = x1 - x0
@ -175,17 +182,17 @@ class turtle:
while (not rev and x0 <= x1) or (rev and x1 <= x0): while (not rev and x0 <= x1) or (rev and x1 <= x0):
if steep: if steep:
self._fg_bitmap[y0, x0] = 1 self._fg_bitmap[y0, x0] = self._pencolor
self._x = y0 self._x = y0
self._y = x0 self._y = x0
self._drawturtle() self._drawturtle()
time.sleep(0.01) time.sleep(0.003)
else: else:
self._fg_bitmap[x0, y0] = 1 self._fg_bitmap[x0, y0] = self._pencolor
self._x = x0 self._x = x0
self._y = y0 self._y = y0
self._drawturtle() self._drawturtle()
time.sleep(0.01) time.sleep(0.003)
err -= dy err -= dy
if err < 0: if err < 0:
y0 += ystep y0 += ystep
@ -231,10 +238,10 @@ class turtle:
return self._heading return self._heading
def pencolor(self, c): def pencolor(self, c):
#if not c in self.fg_palette: if not c in Color.colors:
# raise RuntimeError("Color must be one of the 'color' class items") raise RuntimeError("Color must be one of the 'color' class items")
#print(self._fg_palette[0]) #print(self._fg_palette[0])
self._pencolor = c self._pencolor = 1 + Color.colors.index(c)
# Tell turtle's state # Tell turtle's state
def pos(self): def pos(self):

28
examples/turtle_square.py Normal file
View file

@ -0,0 +1,28 @@
import board
from Adafruit_CircuitPython_turtle.adafruit_turtle import *
turtle = turtle(board.DISPLAY)
print("Turtle time! Lets draw a simple square")
turtle.pencolor(color.WHITE)
print("Position:", turtle.pos())
print("Heading:", turtle.heading())
turtle.penup()
print("Pen down?", turtle.isdown())
turtle.pendown()
print("Pen down?", turtle.isdown())
turtle.forward(25)
print("Position:", turtle.pos())
turtle.left(90)
turtle.forward(25)
print("Position:", turtle.pos())
turtle.left(90)
turtle.forward(25)
print("Position:", turtle.pos())
turtle.left(90)
turtle.forward(25)
print("Position:", turtle.pos())
while True:
pass