first commite from remotecupbase and fix import ref

Author: naderzare

client.py

@@ -0,0 +1,12 @@
+#!/usr/bin/python3
+
+
+import src.Client as Client
+
+
+def main():
+    Client.run()
+
+
+if __name__ == "__main__":
+    main()

src/Client.py

@@ -0,0 +1,76 @@
+import socket
+import random
+from argparse import ArgumentParser
+from src.base.Message import *
+from src.World import *
+import src.ClientGreedy as c_greedy
+import src.ClientRandom as c_random
+import src.ClientBest as c_best
+import src.YourClient as c_your
+import signal
+is_run = True
+
+
+def signal_handler(sig, frame):
+    global is_run
+    print('You pressed Ctrl+C!')
+    is_run = False
+
+
+signal.signal(signal.SIGINT, signal_handler)
+
+
+def run():
+    parser = ArgumentParser()
+    parser.add_argument("-n", "--name", dest="name", type=str, default='team_name' + str(random.randint(0, 10000)),
+                        help="Client Name", metavar="NAME")
+    parser.add_argument("-c", "--client", dest="client_type", type=str, default='auto',
+                        help="greedy, random, hand, best, your", metavar="ClientType")
+    parser.add_argument("-p", "--port", dest="server_port", type=int, default=20002,
+                        help="server port", metavar="ServerPort")
+    parser.add_argument("-s", "--server", dest="server_address", type=str, default='localhost',
+                        help="server address", metavar="ServerAddress")
+    args = parser.parse_args()
+    sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+    sock.settimeout(1)
+    server_address = (args.server_address, args.server_port)
+    world = World()
+    message_snd = MessageClientConnectRequest(args.name).build()
+
+    while is_run:
+        sock.sendto(message_snd, server_address)
+        try:
+            message_rcv = sock.recvfrom(4096)
+        except:
+            continue
+        message = parse(message_rcv[0])
+        if message.type == 'MessageClientConnectResponse':
+            print('my id is ' + str(message.id))
+            world.set_id(message.id, message.ground_config['goal_id'])
+            break
+
+    while is_run:
+        try:
+            r = sock.recvfrom(4096)
+        except:
+            continue
+        message = parse(r[0])
+        if message.type == 'MessageClientDisconnect':
+            break
+        elif message.type == 'MessageClientWorld':
+            world.update(message)
+            world.print()
+
+            if args.client_type == 'greedy' or (args.client_type == 'auto' and world.self_id == 1):
+                action = c_greedy.get_action(world)
+            elif args.client_type == 'random' or (args.client_type == 'auto' and world.self_id >= 2):
+                action = c_random.get_action(world)
+            elif args.client_type == 'best':
+                action = c_best.get_action(world)
+            elif args.client_type == 'your':
+                action = c_your.get_action(world)
+            elif args.client_type == 'hand':
+                action = input('enter action (u or d or l or r:')
+
+            sock.sendto(MessageClientAction(string_action=action).build(), server_address)
+

src/ClientBest.py

@@ -0,0 +1,46 @@
+from src.World import *
+
+
+def get_action(world: World):
+    head_pos = world.get_self().get_head()
+    next_head = []
+    next_head.append(head_pos + Vector2D(1, 0))
+    next_head.append(head_pos + Vector2D(0, 1))
+    next_head.append(head_pos + Vector2D(-1, 0))
+    next_head.append(head_pos + Vector2D(0, -1))
+    next_head_ok = [True, True, True, True]
+
+    h_number = 0
+    for h in next_head:
+        accident = False
+        for s in world.snakes:
+            if h in world.snakes[s].get_body():
+                accident = True
+                break
+        if accident:
+            print(h, 'snake')
+        if h in world.get_walls():
+            accident = True
+            print(h, 'wall')
+        if accident:
+            next_head_ok[h_number] = False
+        h_number += 1
+
+    print(next_head)
+    print(next_head_ok)
+    min_dist = 1000
+    h_best = -1
+    for h in range(4):
+        if not next_head_ok[h]:
+            continue
+        dist = next_head[h].dist(world.goal_position)
+        if dist < min_dist:
+            min_dist = dist
+            h_best = h
+
+    print(h_best)
+
+    actions = ['d', 'r', 'u', 'l']
+    if h_best > 0:
+        return actions[h_best]
+    return actions[0]

src/ClientGreedy.py

