Add annotate file for html output

7c2bc88c · Robert Bradshaw · 3d32b841 · 7c2bc88c
Commit 7c2bc88c authored Nov 18, 2007 by Robert Bradshaw
Hide whitespace changes
Inline Side-by-side

Showing with 163 additions and 0 deletions

Cython/Compiler/Annotate.py Cython/Compiler/Annotate.py +163 -0

No files found.
--- a/Cython/Compiler/Annotate.py
+++ b/Cython/Compiler/Annotate.py
+# Note: Work in progress
+
+
+import re
+from StringIO import StringIO
+
+
+from Code import CCodeWriter
+
+
+class AnnotationCCodeWriter(CCodeWriter):
+
+    def __init__(self, f):
+        CCodeWriter.__init__(self, self)
+        self.buffer = StringIO()
+        self.real_f = f
+        self.annotations = []
+        self.last_pos = None
+        self.code = {}
+        
+    def getvalue(self):
+        return self.real_f.getvalue()
+        
+    def write(self, s):
+        self.real_f.write(s)
+        self.buffer.write(s)
+        
+    def mark_pos(self, pos):
+#        if pos is not None:
+#            CCodeWriter.mark_pos(self, pos)
+#        return
+        print "marking", pos
+        if self.last_pos:
+            try:
+                code = self.code[self.last_pos[1]]
+            except KeyError:
+                code = ""
+            self.code[self.last_pos[1]] = code + self.buffer.getvalue()
+            self.buffer = StringIO()
+        self.last_pos = pos
+
+    def annotate(self, pos, item):
+        self.annotations.append((pos, item))
+        
+    def save_annotation(self, filename):
+        self.mark_pos(None)
+        f = open(filename)
+        lines = f.readlines()
+        for k in range(len(lines)):
+            # there has to be a better way to do this
+            lines[k] = lines[k].replace('  ','\t ')
+            lines[k] = lines[k].replace('  ','\t\t')
+            # TODO: this is incorrect
+            lines[k] = lines[k].replace('<', '~')
+            lines[k] = lines[k].replace('>', '~')
+        f.close()
+        all = []
+        for pos, item in self.annotations:
+            if pos[0] == filename:
+                start = item.start()
+                size, end = item.end()
+                if size:
+                    all.append((pos, start))
+                    all.append(((filename, pos[1], pos[2]+size), end))
+                else:
+                    all.append((pos, start+end))
+                
+        all.sort()
+        all.reverse()
+        for pos, item in all:
+            _, line_no, col = pos
+            line_no -= 1
+            col += 1
+            line = lines[line_no]
+            lines[line_no] = line[:col] + item + line[col:]
+        
+        f = open("%s.html" % filename, "w")
+        f.write('<html>\n')
+        f.write("""
+<head>
+<style type="text/css">
+
+body { font-family: courier; font-size: 12; }
+
+.code  { font-size: 9; color: #444444; display: none; margin-left: 20px; }
+.py_api  { color: red; }
+.pyx_api  { color: #FF3000; }
+.py_macro_api  { color: #FF8000; }
+
+.tag  {  }
+
+.coerce  { color: #008000; border: 1px dotted #008000 }
+
+.py_attr { color: #FF0000; font-weight: bold; }
+.c_attr  { color: #0000FF; }
+
+.py_call { color: #FF0000; font-weight: bold; }
+.c_call  { color: #0000FF; }
+
+</style>
+<script>
+function toggleDiv(id) {
+    theDiv = document.getElementById(id);
+    if (theDiv.style.display == 'none') theDiv.style.display = 'block';
+    else theDiv.style.display = 'none';
+}
+</script>
+</head>
+        """)
+        f.write('<body>\n')
+        k = 0
+        
+        py_c_api = re.compile('(Py[A-Z][a-z]+_[A-Z][a-z][A-Za-z_]+)')
+        pyx_api = re.compile('(__Pyx[A-Za-z_]+)\(')
+        py_marco_api = re.compile('(Py[A-Za-z]*_[A-Z][A-Z_]+)')
+        
+        for line in lines:
+
+            k += 1
+            try:
+                code = self.code[k]
+            except KeyError:
+                code = ''
+                
+            code, c_api_calls = py_c_api.subn(r"<span class='py_api'>\1</span>", code)
+            code, pyx_api_calls = pyx_api.subn(r"<span class='pyx_api'>\1</span>(", code)
+            code, macro_api_calls = py_marco_api.subn(r"<span class='py_macro_api'>\1</span>", code)
+            
+            color = "FFFF%02x" % int(255/(1+(5*c_api_calls+2*pyx_api_calls+macro_api_calls)/10.0))
+            f.write("<div class='line' style='background-color: #%s' onclick='toggleDiv(\"line%s\")'>" % (color, k))
+
+            f.write(" %d: " % k)
+            line = line.replace('\t', '&nbsp;')
+            f.write(line)
+                
+            f.write('</div>\n')
+            f.write("<div id='line%s' class='code' style='background-color: #%s'>%s</div>" % (k, color, code.replace('\n', '\n<br>')))
+        f.write('</body></html>\n')
+        f.close()
+        
+
+# TODO: make this cleaner
+def escape(raw_string):
+    raw_string = raw_string.replace("\'", r"&#146;")
+    raw_string = raw_string.replace('\"', r'&quot;')
+    raw_string = raw_string.replace('\n', r'<br>\n')
+    raw_string = raw_string.replace('\t', r'\t')
+    return raw_string
+
+
+class AnnotationItem:
+    
+    def __init__(self, style, text, tag="", size=0):
+        self.style = style
+        self.text = text
+        self.tag = tag
+        self.size = size
+        
+    def start(self):
+        return "<span class='tag %s' title='%s'>%s" % (self.style, self.text, self.tag)
+    
+    def end(self):
+        return self.size, "</span>"