Commits

Maciej Fijalkowski committed 647ec61

fix counts of loops

Comments (0)

Files changed (1)

_jitviewer/parser.py

     if not input:
         return
     lines = input[-1].splitlines()
-    nums = []
-    i = 0
+    mapping = {}
+    for loop in loops:
+        com = loop.comment
+        if 'Loop' in com:
+            mapping['loop ' + re.search('Loop (\d+)', com).group(1)] = loop
+        else:
+            mapping['bridge ' + re.search('Guard (\d+)', com).group(1)] = loop
     for line in lines:
         if line:
-            num, count = line.split(':')
-            assert int(num) == i
-            count = int(count)
-            nums.append(count)
-            loops[i].count = count
-            i += 1
-    return nums
+            num, count = line.split(':', 2)
+            mapping[num].count = int(count)