Add proper initData parsing for HotS beta replays.

Author: GraylinKim

sc2reader/readers.py

@@ -505,6 +505,103 @@ def __call__(self, data, replay):
         )
 
 
+class InitDataReader_23925(InitDataReader_22612):
+
+    def __call__(self, data, replay):
+        data = BitPackedDecoder(data)
+        return dict(
+            player_init_data=[dict(
+                name=data.read_aligned_bytes(data.read_uint8()),
+                clan_tag=None,
+                highest_league=None,
+                combined_race_levels=None,
+                random_seed=data.read_uint32(),
+                race_preference=data.read_uint8() if data.read_bool() else None,
+                team_preference=data.read_uint8() if data.read_bool() else None,
+                test_map=data.read_bool(),
+                test_auto=data.read_bool(),
+                examine=data.read_bool(),
+                custom_interface=None,
+                observe=data.read_bits(2),
+            ) for i in range(data.read_bits(5))],
+
+            game_description=dict(
+                random_value=data.read_uint32(),
+                game_cache_name=data.read_aligned_bytes(data.read_bits(10)),
+                game_options=dict(
+                    lock_teams=data.read_bool(),
+                    teams_together=data.read_bool(),
+                    advanced_shared_control=data.read_bool(),
+                    random_races=data.read_bool(),
+                    battle_net=data.read_bool(),
+                    amm=data.read_bool(),
+                    competitive=data.read_bool(),
+                    no_victory_or_defeat=data.read_bool(),
+                    fog=data.read_bits(2),
+                    observers=data.read_bits(2),
+                    user_difficulty=data.read_bits(2),
+                    client_debug_flags=data.read_uint64(),
+                ),
+                game_speed=data.read_bits(3),
+                game_type=data.read_bits(3),
+                max_users=data.read_bits(5),
+                max_observers=data.read_bits(5),
+                max_players=data.read_bits(5),
+                max_teams=data.read_bits(4)+1,
+                max_colors=data.read_bits(6),
+                max_races=data.read_uint8()+1,
+                max_controls=data.read_uint8()+1,
+                map_size_x=data.read_uint8(),
+                map_size_y=data.read_uint8(),
+                map_file_sync_checksum=data.read_uint32(),
+                map_file_name=data.read_aligned_bytes(data.read_bits(11)),
+                map_author_name=data.read_aligned_bytes(data.read_uint8()),
+                mod_file_sync_checksum=data.read_uint32(),
+                slot_descriptions=[dict(
+                    allowed_colors=data.read_bits(data.read_bits(6)),
+                    allowed_races=data.read_bits(data.read_uint8()),
+                    allowedDifficulty=data.read_bits(data.read_bits(6)),
+                    allowedControls=data.read_bits(data.read_uint8()),
+                    allowed_observe_types=data.read_bits(data.read_bits(2)),
+                    allowed_ai_builds=data.read_bits(data.read_bits(7)),
+                ) for i in range(data.read_bits(5))],
+                default_difficulty=data.read_bits(6),
+                default_AI_build=data.read_bits(7),
+                cache_handles=[DepotFile(data.read_aligned_bytes(40)) for i in range(data.read_bits(6))],
+                is_blizzardMap=data.read_bool(),
+                is_premade_ffa=data.read_bool(),
+                is_coop_mode=data.read_bool(),
+            ),
+
+            lobby_state=dict(
+                phase=data.read_bits(3),
+                max_users=data.read_bits(5),
+                max_observers=data.read_bits(5),
+                slots=[dict(
+                    control=data.read_uint8(),
+                    user_id=data.read_bits(4) if data.read_bool() else None,
+                    team_id=data.read_bits(4),
+                    colorPref=data.read_bits(5) if data.read_bool() else None,
+                    race_pref=data.read_uint8() if data.read_bool() else None,
+                    difficulty=data.read_bits(6),
+                    ai_build=data.read_bits(7),
+                    handicap=data.read_bits(7),
+                    observe=data.read_bits(2),
+                    working_set_slot_id=None,
+                    rewards=[data.read_uint32() for i in range(data.read_bits(5))],
+                    toon_handle=data.read_aligned_bytes(data.read_bits(7)),
+                    licenses=[data.read_uint32() for i in range(data.read_bits(9))],
+                ) for i in range(data.read_bits(5))],
+                random_seed=data.read_uint32(),
+                host_user_id=data.read_bits(4) if data.read_bool() else None,
+                is_single_player=data.read_bool(),
+                game_duration=data.read_uint32(),
+                default_difficulty=data.read_bits(6),
+                default_ai_build=data.read_bits(7),
+            ),
+        )
+
+
 class InitDataReader_24764(InitDataReader_22612):
 
     def __call__(self, data, replay):

sc2reader/resources.py

@@ -341,7 +341,7 @@ def load_details(self):
             self.game_length = self.length
             self.real_length = utils.Length(seconds=int(self.length.seconds/GAME_SPEED_FACTOR[self.speed]))
             self.start_time = datetime.utcfromtimestamp(self.unix_timestamp-self.real_length.seconds)
-            self.date = self.end_time #backwards compatibility
+            self.date = self.end_time  # backwards compatibility
 
     def load_map(self):
         self.map = self.factory.load_map(self.map_file, **self.opt)
@@ -445,7 +445,7 @@ def createPlayer(pid, pdata, attributes):
 
         # Assign the default region to computer players for consistency
         # We know there will be a default region because there must be
-        # at least 1 human player or we wouldn't have a self.
+        # at least 1 human player or we wouldn't have a replay.
         default_region = self.humans[0].region
         for player in self.players:
             if not player.is_human:
@@ -559,7 +559,8 @@ def register_default_readers(self):
         self.register_reader('replay.initData', readers.InitDataReader_16561(), lambda r: 16561 <= r.base_build < 17326)
         self.register_reader('replay.initData', readers.InitDataReader_17326(), lambda r: 17326 <= r.base_build < 19132)
         self.register_reader('replay.initData', readers.InitDataReader_19132(), lambda r: 19132 <= r.base_build < 22612)
-        self.register_reader('replay.initData', readers.InitDataReader_22612(), lambda r: 22612 <= r.base_build < 24764)
+        self.register_reader('replay.initData', readers.InitDataReader_22612(), lambda r: 22612 <= r.base_build < 23925)
+        self.register_reader('replay.initData', readers.InitDataReader_23925(), lambda r: 23925 <= r.base_build < 24764)
         self.register_reader('replay.initData', readers.InitDataReader_24764(), lambda r: 24764 <= r.base_build)
         self.register_reader('replay.message.events', readers.MessageEventsReader_Base(), lambda r: r.build < 24247 or r.versions[1]==1)
         self.register_reader('replay.message.events', readers.MessageEventsReader_Beta_24247(), lambda r: r.build >= 24247 and r.versions[1]==2)