From df1f75c8764c5d864934d7eaade88fc89fd8c125 Mon Sep 17 00:00:00 2001
From: Geoffrey Frogeye <geoffrey@frogeye.fr>
Date: Thu, 11 Feb 2016 21:40:06 +0100
Subject: [PATCH] Nearest warehouse

---
 main.py | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/main.py b/main.py
index f50d836..d6121f1 100644
--- a/main.py
+++ b/main.py
@@ -198,6 +198,18 @@ def end():
     print("--- End!")
 
 # IA
+def nearestW(p, pos):
+    minDist = math.inf
+    selW = None
+    for w in range(W):
+        if Wi[w][p] > 0:
+            dist = distance(Wp[w], pos)
+            if dist < minDist:
+                minDist = dist
+                selW = w
+    return selW
+
+
 def listNeeds():
     N = []
     # client, product
@@ -207,6 +219,7 @@ def listNeeds():
                 N.append((c, p))
     return N
 
+
 while t < 10:
     N = listNeeds()
     newTurn()