/[escript]/trunk/doc/examples/usersguide/poisson.py
ViewVC logotype

Diff of /trunk/doc/examples/usersguide/poisson.py

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

trunk/esys2/doc/user/examples/mypoisson.py revision 102 by jgs, Wed Dec 15 07:08:39 2004 UTC trunk/doc/examples/usersguide/poisson.py revision 2574 by gross, Mon Aug 3 09:09:20 2009 UTC
# Line 1  Line 1 
1  # $Id$  
2  import esys.finley  ########################################################
3  from esys.linearPDEs import Poisson  #
4    # Copyright (c) 2003-2009 by University of Queensland
5    # Earth Systems Science Computational Center (ESSCC)
6    # http://www.uq.edu.au/esscc
7    #
8    # Primary Business: Queensland, Australia
9    # Licensed under the Open Software License version 3.0
10    # http://www.opensource.org/licenses/osl-3.0.php
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    http://www.opensource.org/licenses/osl-3.0.php"""
20    __url__="https://launchpad.net/escript-finley"
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    import pylab
29  # generate domain:  # generate domain:
30  mydomain = esys.finley.Rectangle(l0=1.,l1=1.,n0=40, n1=20)  mydomain = Rectangle(l0=1.,l1=1.,n0=40, n1=20)
31  # define characteristic function of Gamma^D  # define characteristic function of Gamma^D
32  x = mydomain.getX()  x = mydomain.getX()
33  gammaD = x[0].whereZero()+x[1].whereZero()  gammaD = whereZero(x[0])+whereZero(x[1])
34  # define PDE and get its solution u  # define PDE and get its solution u
35  mypde = Poisson(domain=mydomain,f=1,q=gammaD)  mypde = Poisson(domain=mydomain)
36    mypde.setValue(f=1,q=gammaD)
37  u = mypde.getSolution()  u = mypde.getSolution()
38  # write u to an external file  
39  u.saveDX("u.dx")  # interpolate u to a matplotlib grid:
40    x_grid = numpy.linspace(0.,1.,50)
41    y_grid = numpy.linspace(0.,1.,50)
42    z_grid = matplotlib.mlab.griddata(x=mydomain.getX()[0].toListOfTuples(),y=mydomain.getX()[1].toListOfTuples(),
43                                      z=interpolate(u,mydomain.getX().getFunctionSpace()).toListOfTuples(),
44                                      xi=x_grid,yi=y_grid )
45    # interpolate u to a rectangular grid:
46    matplotlib.pyplot.contourf(x_grid, y_grid, z_grid, 5)
47    matplotlib.pyplot.show()
48    matplotlib.pyplot.savefig("u.png")
49    

Legend:
Removed from v.102  
changed lines
  Added in v.2574

  ViewVC Help
Powered by ViewVC 1.1.26