Continued parsing build orders, nothing completed yet

Author: Prillan

extract.py

@@ -32,6 +32,13 @@
     'Structures Built',
     'Structures Razed Count'
 ]
+# Obviously not complete
+abilities = {
+    0x5602 : 'Warp Gate',
+    0x3402 : 'Extended Thermal Lance',
+    0x4402 : 'Metabolic Boost',
+    
+    }
 
 def getRealm(str):
     if str == '\x00\x00S2':
@@ -95,12 +102,23 @@ def toUnit(bo_unit):
     if bo_unit >> 24 == 1:
         return {'name':unitData.type(i).name, 'id':hex(i)}
     return None
+def toAbility(bo_ability):
+    #print(hex(flip_int(bo_unit, 4)))
+    i = ((bo_ability & 0xff) << 8) | 0x02
+    if bo_ability >> 24 == 2:
+        return {'name':abilities[i] if i in abilities else "Unknown ability", 'type':hex(i), 'id':hex(bo_ability)}
+        
+    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])
+            # is unit
+            if u[0][1] >> 24 == 1:
+                unit = toUnit(u[0][1])
+            elif u[0][1] >> 24 == 2:
+                unit = toAbility(u[0][1])
             if not unit:
                 continue
             for entry in u[1][index]:
@@ -109,7 +127,7 @@ def getBuildOrder(unit_structs, index):
                         'total_supply' : entry[1]&0xff,
                         'time' : toTime(entry[2]),
                         'time_hex' : hex(entry[2]>>8),
-                        'unit' : unit,
+                        'order' : unit,
                         'build_index' : entry[1] >> 16,
                         'struct_as_hex' : {0:hex(entry[0]),
                                            1:hex(entry[1]),
@@ -118,6 +136,20 @@ def getBuildOrder(unit_structs, index):
                         })
     bo.sort(key=lambda x: x['build_index'])
     return bo
+def getBuildOrderOthers(struct):
+    ret = list()
+    for x in struct:
+        if x[1][1] > 0xffff:
+            ret.append({
+                    'type':x[1][1],
+                    'type_hex':hex(x[1][1]),
+                    'player':x[2][0][1],
+                    'others':[x[2][0][2],
+                              #x[2][2][1], 
+                              #x[2][2][2], 
+                              x[4]]
+                    })
+    return ret
 
 def getBuildOrders(data):
     unit_structs = [x[0] for x in data[5:]]
@@ -128,7 +160,7 @@ def getBuildOrders(data):
                 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]
+    return [players, all, getBuildOrderOthers(data[1][0])]
 
 def main():
     arg = sys.argv[2] if len(sys.argv) >= 3 else "pprint"
@@ -146,5 +178,13 @@ def main():
             pprint.PrettyPrinter(indent=2).pprint(data)
         elif arg == "bo":
             pprint.PrettyPrinter(indent=2).pprint(getBuildOrders(data))
-            
+        elif arg == "bo2":
+            bos = getBuildOrders(data)[0]
+            for p in bos:
+                for u in bos[p]:
+                    print("{}:{}  {}  {}/{}".format(u['time'] / 60,
+                                                    u['time'] % 60,
+                                                    u['unit']['name'],
+                                                    u['supply'],
+                                                    u['total_supply']))
 main()