diff --git a/scripts/prismx.py b/scripts/prismx.py index 1111b28..784e24f 100644 --- a/scripts/prismx.py +++ b/scripts/prismx.py @@ -111,13 +111,17 @@ def prism_cmd_cb(data, buffer, args): regex = regex_words if 'w' in opts else regex_chars inc = 'r' not in opts bs = 'e' in opts - stretch = 's' in opts colors = ncolors if 'x' not in opts else (xxcolors if bs or not inc else xcolors) - color_count = len(colors) input = input[::-1] if 'b' in opts else input - output = u"" tokens = re.findall(regex, input) + + if 's' in opts: + color_local = 0 + colors = [colors[int(float(i)/len(tokens)*len(colors))] + for i in xrange(len(tokens))] + + color_count = len(colors) for token in tokens: # prefix each token with a color code c1 = unicode(colors[color_local % color_count]).rjust(2, "0")