class Client(asynchat.async_chat): def __init__(self, host, port): asynchat.async_chat.__init__(self) self.create_socket(socket.AF_INET, socket.SOCK_STREAM) self.host = host self.port = port self.inBuffer = "" self.outBuffer = "" self.connect((host,port)) self.outBuffer += "Tunneler1.0;" self.gameStarted = False self.grid = False self.players = [] self.bullets = [] self.playerID = False self.set_terminator(";") self.timestep = 0 self.shotAtStep = -10 self.movedAtStep = -10 pygame.init() self.screen = pygame.display.set_mode((SCREEN_WIDTH, SCREEN_HEIGHT)) self.HUDSurf = pygame.Surface((SCREEN_WIDTH, HUD_HEIGHT)) pygame.mouse.set_visible(0) self.clock = pygame.time.Clock() pygame.font.init() self.font = pygame.font.Font(None, FONT_SIZE) self.titleFont = pygame.font.Font(None, TITLE_FONT_SIZE) self.displayTitleScreen() def collect_incoming_data(self, data): self.inBuffer += data def found_terminator(self): self.parseChange(self.inBuffer.strip()) self.inBuffer = "" self.input() def handle_connect(self): pass def writable(self): return (len(self.outBuffer) > 0) def handle_write(self): sent = self.send(self.outBuffer) self.outBuffer = self.outBuffer[sent:] def parseInitialConditions(self, msg): ''' The first thing the Server will do when the game is signaled to begin is to send the initial conditions so that each client can set up, getting info about which player ID its been assigned, how large the grid is, and where the bases are for each player on the grid. ''' m = re.match("([0-9]*)\.([0-9]*)\.([0-9]*)\.([0-9]*)\.([0-9]*)\.([0-9]*)\.([0-9]*)\.([0-9]*)", msg) if m and int(m.group(1)) == GAME_START_CODE: # So the client knows which player it's displaying information for playerID = int(m.group(2)) self.playerID = playerID numRows = int(m.group(3)) numCols = int(m.group(4)) aBaseRow = int(m.group(5)) aBaseCol = int(m.group(6)) bBaseRow = int(m.group(7)) bBaseCol = int(m.group(8)) # The player always starts off in the middle of his base myPlayer = False enemyPlayer = False i = 1 while i < 3: playerRow = 0 playerCol = 0 if i == 1: playerRow = aBaseRow + int(math.ceil(BASE_SIZE / 2)) playerCol = aBaseCol + int(math.ceil(BASE_SIZE / 2)) else: playerRow = bBaseRow + int(math.ceil(BASE_SIZE / 2)) playerCol = bBaseCol + int(math.ceil(BASE_SIZE / 2)) self.players.append(PlayerSprite(i, playerRow, playerCol)) if i == self.playerID: myPlayer = self.players[i-1] else: enemyPlayer = self.players[i-1] i+= 1 self.grid = Grid(numRows, numCols) self.grid.addBase(aBaseRow, aBaseCol, A_BASE_UNIT) self.grid.addBase(bBaseRow, bBaseCol, B_BASE_UNIT) self.gameStarted = True self.viewport = Viewport(self.grid, myPlayer, enemyPlayer) else: print "invalid initial conditions" def parseChange(self, msg): ''' This is where the client picks apart a Change message sent from the server, and applies it to its own internal representation of what's going on in the game. It figures out which type of change each one is, then calls methods with the appropriate arguments for that particular change. ''' if not self.gameStarted: self.parseInitialConditions(msg) else: # The ; gets removed by the asynchat class msg += ";" tm = re.search("([0-9]*);", msg) changeTimestep = int(tm.group(1)) if self.timestep == changeTimestep: msg = msg.rstrip(tm.group(0)) m = re.match("([0-9]*\.)*[0-9]*,", msg) while m: cmd = m.group(0) i = 0 type = False args = [] # take the command apart and get its type and arguments while cmd: match = re.match("([0-9]*).", cmd) if i == 0: type = int(match.group(1)) else: args.append(int(match.group(1))) i += 1 cmd = cmd[len(match.group(0)):] if type == MOBILE_MOVE_CODE: self.moveSprite(args[0], args[1], args[2]) elif type == MOBILE_DIRECTION_CODE: self.makePlayerChangeDirection(args[0], args[1]) elif type == MOBILE_SPAWN_CODE: self.spawnBullet(args[0], args[1], args[2]) elif type == MOBILE_DIE_CODE: if args[0] > 2: self.destroyBullet(args[0]) elif type == DIRT_LOSE_HEALTH_CODE: self.grid.damageDirt(args[0], args[1], args[2]) elif type == DIRT_DIE_CODE: self.grid.setObject(args[0], args[1], 0) elif type == SET_HEALTH_CODE: self.setPlayerHealth(args[0], args[1]) elif type == SET_ENERGY_CODE: self.setPlayerEnergy(args[0], args[1]) elif type == GAME_OVER_CODE: self.gameover(args[0]) msg = msg[len(m.group(0)):] m = re.match("([0-9]*\.)*[0-9]*,", msg) self.timestep += 1 self.displayTimestep() def spawnBullet(self, id, row, col): bullet = MobileSprite(id, row, col) bullet.setImage("bullet.png") self.viewport.addBullet(bullet) def destroyBullet(self, id): self.viewport.destroyBulletWithID(id) def setPlayerHealth(self, id, health): self.players[id-1].health = health def setPlayerEnergy(self, id, energy): self.players[id-1].energy = energy def moveSprite(self, id, row, col): if id < 3: player = self.players[id - 1] player.row = row player.col = col else: # bullets for bullet in self.viewport.bullets: if bullet.id == id: bullet.row = row bullet.col = col def makePlayerChangeDirection(self, id, direction): self.players[id-1].changeDirection(direction) def input(self): ''' This is where the Client gets the player's input to send off to the server. It gets the keys that the player has pressed, and when the timer goes off, it packages those commands up and sends them off to the server. ''' cmd = "" endInput = False noInput = False tick = 0 commandHasMove = False while not endInput: movePart = False shootPart = False keydown = False for event in pygame.event.get(): if event.type == pygame.QUIT: print "Exiting" self.exit() if event.type == KEYDOWN: keydown = event.key if event.key == LEFT_KEY: self.movedAtStep = self.timestep movePart = "%d" % LEFT elif event.key == RIGHT_KEY: self.movedAtStep = self.timestep movePart = "%d" % RIGHT elif event.key == UP_KEY: self.movedAtStep = self.timestep movePart = "%d" % UP elif event.key == DOWN_KEY: self.movedAtStep = self.timestep movePart = "%d" % DOWN elif event.key == SHOOT_KEY: if self.shotAtStep not in range(self.timestep - SHOOT_TIMESTEP_WAIT, self.timestep): self.shotAtStep = self.timestep shootPart = "%d" % SHOOT elif event.key == K_ESCAPE: print "Escape key pressed" self.exit() key = pygame.key.get_pressed() if key[LEFT_KEY]: if self.movedAtStep not in range(self.timestep - MOVE_TIMESTEP_WAIT, self.timestep): self.movedAtStep = self.timestep movePart = "%d" % LEFT if key[RIGHT_KEY]: if self.movedAtStep not in range(self.timestep - MOVE_TIMESTEP_WAIT, self.timestep): self.movedAtStep = self.timestep movePart = "%d" % RIGHT if key[UP_KEY]: if self.movedAtStep not in range(self.timestep - MOVE_TIMESTEP_WAIT, self.timestep): self.movedAtStep = self.timestep movePart = "%d" % UP if key[DOWN_KEY]: if self.movedAtStep not in range(self.timestep - MOVE_TIMESTEP_WAIT, self.timestep): self.movedAtStep = self.timestep movePart = "%d" % DOWN if key[SHOOT_KEY]: if self.shotAtStep not in range(self.timestep - SHOOT_TIMESTEP_WAIT, self.timestep): self.shotAtStep = self.timestep shootPart = "%d" % SHOOT tick += self.clock.tick_busy_loop(FPS) # So the game doesn't wait forever if tick > TICK_LIMIT: endInput = True cmd = "" if movePart: cmd += movePart else: cmd += "0" if shootPart: cmd += shootPart else: cmd += "0" cmd += ".%d;" % self.timestep self.outBuffer += cmd def updateHUD(self): ''' Shows the player's health and energy at the bottom of the screen. ''' healthFontSurf = self.font.render("Health: %d" % self.viewport.player.health, True, (255, 255, 255)) energyFontSurf = self.font.render("Energy: %d" % self.viewport.player.energy, True, (255, 255, 255)) self.screen.blit(healthFontSurf, (7, VP_HEIGHT + 9)) self.screen.blit(energyFontSurf, (7, VP_HEIGHT + 5 + FONT_SIZE)) def displayTimestep(self): ''' Gets the viewport to update to display the latest timestep. ''' self.screen.fill((50, 50, 50)) self.updateHUD() self.viewport.updateDisplay() self.screen.blit(self.viewport.surf, (0, 0)) pygame.display.flip() def displayTitleScreen(self): ''' When the user first opens the Client, they are smacked in the face with this Title screen. ''' titleScreenImage = imageLoad("TitleScreen.png") breakLoop = False c = 0 while not breakLoop: for event in pygame.event.get(): if event.type == pygame.QUIT: print "Exiting" self.exit() if event.type == KEYDOWN: if event.key == K_ESCAPE: print "Escape key pressed" self.exit() if event.key == K_RETURN: self.indicateReadiness() breakLoop = True self.screen.fill((0,0,0)) self.screen.blit(titleScreenImage, (0,0)) if not breakLoop: cs = self.titleFont.render("You are connected to %s." % self.host, False, (255, 255, 255)) es = self.titleFont.render("Press ENTER when you are ready to play.", True, (255, 255, 255)) self.screen.blit(cs, (45, 350)) self.screen.blit(es, (10, 375)) else: rs = self.font.render("READY...", True, (255, 255, 255)) ws = self.titleFont.render("Waiting for other player...", True, (255,255,255)) self.screen.blit(rs, (102, 230)) self.screen.blit(ws, (70, 270)) pygame.display.flip() def gameover(self, idOfLoser): ''' When the game ends, the game displays the full grid and the loser. ''' self.screen.fill((0,0,0)) renderedFont = False if idOfLoser == self.viewport.player.id: renderedFont = self.font.render("LOSS.", False, (255,255,255)) else: renderedFont = self.font.render("VICTORY.", False, (255,255,255)) helpfulFont = self.titleFont.render("Press ESC to quit.", False, (255,255,255)) self.screen.blit(renderedFont, (5, 5)) self.screen.blit(helpfulFont, (5, 40)) pygame.display.flip() while True: for event in pygame.event.get(): if event.type == pygame.QUIT: print "Exiting" self.exit() elif event.type == KEYDOWN: if event.key == K_ESCAPE: print "Escape key pressed" self.exit() def indicateReadiness(self): ''' Tell the server that this client is ready to play a game. ''' self.outBuffer += "%d;" % GAME_START_CODE def exit(self): self.close() sys.exit(0)