test/python/convection.py

# $Id$

#
#
#    a very simple convection model in model frame:
#

from escript.modelframe import Link,Simulation
from modellib.geometry import RectangularDomain,ScalarConstrainer,VectorConstrainer
from modellib.input import Sequencer,InterpolateOverBox,GaussianProfile,LinearCombination
from modellib.flow import SteadyIncompressibleFlow
from modellib.temperature import TemperatureAdvection
from modellib.materials import SimpleEarthModel,GravityForce
from modellib.visualization import WriteVTK

# from esys.modelframe.geometry import RectangularDomain,ScalarConstrainer,VectorConstrainer
# from esys.modelframe.input import Sequencer,InterpolateOverBox
# from esys.modelframe.flow import SteadyIncompressibleFlow
# from esys.modelframe.temperature import TemperatureAdvection
# from esys.modelframe.materials import SimpleEarthModel,GravityForce

dom=RectangularDomain()
dom.order=2

temp_constraints=ScalarConstrainer()
temp_constraints.domain=Link(dom)
temp_constraints.top=1
temp_constraints.bottom=1

vel_constraints=VectorConstrainer()
vel_constraints.domain=Link(dom)
vel_constraints.left=[1,0,0]
vel_constraints.right=[1,0,0]
vel_constraints.top=[0,1,0]
vel_constraints.bottom=[0,1,0]
vel_constraints.front=[0,0,1]
vel_constraints.back=[0,0,1]


temp_val=InterpolateOverBox()
temp_val.domain=Link(dom,"domain")
temp_val.right_top_back=Link(dom,"l")
temp_val.value_left_bottom_front=1.
temp_val.value_right_bottom_front=1.
temp_val.value_left_top_front=0.
temp_val.value_right_top_front=0.
temp_val.value_left_bottom_back=1.
temp_val.value_right_bottom_back=1.
temp_val.value_left_top_back=0.
temp_val.value_right_top_back=0.

mat=SimpleEarthModel()
mat.density0=1.
mat.viscocity0=1.
mat.rayleigh_number=10000.
mat.alpha=0.001

temp=TemperatureAdvection(debug=True)
temp.domain=Link(dom)
temp.density=Link(mat,"density0")
temp.heat_capacity=Link(mat,"heat_capacity")
temp.location_fixed_temperature=Link(temp_constraints,"location_of_constraint")
temp.fixed_temperature=Link(temp_val,"out")
temp.safety_factor=0.01
mat.temperature=Link(temp,"temperature")


grav=GravityForce()
grav.domain=Link(dom,"domain")
grav.direction=[0.,-1.,0.]
grav.density=Link(mat,"density")
grav.gravity=Link(mat,"gravity")


vel=SteadyIncompressibleFlow(debug=True)
vel.domain=Link(dom)
vel.internal_force=Link(grav,"gravity_force")
vel.viscosity=Link(mat,"viscosity")
vel.location_prescribed_velocity=Link(vel_constraints,"location_of_constraint")
vel.rel_tol=1.e-6
temp.velocity=Link(vel,"velocity")

sq=Sequencer()
sq.t_end=0.005

vis=WriteVTK()
vis.t=Link(sq)
vis.scalar=Link(temp,"temperature")
vis.vector=Link(vel,"velocity")
vis.stride=5

per=GaussianProfile()
per.domain=Link(dom)
per.x_c=[0.5,0.5,0.5]
per.A=0.0001
per.width=0.01
per.r=0

lc=LinearCombination()
lc.f0=1.
lc.v0=Link(per,"out")
lc.f1=1.
lc.v1=Link(temp_val,"out")
temp.temperature=Link(lc,"out")

