@@ -217,10 +217,9 @@ class GameEventsReader_Base(object):
217
217
HOTKEY_OVERLAY = 0
218
218
POFFSET = - 1
219
219
220
- def __call__ (self , data , replay ):
221
- EVENT_DISPATCH = {
220
+ def __init__ (self ):
221
+ self . EVENT_DISPATCH = {
222
222
0x05 : self .game_start_event ,
223
- 0x07 : self .beta_join_event ,
224
223
0x0B : self .player_join_event ,
225
224
0x0C : self .player_join_event ,
226
225
0x19 : self .player_leave_event ,
@@ -230,9 +229,9 @@ def __call__(self, data, replay):
230
229
0x1F : self .player_send_resource_event ,
231
230
0x31 : self .camera_event ,
232
231
0x46 : self .player_request_resource_event ,
233
- 0x65 : self .beta_win_event ,
234
232
}
235
233
234
+ def __call__ (self , data , replay ):
236
235
game_events = list ()
237
236
fstamp = 0
238
237
debug = replay .opt .debug
@@ -254,8 +253,8 @@ def __call__(self, data, replay):
254
253
event_type = read_bits (7 )
255
254
256
255
# Check for a lookup
257
- if event_type in EVENT_DISPATCH :
258
- event = EVENT_DISPATCH [event_type ](data , fstamp , pid , event_type )
256
+ if event_type in self . EVENT_DISPATCH :
257
+ event = self . EVENT_DISPATCH [event_type ](data , fstamp , pid , event_type )
259
258
if debug :
260
259
event .bytes = data .read_range (event_start , tell ())
261
260
append (event )
@@ -510,6 +509,12 @@ class GameEventsReader_22612(GameEventsReader_19595):
510
509
UNIT_INDEX_BITS = 9 # Now can select up to 512 units
511
510
512
511
class GameEventsReader_Beta (GameEventsReader_22612 ):
512
+
513
+ def __init__ (self ):
514
+ super (GameEventsReader_Beta , self ).__init__ ()
515
+ self .EVENT_DISPATCH [0x07 ] = self .beta_join_event
516
+ self .EVENT_DISPATCH [0x65 ] = self .beta_win_event
517
+
513
518
def beta_join_event (self , data , fstamp , pid , event_type ):
514
519
flags = data .read_bytes (5 )
515
520
return BetaJoinEvent (fstamp , pid , event_type , flags )
0 commit comments