@@ -0,0 +1,25 @@
+from src.World import *
+
+
+def get_action(world: World):
+    head_pos = world.get_self().get_head()
+    next_head = []
+    next_head.append(head_pos + Vector2D(1, 0))
+    next_head.append(head_pos + Vector2D(0, 1))
+    next_head.append(head_pos + Vector2D(-1, 0))
+    next_head.append(head_pos + Vector2D(0, -1))
+
+    min_dist = 1000
+    h_best = -1
+    for h in range(4):
+        dist = next_head[h].dist(world.goal_position)
+        if dist < min_dist:
+            min_dist = dist
+            h_best = h
+
+    print(h_best)
+
+    actions = ['d', 'r', 'u', 'l']
+    if h_best > 0:
+        return actions[h_best]
+    return actions[0]

src/ClientRandom.py

@@ -0,0 +1,8 @@
+from src.World import *
+import random
+
+
+def get_action(world: World):
+    actions = ['u', 'd', 'l', 'r']
+    action = actions[random.randint(0, 3)]
+    return action

src/World.py

@@ -0,0 +1,86 @@
+from src.base.Math import *
+
+class Snake:
+    def __init__(self, id):
+        self.id = id
+        self.head = None
+        self.body = []
+        self.name = ''
+
+    def get_body(self):
+        return self.body
+
+    def add_body(self, pos):
+        self.body.append(pos)
+
+    def get_head(self):
+        return self.head
+
+    def set_head(self, pos):
+        self.head = pos
+
+    def get_id(self):
+        return self.id
+
+    def reset(self, name):
+        self.head = None
+        self.body.clear()
+        self.name = name
+
+
+class World:
+    def __init__(self):
+        self.board = None
+        self.cycle = None
+        self.self_id = None
+        self.goal_id = None
+        self.goal_position = None
+        self.snakes = {}
+        for s in range(1, 5):
+            self.snakes[s] = Snake(s)
+        self.walls = []
+
+    def set_id(self, self_id, goal_id):
+        self.self_id = self_id
+        self.goal_id = goal_id
+
+    def update(self, message):
+        self.board = message.world['board']
+        self.cycle = message.cycle
+        self.walls.clear()
+        n = 0
+        print(message.world)
+        for s in self.snakes:
+            self.snakes[s].reset(list(message.score.keys())[n])
+            n += 1
+
+        for i in range(len(self.board)):
+            for j in range(len(self.board[i])):
+                if self.board[i][j] == self.goal_id:
+                    self.goal_position = Vector2D(i, j)
+                elif self.board[i][j] > 0:
+                    self.snakes[self.board[i][j]].add_body(Vector2D(i, j))
+                elif self.board[i][j] == -1:
+                    self.walls.append(Vector2D(i, j))
+
+        for s in self.snakes:
+            self.snakes[s].set_head(Vector2D(message.world['heads'][self.snakes[s].name][0], message.world['heads'][self.snakes[s].name][1]))
+
+    def get_self(self):
+        return self.snakes[self.self_id]
+
+    def get_snake(self, id):
+        return self.snakes[self.id]
+
+    def get_walls(self):
+        return self.walls
+
+    def print(self):
+        print('------------------------------------')
+        print('cycle: {}'.format(self.cycle))
+        for f in self.board:
+            print(f)
+        print('$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$')
+        for s in self.snakes:
+            print(self.snakes[s].get_id(), self.snakes[s].get_head(), self.snakes[s].get_body())
+        print('------------------------------------')

src/YourClient.py

@@ -0,0 +1,8 @@
+from src.World import *
+import random
+
+
+def get_action(world: World):
+    actions = ['u', 'd', 'l', 'r']
+    action = actions[random.randint(0, 3)]
+    return action

src/base/Math.py

@@ -0,0 +1,33 @@
+simple_color = {0: 'black', 1: 'red', 2: 'blue', 3: 'orange', 4: 'pink', 'g': 'green', 'w': 'brown'}
+advance_color = {0: ['black'], 1: ['red', 'red4'], 2: ['blue', 'blue4'], 3: ['orange', 'dark goldenrod'],
+                 4: ['pink', 'pink4'], 'g': ['green'], 'w': ['brown']}
+
+
+class Vector2D:
+    def __init__(self, i, j):
+        self.i = i
+        self.j = j
+
+    def __str__(self):
+        return '(' + str(self.i) + ',' + str(self.j) + ')'
+
+    def __repr__(self):
+        return '(' + str(self.i) + ',' + str(self.j) + ')'
+
+    def __eq__(self, other):
+        if self.i == other.i and self.j == other.j:
+            return True
+        return False
+
+    def is_near(self, other):
+        dist = abs(self.i - other.i)
+        dist += abs(self.j - other.j)
+        if dist < 3:
+            return True
+        return False
+
+    def dist(self, other):
+        return abs(self.i - other.i) + abs(self.j - other.j)
+
+    def __add__(self, other):
+        return Vector2D(self.i + other.i, self.j + other.j)