# s=Simulation([dom,sq,temp_constraints,vel_constraints,temp_val,per,lc,mat,grav,Simulation([vel],debug=True),temp,vis],debug=True)
s=Simulation([dom,sq,Simulation([vel],debug=True),temp,vis],debug=True)
s.writeXML()
s.run()
1	jgs	147	# $Id$
2
3			#
4			#
5			# a very simple convection model in model frame:
6			#
7
8			from escript.modelframe import Link,Simulation
9			from modellib.geometry import RectangularDomain,ScalarConstrainer,VectorConstrainer
10	jgs	148	from modellib.input import Sequencer,InterpolateOverBox,GaussianProfile,LinearCombination
11	jgs	147	from modellib.flow import SteadyIncompressibleFlow
12			from modellib.temperature import TemperatureAdvection
13			from modellib.materials import SimpleEarthModel,GravityForce
14			from modellib.visualization import WriteVTK
15
16			# from esys.modelframe.geometry import RectangularDomain,ScalarConstrainer,VectorConstrainer
17			# from esys.modelframe.input import Sequencer,InterpolateOverBox
18			# from esys.modelframe.flow import SteadyIncompressibleFlow
19			# from esys.modelframe.temperature import TemperatureAdvection
20			# from esys.modelframe.materials import SimpleEarthModel,GravityForce
21
22			dom=RectangularDomain()
23			dom.order=2
24
25			temp_constraints=ScalarConstrainer()
26			temp_constraints.domain=Link(dom)
27			temp_constraints.top=1
28			temp_constraints.bottom=1
29
30			vel_constraints=VectorConstrainer()
31			vel_constraints.domain=Link(dom)
32			vel_constraints.left=[1,0,0]
33			vel_constraints.right=[1,0,0]
34			vel_constraints.top=[0,1,0]
35			vel_constraints.bottom=[0,1,0]
36			vel_constraints.front=[0,0,1]
37			vel_constraints.back=[0,0,1]
38
39
40			temp_val=InterpolateOverBox()
41			temp_val.domain=Link(dom,"domain")
42			temp_val.right_top_back=Link(dom,"l")
43			temp_val.value_left_bottom_front=1.
44			temp_val.value_right_bottom_front=1.
45			temp_val.value_left_top_front=0.
46			temp_val.value_right_top_front=0.
47			temp_val.value_left_bottom_back=1.
48			temp_val.value_right_bottom_back=1.
49			temp_val.value_left_top_back=0.
50			temp_val.value_right_top_back=0.
51
52			mat=SimpleEarthModel()
53			mat.density0=1.
54			mat.viscocity0=1.
55			mat.rayleigh_number=10000.
56			mat.alpha=0.001
57
58			temp=TemperatureAdvection(debug=True)
59			temp.domain=Link(dom)
60			temp.density=Link(mat,"density0")
61			temp.heat_capacity=Link(mat,"heat_capacity")
62			temp.location_fixed_temperature=Link(temp_constraints,"location_of_constraint")
63			temp.fixed_temperature=Link(temp_val,"out")
64			temp.safety_factor=0.01
65			mat.temperature=Link(temp,"temperature")
66
67
68			grav=GravityForce()
69			grav.domain=Link(dom,"domain")
70			grav.direction=[0.,-1.,0.]
71			grav.density=Link(mat,"density")
72			grav.gravity=Link(mat,"gravity")
73
74
75			vel=SteadyIncompressibleFlow(debug=True)
76			vel.domain=Link(dom)
77			vel.internal_force=Link(grav,"gravity_force")
78			vel.viscosity=Link(mat,"viscosity")
79			vel.location_prescribed_velocity=Link(vel_constraints,"location_of_constraint")
80			vel.rel_tol=1.e-6
81			temp.velocity=Link(vel,"velocity")
82
83			sq=Sequencer()
84			sq.t_end=0.005
85
86			vis=WriteVTK()
87			vis.t=Link(sq)
88			vis.scalar=Link(temp,"temperature")
89			vis.vector=Link(vel,"velocity")
90			vis.stride=5
91
92	jgs	148	per=GaussianProfile()
93	jgs	147	per.domain=Link(dom)
94			per.x_c=[0.5,0.5,0.5]
95			per.A=0.0001
96			per.width=0.01
97			per.r=0
98
99			lc=LinearCombination()
100			lc.f0=1.
101			lc.v0=Link(per,"out")
102			lc.f1=1.
103			lc.v1=Link(temp_val,"out")
104			temp.temperature=Link(lc,"out")
105
106	jgs	148	# s=Simulation([dom,sq,temp_constraints,vel_constraints,temp_val,per,lc,mat,grav,Simulation([vel],debug=True),temp,vis],debug=True)
107			s=Simulation([dom,sq,Simulation([vel],debug=True),temp,vis],debug=True)
108	jgs	147	s.writeXML()
109			s.run()
Name	Value
svn:eol-style	native
svn:keywords	Author Date Id Revision