summaryrefslogtreecommitdiff
path: root/stats.py
diff options
context:
space:
mode:
authorFlorian Jung <flo@windfisch.org>2015-08-30 17:19:41 +0200
committerFlorian Jung <flo@windfisch.org>2015-08-30 17:19:41 +0200
commitf88995ed18fa57ddb7ce7b00003e025f67d0db11 (patch)
tree3e3f3a850354bda1aca662ebb58bf0c9c6af405e /stats.py
parent87ed12e22d6319ad43bfb11a6da1ded39d7ba74e (diff)
collect more data
Diffstat (limited to 'stats.py')
-rw-r--r--stats.py52
1 files changed, 48 insertions, 4 deletions
diff --git a/stats.py b/stats.py
index bb88c3e..9bf834a 100644
--- a/stats.py
+++ b/stats.py
@@ -63,6 +63,16 @@ def return_zero():
def return_defaultdict_with_zeros():
return defaultdict(return_zero)
+class ReMerging:
+ def __init__(self, size1, size2, birth1, birth2, is_parent_child, begin_time):
+ self.size1 = size1
+ self.size2 = size2
+ self.birth1 = birth1
+ self.birth2 = birth2
+ self.is_parent_child = is_parent_child
+ self.begin_time = begin_time
+ self.end_time = None
+
class Stats:
def __init__(self,c,data=None):
self.c = c
@@ -71,7 +81,7 @@ class Stats:
if data == None:
self.data = StatData()
- self.data.version = 3
+ self.data.version = 4
self.data.min_mass = 0
self.data.max_mass = 0
@@ -89,6 +99,9 @@ class Stats:
self.data.eject_distlogs = {"virus" : [], "split cell" : [], "ejected mass" : []}
self.data.eject_deviations = {"virus" : [], "split cell" : [], "ejected mass" : []}
+
+ self.data.observed_virus_sizes = defaultdict(return_zero)
+ self.data.remerging = {}
else:
self.data = data
@@ -158,6 +171,30 @@ class Stats:
self.data.mass_vs_visible_window[n_own_cells][own_total_mass].append((visible_width,visible_height))
+ # log virus sizes
+ for cell in cells:
+ if cell.is_virus:
+ self.data.observed_virus_sizes[cell.size] += 1
+
+ # detect re-merging cells
+ for cell in own_cells:
+ for cell2 in own_cells:
+ if cell2 != cell:
+ dist = (cell.pos - cell2.pos).len()
+ expected_dist = cell.size + cell2.size
+ min_dist = max(cell.size, cell2.size)
+
+ if (dist < (0.9 * expected_dist + 0.1 * min_dist)):
+ is_parent_child = (cell == cell2.parent or cell2 == cell.parent)
+ print("cells seem to be merging! they are "+ ("" if is_parent_child else "NOT ") + "parent and child")
+ pair_id = (min(cell.cid,cell2.cid), max(cell.cid,cell2.cid))
+
+ if pair_id not in self.data.remerging:
+ self.data.remerging[pair_id] = ReMerging(cell.size, cell2.size, cell.spawntime, cell2.spawntime, is_parent_child, self.c.world.time)
+ else:
+ self.data.remerging[pair_id].end_time = self.c.world.time
+
+
# find ejected mass, split cells or viruses that have come to rest
for cell in cells:
@@ -184,11 +221,18 @@ class Stats:
cell.calmed_down = True
# TODO: speed log
+ # distance is calculated naively
distance = (cell.spawnpoint - cell.pos).len()
+
+ # distance2 is calculated along the cell's path (will differ if the flight was not colinear)
+ poslog = list(cell.poslog)
+ speeds = list(map(lambda vecs : (vecs[0]-vecs[1]).len(), zip(poslog, poslog[1:])))
+ distance2 = sum(speeds)
+
distance_from_parent = (cell.parentpos_when_spawned - cell.pos).len()
- self.data.eject_distlogs[celltype] += [(distance, distance_from_parent, cell.parentsize_when_spawned)]
- print(" flown distance = "+str(distance))
+ self.data.eject_distlogs[celltype] += [(distance, distance2, distance_from_parent, cell.parentsize_when_spawned, len(cell.poslog), speeds)]
+ print(" flown distance = %.2f / %.2f"%(distance,distance2))
if len(cell.poslog) == 5:
# calculate movement direction from the first 5 samples
@@ -284,7 +328,7 @@ class Stats:
maxheight = quantile(sorted(map(lambda x:x[1], rects)), 0.75)
if math.sqrt(maxwidth**2+maxheight**2) < 4000:
- # TODO FIXME
+ # TODO FIXME
svw[size] = (maxwidth,maxheight)
ratios += [maxwidth/maxheight]