hack to make HotS 3.0 / 38215 use the LotV datapack

so that mineral fields will be recognized, which we use for base placement logic
adjust APM upward for 3.0 because people expect 1.4x their true APM.

Author: dsjoerg

sc2reader/data/__init__.py

@@ -401,6 +401,8 @@ def load_build(expansion, version):
 hots_builds = dict()
 for version in ('base', '23925', '24247', '24764'):
     hots_builds[version] = load_build('HotS', version)
+hots_builds['38215'] = load_build('LotV', 'base')
+hots_builds['38215'].id = '38215'
 
 # Load LotV Data
 lotv_builds = dict()

sc2reader/engine/plugins/apm.py

@@ -24,8 +24,8 @@ def handleInitGame(self, event, replay):
             human.seconds_played = replay.length.seconds
 
     def handlePlayerActionEvent(self, event, replay):
-        event.player.aps[event.second] += 1
-        event.player.apm[int(event.second/60)] += 1
+        event.player.aps[event.second] += 1.4
+        event.player.apm[int(event.second/60)] += 1.4
 
     def handlePlayerLeaveEvent(self, event, replay):
         event.player.seconds_played = event.second

sc2reader/factories/plugins/replay.py

@@ -107,8 +107,8 @@ def APMTracker(replay):
 
         for event in player.events:
             if event.name == 'SelectionEvent' or 'AbilityEvent' in event.name or 'Hotkey' in event.name:
-                player.aps[event.second] += 1
-                player.apm[int(event.second/60)] += 1
+                player.aps[event.second] += 1.4
+                player.apm[int(event.second/60)] += 1.4
 
             elif event.name == 'PlayerLeaveEvent':
                 player.seconds_played = event.second

sc2reader/resources.py

@@ -585,7 +585,8 @@ def register_default_datapacks(self):
         self.register_datapack(datapacks['HotS']['base'], lambda r: r.expansion == 'HotS' and r.build < 23925)
         self.register_datapack(datapacks['HotS']['23925'], lambda r: r.expansion == 'HotS' and 23925 <= r.build < 24247)
         self.register_datapack(datapacks['HotS']['24247'], lambda r: r.expansion == 'HotS' and 24247 <= r.build <= 24764)
-        self.register_datapack(datapacks['HotS']['24764'], lambda r: r.expansion == 'HotS' and 24764 <= r.build)
+        self.register_datapack(datapacks['HotS']['24764'], lambda r: r.expansion == 'HotS' and 24764 <= r.build < 38215)
+        self.register_datapack(datapacks['HotS']['38215'], lambda r: r.expansion == 'HotS' and 38215 <= r.build)
         self.register_datapack(datapacks['LotV']['base'], lambda r: r.expansion == 'LotV' and 34784 <= r.build)
 
     # Internal Methods

test_replays/3.0.0.38215/fourth.SC2Replay

@@ -439,7 +439,18 @@ def test_30_map(self):
         factory = sc2reader.factories.SC2Factory()
         replay =factory.load_replay(replayfilename,load_level=1,load_map= True)
 
-      
+    def test_30_apms(self):
+        from sc2reader.factories.plugins.replay import APMTracker, SelectionTracker, toJSON
+
+        factory = sc2reader.factories.SC2Factory()
+        factory.register_plugin("Replay", APMTracker())
+        replay = factory.load_replay("test_replays/3.0.0.38215/fourth.SC2Replay")
+        for player in replay.players:
+            if player.name == 'Owl':
+                print player.name, player.avg_apm
+                self.assertTrue(player.avg_apm > 110)
+
+          
 class TestGameEngine(unittest.TestCase):
     class TestEvent(object):
         name='TestEvent'