В прошлом эксперименте у нас уже было игровое поле, кладовщик, цели и ящики. Кроме того мы подключили кнопки и кладовщик реагировал на их нажатия. Теперь нам нужно реализовать логику игры, описать ограничения (такие как запрет на прохождение сквозь стены и ящики).
from machine import Pin, SPI from tft import TFT_GREEN _init() machine.freq(160000000) dc = Pin(4, Pin.OUT) cs = Pin(2, Pin.OUT) rst = Pin(5, Pin.OUT) spi = SPI(1, baudrate=40000000, polarity=0, phase=0) tft = TFT_GREEN(128, 160, spi, dc, cs, rst, rotate=0) Map = [ [1,1,0,1,1,1,0,1], [0,1,1,1,1,1,1,0], [1,1,0,0,0,1,1,1], [0,1,0,1,0,1,0,1], [0,1,0,0,0,1,0,1], [1,1,1,1,0,0,0,1], [1,0,0,0,0,0,0,1], [1,0,0,0,1,0,0,1], [1,0,0,0,1,1,1,1], [1,1,1,1,1,0,0,0] ] Gates = [] Boxes = [] class Box: def __init__(self, tft, x, y): self.tft = tft self.x = x self.y = y self.picture = 'box.bmp' self.picture_onGate = 'boxngate.bmp' self.onGate = False self.draw() def draw(self): if (self.onGate): self.tft.draw_bmp(self.x * 16,self.y * 16, self.picture_onGate) else: self.tft.draw_bmp(self.x * 16,self.y * 16, self.picture) def setOnGate(self, state): self.onGate = state def getOnGate(self): return self.onGate def getPos(self): return (self.x, self.y) def setPos(self, x, y): self.x = x self.y = y self.draw() class Gate: def __init__(self, tft, x, y): self.tft = tft self.x = x self.y = y self.picture = 'gate.bmp' self.draw() def draw(self): self.tft.draw_bmp(self.x * 16,self.y * 16, self.picture) def getPos(self): return (self.x, self.y) class Man: def __init__(self, tft, x, y): self.tft = tft self.x = x self.y = y self.picture = 'man.bmp' self.draw() def draw(self): self.tft.draw_bmp(self.x * 16,self.y * 16, self.picture) def getPos(self): return (self.x, self.y) def setPos(self, x, y): self.tft.rect(self.x * 16, self.y * 16, 16, 16, tft.COLOR_BLACK) self.x = x self.y = y self.draw() class Button: def __init__(self, p, pressSate): self.pin = Pin(p, Pin.IN) self.pressSate = pressSate self.oldState = 0 def onPress(self): state = self.pin.value() if state != self.oldState: self.oldState = state if state == self.pressSate: return True return False tft.initr(tft.BGR) # tft.initr(tft.RGB) #Если вместо синего цвета отображается красный, а вместо красного синий tft.clear(tft.COLOR_BLACK) x = 0 y = 0 for row in Map: for col in row: if col: tft.draw_bmp(x * 16, y * 16,'brick.bmp') x+=1 x=0 y+=1 Boxes.append(Box(tft, 3,4)) Boxes.append(Box(tft, 4,6)) Boxes.append(Box(tft, 2,7)) Gates.append(Gate(tft, 6,3)) Gates.append(Gate(tft, 6,4)) Gates.append(Gate(tft, 6,5)) man = Man(tft, 5, 6) btnUp = Button(16, 1) btnDown = Button(15, 1) btnLeft = Button(12, 1) btnRight = Button(0, 0) def canMove(x,y): if (Map[y][x]): return False else: return True def feelBox(x,y): for B in Boxes: if B.getPos() == (x,y): return B return False def boxFeelGate(x,y): for G in Gates: if G.getPos() == (x,y): return True return False while True: mPos = man.getPos() newPos = (-1,-1) if btnUp.onPress(): newPos = (mPos[0], mPos[1]-1) newPosNext = (mPos[0], mPos[1]-2) if btnDown.onPress(): newPos = (mPos[0], mPos[1]+1) newPosNext = (mPos[0], mPos[1]+2) if btnLeft.onPress(): newPos = (mPos[0]-1, mPos[1]) newPosNext = (mPos[0]-2, mPos[1]) if btnRight.onPress(): newPos = (mPos[0]+1, mPos[1]) newPosNext = (mPos[0]+2, mPos[1]) if newPos != (-1,-1): B = feelBox(newPos[0], newPos[1]) if B and canMove(newPosNext[0], newPosNext[1]): if boxFeelGate(newPosNext[0], newPosNext[1]): B.setOnGate(True) else: B.setOnGate(False) B.setPos(newPosNext[0], newPosNext[1]) man.setPos(newPos[0], newPos[1]) if not B and canMove(newPos[0], newPos[1]): man.setPos(newPos[0], newPos[1]) for G in Gates: gPos = G.getPos() if not (feelBox(gPos[0], gPos[1])) and gPos != man.getPos(): G.draw() win = 1 for B in Boxes: if (not B.getOnGate()): win = 0 break if (win): tft.draw_bmp(0, 16, 'win.bmp') raise SystemExit
Мы добавили несколько дополнительных функций. А именно:
canMove(x, y)
— проверяет возможность перемещения кладовщика или ящика в заданную координату. Эта функция просто проверяет наличие стены в данном месте. Если ее там нет — значит переместиться в эту координату можно. Если стена есть, то нельзя.
def canMove(x,y): if (Map[y][x]): return False else: return True
Функция feelBox(x, y)
— проверяет наличие ящика в заданной координате. Эта проверка происходит при движении кладовщика. Если перед ним находится ящик, то он должен сдвинуться. Функция перебирает все объекты ящиков и сравнивает их координаты с заданной. Если на пути есть ящик, то возвращается ссылка на объект этого ящика, иначе возвращается False.
def feelBox(x,y): for B in Boxes: if B.getPos() == (x,y): return B return False
Функция boxFeelGate(x,y)
— проверяет находится ли ящик на цели. В качестве параметров функция принимает координату ящика и проверяет наличие цели с такой же координатой.
def boxFeelGate(x,y): for G in Gates: if G.getPos() == (x,y): return True return False
В основном цикле программы мы получаем текущую координату кладовщика и задаем новую координату как (-1, -1). При обнаружении нажатия на кнопку, новая координата пересчитывается исходя из текущей координаты и нажатой кнопки. Кроме того высчитывается следующая координата в направлении движения после новой. Например если мы перемещаемся вверх, то вычисляем координаты на 1 клетку вверх и на 2 клетки вверх. Вторая координата нужна на случай если перед кладовщиком окажется ящик. Тогда он должен будет перейти на 1 клетку вверх, и ящик тоже (на 2 клетки вверх относительно текущей координаты кладовщика). Если нажатия на кнопку не было, то на этом итерация цикла завершается.
Если нажатие было и новая координата определена, то происходит проверка новой координаты. Сначала мы проверяем есть ли на пути кладовщика ящик:
B = feelBox(newPos[0], newPos[1])
Если ящик есть, значит мы должны двигаться вместе с ящиком. Но для этого нужно проверить может ли передвинуться ящик (нет ли стены перед ящиком).
if B and canMove(newPosNext[0], newPosNext[1]):
Если все в порядке и ящик может передвинуться, то проверяем еще одно условие — находится ли ящик на цели. Если да, то B.setOnGate(True), если нет, то B.setOnGate(False).
После чего передвигаем ящик и кладовщика в новую координату.
Этим описывается вся логика перемещения кладовщика и ящиков.
Далее идет вспомогательная логика. В частности мы перерисовываем цели, если на них нет ящика или кладовщика. Это нужно, чтобы после прохождения ящика или кладовщика по цели они прорисовывались заново, но не прорисовывались поверх ящика или кладовщика:
for G in Gates: gPos = G.getPos() if not (feelBox(gPos[0], gPos[1])) and gPos != man.getPos(): G.draw()
Далее идет проверка условия выигрыша. Игра считается выигранной когда все ящики установлены на свои цели. Для этого мы проверяем условие нахождения ящика на цели для каждого ящика. Если у всех ящиков соблюдается данное условие, то игра выиграна. А когда она выиграна на дисплее отображается победная картинка осуществляется выход из программы:
win = 1 for B in Boxes: if (not B.getOnGate()): win = 0 break if (win): tft.draw_bmp(0, 16, 'win.bmp') raise SystemExit
Таким образом мы реализовали известную игру Sokoban.
Попробуй нарисовать свой уровень с другой конфигурацией стен и другими положениями ящиков и целей