test/python/OutTest.py

# $Id$
"""
Test writing data object to various file formats 

by Lutz Gross, ACcESS, University of Queensland, Australia, 2005.
"""

from esys.escript import *
from esys.finley import Rectangle,Brick


ne=2
work_dir="/tmp"

def  writeInFormat(fs,format,filetype):
        d=fs.getDim()
        x=length(fs.getX())
        # generate scalar data:
        filename="%s/scalar.%s.%s"%(work_dir,filetype,format)
        print "file ",filename," is generated"
        try:
           eval("x.save%s(\"%s\")"%(format,filename))
        except StandardError, msg:
           print "%% failed because of ",msg
        # generate vector data:
        print "file ",filename," is generated"
        filename="%s/vector.%s.%s"%(work_dir,filetype,format)
        if d==2:
           m=[1.,2.]
        else:
           m=[1.,2.,3.]
        try:
           eval("(x*m).save%s(\"%s\")"%(format,filename))
        except StandardError, msg:
           print "%% failed because of ",msg
        # generate tensor data:
        filename="%s/tensor.%s.%s"%(work_dir,filetype,format)
        print "file ",filename," is generated"
        if d==2:
           m=[[11.,12.],[21.,22.]]
        else:
           m=[[11.,12.,13.],[21.,22.,23.],[31.,32.,33.]]
        try:
           eval("(x*m).save%s(\"%s\")"%(format,filename))
        except StandardError, msg:
           print "%% failed because of ",msg

for format in ["VTK","DX"]:
   for d in [2,3]:
      for order in [1,2]:
        if (d == 2):
            mesh = Rectangle(ne, ne,order,l0=order*ne,l1=order*ne)
        elif (d == 3):
            mesh = Brick(ne,ne,ne,order,l0=order*ne,l1=order*ne,l2=order*ne)
        for fs in ["ContinuousFunction","Function","FunctionOnBoundary","Solution","ReducedSolution"]:
              filetype="%s.o%d.d%d"%(fs,order,d)
              writeInFormat(eval("%s(mesh)"%fs),format,filetype)

1	# $Id$
2	"""
3	Test writing data object to various file formats
4
5	by Lutz Gross, ACcESS, University of Queensland, Australia, 2005.
6	"""
7
8	from esys.escript import *
9	from esys.finley import Rectangle,Brick
10
11
12	ne=2
13	work_dir="/tmp"
14
15	def writeInFormat(fs,format,filetype):
16	d=fs.getDim()
17	x=length(fs.getX())
18	# generate scalar data:
19	filename="%s/scalar.%s.%s"%(work_dir,filetype,format)
20	print "file ",filename," is generated"
21	try:
22	eval("x.save%s(\"%s\")"%(format,filename))
23	except StandardError, msg:
24	print "%% failed because of ",msg
25	# generate vector data:
26	print "file ",filename," is generated"
27	filename="%s/vector.%s.%s"%(work_dir,filetype,format)
28	if d==2:
29	m=[1.,2.]
30	else:
31	m=[1.,2.,3.]
32	try:
33	eval("(x*m).save%s(\"%s\")"%(format,filename))
34	except StandardError, msg:
35	print "%% failed because of ",msg
36	# generate tensor data:
37	filename="%s/tensor.%s.%s"%(work_dir,filetype,format)
38	print "file ",filename," is generated"
39	if d==2:
40	m=[[11.,12.],[21.,22.]]
41	else:
42	m=[[11.,12.,13.],[21.,22.,23.],[31.,32.,33.]]
43	try:
44	eval("(x*m).save%s(\"%s\")"%(format,filename))
45	except StandardError, msg:
46	print "%% failed because of ",msg
47
48	for format in ["VTK","DX"]:
49	for d in [2,3]:
50	for order in [1,2]:
51	if (d == 2):
52	mesh = Rectangle(ne, ne,order,l0=orderne,l1=orderne)
53	elif (d == 3):
54	mesh = Brick(ne,ne,ne,order,l0=orderne,l1=orderne,l2=order*ne)
55	for fs in ["ContinuousFunction","Function","FunctionOnBoundary","Solution","ReducedSolution"]:
56	filetype="%s.o%d.d%d"%(fs,order,d)
57	writeInFormat(eval("%s(mesh)"%fs),format,filetype)
58
Name	Value
svn:eol-style	native
svn:keywords	Author Date Id Revision