# Contents of /trunk/doc/examples/usersguide/poisson_matplotlib.py

Revision 2600 - (show annotations)
Fri Aug 7 04:47:49 2009 UTC (13 years, 4 months ago) by jfenwick
File MIME type: text/x-python
File size: 2085 byte(s)
```Using the 'agg' backend in matplotlib to avoid reference to the X Display.

```
 1 2 ######################################################## 3 # 4 # Copyright (c) 2003-2009 by University of Queensland 5 # Earth Systems Science Computational Center (ESSCC) 6 7 # 8 # Primary Business: Queensland, Australia 9 # Licensed under the Open Software License version 3.0 10 11 # 12 ######################################################## 13 14 __copyright__="""Copyright (c) 2003-2009 by University of Queensland 15 Earth Systems Science Computational Center (ESSCC) 16 http://www.uq.edu.au/esscc 17 Primary Business: Queensland, Australia""" 18 __license__="""Licensed under the Open Software License version 3.0 19 20 __url__= 21 22 from esys.escript import * 23 from esys.escript.linearPDEs import Poisson 24 from esys.finley import Rectangle 25 import numpy 26 import matplotlib 27 28 matplotlib.use('agg') #For interactive use, you can comment out this line 29 #It's just here to make testing easier 30 31 import pylab 32 33 #Testing whether we have a late enough version of matplotlib 34 try: 35 matplotlib.mlab.griddata 36 # generate domain: 37 mydomain = Rectangle(l0=1.,l1=1.,n0=40, n1=20) 38 # define characteristic function of Gamma^D 39 x = mydomain.getX() 40 gammaD = whereZero(x[0])+whereZero(x[1]) 41 # define PDE and get its solution u 42 mypde = Poisson(domain=mydomain) 43 mypde.setValue(f=1,q=gammaD) 44 u = mypde.getSolution() 45 46 # interpolate u to a matplotlib grid: 47 x_grid = numpy.linspace(0.,1.,50) 48 y_grid = numpy.linspace(0.,1.,50) 49 x=mydomain.getX()[0].toListOfTuples() 50 y=mydomain.getX()[1].toListOfTuples() 51 z=interpolate(u,mydomain.getX().getFunctionSpace()).toListOfTuples() 52 z_grid = matplotlib.mlab.griddata(x,y,z,xi=x_grid,yi=y_grid ) 53 # interpolate u to a rectangular grid: 54 matplotlib.pyplot.contourf(x_grid, y_grid, z_grid, 5) 55 matplotlib.pyplot.savefig("u.png") 56 # uncommend this line if you want to interact with a plot window 57 #matplotlib.pyplot.show() 58 59 except AttributeError: 60 print "Your version of matplotlib does not provide the griddata method.\nSkipping example.\n"

 ViewVC Help Powered by ViewVC 1.1.26