initial release

2 files changed, 225 insertions, 0 deletions

diff --git a/start.sh b/start.sh
new file mode 100755
index 0000000..5cd8bd7
--- /dev/null
+++ b/start.sh
@@ -0,0 +1,2 @@
+#! /bin/bash
+python strategy.py < /dev/stdin
diff --git a/strategy.py b/strategy.py
new file mode 100755
index 0000000..ebbf08d
--- /dev/null
+++ b/strategy.py
@@ -0,0 +1,223 @@
+#!/bin/python
+
+import sys, re
+
+entity=0
+memory = []
+memory_nest = []
+pheromone_type = 0
+see_food_id = 0
+see_food_zone = ""
+see_food_distance = 0
+see_food_amount = 0
+see_nest_id = 0
+see_nest_zone = ""
+see_nest_dist = 0
+see_nest_friend = 0
+ant_stock = 0
+ant_stamina = 0
+ant_sorted = 0
+ant_exist = 0
+explo = 0
+
+def log(string):
+    with open('out.txt', 'a') as f:
+        f.write(str(string)+"\n")
+
+for line in sys.stdin:
+    line = line.strip()
+
+    if line == "BEGIN ANT":
+        food_see_list = []
+        nest_see_list = []
+        pheromone_see_list = []
+        entity = 1
+    if line == "BEGIN NEST":
+        entity = 2
+    if line.startswith("MEMORY"):
+        line_splited = line.split()
+        m = []
+
+        m.append(line_splited[1])
+        m.append(line_splited[2])
+
+        if entity == 1:
+            if len(memory) == 0:
+                memory.append(m[0])
+                memory.append(m[1])
+            else:
+                memory[0] = m[0]
+                memory[1] = m[1]
+        elif entity == 2:
+            if len(memory_nest) == 0:
+                for x in range(0, 21):
+                    if x != 0:
+                        memory_nest.append(line_splited[x])
+            else:
+                for x in range(0, 21):
+                    if x != 0:
+                        memory_nest[x-1] = line_splited[x]
+    if line.startswith("SEE_FOOD"):
+        line_splited = line.split()
+        see_food_id = line_splited[1]
+        see_food_zone = line_splited[2]
+        see_food_distance = line_splited[3]
+        see_food_amount = line_splited[4]
+        liste = []
+        liste.append(see_food_id)
+        liste.append(see_food_zone)
+        liste.append(see_food_distance)
+        liste.append(see_food_amount)
+        food_see_list.append(liste)
+    if line.startswith("SEE_NEST"):
+        line_splited = line.split()
+        see_nest_id = line_splited[1]
+        see_nest_zone = line_splited[2]
+        see_nest_dist = line_splited[3]
+        see_nest_friend = line_splited[4]
+        liste = []
+        liste.append(see_nest_id)
+        liste.append(see_nest_zone)
+        liste.append(see_nest_dist)
+        liste.append(see_nest_friend)
+        nest_see_list.append(liste)
+    if line.startswith("SEE_PHEROMONE"):
+        line_splited = line.split()
+        see_pheromone_id = line_splited[1]
+        see_pheromone_zone = line_splited[2]
+        see_pheromone_dist = line_splited[3]
+        see_pheromone_type = line_splited[4]
+        liste = []
+        liste.append(see_pheromone_id)
+        liste.append(see_pheromone_zone)
+        liste.append(see_pheromone_dist)
+        liste.append(see_pheromone_type)
+        pheromone_see_list.append(liste)
+    if line.startswith("STOCK"):
+        line_splited = line.split()
+        ant_stock = line_splited[1]
+    if line.startswith("STAMINA"):
+        line_splited = line.split()
+        ant_stamina = line_splited[1]
+    if line == "END":
+        if entity == 1:
+            if int(ant_stock) < 10 and int(memory[1]) < 255:
+                if int(memory[0]) >= 8:
+                    print("PUT_PHEROMONE "+str(memory[1]))
+                    memory[0]=0
+                    if int(memory[1]) < 3:
+                        memory[1]=str(int(memory[1])+1)
+                    else:
+                        memory[1] = "1"
+                    print("SET_MEMORY 0 "+str(memory[1]))
+                elif see_food_zone != "":
+                    dist_close_food = 100
+                    id_food = 0
+                    probably_near_food_zone = ""
+                    food_lvl = 0
+                    for food_see in food_see_list:
+                        if int(food_see[2]) <= int(dist_close_food):
+                            dist_close_food = food_see[2]
+                            id_food = food_see[0]
+                            food_lvl = food_see[3]
+                            probably_near_food_zone = food_see[1]
+                    if probably_near_food_zone == "NEAR" and int(id_food) != 0 and int(food_lvl) > 10:
+                        print("COLLECT "+id_food+" 10")
+                    elif probably_near_food_zone == "FAR" and int(id_food) != 0:
+                        print("MOVE_TO "+str(id_food))
+                elif see_food_zone == "":
+                    print("EXPLORE")
+                    memory[0]=str(int(memory[0])+1)
+                    print("SET_MEMORY "+str(memory[0])+" "+str(memory[1]))
+            elif int(memory[1]) >= 255:
+                print("SUICIDE")
+            else:
+                if int(ant_stamina) <= 10:
+                    print("EAT")
+                else:
+                    nest_found_friend = 0
+                    nest_friend_zone = "FAR"
+                    nest_friend_id = 0
+                    for nest_see in nest_see_list:
+                        if nest_see[3] == "FRIEND":
+                            nest_found_friend = 1
+                            nest_friend_zone = nest_see[1]
+                            nest_friend_id = nest_see[0]
+                    if nest_found_friend == 0:
+                        dist_closest_pheromone = 100
+                        id_pheromone = 0
+                        probably_near_food_zone = ""
+                        no_pheromone = 0
+                        type_pheromone = ""
+                        dist_closest_pheromone_good_type = 100
+                        id_pheromone_good_type = 0
+                        probably_near_food_zone_good_type = ""
+                        no_pheromone_good_type = 0
+                        type_pheromone_good_type = 0
+
+                        if int(memory[1]) > 1:
+                            m1tmp=int(memory[1])-1
+                        else:
+                            m1tmp=3
+
+
+                        log(pheromone_see_list)
+                        pheromone_see_good_type_list = []
+                        for pheromone_see in pheromone_see_list:
+                            if str(pheromone_see[3]) == str(m1tmp):
+                                pheromone_see_good_type_list.append(pheromone_see)
+                        if pheromone_see_good_type_list != []:
+                            near_phero = min(pheromone_see_good_type_list, key=lambda p: p[2])
+                            type_pheromone_good_type = near_phero[3]
+                            id_pheromone_good_type = near_phero[0]
+                            dist_closest_pheromone_good_type = near_phero[2]
+                            log(str(id_pheromone_good_type)+": "+str(dist_closest_pheromone_good_type))
+                        if pheromone_see_list != []:
+                            near_phero = min(pheromone_see_list, key=lambda p: p[2])
+                            type_pheromone = near_phero[3]
+                            id_pheromone = near_phero[0]
+                            dist_closest_pheromone = near_phero[2]
+                            log(str(id_pheromone)+": "+str(dist_closest_pheromone))
+
+                        log(str(id_pheromone_good_type)+" != 0"+str(type_pheromone_good_type)+" == "+str(m1tmp))
+                        log(str(id_pheromone)+" != 0"+str(id_pheromone_good_type)+" == 0")
+                        if str(id_pheromone_good_type) != "0" and str(type_pheromone_good_type) == str(m1tmp):
+                            log("good type:"+str(id_pheromone_good_type)+":"+str(dist_closest_pheromone_good_type)+" > 1")
+                            if int(dist_closest_pheromone_good_type) > 1:
+                                print("MOVE_TO "+id_pheromone_good_type)
+                                no_pheromone = 0
+                            else:
+                                print("SET_MEMORY "+str(memory[0])+" "+str(m1tmp))
+                                no_pheromone = 0
+                        elif str(id_pheromone) != "0" and str(id_pheromone_good_type) == "0":
+                            log(str(id_pheromone)+":"+str(dist_closest_pheromone)+" > 1")
+                            if int(dist_closest_pheromone) > 1:
+                                print("MOVE_TO "+id_pheromone)
+                                no_pheromone = 0
+                            else:
+                                print("SET_MEMORY "+str(memory[0])+" "+str(m1tmp))
+                                no_pheromone = 0
+                        else:
+                            no_pheromone = 1
+                        if no_pheromone == 1:
+                            if int(memory[0]) == 10:
+                                print("EXPLORE")
+                                explo = 1
+                            else:
+                                print("TURN 180")
+                                print("SET_MEMORY 10 "+str(memory[1]))
+                                no_pheromone == 0
+                    else:
+                        if nest_friend_zone == "NEAR":
+                            print("NEST "+nest_friend_id)
+                            print("SET MEMORY 0 "+str(memory[1]))
+                        elif nest_friend_id != 0:
+                            print("MOVE_TO "+nest_friend_id)
+        elif entity == 2:
+            if memory_nest[0] == "0" and memory_nest[1] == "0":
+                print("ANT_NEW 0")
+                print("SET_MEMORY 0 1 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0")
+            elif memory_nest[0] == "0" and memory_nest[1] == "1":
+                print("ANT_OUT 0 0 0 1")
+                print("SET_MEMORY 1 1 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0")
+        print("END", flush=True)