examples/cookbook/onedheatdiff.py


########################################################
#
# Copyright (c) 2009 by University of Queensland
# Earth Systems Science Computational Center (ESSCC)
# http://www.uq.edu.au/esscc
#
# Primary Business: Queensland, Australia
# Licensed under the Open Software License version 3.0
# http://www.opensource.org/licenses/osl-3.0.php
#
########################################################

__copyright__="""Copyright (c) 2009 by University of Queensland
Earth Systems Science Computational Center (ESSCC)
http://www.uq.edu.au/esscc
Primary Business: Queensland, Australia"""
__license__="""Licensed under the Open Software License version 3.0
http://www.opensource.org/licenses/osl-3.0.php"""
__url__="https://launchpad.net/escript-finley"

"""
Author: Antony Hallam antony.hallam@uqconnect.edu.au
"""

# To solve the problem it is necessary to import the modules we require.
from esys.escript import * # This imports everything from the escript library
from esys.escript.linearPDEs import LinearPDE # This defines LinearPDE as LinearPDE
from esys.finley import Rectangle # This imports the rectangle domain function from finley
import os #This package is necessary to handle saving our data.
#from cblib import needdirs
from cblib1 import needdirs

##ESTABLISHING VARIABLES
#PDE related
q=50.e6 #our heat source temperature
Tref=0. #the starting temperature of our iron bar
rho=2.6e6
eta=0#75.
kappa=240.
#Script/Iteration Related
t=0 #our start time, usually zero
tend=5.#the time we want to end the simulation
h=0.05 #size of time step

print "Expected Number of Output Files is: ", (tend-t)/h

i=0 #loop counter 
save_path = "data/onedheatdiff" #the folder to put our outputs in, leave blank "" for script path - note this folder path must exist to work
needdirs([save_path])

#... generate domain ...
rod = Rectangle(l0=0.05,l1=.01,n0=500, n1=1)
# extract finite points
x=rod.getX()
#... open PDE ...
mypde=LinearPDE(rod)
mypde.setSymmetryOn()
mypde.setValue(A=kappa*kronecker(rod),D=rho/h,d=eta,y=eta*Tref)
# ... set heat source: ....

qH=q*whereZero(x[0])
# ... set initial temperature ....
T=Tref

# ... start iteration:
while t<=tend:
      i+=1
      t+=h
      mypde.setValue(Y=qH+rho/h*T)
      T=mypde.getSolution()
      print T
      saveVTK(os.path.join(save_path,"data%03d.vtu") %i,sol=T)
      
#~ command = ('mencoder',
           #~ 'mf://*.png',
           #~ '-mf',
           #~ 'type=png:w=800:h=600:fps=25',
           #~ '-ovc',
           #~ 'lavc',
           #~ '-lavcopts',
           #~ 'vcodec=mpeg4',
           #~ '-oac',
           #~ 'copy',
           #~ '-o',
           #~ 'output.avi')

1
2	########################################################
3	#
4	# Copyright (c) 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) 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	"""
23	Author: Antony Hallam antony.hallam@uqconnect.edu.au
24	"""
25
26	# To solve the problem it is necessary to import the modules we require.
27	from esys.escript import * # This imports everything from the escript library
28	from esys.escript.linearPDEs import LinearPDE # This defines LinearPDE as LinearPDE
29	from esys.finley import Rectangle # This imports the rectangle domain function from finley
30	import os #This package is necessary to handle saving our data.
31	#from cblib import needdirs
32	from cblib1 import needdirs
33
34	##ESTABLISHING VARIABLES
35	#PDE related
36	q=50.e6 #our heat source temperature
37	Tref=0. #the starting temperature of our iron bar
38	rho=2.6e6
39	eta=0#75.
40	kappa=240.
41	#Script/Iteration Related
42	t=0 #our start time, usually zero
43	tend=5.#the time we want to end the simulation
44	h=0.05 #size of time step
45
46	print "Expected Number of Output Files is: ", (tend-t)/h
47
48	i=0 #loop counter
49	save_path = "data/onedheatdiff" #the folder to put our outputs in, leave blank "" for script path - note this folder path must exist to work
50	needdirs([save_path])
51
52	#... generate domain ...
53	rod = Rectangle(l0=0.05,l1=.01,n0=500, n1=1)
54	# extract finite points
55	x=rod.getX()
56	#... open PDE ...
57	mypde=LinearPDE(rod)
58	mypde.setSymmetryOn()
59	mypde.setValue(A=kappakronecker(rod),D=rho/h,d=eta,y=etaTref)
60	# ... set heat source: ....
61
62	qH=q*whereZero(x[0])
63	# ... set initial temperature ....
64	T=Tref
65
66	# ... start iteration:
67	while t<=tend:
68	i+=1
69	t+=h
70	mypde.setValue(Y=qH+rho/h*T)
71	T=mypde.getSolution()
72	print T
73	saveVTK(os.path.join(save_path,"data%03d.vtu") %i,sol=T)
74
75	#~ command = ('mencoder',
76	#~ 'mf://*.png',
77	#~ '-mf',
78	#~ 'type=png:w=800:h=600:fps=25',
79	#~ '-ovc',
80	#~ 'lavc',
81	#~ '-lavcopts',
82	#~ 'vcodec=mpeg4',
83	#~ '-oac',
84	#~ 'copy',
85	#~ '-o',
86	#~ 'output.avi')
87