Add new listeners, primarily used for debugging.

Author: GraylinKim

sc2reader/listeners/__init__.py

@@ -3,4 +3,6 @@
 from collections import defaultdict
 
 from sc2reader.listeners.apm import APMTracker
-from sc2reader.listeners.selection import SelectionListener
+from sc2reader.listeners.selection import SelectionListener
+from sc2reader.listeners.location import LocationListener
+from sc2reader.listeners.target import TargetListener

sc2reader/listeners/location.py

@@ -0,0 +1,21 @@
+from __future__ import absolute_import
+
+from sc2reader import utils
+from sc2reader.events import LocationAbilityEvent
+from sc2reader.listeners.utils import ListenerBase
+
+
+class LocationListener(ListenerBase):
+
+	def accepts(self, event):
+		return isinstance(event, LocationAbilityEvent) and event.ability!=0x3601 # and event.ability # and event.ability != 0x3601
+
+	def __call__(self, event, replay):
+		if replay.opt.debug:
+			print event.bytes.encode('hex')
+
+		ability = event.ability
+		if ability is not None and ability in replay.datapack.abilities:
+			ability = replay.datapack.ability(event.ability)
+
+		print "[{0}] {1} used ability {2} at location: {3}, {4}".format(utils.Length(seconds=event.second),event.player.name, ability, event.location[0], event.location[1])

sc2reader/listeners/target.py

@@ -0,0 +1,21 @@
+from __future__ import absolute_import
+
+from sc2reader import utils
+from sc2reader.events import TargetAbilityEvent
+from sc2reader.listeners.utils import ListenerBase
+
+
+class TargetListener(ListenerBase):
+
+	def accepts(self, event):
+		return isinstance(event, TargetAbilityEvent) #and event.ability in (0x3700, 0x5700)
+
+	def __call__(self, event, replay):
+		if replay.opt.debug:
+			print event.bytes.encode('hex')
+
+		ability = event.ability
+		if ability is not None and ability in replay.datapack.abilities:
+			ability = replay.datapack.ability(event.ability)
+
+		print "[{0}] {1} used ability {2} on {3}".format(utils.Length(seconds=event.second),event.player.name, ability, event.target)