Fix more regressions from the event module changes.

Author: GraylinKim

sc2reader/events/__init__.py

@@ -0,0 +1,4 @@
+import base, game, message
+from base import *
+from game import *
+from message import *

sc2reader/events/base.py

@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 from __future__ import absolute_import
 
+from sc2reader.utils import Length
 from sc2reader.log_utils import loggable
 
 @loggable

sc2reader/events/game.py

@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 from __future__ import absolute_import
 
+from sc2reader.data import Unit
 from sc2reader.events.base import Event
 from sc2reader.log_utils import loggable
 
@@ -174,21 +175,21 @@ def load_context(self, replay):
         if not replay.datapack:
             return
 
-        uid = (self.target_id, self.target_type)
-        if uid in replay.objects:
-            self.target = replay.objects[uid]
+        if self.target_id in replay.objects:
+            self.target = replay.objects[self.target_id]
+            if not self.target.is_type(self.target_type):
+                replay.datapack.change_type(self.target, self.target_type, self.frame)
 
         else:
             if self.target_type not in replay.datapack.units:
                 self.logger.error("{0}\t{1}\tMissing unit {2} from {3}".format(self.frame, self.player.name, hex(self.target_type), replay.datapack.id))
                 unit = Unit(self.target_id, 0x00)
 
             else:
-                unit_class = replay.datapack.units[self.target_type]
-                unit = unit_class(self.target_id, 0x00)
+                unit = replay.datapack.create_unit(self.target_id, self.target_type, 0x00, self.frame)
 
             self.target = unit
-            replay.objects[uid] = unit
+            replay.objects[self.target_id] = unit
 
     def __str__(self):
         if self.target: