[pypy-svn] r41124 - pypy/dist/pypy/tool

[xoraxax at codespeak.net]

Author: xoraxax
Date: Thu Mar 22 21:31:41 2007
New Revision: 41124

Added:
   pypy/dist/pypy/tool/ansi_mandelbrot.py
Modified:
   pypy/dist/pypy/tool/ansi_print.py
Log:
Added mandelbrot set printing instead of dots only.
Feel free to tweak the colours, characters etc., currently they are not optimal yet.



Added: pypy/dist/pypy/tool/ansi_mandelbrot.py
==============================================================================
--- (empty file)
+++ pypy/dist/pypy/tool/ansi_mandelbrot.py	Thu Mar 22 21:31:41 2007
@@ -0,0 +1,119 @@
+import sys
+
+from py.__.misc.terminal_helper import ansi_print, get_terminal_width
+
+"""
+Black       0;30     Dark Gray     1;30
+Blue        0;34     Light Blue    1;34
+Green       0;32     Light Green   1;32
+Cyan        0;36     Light Cyan    1;36
+Red         0;31     Light Red     1;31
+Purple      0;35     Light Purple  1;35
+Brown       0;33     Yellow        1;33
+Light Gray  0;37     White         1;37
+"""
+
+
+palette = [37, 34, 35, 36, 31, 33, 32, 37]
+
+
+def print_pixel(colour, value_range, invert=1):
+    chars = [".", ".", "+", "*", "%", "#"]
+    idx = lambda chars: (colour+1) * (len(chars) - 1) / value_range
+    if invert:
+        idx = lambda chars, idx=idx:len(chars) - 1 - idx(chars)
+    char = chars[idx(chars)]
+    ansi_colour = palette[idx(palette)]
+    ansi_print(char, ansi_colour, newline=False, flush=True)
+
+
+class Mandelbrot:
+    def __init__ (self, width=100, height=28):
+        self.xpos = -0.5
+        self.ypos = 0
+        aspect_ratio = 1/3.
+        factor = 6.75 / width
+        self.xscale = factor * aspect_ratio
+        self.yscale = factor
+        self.iterations = 100
+        self.x = width
+        self.y = height
+        self.z0 = complex(0, 0)
+
+    def init(self):
+        self.do_reset = False
+        xmin = self.xpos - self.xscale * self.x / 2
+        ymin = self.ypos - self.yscale * self.y / 2
+        self.x_range = [xmin + self.xscale * ix for ix in range(self.x)]
+        self.y_range = [ymin + self.yscale * iy for iy in range(self.y)]
+
+    def reset(self):
+        self.do_reset = True
+
+    def generate(self):
+        for iy in range(self.y):
+            ix = 0
+            while ix < self.x:
+                c = complex(self.x_range[ix], self.y_range[iy])
+                z = self.z0
+                colour = 0
+                mind = 2
+
+                for i in range(self.iterations):
+                    z = z * z + c
+                    d = abs(z)
+                    if d >= 2:
+                        colour = min(int(mind / 0.001), 254) + 1
+                        break
+                    else:
+                        mind = min(d, mind)
+
+                yield ix, iy, colour
+                if self.do_reset: # jump to the beginning of the line
+                    self.do_reset = False
+                    ix = 0
+                else:
+                    ix += 1
+
+
+class Driver(object):
+    def __init__(self):
+        self.init()
+
+    def init(self):
+        self.width = get_terminal_width()
+        self.mandelbrot = Mandelbrot(width=self.width)
+        self.mandelbrot.init()
+        self.gen = self.mandelbrot.generate()
+
+    def reset(self):
+        self.mandelbrot.reset()
+
+    def dot(self):
+        x = c = 0
+        try:
+            x, y, c = self.gen.next()
+            if x == 0:
+                width = get_terminal_width()
+                if width != self.width:
+                    self.init()
+        except StopIteration:
+            print
+            self.init()
+        if x == self.width - 1:
+            print
+        print_pixel(c, 256, 1)
+
+
+if __name__ == '__main__':
+    import random
+    from time import sleep
+
+    d = Driver()
+    for x in xrange(4000):
+        sleep(random.random() / 300)
+        d.dot()
+        if random.random() < 0.001:
+            print
+            d.reset()
+

Modified: pypy/dist/pypy/tool/ansi_print.py
==============================================================================
--- pypy/dist/pypy/tool/ansi_print.py	(original)
+++ pypy/dist/pypy/tool/ansi_print.py	Thu Mar 22 21:31:41 2007
@@ -5,6 +5,7 @@
 import sys
 
 from py.__.misc.terminal_helper import ansi_print
+from pypy.tool.ansi_mandelbrot import Driver
 
 class AnsiLog:
     wrote_dot = False # XXX sharing state with all instances
@@ -24,11 +25,13 @@
         self.kw_to_color = self.KW_TO_COLOR.copy()
         self.kw_to_color.update(kw_to_color)
         self.file = file
+        self.mandelbrot_driver = Driver()
 
     def __call__(self, msg):
         tty = getattr(sys.stderr, 'isatty', lambda: False)()
         flush = False
         newline = True
+        fancy = True
         keywords = []
         esc = []
         for kw in msg.keywords:
@@ -48,7 +51,12 @@
                 return
         elif 'dot' in keywords:
             if tty:
-                ansi_print(".", tuple(esc), file=self.file, newline=False, flush=flush)
+                if fancy:
+                    if not AnsiLog.wrote_dot:
+                        self.mandelbrot_driver.reset()
+                    self.mandelbrot_driver.dot()
+                else:
+                    ansi_print(".", tuple(esc), file=self.file, newline=False, flush=flush)
                 AnsiLog.wrote_dot = True
                 return
         if AnsiLog.wrote_dot:
@@ -58,7 +66,7 @@
         for line in msg.content().splitlines():
             ansi_print("[%s] %s" %(":".join(keywords), line), esc, 
                        file=self.file, newline=newline, flush=flush)
- 
+
 ansi_log = AnsiLog()
 
 # ____________________________________________________________