#转自:http://www.oschina.net/code/snippet_162893_24130
#!/usr/bin/python
# -*- coding: UTF-8 -*-
# snakegame
import sys
from PyQt4.QtCore import *
from PyQt4.QtGui import *
import random
class Snake:
def __init__(self):
self.length = 4
self.posArray = [(3,0),(2,0),(1,0),(0,0)]
self.isDead = 0
class Food:
def __init__(self):
self.isExist = 0
self.pos = (0,0)
class SnakeGame(QWidget):
def __init__(self):
super(SnakeGame, self).__init__()
self.timer = QTimer(self)
#self.connect(timer, SIGNAL("timeout()"),self,SLOT("update()"))
self.timer.timeout.connect(self.update)
self.initGame()
self.initUI()
def initUI(self):
#self.setGeometry(300, 300, 500, 300)
self.setMaximumSize(500,330)
self.setMinimumSize(500,330)
self.show()
def paintEvent(self, event):
qp = QPainter()
qp.begin(self)
if self.isStart == -1:
self.pause(event, qp)
elif self.isStart == 0:
self.readyGame(event, qp)
elif self.isStart == 1:
self.proceedGame(event, qp)
else:
self.overGame(event, qp)
qp.end()
def keyPressEvent(self, event):
key = event.key()
if self.isStart == 1:
if key == Qt.Key_Left and self.tag != 'Right':
self.xch = -1
self.ych = 0
self.tag = 'Left'
if key == Qt.Key_Right and self.tag != 'Left':
self.xch = 1
self.ych = 0
self.tag = 'Right'
if key == Qt.Key_Down and self.tag != 'Up':
self.ych = 1
self.xch = 0
self.tag = 'Down'
if key == Qt.Key_Up and self.tag != 'Down':
self.ych = -1
self.xch = 0
self.tag = 'Up'
if key == Qt.Key_Space:
if self.isStart == 0:
self.isStart = 1
elif self.isStart == 2:
self.initGame()
self.isStart = 0
else:
self.isStart = -self.isStart
def drawSquare(self, qp, xx, yy):
qp.drawRect(xx*10, yy*10, 10, 10)
def drawBlackSquare(self, qp, xx, yy):
qp.setBrush(QColor(0,0,0))
qp.drawRect(xx*10,yy*10,10,10)
def drawSnake(self,qp):
for a in self.snake.posArray:
xx = a[0]
yy = a[1]
self.drawSquare(qp, xx, yy)
def moveSnake(self):
l = len(self.snake.posArray)
for i in xrange(l-1, 0, -1):
self.snake.posArray[i] = self.snake.posArray[i-1]
xx = self.snake.posArray[0][0]
yy = self.snake.posArray[0][1]
self.snake.posArray[0] = (xx+self.xch, yy+self.ych)
#def isDead():
def produceFood(self):
if self.food.isExist == 0:
#random.seed()
xx = random.randint(0, 49)
yy = random.randint(0, 29)
while (xx, yy) in self.snake.posArray:
xx = random.randint(0, 49)
yy = random.randint(0, 29)
self.food.pos = (xx, yy)
self.food.isExist = 1
def drawFood(self, qp):
self.drawBlackSquare(qp, self.food.pos[0], self.food.pos[1])
def eatFood(self):
l = len(self.snake.posArray)
if self.snake.posArray[0] == self.food.pos:
self.snake.posArray.append(self.snake.posArray[l-1])
self.food.isExist = 0
self.score = self.score + self.lvl * 100
def drawCoverText(self, event, qp, text, color):
qp.setPen(color)
qp.setFont(QFont('Decorative', 30))
qp.drawText(event.rect(), Qt.AlignCenter, text)
def drawStatusText(self, qp, text, xx, yy):
qp.setPen(QColor(0,0,0))
qp.setFont(QFont('Decorative', 15))
qp.drawText(xx, yy, text)
def drawStatus(self, qp):
text1 = 'SCORE: ' + str(self.score)
self.drawStatusText(qp, text1, 10, 315)
text2 = 'LVL: ' + str(self.lvl)
self.drawStatusText(qp, text2, 300, 315)
def pause(self, event, qp):
color = QColor(0, 0, 0)
text = 'Pause'
self.drawCoverText(event, qp, text, color)
def overGame(self, event, qp):
color = QColor(0, 0, 0)
text = 'Game Over'
self.drawCoverText(event, qp, text, color)
self.drawSnake(qp)
self.drawBorder(qp)
text1 = 'SCORE: ' + str(self.score)
self.drawStatusText(qp, text1, 185, 220)
self.drawFood(qp)
self.isStart = 2
def initGame(self):
self.xch, self.ych = 1,0
self.tag = 'Right'
self.isStart = 0
self.snake = Snake()
self.food = Food()
self.score = 0
self.lvl = 1
self.timer.start(240)
def readyGame(self, event, qp):
self.drawSnake(qp)
self.drawBorder(qp)
text = 'Press Space To Play'
color = QColor(0, 0, 0)
self.drawCoverText(event, qp, text, color)
def checkLiveOrDead(self):
xx = self.snake.posArray[0][0] + self.xch
yy = self.snake.posArray[0][1] + self.ych
if xx < 0 or yy < 0 or xx >49 or yy >29 or (xx, yy) in self.snake.posArray:
self.snake.isDead = 1
def checkLvl(self):
if self.score >= self.lvl * self.lvl *1000:
time = 240 - self.lvl * 30
self.lvl = self.lvl + 1
if time < 0:
time = 1
self.timer.start(time)
def drawBorder(self, qp):
qp.setPen(QColor(0,0,0))
qp.drawRect(0,0,500,300)
def proceedGame(self, event, qp):
self.checkLiveOrDead()
self.checkLvl()
if self.snake.isDead:
self.overGame(event, qp)
else:
self.drawBorder(qp)
self.moveSnake()
self.eatFood()
self.drawSnake(qp)
self.produceFood()
self.drawFood(qp)
self.drawStatus(qp)
def main():
app = QApplication(sys.argv)
ex = SnakeGame()
sys.exit(app.exec_())
if __name__ == '__main__':
main()
