source: qesdi/linplot/trunk/tests/linplot_tests/draw/draw_trig_functions.py @ 6161

Subversion URL: http://proj.badc.rl.ac.uk/svn/ndg/qesdi/linplot/trunk/tests/linplot_tests/draw/draw_trig_functions.py@6161
Revision 6161, 1.2 KB checked in by pnorton, 11 years ago (diff)

Added the first version of the linplot code.

Line 
1'''
2Created on 15 Dec 2009
3
4@author: pnorton
5'''
6import logging
7import math
8import os
9import numpy
10from linplot import Plot
11from linplot_tests.draw import getDrawOutputDir, SHORT_LOG_FORMAT_STRING
12
13def draw_trig_functions():
14   
15    fileName = 'trig_functions.png'
16    outputDir = getDrawOutputDir()
17    outputFile = os.path.join(outputDir, fileName)
18       
19    xdata = numpy.arange(0, 3*math.pi, 0.1)
20
21    colours = ['red', 'blue', 'yellow', 'green', 'orange', 'purple', 'cyan', 'black', 'brown', 'pink', 'grey', 'lightblue', 'darkblue', 'lightgreen', 'darkgreen']
22    funcs = ['sin', 'cos', 'tan', 'sinc', 'arccos', 'arcsin',  'arctan', 'cosh', 'sinh', 'tanh', 'arccosh', 'arctanh','arcsinh']
23    plt = Plot()
24   
25    for i, f in enumerate(funcs):
26        fn = getattr(numpy, f)
27        ydata = [fn(x) for x in xdata]
28        label = "Y = %s(X)" % f
29        plt.draw(xdata, ydata, label=label, color=colours[i])
30       
31   
32    plt.setYRange(-2.5, 2.5)
33    plt.setXRange(0, 3*math.pi)
34   
35    plt.saveImage(outputFile)
36    print "Wrote %s" % (outputFile)
37   
38
39if __name__ == '__main__':
40   
41    logging.basicConfig(level=logging.DEBUG, format=SHORT_LOG_FORMAT_STRING)
42    draw_trig_functions()
Note: See TracBrowser for help on using the repository browser.