--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/CADrawer.py	Thu Jul 25 21:27:36 2013 -0500
@@ -0,0 +1,181 @@
+""" Code example from Complexity and Computation, a book about
+exploring complexity science with Python.  Available free from
+
+http://greenteapress.com/complexity
+
+Copyright 2011 Allen B. Downey.
+Distributed under the GNU General Public License at gnu.org/licenses/gpl.html.
+"""
+
+import numpy
+
+
+class UnimplementedMethodException(Exception):
+    """Used to indicate that a child class has not implemented an
+    abstract method."""
+
+
+class Drawer(object):
+    """Drawer is an abstract class that should not be instantiated.
+    It defines the interface for a CA drawer; child classes of Drawer
+    should implement draw, show and save.
+
+    If draw_array is not overridden, the child class should provide
+    draw_cell.
+    """
+    def __init__(self):
+        msg = 'CADrawer is an abstract type and should not be instantiated.'
+        raise UnimplementedMethodException, msg
+
+    def draw(self, ca):
+        """Draws a representation of cellular automaton (CA).
+        This function generally has no visible effect."""
+        raise UnimplementedMethodException
+    
+    def draw_array(self, a):
+        """Iterate through array (a) and draws any non-zero cells."""
+        for i in xrange(self.rows):
+            for j in xrange(self.cols):
+                if a[i,j]:
+                    self.draw_cell(j, self.rows-i-1)
+
+    def draw_cell(self, ca):
+        """Draws a single cell.
+        Not required for all implementations."""
+        raise UnimplementedMethodException
+    
+    def show(self):
+        """Displays the representation on the screen, if possible."""
+        raise UnimplementedMethodException
+
+    def save(self, filename):
+        """Saves the representation of the CA in filename."""
+        raise UnimplementedMethodException
+        
+
+class PyplotDrawer(Drawer):
+    """Implementation of Drawer using matplotlib."""
+
+    def __init__(self):
+        # we only need to import pyplot if a PyplotDrawer
+        # gets instantiated
+        global pyplot
+        import matplotlib.pyplot as pyplot
+
+    def draw(self, ca, start=0, end=None):
+        """Draws the CA using pyplot.pcolor."""
+        pyplot.gray()
+        a = ca.get_array(start, end)
+        rows, cols = a.shape
+
+        # flipud puts the first row at the top; 
+        # negating it makes the non-zero cells black.
+        pyplot.pcolor(-numpy.flipud(a))
+        pyplot.axis([0, cols, 0, rows])
+
+        # empty lists draw no ticks
+        pyplot.xticks([])
+        pyplot.yticks([])
+
+    def show(self):
+        """display the pseudocolor representation of the CA"""
+        pyplot.show()
+
+    def save(self, filename='ca.png'):
+        """save the pseudocolor representation of the CA in (filename)."""
+        pyplot.savefig(filename)
+    
+
+class PILDrawer(Drawer):
+    """Implementation of Drawer using PIL and Swampy."""
+
+    def __init__(self, csize=4, color='black'):
+        # we only need to import these modules if a PILDrawer
+        # gets instantiated
+        global Image, ImageDraw, ImageTk, Gui
+        import Image
+        import ImageDraw
+        import ImageTk
+        try:
+            import Gui
+        except ImportError:
+            import swampy.Gui
+        self.csize = csize
+        self.color = color
+
+    def draw(self, ca, start=0, end=None):
+        a = ca.get_array(start, end)
+        self.rows, self.cols = a.shape
+        size = [self.cols * self.csize, self.rows * self.csize]
+
+        self.gui = Gui.Gui()
+        self.button = self.gui.bu(command=self.gui.quit)
+
+        self.image = Image.new(mode='1', size=size, color='white')
+        self.drawable = ImageDraw.Draw(self.image)
+        self.draw_array(numpy.flipud(a))
+
+    def draw_cell(self, i, j):
+        size = self.csize
+        x, y = i*size, j*size
+        self.drawable.rectangle([x, y, x+size, y+size], fill=self.color)
+
+    def show(self):
+        self.tkpi = ImageTk.PhotoImage(self.image)
+        self.button.config(image=self.tkpi)
+        self.gui.mainloop()
+ 
+    def save(self, filename='ca.gif'):
+        self.image.save(filename)
+
+
+class EPSDrawer(Drawer):
+    """Implementation of Drawer using encapsulated Postscript (EPS)."""
+
+    def __init__(self):
+        self.cells = []
+
+    def draw(self, ca, start=0, end=None):
+        a = ca.get_array(start, end)
+        self.rows, self.cols = a.shape
+        self.draw_array(a)
+
+    def draw_cell(self, i, j):
+        self.cells.append((i,j))
+        
+    def show(self):
+        raise UnimplementedMethodException
+
+    def save(self, filename='ca.eps'):
+        fp = open(filename, 'w')
+        self.print_header(fp)
+        self.print_outline(fp)
+        self.print_cells(fp)
+        self.print_footer(fp)
+
+    def print_cells(self, fp):
+        for i, j in self.cells:
+            fp.write('%s %s c\n' % (i, j))
+
+    def print_header(self, fp, size=0.9):
+        fp.write('%!PS-Adobe-3.0 EPSF-3.0\n')
+        fp.write('%%%%BoundingBox: -2 -2 %s %s\n' % (self.cols+2, self.rows+2))
+
+        fp.write('/c {\n')
+        fp.write('   newpath moveto\n')
+        fp.write('   0 %g rlineto\n' % size)
+        fp.write('   %g 0 rlineto\n' % size)
+        fp.write('   0 -%g rlineto\n' % size)
+        fp.write('   closepath fill\n')
+        fp.write('} def\n')
+
+    def print_outline(self, fp):
+        fp.write('newpath 0.1 setlinewidth 0 0 moveto\n')
+        fp.write('0 %s rlineto\n' % self.rows)
+        fp.write('%s 0 rlineto\n' % self.cols)
+        fp.write('0 -%s rlineto\n' % self.rows)
+        fp.write('closepath stroke\n')
+
+    def print_footer(self, fp):
+        fp.write('%%EOF\n')
+