Commit 91525792 authored by Stefan Behnel's avatar Stefan Behnel

some profiler guided streamlining in compiler output writers

parent f630bc37
......@@ -603,10 +603,16 @@ class CCodeWriter(object):
def put(self, code):
fix_indent = False
dl = code.count("{") - code.count("}")
if "{" in code:
dl = code.count("{")
else:
dl = 0
if "}" in code:
dl -= code.count("}")
if dl < 0:
self.level += dl
elif dl == 0 and code.startswith('}'):
elif dl == 0 and code[0] == "}":
# special cases like "} else {" need a temporary dedent
fix_indent = True
self.level -= 1
if self.bol:
......
......@@ -156,12 +156,13 @@ class Node(object):
self.body.annotate(code)
def end_pos(self):
if not self.child_attrs:
return self.pos
try:
return self._end_pos
except AttributeError:
pos = self.pos
if not self.child_attrs:
self._end_pos = pos
return pos
for attr in self.child_attrs:
child = getattr(self, attr)
# Sometimes lists, sometimes nodes
......
......@@ -7,31 +7,32 @@ class StringIOTree(object):
def __init__(self, stream=None):
self.prepended_children = []
self.stream = stream # if set to None, it will be constructed on first write
if stream is None:
stream = StringIO()
self.stream = stream
self.write = stream.write
def getvalue(self):
return ("".join([x.getvalue() for x in self.prepended_children]) +
self.stream.getvalue())
content = [x.getvalue() for x in self.prepended_children]
content.append(self.stream.getvalue())
return "".join(content)
def copyto(self, target):
"""Potentially cheaper than getvalue as no string concatenation
needs to happen."""
for child in self.prepended_children:
child.copyto(target)
if self.stream:
target.write(self.stream.getvalue())
def write(self, what):
if not self.stream:
self.stream = StringIO()
self.stream.write(what)
stream_content = self.stream.getvalue()
if stream_content:
target.write(stream_content)
def commit(self):
# Save what we have written until now so that the buffer
# itself is empty -- this makes it ready for insertion
if self.stream:
if self.stream.tell():
self.prepended_children.append(StringIOTree(self.stream))
self.stream = None
self.stream = StringIO()
self.write = self.stream.write
def insert(self, iotree):
"""
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment