wxPython：实现俄罗斯方块

#!/usr/bin/python# -*- coding: utf-8 -*-# tetris.pyimport wximport randomclass Tetris(wx.Frame):    speed = 300        def __init__(self, parent, title):        wx.Frame.__init__(self, parent, title=title, size=(500, 600))                self.menuBar = wx.MenuBar()        self.menu = wx.Menu()        self.menu.Append(200, u"游戏模式：简单")        self.menu.Append(201, u"游戏模式：入门")        self.menu.Append(202, u"游戏模式：困难")        self.menu.Append(203, u"查看耗时(游戏暂停后才可查看)")        self.menu.Append(204, u"重新开始(简单模式)")        self.Bind(wx.EVT_MENU, self.EasyMode, id=200)        self.Bind(wx.EVT_MENU, self.MiddleMode, id=201)        self.Bind(wx.EVT_MENU, self.HardMode, id=202)        self.Bind(wx.EVT_MENU, self.CostTime, id=203)        self.Bind(wx.EVT_MENU, self.Restart, id=204)        self.menuBar.Append(self.menu, u"&功能菜单")         self.menuBar.Enable(203, False)               self.SetMenuBar(self.menuBar)                self.initFrame()        def initFrame(self):                self.statusbar = self.CreateStatusBar()        self.statusbar.SetStatusText('0')        self.board = Board(self, Tetris.speed)        self.board.SetFocus()        self.board.start()        self.Centre()        self.Show(True)        def EasyMode(self, evt):       # 简单模式        Tetris.speed = 300        self.Destroy()        Tetris(None, title=u'俄罗斯方块')        def MiddleMode(self, evt):    # 入门模式        Tetris.speed = 150        self.Destroy()        Tetris(None, title=u'俄罗斯方块')            def HardMode(self, evt):     # 困难模式        Tetris.speed = 50        self.Destroy()        Tetris(None, title=u'俄罗斯方块')            def CostTime(self, evt):     # 查看耗时        wx.MessageBox(u'勇士不要急，时间还早，现在才过了 %d s！'%(self.board.costTime * self.board.Speed / 1000.0),           u"2333333", wx.OK | wx.ICON_INFORMATION, self)        def Restart(self, evt):     # 重新开始        Tetris.speed = 300        self.Destroy()        Tetris(None, title=u'俄罗斯方块')       class Board(wx.Panel):        BoardWidth = 10    BoardHeight = 22    Speed = 300    ID_TIMER = 1    costTime = 0    def __init__(self, parent, MySpeed):        wx.Panel.__init__(self, parent,style=wx.WANTS_CHARS)        Board.Speed = MySpeed                self.initBoard()            def initBoard(self):            Board.costTime = 0        self.timer = wx.Timer(self, Board.ID_TIMER)        self.isWaitingAfterLine = False        self.curPiece = Shape()        self.nextPiece = Shape()        self.curX = 0        self.curY = 0        self.numLinesRemoved = 0        self.board = []        self.isStarted = False        self.isPaused = False        self.Bind(wx.EVT_PAINT, self.OnPaint)        self.Bind(wx.EVT_KEY_DOWN, self.OnKeyDown)        self.Bind(wx.EVT_TIMER, self.OnTimer, self.timer)        self.clearBoard()    def shapeAt(self, x, y):                return self.board[(y * Board.BoardWidth) + x]    def setShapeAt(self, x, y, shape):                self.board[(y * Board.BoardWidth) + x] = shape    def squareWidth(self):                return self.GetClientSize().GetWidth() / Board.BoardWidth    def squareHeight(self):                return self.GetClientSize().GetHeight() / Board.BoardHeight    def start(self):                if self.isPaused:            return        self.isStarted = True        self.isWaitingAfterLine = False        self.numLinesRemoved = 0        self.clearBoard()        self.newPiece()        self.timer.Start(Board.Speed)    def pause(self):                if not self.isStarted:            return        self.isPaused = not self.isPaused                # 只有游戏暂停的时候“查看耗时”菜单选项才可以选择，否则处于灰化状态        if self.isPaused:            self.GetParent().GetMenuBar().Enable(203, True)        else:            self.GetParent().GetMenuBar().Enable(203, False)                    statusbar = self.GetParent().statusbar        if self.isPaused:            self.timer.Stop()            statusbar.SetStatusText('paused')        else:            self.timer.Start(Board.Speed)            statusbar.SetStatusText(str(self.numLinesRemoved))        self.Refresh()    def clearBoard(self):                for i in range(Board.BoardHeight * Board.BoardWidth):            self.board.append(Tetrominoes.NoShape)    def OnPaint(self, event):        dc = wx.PaintDC(self)                size = self.GetClientSize()        boardTop = size.GetHeight() - Board.BoardHeight * self.squareHeight()                for i in range(Board.BoardHeight):            for j in range(Board.BoardWidth):                shape = self.shapeAt(j, Board.BoardHeight - i - 1)                if shape != Tetrominoes.NoShape:                    self.drawSquare(dc,                        0 + j * self.squareWidth(),                        boardTop + i * self.squareHeight(), shape)        if self.curPiece.shape() != Tetrominoes.NoShape:            for i in range(4):                x = self.curX + self.curPiece.x(i)                y = self.curY - self.curPiece.y(i)                self.drawSquare(dc, 0 + x * self.squareWidth(),                    boardTop + (Board.BoardHeight - y - 1) * self.squareHeight(),                    self.curPiece.shape())    def OnKeyDown(self, event):                if not self.isStarted or self.curPiece.shape() == Tetrominoes.NoShape:            event.Skip()            return        keycode = event.GetKeyCode()        if keycode == ord('P') or keycode == ord('p'):            self.pause()            return        if self.isPaused:            return        elif keycode == wx.WXK_LEFT:            self.tryMove(self.curPiece, self.curX - 1, self.curY)        elif keycode == wx.WXK_RIGHT:            self.tryMove(self.curPiece, self.curX + 1, self.curY)        elif keycode == wx.WXK_DOWN:            self.tryMove(self.curPiece.rotatedRight(), self.curX, self.curY)        elif keycode == wx.WXK_UP:            self.tryMove(self.curPiece.rotatedLeft(), self.curX, self.curY)        elif keycode == wx.WXK_SPACE:            self.dropDown()        elif keycode == ord('D') or keycode == ord('d'):            self.oneLineDown()        else:            event.Skip()    def OnTimer(self, event):                if event.GetId() == Board.ID_TIMER:            if self.isWaitingAfterLine:                self.isWaitingAfterLine = False                self.newPiece()            else:                self.oneLineDown()        else:            event.Skip()                Board.costTime = Board.costTime + 1    def dropDown(self):                newY = self.curY                while newY > 0:            if not self.tryMove(self.curPiece, self.curX, newY - 1):                break            newY -= 1        self.pieceDropped()    def oneLineDown(self):                if not self.tryMove(self.curPiece, self.curX, self.curY - 1):            self.pieceDropped()                def pieceDropped(self):                for i in range(4):            x = self.curX + self.curPiece.x(i)            y = self.curY - self.curPiece.y(i)            self.setShapeAt(x, y, self.curPiece.shape())        self.removeFullLines()        if not self.isWaitingAfterLine:            self.newPiece()    def removeFullLines(self):            numFullLines = 0        statusbar = self.GetParent().statusbar        rowsToRemove = []        for i in range(Board.BoardHeight):            n = 0            for j in range(Board.BoardWidth):                if not self.shapeAt(j, i) == Tetrominoes.NoShape:                    n = n + 1            if n == 10:                rowsToRemove.append(i)        rowsToRemove.reverse()        for m in rowsToRemove:            for k in range(m, Board.BoardHeight):                for l in range(Board.BoardWidth):                        self.setShapeAt(l, k, self.shapeAt(l, k + 1))            numFullLines = numFullLines + len(rowsToRemove)            if numFullLines > 0:                self.numLinesRemoved = self.numLinesRemoved + numFullLines                statusbar.SetStatusText(str(self.numLinesRemoved))                 self.isWaitingAfterLine = True                self.curPiece.setShape(Tetrominoes.NoShape)                self.Refresh()    def newPiece(self):                self.curPiece = self.nextPiece        statusbar = self.GetParent().statusbar        self.nextPiece.setRandomShape()        self.curX = Board.BoardWidth / 2 + 1        self.curY = Board.BoardHeight - 1 + self.curPiece.minY()        if not self.tryMove(self.curPiece, self.curX, self.curY):                        self.curPiece.setShape(Tetrominoes.NoShape)            self.timer.Stop()            self.isStarted = False            statusbar.SetStatusText('Game over')                def tryMove(self, newPiece, newX, newY):                for i in range(4):                        x = newX + newPiece.x(i)            y = newY - newPiece.y(i)                        if x < 0 or x >= Board.BoardWidth or y < 0 or y >= Board.BoardHeight:                return False            if self.shapeAt(x, y) != Tetrominoes.NoShape:                return False        self.curPiece = newPiece        self.curX = newX        self.curY = newY        self.Refresh()                return True    def drawSquare(self, dc, x, y, shape):                colors = ['#000000', '#CC6666', '#66CC66', '#6666CC',                  '#CCCC66', '#CC66CC', '#66CCCC', '#DAAA00']        light = ['#000000', '#F89FAB', '#79FC79', '#7979FC',                  '#FCFC79', '#FC79FC', '#79FCFC', '#FCC600']        dark = ['#000000', '#803C3B', '#3B803B', '#3B3B80',                  '#80803B', '#803B80', '#3B8080', '#806200']        pen = wx.Pen(light[shape])        pen.SetCap(wx.CAP_PROJECTING)        dc.SetPen(pen)        dc.DrawLine(x, y + self.squareHeight() - 1, x, y)        dc.DrawLine(x, y, x + self.squareWidth() - 1, y)        darkpen = wx.Pen(dark[shape])        darkpen.SetCap(wx.CAP_PROJECTING)        dc.SetPen(darkpen)        dc.DrawLine(x + 1, y + self.squareHeight() - 1,            x + self.squareWidth() - 1, y + self.squareHeight() - 1)        dc.DrawLine(x + self.squareWidth() - 1,         y + self.squareHeight() - 1, x + self.squareWidth() - 1, y + 1)        dc.SetPen(wx.TRANSPARENT_PEN)        dc.SetBrush(wx.Brush(colors[shape]))        dc.DrawRectangle(x + 1, y + 1, self.squareWidth() - 2,         self.squareHeight() - 2)class Tetrominoes(object):        NoShape = 0    ZShape = 1    SShape = 2    LineShape = 3    TShape = 4    SquareShape = 5    LShape = 6    MirroredLShape = 7class Shape(object):        coordsTable = (        ((0, 0),     (0, 0),     (0, 0),     (0, 0)),        ((0, -1),    (0, 0),     (-1, 0),    (-1, 1)),        ((0, -1),    (0, 0),     (1, 0),     (1, 1)),        ((0, -1),    (0, 0),     (0, 1),     (0, 2)),        ((-1, 0),    (0, 0),     (1, 0),     (0, 1)),        ((0, 0),     (1, 0),     (0, 1),     (1, 1)),        ((-1, -1),   (0, -1),    (0, 0),     (0, 1)),        ((1, -1),    (0, -1),    (0, 0),     (0, 1))    )    def __init__(self):                self.coords = [[0,0] for i in range(4)]        self.pieceShape = Tetrominoes.NoShape        self.setShape(Tetrominoes.NoShape)    def shape(self):                return self.pieceShape    def setShape(self, shape):                table = Shape.coordsTable[shape]        for i in range(4):            for j in range(2):                self.coords[i][j] = table[i][j]        self.pieceShape = shape    def setRandomShape(self):                self.setShape(random.randint(1, 7))    def x(self, index):                return self.coords[index][0]    def y(self, index):                return self.coords[index][1]    def setX(self, index, x):                self.coords[index][0] = x    def setY(self, index, y):                self.coords[index][1] = y    def minX(self):                m = self.coords[0][0]        for i in range(4):            m = min(m, self.coords[i][0])        return m    def maxX(self):                m = self.coords[0][0]        for i in range(4):            m = max(m, self.coords[i][0])        return m    def minY(self):                m = self.coords[0][1]        for i in range(4):            m = min(m, self.coords[i][1])        return m    def maxY(self):                m = self.coords[0][1]                for i in range(4):            m = max(m, self.coords[i][1])        return m    def rotatedLeft(self):                if self.pieceShape == Tetrominoes.SquareShape:            return self        result = Shape()        result.pieceShape = self.pieceShape                for i in range(4):            result.setX(i, self.y(i))            result.setY(i, -self.x(i))        return result    def rotatedRight(self):                if self.pieceShape == Tetrominoes.SquareShape:            return self        result = Shape()        result.pieceShape = self.pieceShape                for i in range(4):            result.setX(i, -self.y(i))            result.setY(i, self.x(i))        return resultapp = wx.App()Tetris(None, title=u'俄罗斯方块')app.MainLoop()