Merges in the bulk of obslib code via a substantial reorganization.

Author: GraylinKim

sc2reader/__init__.py

@@ -2,8 +2,27 @@
 
 from mpyq import MPQArchive
 from config import DefaultConfig
-from utils import read_header
+from utils import ReplayBuffer, LITTLE_ENDIAN
 
+def read_header(file):
+    buffer = ReplayBuffer(file)
+    
+    #Check the file type for the MPQ header bytes
+    if buffer.read_hex(4).upper() != "4D50511B":
+        print "Header Hex was: %s" % buffer.read_hex(4).upper()
+        raise ValueError("File '%s' is not an MPQ file" % file.name)
+    
+    #Extract replay header data, we don't actually use this for anything
+    max_data_size = buffer.read_int(LITTLE_ENDIAN) #possibly data max size
+    header_offset = buffer.read_int(LITTLE_ENDIAN) #Offset of the second header
+    data_size = buffer.read_int(LITTLE_ENDIAN)     #possibly data size
+    
+    #Extract replay attributes from the mpq
+    data = buffer.read_data_struct()
+    
+    #return the release and frames information
+    return data[1],data[3]
+    
 def read(location,config=DefaultConfig()):
     if not os.path.exists(location):
         raise ValueError("Location must exist")

sc2reader/config.py

@@ -44,7 +44,7 @@ class DefaultConfig(Config):
             ('replay.details', [ReplayDetailsReader()]),
             ('replay.attributes.events', [AttributeEventsReader_17326(), AttributeEventsReader()]),
             ('replay.message.events', [MessageEventsReader()]),
-            ('replay.game.events', [GameEventsReader_17326(), GameEventsReader_16561(), GameEventsReader()]),
+            ('replay.game.events', [GameEventsReader()]),
         ])
 
     processors = [
@@ -77,3 +77,16 @@ class NoEventsConfig(DefaultConfig):
             RecorderProcessor(),
         ]
 
+#########################################################
+
+class IntegrationConfig(Config):
+    ReplayClass = Replay
+    readers = OrderedDict([
+            ('replay.initData', [ReplayInitDataReader()]),
+            ('replay.details', [ReplayDetailsReader()]),
+            ('replay.attributes.events', [AttributeEventsReader_17326(), AttributeEventsReader()]),
+            ('replay.message.events', [MessageEventsReader()]),
+            ('replay.game.events', [GameEventsReader()]),
+        ])
+        
+    processors = []

sc2reader/constants.py

@@ -0,0 +1,121 @@
+# -*- coding: utf-8 -*-
+
+# These are found in Repack-MPQ/fileset.{locale}#Mods#Core.SC2Mod#{locale}.SC2Data/LocalizedData/Editor/EditorCategoryStrings.txt
+# EDSTR_CATEGORY_Race
+# EDSTR_PLAYERPROPS_RACE
+# question mark means not confirmed data
+races = {
+
+    # deDE
+    # enUS
+    # esES
+    # esMX
+    # frFR
+    # itIT
+    
+    # koKR
+    '프로토스': 'Protoss',
+    '테란': 'Terran',
+    '저그': 'Zerg',
+    
+    # plPL
+    # ptBR
+    # ruRU
+    'Протосс': 'Protoss',
+    'Терран': 'Terran',
+    'Зерг': 'Zerg',
+    # zhCN
+    # zhTW
+    
+    # Uncategorized
+    '神族': 'Protoss',
+    '蟲族': 'Zerg',
+    '人類': 'Terran'
+}
+
+RACES = {
+    'rreT': 'Terran',
+    'greZ': 'Zerg',
+    'torP': 'Protoss',
+    'DNAR': 'Random',
+}
+MESSAGE_CODES = {
+    '0': 'All',
+    '2': 'Allies',
+    '128': 'Header',
+    '125': 'Ping',
+}
+
+TEAM_COLORS = {
+    '10ct': "Red",
+    '20ct': "Blue",
+    '30ct': "Teal",
+    '40ct': "Purple",
+    '50ct': "Yellow",
+    '60ct': "Orange",
+    '70ct': "Green",
+    '80ct': "Pink"
+}
+DIFFICULTIES = {
+    'yEyV': 'Very easy',
+    'ysaE': 'Easy',
+    'ideM': 'Medium',
+    'draH': 'Hard',
+    'dHyV': 'Very hard',
+    'asnI': 'Insane',
+}
+
+GAME_TYPES = {
+    'virP': 'Private',
+    'buP': 'Public',
+    'mmA': 'Ladder',
+    '': 'Single Player',
+}
+
+# (name, key for team ids)
+GAME_FORMATS = {
+    '1v1': ('1v1', 0x07d2,),
+    '2v2': ('2v2', 0x07d3,),
+    '3v3': ('3v3', 0x07d4,),
+    '4v4': ('4v4', 0x07d5,),
+    'AFF': ('FFA', 0x07d6,),
+}
+
+GAME_SPEEDS = {
+    'rolS': 'Slower',
+    'wolS': 'Slow',
+    'mroN': 'Normal',
+    'tsaF': 'Fast',
+    'rsaF': 'Faster',
+}
+
+PLAYER_TYPES = {
+    'nmuH': 'Human',
+    'pmoC': 'Computer',
+}
+
+GATEWAYS = {
+    'US': 'Americas',
+    'KR': 'Asia',
+    'EU': 'Europe',
+    'SG': 'South East Asia',
+    'XX': 'Public Test',
+}
+
+COLOURS = {
+    'B4141E': 'Red',
+    '0042FF': 'Blue',
+    '1CA7EA': 'Teal',
+    'EBE129': 'Yellow',
+    '540081': 'Purple',
+    'FE8A0E': 'Orange',
+    '168000': 'Green',
+    'CCA6FC': 'Light pink',
+    '1F01C9': 'Violet',
+    '525494': 'Light grey',
+    '106246': 'Dark green',
+    '4E2A04': 'Brown',
+    '96FF91': 'Light green',
+    '232323': 'Dark grey',
+    'E55BB0': 'Pink'
+}