better colour stacking

This commit is contained in:
Mattias Andrée 2012-09-01 06:52:42 +02:00
parent 1b78d94ef4
commit 1601dda9d3

22
ponysay
View file

@ -1246,8 +1246,8 @@ class Backend():
h = int(props[props.index(',') + 1:]) h = int(props[props.index(',') + 1:])
else: else:
w = int(props) w = int(props)
balloon = self.__getballoon(w, h, indent).split('\n') balloon = self.__getballoon(w, h, indent)
balloon = [AUTO_PUSH + item + AUTO_POP for item in balloon] balloon = balloon.split('\n')
for b in balloon[0]: for b in balloon[0]:
self.output += b + colourstack.feed(b) self.output += b + colourstack.feed(b)
if lineindex == 0: if lineindex == 0:
@ -1520,32 +1520,20 @@ class ColourStack():
self.stack = [[self.bufproto, None, None, [False] * 9]] + self.stack self.stack = [[self.bufproto, None, None, [False] * 9]] + self.stack
if len(self.stack) == 1: if len(self.stack) == 1:
return None return None
old = self.stack[1] return '\033[0m'
rc = '\033['
if old[1] is not None: rc += '39;'
if old[2] is not None: rc += '49;'
for i in range(0, 9):
if old[3][i]:
rc += '2%i;' % (i + 1)
return '' if len(rc) == 2 else (rc[:-1] + 'm')
def pop(self): def pop(self):
old = self.stack[0] old = self.stack[0]
self.stack = self.stack[1:] self.stack = self.stack[1:]
rc = '\033[' rc = '\033[0;'
if old[1] is not None: rc += '39;'
if old[2] is not None: rc += '49;'
for i in range(0, 9):
if old[3][i]:
rc += str(i + 21) + ';'
new = self.stack[0] new = self.stack[0]
if new[1] is not None: rc += new[1] + ';' if new[1] is not None: rc += new[1] + ';'
if new[2] is not None: rc += new[2] + ';' if new[2] is not None: rc += new[2] + ';'
for i in range(0, 9): for i in range(0, 9):
if new[3][i]: if new[3][i]:
rc += str(i + 1) + ';' rc += str(i + 1) + ';'
return '' if len(rc) == 2 else (rc[:-1] + 'm') return rc[:-1] + 'm'
def feed(self, char): def feed(self, char):