virus direction stats

1 files changed, 9 insertions, 0 deletions

diff --git a/gui.py b/gui.py
index 6ba6f36..2040f32 100644
--- a/gui.py
+++ b/gui.py
@@ -319,8 +319,12 @@ def draw_markers():
 def draw_debug():
     for cell in c.world.cells.values():
         parent = None
+        shoot_vec = None
+        settled = None
         try:
             parent = cell.parent
+            shoot_vec = cell.shoot_vec
+            settled = cell.calmed_down
         except AttributeError:
             pass
 
@@ -329,6 +333,11 @@ def draw_debug():
             draw_circle(parent.pos,3,(255,0,0),True)
 
 
+        if shoot_vec != None and settled != True:
+            shoot_vec = 400 * shoot_vec / shoot_vec.len()
+            draw_line(cell.pos, cell.pos+shoot_vec, (0,255,255))
+
+
 def draw_frame():
     global screen, movement, zoom, screensize, input, bot_input, marker, marker_updated, running