pikapython/examples/LCD/main_snake.py

from PikaObj import *
import PikaStdLib
import PikaPiZero
import STM32G0

# hardware init
lcd = PikaPiZero.LCD()
lcd.init()
lcd.clear('white')
key = PikaPiZero.KEY()
key.init()
time = STM32G0.Time()
x_max = 120
y_max = 150

# snake init
s = PikaPiZero.Point()
w = 9
h = 9
s.x = 50
s.y = 10
len = 0
while len < 3:
    b = s
    i = 0
    while i < len:
        b = b.next
        i = i + 1
    b.next = PikaPiZero.Point()
    b.next.x = b.x - 10
    b.next.y = b.y
    b.next.prev = b
    len = len + 1

i = 0
b = s
while i < len:
    lcd.fill(b.x, b.y, w, h, 'blue')
    b = b.next
    i = i + 1

print('snake lengh')
print(len)

# fruit init
f = PikaPiZero.Point()
f.x = 30
f.y = 20
lcd.fill(f.x, f.y, w, h, 'green')

# memory check
mem = PikaStdLib.MemChecker()
print('mem used max:')
mem.max()

# main loop
d = 0
isUpdate = 0
while True:
    if isUpdate:
        isUpdate = 0
        # check eat fruit
        if f.x == s.x:
            if f.y == s.y:
                # have eat fruit
                b = s
                i = 0
                while i < len:
                    b = b.next
                    i = i + 1
                b.next = PikaPiZero.Point()
                b.next.prev = b
                len = len + 1
                print('snake lengh')
                print(len)
                print('mem used max:')
                mem.max()
                f.x = f.x + 30
                if f.x > x_max:
                    f.x = f.x - x_max
                f.y = f.y + 30
                if f.y > y_max:
                    f.y = f.y - y_max
                lcd.fill(f.x, f.y, w, h, 'green')
        # update snake body
        b = s
        i = 0
        while i < len:
            b = b.next
            i = i + 1
        i = 0
        s.prev = b
        while i < len:
            b = b.prev
            b.next.x = b.x
            b.next.y = b.y
            i = i + 1
        # move snake by the direction
        if d == 0:
            s.x = s.x + 10
            if s.x > x_max:
                s.x = 0
        if d == 1:
            s.y = s.y - 10
            if s.y < 0:
                s.y = y_max
        if d == 2:
            s.y = s.y + 10
            if s.y > y_max:
                s.y = 0
        if d == 3:
            s.x = s.x - 10
            if s.x < 0:
                s.x = x_max
        # drow the snake and fruit
        b = s
        i = 0
        lcd.fill(b.x, b.y, w, h, 'blue')
        lcd.fill(s.prev.x, s.prev.y, w, h, 'white')
    key_val = key.get()
    if key_val == 0:
        d = 0
        isUpdate = 1
    if key_val == 1:
        d = 1
        isUpdate = 1
    if key_val == 2:
        d = 2
        isUpdate = 1
    if key_val == 3:
        d = 3
        isUpdate = 1