plotter/hpgl_mpl.py

Mon, 04 Feb 2008 19:19:57 +0100

author
David Douard <david.douard@logilab.fr>
date
Mon, 04 Feb 2008 19:19:57 +0100
changeset 38
73bd2fe81005
parent 32
59da528bc470
permissions
-rw-r--r--

small improvement; display first plot at startup

# -*- coding: utf-8 -*-
import numpy
import pylab

from hpgl_parser import HPGLParser

class HPGLmplPlotter(HPGLParser):
    pens = "krbgcmyw"
    lines = ['-', ':', '--', '-.', '.', '-', '-', '-']
    def __init__(self, data):        
        super(HPGLmplPlotter, self).__init__(data)
        pylab.show()
        
    def plot_symbols(self, points):
        x, y = points.T
        pylab.plot(x, y, 'o')

    def plot_lines(self, points):
        x, y = points.T
        pylab.plot(x, y, self.pens[self.pen]+self.lines[self.line_type])
        
if __name__ == "__main__":
    import sys
    data = open(sys.argv[1]).read()
    HPGLmplPlotter(data)
    

mercurial