class Client(object):
"""Client object.
"""
def __init__(self):
# Maak screen en font aan.
self.screen = pygame.display.set_mode((800, 600))
self.font = pygame.font.Font(None, 20)
# Maak socket aan.
self.sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
self.sock.setblocking(0)
# ID van onze connectie met de server.
self.connectionId = None
# Gamestate object.
self.gameState = GameState()
# Sla gegevens van de user op.
self.serverAddress = sys.argv[1]
self.playerName = sys.argv[2]
# Variabelen voor autopilot
self.autopilot = False
self.autopilotVelX = 0
self.autopilotVelY = 0
self.autopilotLastTick = 0
self.autopilotDelay = 1000
def handleEvents(self):
"""Handle pygame events af.
"""
events = pygame.event.get()
# Vang key events af.
for event in events:
if event.type == pygame.KEYDOWN:
# Q is afsluiten.
if event.key == pygame.K_q:
self.sock.sendto("%i disconnect %s" % (self.connectionId, self.playerName), (self.serverAddress, 12221))
sys.exit(0)
# A is autopilot toggle.
if event.key == pygame.K_a:
self.autopilot = not self.autopilot
# Stuur events door naar input state.
inputState.handleInput(events)
def move(self):
"""Voer movement uit.
"""
if self.autopilot == True:
# Autopilot movement.
self.moveAutopilot()
else:
# Manual movement.
self.moveManual()
def moveAutopilot(self):
"""Movement voor als de autopilot aan staat.
"""
# Check of er van richting veranderd moet worden.
ticks = pygame.time.get_ticks()
if ticks > self.autopilotLastTick + self.autopilotDelay:
self.autopilotLastTick = ticks
self.autopilotVelX = random.randint(-1, 1) * MOVE_SPEED
self.autopilotVelY = random.randint(-1, 1) * MOVE_SPEED
self.autopilotDelay = random.randint(200, 1000)
# Stuur movement door naar de server.
self.sendMove(self.autopilotVelX, self.autopilotVelY)
def moveManual(self):
"""Movement voor als je op manual control staat.
"""
# Bepaal velocity aan de hand van de input state.
velX = velY = 0
if inputState.getMovementState("left") == True:
velX = -MOVE_SPEED
if inputState.getMovementState("right") == True:
velX = MOVE_SPEED
if inputState.getMovementState("up") == True:
velY = -MOVE_SPEED
if inputState.getMovementState("down") == True:
velY = MOVE_SPEED
# Stuur movement door naar de server.
self.sendMove(velX, velY)
def sendMove(self, velX, velY):
"""Verstuur movement naar de server.
"""
# Als er een velocity is, stuur deze door naar de server.
if velX != 0 or velY != 0:
self.sock.sendto("%i move %s %i %i" % (self.connectionId, self.playerName, velX, velY), (self.serverAddress, 12221))
def handleNetworkRequest(self, data, address):
"""Handel een netwerk request af.
"""
# Lees bericht uit.
id = int(data.split(" ")[0])
command = data.split(" ")[1]
args = " ".join(data.split(" ")[2:])
# Check of het ons ID heeft.
if id not in [0, self.connectionId]:
return
# Welcome command bij inloggen.
if command == "welcome":
playerName = args.split(" ")[0]
if playerName == self.playerName:
connectionId = int(args.split(" ")[1])
self.connectionId = connectionId
print "Connected! Connection ID = %i" % connectionId
# Denied command.
elif command == "denied":
# Dit gaat alleen op als je nog geen verbinding hebt.
if self.connectionId == None:
# Check of het om onze speler gaat.
playerName = args.split(" ")[0]
if playerName == self.playerName:
# Sluit af met de reden die de server ons gaf.
reason = " ".join(args.split(" ")[1:])
print "Connection denied: %s" % reason
sys.exit(0)
# Updaten van de posities.
elif command == "update":
self.gameState.fromString(args)
# Onbekend request.
else:
print "Unknown network command: %s" % data
def handleConnection(self):
"""Handel server connection shit af.
"""
try:
data, address = self.sock.recvfrom( 1024 )
except socket.error:
pass
else:
#print "Received message:", data
self.handleNetworkRequest(data, address)
def draw(self):
"""Teken het scherm.
"""
# Zwarte achtergrond.
self.screen.fill((0,0,0))
# Teken alle players.
for player in self.gameState.getPlayers():
ding = pygame.Surface((16, 16))
ding.fill((255, 255, 255))
self.screen.blit(ding, player.getPosition())
# Teken de autopilot als die aan staat.
if self.autopilot == True:
text = self.font.render("Autopilot enabled", True, (255, 255, 255))
self.screen.blit(text, (10, 10))
# Display updaten.
pygame.display.flip()
def run(self):
"""Start de main loop.
"""
# Connect met de server.
self.sock.sendto("0 connect %s" % self.playerName, (self.serverAddress, 12221))
# Ga eeuwig door.
while True:
# Events afhandelen.
self.handleEvents()
# Bewegen doorsturen naar server.
self.move()
# Server network input afhandelen.
self.handleConnection()
# Tekenen.
self.draw()
# Wachten.
#pygame.time.delay(10)
pygame.time.delay(50)
