From 3a35e20567ac23dd9b3d1a4bc5d94e08d99bda16 Mon Sep 17 00:00:00 2001
From: phwitte <phillip_witte@gmx.de>
Date: Tue, 25 Jul 2017 08:35:19 +0000
Subject: [PATCH] Update agent.py, now with arrows!

---
 agent.py | 16 +++++++++++++---
 1 file changed, 13 insertions(+), 3 deletions(-)

diff --git a/agent.py b/agent.py
index ebf2874..5b73389 100644
--- a/agent.py
+++ b/agent.py
@@ -66,6 +66,7 @@ def checkBoundary(agent, winWidth, winHeight):
     elif y <= 0 or y >= winHeight:
         agent[2] = agent[2] * (-1)
         agent[0].move(agent[1],agent[2])
+    return agent
 
 def main():
     winWidth = 1000
@@ -75,8 +76,8 @@ def main():
 
     maxTime = 4000
     maxV = 8
-    agentNum = 50
-    agents = [[0 for x in range(3)] for y in range(agentNum)]
+    agentNum = 75
+    agents = [[0 for x in range(4)] for y in range(agentNum)]
 
     #generate point
     for agent in agents:
@@ -84,7 +85,11 @@ def main():
 
         agent[1] = random.uniform(-2,2)
         agent[2] = random.uniform(-2,2)
+
         agent[0].draw(window)
+        agent[3] = Line(agent[0], Point(agent[0].getX() + agent[1], agent[0].getY() + agent[2]))
+        agent[3].setArrow("last")
+        agent[3].draw(window)
 
     #update points
     for i in range(maxTime):
@@ -94,8 +99,13 @@ def main():
             agent = updateV(agent, nn, maxV)
             agent = checkBoundary(agent, winWidth, winHeight)
 
+            agent[3].undraw()
+            agent[3] = Line(agent[0], Point(agent[0].getX() + agent[1], agent[0].getY() + agent[2]))
+            agent[3].setArrow("last")
+            agent[3].draw(window)
+
         time.sleep(0.01)
-        
+
     window.getMouse()
     window.close()
 
-- 
GitLab