Build order parsing completed.

We still need all upgrades/research ints.

Author: Prillan

sc2reader/constants.py

@@ -155,6 +155,54 @@
     'E55BB0': 'Pink'
 }
 
+# Obviously not complete
+# TODO: Complete this!
+BUILD_ORDER_UPGRADES = {
+    # Protoss
+    
+    ## Forge
+    0x2902 : 'Protoss Ground Weapons Level 1',
+    0x2a02 : 'Protoss Ground Weapons Level 2', #extrapolated
+    0x2b02 : 'Protoss Ground Weapons Level 3', #extrapolated 
+    
+    ## Twilight
+    0x5902 : 'Blink',
+    # Charge
+    
+    ## Cyber core
+    0x5602 : 'Warp Gate Research',
+    0x5702 : 'Hallucination',
+    # Air Weapons 1-3
+    # Air Armor 1-3
+    # Hallucination
+
+    ## Robo bay
+    # Obs, speed
+    # Prism, speed
+    0x3402 : 'Extended Thermal Lance',
+    
+    #Zerg
+    ## Pool
+    0x4402 : 'Metabolic Boost',
+    # Adrenal glands
+    
+    ## Evo. chamber
+    
+    0x3a02 : 'Zerg Ground Carapace Level 1',
+    0x3b02 : 'Zerg Ground Carapace Level 2', #extrapolated
+    0x3c02 : 'Zerg Ground Carapace Level 3', #extrapolated
+    0x3d02 : 'Zerg Missile Attacks Level 1',
+    0x3e02 : 'Zerg Missile Attacks Level 2', #extrapolated
+    0x3f02 : 'Zerg Missile Attacks Level 3', #extrapolated
+    
+    ## Lair
+    0x4202 : 'Burrow',
+    
+    #Terran
+    ## Factory tech lab
+    0x1402 : 'Siege Tech'
+    }
+
 # TODO: Not sure if this is a complete mapping
 #
 # Assuming only 1 Public Test Realm subregion on the following basis:

sc2reader/factories.py

@@ -277,7 +277,7 @@ def load_map_header(self, header_file, options=None, **new_options):
         """
         options = options or utils.merged_dict(self.options, new_options)
         resource, name = self.load_resource(header_file, options=options)
-        print name
+        #print name
         s2mh = MapHeader(resource, name, **options)
 
         # Load summary procedure here!

sc2reader/resources.py

@@ -429,9 +429,12 @@ class GameSummary(Resource):
     #: Game completion time
     time = int()
 
-    #: Players, a dict of :class`PlayerSummary` from the game
+    #: Players, a list of :class`PlayerSummary` from the game
     players = list()
 
+    #: Build orders, a dict of build orders indexed by player id
+    build_orders = dict()
+
     #: Map image urls
     image_urls = list()
 
@@ -442,6 +445,7 @@ def __init__(self, summary_file, filename=None, **options):
         super(GameSummary, self).__init__(summary_file, filename,**options)
 
         self.players = list()
+        self.build_orders = dict()
         self.image_urls = list()
         self.localization_urls = dict()
 
@@ -531,6 +535,37 @@ def __init__(self, summary_file, filename=None, **options):
             parsed_hash = utils.parse_hash(hash)
             self.image_urls.append(self.base_url_template.format(parsed_hash['server'], parsed_hash['hash'], parsed_hash['type']))
 
+
+        # Parse build orders
+        
+        bo_structs = [x[0] for x in self.parts[5:]]
+        bo_structs.append(self.parts[4][0][3:])
+
+        # This might not be the most effective way, but it works
+        for p in self.players:
+            bo = list()
+            for bo_struct in bo_structs:
+                for order in bo_struct:
+                    
+                    if order[0][1] >> 24 == 0x01:
+                        # unit
+                        parsed_order = utils.get_unit(order[0][1])
+                    elif order[0][1] >> 24 == 0x02:
+                        # research
+                        parsed_order = utils.get_research(order[0][1])
+                    
+                    for entry in order[1][p.pid]:
+                        bo.append({
+                                'supply' : entry[0],
+                                'total_supply' : entry[1]&0xff,
+                                'time' : (entry[2] >> 8) / 16,
+                                'order' : parsed_order,
+                                'build_index' : entry[1] >> 16,
+                                })
+            bo.sort(key=lambda x: x['build_index'])
+            self.build_orders[p.pid] = bo
+                        
+
     def __str__(self):
         return "{} - {}:{}:{} {}".format(time.ctime(self.time),
                                          int(self.game_length)/3600,

sc2reader/utils.py

@@ -12,7 +12,8 @@
 from datetime import timedelta
 
 from sc2reader import exceptions
-from sc2reader.constants import COLOR_CODES
+from sc2reader.constants import COLOR_CODES, BUILD_ORDER_UPGRADES
+from sc2reader.data.build19595 import Data_19595 as Data
 
 LITTLE_ENDIAN,BIG_ENDIAN = '<','>'
 
@@ -720,6 +721,38 @@ def get_files(path, extensions=['.sc2replay'], exclude=[], depth=-1, followlinks
 
     return files
 
+def get_unit(type_int):
+    """
+    Takes an int, i, with (i & 0xff000000) = 0x01000000
+    and returns the corresponding unit/structure
+    """
+    # Test if we have used data_obj before
+    try:
+        data_obj
+    except:
+        #Nope, create
+        data_obj = Data()
+    
+    # Try to parse a unit
+    try:
+        unit = data_obj.type(((type_int & 0xff) << 8) | 0x01)
+    except:
+        unit = None
+    
+    return {
+        'name': unit.name if unit else "Unknown unit ({})".format(hex(type_int)) ,
+        'type_int':hex(type_int)
+        }
+def get_research(type_int):
+    """
+    Takes an int, i, with (i & 0xff000000) = 0x02000000
+    and returns the corresponding research/upgrade
+    """
+    t = ((type_int & 0xff) << 8) | 0x02
+    return {
+        'name': BUILD_ORDER_UPGRADES[t] if t in BUILD_ORDER_UPGRADES else "Unknown upgrade ({})".format(hex(t)),
+        'type_int': hex(type_int)}
+
 def parse_hash(hash_string):
     """
     Parse a hash to useful data