Chapter ? - Square Dodge

SquareDodge.py

1. import pygame, time, random, sys
2. from pygame.locals import *
4. windowWidth = 600
5. windowHeight = 600
7. topScore = 0
9. displayWindow = None
10. mainSurface = None
11. font = None
13. def init():
14.     global mainSurface, displayWindow, font
15.     
16.     displayWindow = pygame.display.set_mode((windowWidth, windowHeight))
17.     
18.     mainSurface = pygame.Surface(displayWindow.get_size())
19.     pygame.display.set_caption('Square Dodge')
20.     
21.     pygame.font.init()
22.     font = pygame.font.SysFont(None, 48)
25. def showStartScreen():
26.     global mainSurface, displayWindow, font
27.     
28.     mainSurface.fill((0,0,0))
29.     text = font.render("Square Dodge", 1, (255,255,255) )
30.     textpos = text.get_rect()
31.     textpos.centerx = mainSurface.get_rect().centerx
32.     mainSurface.blit(text, textpos)
33.     
34.     displayWindow.blit(mainSurface, (0, 0))
35.     pygame.display.flip()
36.     
37.     while True:
38.         for event in pygame.event.get():
39.             if event.type == QUIT:
40.                 sys.exit()
41.             if event.type == KEYDOWN:
42.                 return
43.     
44.     
45. def showGameOverScreen():
46.     global mainSurface, displayWindow, font
47.     
48.     #mainSurface.fill((0,0,0))
49.     text = font.render("GAME OVER", 1, (255,255,255) )
50.     textpos = text.get_rect()
51.     textpos.centerx = mainSurface.get_rect().centerx
52.     textpos.centery = mainSurface.get_rect().centery
53.     mainSurface.blit(text, textpos)
54.     
55.     text = font.render("(Press a key to play again.)", 1, (255,255,255) )
56.     textpos = text.get_rect()
57.     textpos.centerx = mainSurface.get_rect().centerx
58.     textpos.centery = mainSurface.get_rect().centery + 50
59.     mainSurface.blit(text, textpos)
60.     
61.     displayWindow.blit(mainSurface, (0, 0))
62.     pygame.display.flip()
63.     
64.     while True:
65.         for event in pygame.event.get():
66.             if event.type == QUIT:
67.                 sys.exit()
68.             if event.type == KEYDOWN:
69.                 mainSurface.fill((0,0,0))
70.                 displayWindow.blit(mainSurface, (0, 0))
71.                 pygame.display.flip()
72.                 return
74. def enterGameLoop():
75.     global windowWidth, windowHeight, topScore
76.     
77.     blocks = []
78.     
79.     gameSpeed = 0.001
80.     addNewBlockRate = 4
81.     playerMoveRate = 8
82.     
83.     minBlockSpeed = 1
84.     maxBlockSpeed = 8
85.     
86.     score = 0
87.     
88.     playerRect = pygame.Rect(windowWidth / 2, windowHeight - 30, 20, 20)
89.     moveLeft = False
90.     moveRight = False
91.     moveUp = False
92.     moveDown = False
93.     shiftDown = False
94.     
95.     lastWorldUpdate = time.time()
96.     blockAddCounter = 0
97.     lastScoreAdd = time.time()
98.     
99.     while True:
100.         # Update the score
101.         if time.time() > lastScoreAdd + 0.1:
102.             lastScoreAdd = time.time()
103.             score += 1
104.             
105.         # Update the blocks and the user's position.
106.         for event in pygame.event.get():
107.             if event.type == QUIT:
108.                 sys.exit()
109.             if event.type == KEYDOWN:
110.                 if event.key == 276: # 276 is the left key
111.                     moveRight = False
112.                     moveLeft = True
113.                 if event.key == 275: # 275 is the right key
114.                     moveLeft = False
115.                     moveRight = True
116.                 if event.key == 273: # 273 is the up key
117.                     moveDown = False
118.                     moveUp = True
119.                 if event.key == 274: # 274 is the down key
120.                     moveUp = False
121.                     moveDown = True
122.                 if event.key in [303,304]:
123.                     shiftDown = True
124.             if event.type == KEYUP:
125.                 if event.key == 276: # 276 is the left key
126.                     moveLeft = False
127.                 if event.key == 275: # 275 is the right key
128.                     moveRight = False
129.                 if event.key == 273: # 273 is the up key
130.                     moveUp = False
131.                 if event.key == 274: # 274 is the down key
132.                     moveDown = False
133.                 if event.key == 27: # 27 is the Escape key
134.                     sys.exit()
135.                 if event.key in [303,304]:
136.                     shiftDown = False
137.             
138.         # Update the game world, if needed.
139.         if time.time() > lastWorldUpdate + gameSpeed:
140.             lastWorldUpdate = time.time()
141.             
142.             # Add new blocks at the top of the screen, if needed.
143.             blockAddCounter += 1
144.             if blockAddCounter > addNewBlockRate:
145.                 blockAddCounter = 0
146.                 blockSize = random.randint(10, 30)
147.                 newBlock = {'rect': pygame.Rect(random.randint(0, windowWidth-blockSize), -1 * blockSize, blockSize, blockSize),
148.                             'speed': random.randint(minBlockSpeed, maxBlockSpeed),
149.                             'size': blockSize,
150.                             'color': (random.randint(100,255),0,0),
151.                             }
152.                 blocks.append(newBlock)
153.             
154.             # Move the player around.
155.             if moveLeft and playerRect.left > 0:
156.                 playerRect.move_ip(-playerMoveRate, 0)
157.             if moveRight and playerRect.right < windowWidth:
158.                 playerRect.move_ip(playerMoveRate, 0)
159.             if moveUp and playerRect.top > 0:
160.                 playerRect.move_ip(0, -playerMoveRate)
161.             if moveDown and playerRect.bottom < windowHeight:
162.                 playerRect.move_ip(0, playerMoveRate)
163.             
164.             # Move the blocks down.
165.             for b in blocks:
166.                 b['rect'].move_ip(0, b['speed'])
167.             
168.              # Delete blocks that have fallen past the bottom.
169.             delBlocks = []
170.             for i in range(len(blocks)):
171.                 if blocks[i]['rect'].top > windowHeight:
172.                     delBlocks.append(i)
173.             delBlocks.reverse()
174.             for i in delBlocks:
175.                 del blocks[i]
176.             
177.             # Draw the game world on the window.
178.             mainSurface.fill((0,0,0)) # black out everything
179.             
180.             # Draw the score.
181.             text = font.render('Score: %s' % (score), 1, (255,255,255) )
182.             textpos = text.get_rect()
183.             textpos.left = mainSurface.get_rect().left
184.             textpos.top = mainSurface.get_rect().top
185.             mainSurface.blit(text, textpos)
186.             
187.             text = font.render('Top Score: %s' % (topScore), 1, (255,255,255) )
188.             textpos = text.get_rect()
189.             textpos.left = mainSurface.get_rect().left
190.             textpos.top = mainSurface.get_rect().top + 50
191.             mainSurface.blit(text, textpos)
192.             
193.             # Draw the player's rectangle    
194.             mainSurface.fill((255,255,255), playerRect)
195.             
196.             # Draw each block
197.             for b in blocks:
198.                 mainSurface.fill(b['color'], b['rect'])
199.                         
200.             # Draw the surface to the window
201.             displayWindow.blit(mainSurface, (0, 0))
202.             pygame.display.flip()
203.             
204.             # Check if any of the blocks have hit the player.
205.             for b in blocks:
206.                 if doRectsOverlap(playerRect, b['rect']):
207.                     if score > topScore:
208.                         topScore = score
209.                     score = 0
210.                     
211.                     blocks = [] # delete all the blocks
212.                     
213.                     # reset the player's location
214.                     playerRect = pygame.Rect(windowWidth / 2, windowHeight - 30, 20, 20)
215.                     moveLeft = False
216.                     moveRight = False
217.                     moveUp = False
218.                     moveDown = False
219.                     return
220.                     
222. def doRectsOverlap(rect1, rect2):
223.     # Check if rect1's corners are inside rect
224.     if ((isPointInsideRect(rect1.left, rect1.top, rect2)) or
225.         (isPointInsideRect(rect1.left, rect1.bottom, rect2)) or
226.         (isPointInsideRect(rect1.right, rect1.top, rect2)) or
227.         (isPointInsideRect(rect1.right, rect1.bottom, rect2))):
228.         return True
229.     
230.     # Check if rect2's corners are inside rect1
231.     if ((isPointInsideRect(rect2.left, rect2.top, rect1)) or
232.         (isPointInsideRect(rect2.left, rect2.bottom, rect1)) or
233.         (isPointInsideRect(rect2.right, rect2.top, rect1)) or
234.         (isPointInsideRect(rect2.right, rect2.bottom, rect1))):
235.         return True
236.     
237.     return False
239. def isPointInsideRect(x, y, rect):
240.     if (x > rect.left) and (x < rect.right) and (y > rect.top) and (y < rect.bottom):
241.         return True
242.     else:
243.         return False
244.         
246. def main():
247.     global mainSurface, displayWindow, font
248.     
249.     init()
250.     showStartScreen()
251.     
252.     while True:
253.         enterGameLoop()
254.         showGameOverScreen()
256. if __name__ == '__main__': main()

Things Covered In This Chapter: