def go_for_food(self) -> None:
if self.food is not None and (
self.food.x() != self.client.player.position.x()
or self.food.y() != self.client.player.position.y()):
# print('taking previous food')
pos = self.food
else:
self.client.look()
found = find(self.client.player.vision, key=self.has_food)
if found is not None:
# print('found food !')
idx = index(self.client.player.vision, key=self.has_food)
# print(f'found index {idx}')
pos = Vec2d(idx % self.client.mapSize.x(),
int(idx / self.client.mapSize.x()))
self.food = pos
if self.food.x() == self.client.player.position.x(
) and self.food.y() == self.client.player.position.y():
# print(f'taking all food at {pos.x()};{pos.y()}')
self.take_all()
return
else:
# print('taking random food')
pos = Vec2d(randint(0, self.client.mapSize.x()),
randint(0, self.client.mapSize.y()))
# print(f'going for food at {pos.x()};{pos.y()} from {self.client.player.position.x()};{self.client.player.position.y()}')
self.best_direction(pos)
def __init__(self, port: int, name: str, host: str):
self.r_th = Thread(target=self.read,
args=(self.sock, self.responses, self.messages))
self.host = host
self.port = port
self.team = name
self.player = Player(pid=0, pos=Vec2d(0, 0))
def best_direction(self, pos: Vec2d) -> None:
deltax = pos.x() - self.client.player.position.x()
deltay = pos.y() - self.client.player.position.y()
if deltax > self.client.mapSize.x() / 2:
deltax = self.client.mapSize.x() - deltax
if deltay > self.client.mapSize.y() / 2:
deltay = self.client.mapSize.y() - deltay
# print(f'deltax {deltax}, deltay {deltay}')
if abs(deltax) > abs(deltay):
direction = 2 + -1 * sign(deltax)
else:
direction = 1 + -1 * sign(deltay)
# print(f'should go {direction}, is looking at {self.client.player.orientation}')
l_turns = (self.client.player.orientation - direction + 4) % 4
r_turns = (direction - self.client.player.orientation + 4) % 4
turns = min((l_turns, self.client.turn_left),
(r_turns, self.client.turn_right),
key=lambda x: x[0])
if turns[0] == 0:
self.client.move_forward()
self.take_all()
else:
turns[1]()
class Player:
level = 1
position: Vec2d = Vec2d(0, 0)
orientation = 0
inventory = Inventory()
vision = []
def __init__(self, pid: int, pos: Vec2d or None):
self.id = pid
self.position = pos if not None else Vec2d(0, 0)
for x in range(4):
self.vision.append(Inventory())
def to_str(self) -> str:
return f'{self.level};{self.inventory}'
def __init__(self, pid: int, pos: Vec2d or None): self.id = pid self.position = pos if not None else Vec2d(0, 0) for x in range(4): self.vision.append(Inventory())
def get_initial_data(self):
self.slotsLeft = int(self.responses.get())
x, y = self.responses.get().split(' ')
self.mapSize = Vec2d(int(x), int(y))
class Client:
player: Player = None
sock = socket.socket()
port = -1
team = ''
mapSize = Vec2d(0, 0)
host: str
slotsLeft: int
responses = Queue()
messages = Queue()
def __init__(self, port: int, name: str, host: str):
self.r_th = Thread(target=self.read,
args=(self.sock, self.responses, self.messages))
self.host = host
self.port = port
self.team = name
self.player = Player(pid=0, pos=Vec2d(0, 0))
def connect(self):
self.sock.connect((self.host, self.port))
self.r_th.daemon = True
self.r_th.start()
@staticmethod
def read(sock: socket, responses: Queue, messages: Queue):
buf = ''
while True:
buf += sock.recv(_max_buffer_size).decode()
if len(buf) == 0:
return
while buf.find('\n') is not -1:
res, buf = buf.split('\n', maxsplit=1)
if res.split(maxsplit=1)[0] == 'message':
# print(f'received msg [{res}]')
messages.put(res)
else:
# print(f'received response [{res}]')
responses.put(res)
def write(self, data):
# print(f' sending [{data}]')
if not data.endswith('\n'):
data += '\n'
self.sock.send(data.encode())
def terminate(self):
self.sock.close()
def get_initial_data(self):
self.slotsLeft = int(self.responses.get())
x, y = self.responses.get().split(' ')
self.mapSize = Vec2d(int(x), int(y))
def move_forward(self):
# print(f'Forward in {self.player.orientation} direction')
self.write('Forward')
if self.responses.get() != 'ok':
exit(0)
if self.player.orientation == 0: # NORTH
# print('NORTH')
self.player.position.set_y(
clamp(self.player.position.y() + 1, self.mapSize.y()))
elif self.player.orientation == 2: # SOUTH
# print('SOUTH')
self.player.position.set_y(
clamp(self.player.position.y() - 1, self.mapSize.y()))
elif self.player.orientation == 1: # EAST
# print('EAST')
self.player.position.set_x(
clamp(self.player.position.x() + 1, self.mapSize.x()))
elif self.player.orientation == 3: # WEST
# print('WEST')
self.player.position.set_x(
clamp(self.player.position.x() - 1, self.mapSize.x()))
def turn_right(self):
# print('Right')
self.write('Right')
if self.responses.get() != 'ok':
exit(0)
self.player.orientation = (self.player.orientation + 1 + 4) % 4
def turn_left(self):
# print('Left')
self.write('Left')
if self.responses.get() != 'ok':
exit(0)
self.player.orientation = (self.player.orientation - 1 + 4) % 4
def look(self):
self.write('Look')
data = parse_response_array(self.responses.get())
self.player.vision = []
for s in data:
vision = Inventory([0] * len(_ITEMS))
segment = s.strip().split(' ')
for key in segment:
if find(vision.keys(), key=lambda x: x == key) is not None:
vision[key] += 1
self.player.vision.append(vision)
def get_inventory(self):
self.write('Inventory')
data = parse_response_array(self.responses.get())
for s in data:
item, val = s.strip().split(' ')
self.player.inventory[item] = int(val)
def send_information(self):
text = f'{self.team};{self.player.to_str()}'
self.broadcast(text)
def broadcast(self, text: str):
self.write('Broadcast ' + text)
if self.responses.get() != 'ok':
exit(0)
def get_remaining_slots(self):
self.write('Connect_nbr')
self.slotsLeft = int(self.responses.get())
def fork(self):
if self.slotsLeft > 0:
self.write('Fork')
if self.responses.get() != 'ok':
exit(0)
def eject(self):
self.write('Eject')
if self.responses.get() != 'ok':
exit(0)
def take(self, item: str) -> bool:
self.write('Take ' + item)
res = self.responses.get()
if res == 'dead':
exit(0)
if res == 'ok':
self.player.inventory[item] += 1
return True
return False
def set(self, item: str):
if self.player.inventory[item] <= 0:
return
self.write('Set ' + item)
if self.responses.get().strip() == 'ok':
self.player.inventory[item] -= 1
self.player.vision[0][item] += 1
else:
exit(0)
def incantation(self):
self.write('Incantation')
response = self.responses.get()
if response != 'ok':
exit(0)
self.player.timeout = 300