summaryrefslogtreecommitdiff
path: root/stats.py
diff options
context:
space:
mode:
Diffstat (limited to 'stats.py')
-rw-r--r--stats.py36
1 files changed, 36 insertions, 0 deletions
diff --git a/stats.py b/stats.py
new file mode 100644
index 0000000..f850339
--- /dev/null
+++ b/stats.py
@@ -0,0 +1,36 @@
+import time
+
+class Stats:
+ def __init__(self):
+ self.min_mass = 0
+ self.max_mass = 0
+ self.current_mass = 0
+
+ self.mass_history = []
+ self.pos_history = []
+ self.cell_aggressivity = {}
+ self.cell_split_frequency = {}
+ self.cell_defensiveness = {}
+
+ def log_mass(self, mass):
+ self.mass_history.append((time.time(), mass))
+ self.current_mass = mass
+ if mass > self.max_mass:
+ self.max_mass = mass
+ if mass < self.min_mass:
+ self.min_mass = mass
+
+ def log_pos(self, pos):
+ self.pos_history.append((time.time(), (pos[0], pos[1])))
+
+ def update_cell_aggressivity(self, cell, value):
+ self.cell_aggressivity[cell] = value
+
+ def update_cell_split_frequency(self, cell, value):
+ self.cell_split_frequency[cell] = value
+
+ def update_cell_defensiveness(self, cell, value):
+ self.cell_defensiveness[cell] = value
+
+ def get_last_steps(self, list, steps = 10):
+ return list[-steps:] \ No newline at end of file