Make links between players and killed units.

Author: GraylinKim

sc2reader/data/__init__.py

@@ -38,6 +38,9 @@ def __init__(self, unit_id, flags):
         #: The frame the unit died at
         self.died_at = None
 
+        #: A reference to the player that killed this unit. Not always available.
+        self.killed_by = None
+
         self.id = unit_id
         self.flags = flags
         self._type_class = None

sc2reader/events/tracker.py

@@ -256,6 +256,9 @@ def load_context(self, replay):
 
         if self.killer_pid in replay.player:
             self.killer = replay.player[self.killer_pid]
+            if self.unit:
+                self.unit.killed_by = self.killer
+                self.killer.killed_units.append(self.unit)
         elif self.killer_pid:
             pass#print "Unknown killer pid", self.killer_pid
 

sc2reader/objects.py

@@ -204,6 +204,9 @@ def __init__(self, pid, name):
         #: A list of references to the units the player had this game
         self.units = list()
 
+        #: A list of references to the units that the player killed this game
+        self.killed_units = list()
+
 
     @property
     def url(self):