[pypy-commit] jitviewer default: fix counts of loops

[fijal]

Author: Maciej Fijalkowski <fijall at gmail.com>
Branch: 
Changeset: r167:647ec61d599d
Date: 2011-08-01 18:12 +0200
http://bitbucket.org/pypy/jitviewer/changeset/647ec61d599d/

Log:	fix counts of loops

diff --git a/_jitviewer/parser.py b/_jitviewer/parser.py
--- a/_jitviewer/parser.py
+++ b/_jitviewer/parser.py
@@ -148,14 +148,14 @@
     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)