/[escript]/trunk/pyvisi/test/python/gen_data.py
ViewVC logotype

Annotation of /trunk/pyvisi/test/python/gen_data.py

Parent Directory Parent Directory | Revision Log Revision Log


Revision 811 - (hide annotations)
Mon Aug 14 05:36:17 2006 UTC (14 years, 1 month ago) by gross
File MIME type: text/x-python
File size: 1342 byte(s)
generator for test data
1 gross 811 # $Id:$
2     #
3     # simple script to generate some test data for pyvisi
4     #
5    
6     __copyright__=""" Copyright (c) 2006 by ACcESS MNRF
7     http://www.access.edu.au
8     Primary Business: Queensland, Australia"""
9     __license__="""Licensed under the Open Software License version 3.0
10     http://www.opensource.org/licenses/osl-3.0.php"""
11     from esys.escript import *
12     from esys import finley
13     #
14     #
15     #
16     DIM=2
17     L0=3
18     L1=3.
19     L2=1.
20     NE0=15
21     NE1=int(NE0/L0*L1)
22     NE2=int(NE0/L0*L2)
23     #
24     # generate domains:
25     #
26     if DIM == 2:
27     domain=finley.Rectangle(NE0,NE1,1,l0=L0,l1=L1)
28     trafo=numarray.array([[0.,1.],[-1.,0.]])
29     cen=numarray.array([L0/2.,L1/2.])
30     else:
31     domain=finley.Brick(NE0,NE1,NE2,1,l0=L0,l1=L1,l2=L2)
32     trafo=numarray.array([[0.,1.,0.],[-1.,0.,0.],[0.,0.,1.]])
33     cen=numarray.array([L0/2.,L1/2.,L2/2.])
34     #
35     # get function spaces:
36     #
37     c=ContinuousFunction(domain)
38     f=Function(domain)
39     b=FunctionOnBoundary(domain)
40     #
41     # get coordinates
42     #
43     c_x=c.getX()-cen
44     c_r=length(c_x)
45    
46     f_x=f.getX()-cen
47     f_r=length(f_x)
48    
49     b_x=b.getX()-cen
50     b_r=length(b_x)
51     #
52     #
53     #
54     saveVTK("interior_%dD.xml"%DIM,temperature=sin(2*c_r),temperature_cell=sin(2*f_r), \
55     velocity=matrix_mult(trafo,c_x/(c_r+1.e-15)), velocity_cell=matrix_mult(trafo,f_x/f_r))
56     saveVTK("boundary_%dD.xml"%DIM,temperature=sin(b_r),velocity=matrix_mult(trafo,b_x/(b_r+1.e-15)))

  ViewVC Help
Powered by ViewVC 1.1.26