Added extract.py. I use it to test the parsing as I work (don't expect it to work)

Author: Prillan

extract.py

@@ -0,0 +1,150 @@
+import zlib, sys, pprint
+from sc2reader.utils import ReplayBuffer
+from sc2reader.data.build19595 import Data_19595 as DataObject
+
+unitData = DataObject()
+
+races = {'Prot':'Protoss','Zerg':'Zerg','Terr':'Terran','RAND':'Random'}
+data_names = [
+    'R',
+    'U',
+    'S',
+    'O',
+    'AUR',
+    'RCR',
+    'WC',
+    'UT',
+    'KUC',
+    'SB',
+    'SRC',
+    
+]
+data_names_pretty = [
+    'Resources',
+    'Units',
+    'Structures',
+    'Overview',
+    'Average Unspent Resources',
+    'Resource Collection Rate',
+    'Workers Created',
+    'Units Trained',
+    'Killed Unit Count',
+    'Structures Built',
+    'Structures Razed Count'
+]
+
+def getRealm(str):
+    if str == '\x00\x00S2':
+        return "EU"
+    
+    return "?"
+def getPlayers(data):
+    players = []
+    parr = data[0][3]
+    for i in range(16):
+        if not (parr[i][0][1] == 0):
+            players.append(getPlayer(data, i))
+
+    return players
+def getIncomeGraph(data, index):
+    return getGraph(data[4][0][1][1], index)
+def getArmyGraph(data, index):
+    return getGraph(data[4][0][2][1], index)
+def getGraph(graph, index):
+    return [(o[2], o[0]) for o in graph[index]] 
+def getPlayer(data, index):
+    pinfo = data[0][3][index]
+    pdata = data[3][0]
+
+    player = {
+        'id': "{}/{}/{}".format(getRealm(pinfo[0][1][0][1]), pinfo[0][1][0][2], pinfo[0][1][0][3]),
+        'race' : races[pinfo[2]]
+        }
+
+    stats = {}
+
+    for i in range(len(pdata)):
+        stats[data_names[i]] = pdata[i][1][index][0][0]
+    stats[data_names[len(pdata)]] = data[4][0][0][1][index][0][0]
+    player['stats'] = stats
+    player['graphs'] = {}
+    player['graphs']['income'] = getIncomeGraph(data, index)
+    player['graphs']['army'] = getArmyGraph(data, index)
+
+    return player
+def flip_int(num, b):
+    """
+    Flips the b first bytes in num
+    Example:
+    (0x12345, 3) -> 0x452301
+    (0x00112233, 4) -> 0x33221100
+    """
+    o = 0
+    for i in range(b/2):
+        o |= ((num & (0xff << i*8)) << (b-(2*i+1))*8)
+        o |= ((num & (0xff << (b-(i+1))*8)) >> (b-(2*i+1)) * 8)
+    if b % 2 == 1:
+        o |= (num & (0xff << (b/2)*8))
+    return o
+    
+def toTime(bo_time):
+    return (bo_time >> 8) / 16
+def toUnit(bo_unit):
+    #print(hex(flip_int(bo_unit, 4)))
+    i = ((bo_unit & 0xff) << 8) | 0x01
+    if bo_unit >> 24 == 1:
+        return {'name':unitData.type(i).name, 'id':hex(i)}
+    return None
+def getBuildOrder(unit_structs, index):
+    ''' [ {supply, time, unit} ] '''
+    bo = []
+    for unit_struct in unit_structs:
+        for u in unit_struct:
+            unit = toUnit(u[0][1])
+            if not unit:
+                continue
+            for entry in u[1][index]:
+                bo.append({
+                        'supply' : entry[0],
+                        'total_supply' : entry[1]&0xff,
+                        'time' : toTime(entry[2]),
+                        'time_hex' : hex(entry[2]>>8),
+                        'unit' : unit,
+                        'build_index' : entry[1] >> 16,
+                        'struct_as_hex' : {0:hex(entry[0]),
+                                           1:hex(entry[1]),
+                                           2:hex(entry[2]),
+                                                 }
+                        })
+    bo.sort(key=lambda x: x['build_index'])
+    return bo
+
+def getBuildOrders(data):
+    unit_structs = [x[0] for x in data[5:]]
+    players = {}
+    for i in range(15):
+            bo = getBuildOrder(unit_structs, i)
+            if len(bo) > 0:
+                players[i] = bo
+    all = [u for p in players for u in players[p]]
+    all.sort(key=lambda x: x['build_index'])
+    return [players, all]
+
+def main():
+    arg = sys.argv[2] if len(sys.argv) >= 3 else "pprint"
+    with open(sys.argv[1],"rb") as s2gs:
+        raw_data = zlib.decompress(s2gs.read()[16:])
+        data = list()
+        buffer = ReplayBuffer(raw_data)
+        while buffer.left:
+            part = buffer.read_data_struct()
+            data.append(part)
+                
+        if arg == "players":
+            pprint.PrettyPrinter(indent=2).pprint(getPlayers(data))
+        elif arg == "pprint":
+            pprint.PrettyPrinter(indent=2).pprint(data)
+        elif arg == "bo":
+            pprint.PrettyPrinter(indent=2).pprint(getBuildOrders(data))
+            
+main()