doc/examples/diffusion_with_pyvisi.py

# $Id: diffusion.py 575 2006-03-03 03:33:07Z lkettle $
from esys.escript import *
from esys.escript.linearPDEs import LinearPDE
from esys.finley import Rectangle
from esys.pyvisi import Scene, DataCollector, Map
from esys.pyvisi.constant import *

#... set some parameters ...
xc=[0.02,0.002]
r=0.001
qc=50.e6
Tref=0.
rhocp=2.6e6
eta=75.
kappa=240.
tend=5.
# ... time, time step size and counter ...
t=0
h=0.1
i=0
#... generate domain ...
mydomain = Rectangle(l0=0.05,l1=0.01,n0=250, n1=50)
#... open PDE ...
mypde=LinearPDE(mydomain)
mypde.setSymmetryOn()
mypde.setValue(A=kappa*kronecker(mydomain),D=rhocp/h,d=eta,y=eta*Tref)
# ... set heat source: ....
x=mydomain.getX()
qH=qc*whereNegative(length(x-xc)-r)
# ... set initial temperature ....
T=Tref

s = Scene(renderer = Renderer.DISPLAY, x_size = 800, y_size = 600)
dc = DataCollector(source = Source.ESCRIPT)

# ... start iteration:
while t<tend:
      i+=1
      t+=h
      print "time step :",t
      mypde.setValue(Y=qH+rhocp/h*T)
      T=mypde.getSolution()
      #saveVTK("T.%d.xml"%i,temp=T)

      dc.setData(temp = T)
      Map(scene = s, data_collector = dc, viewport = Viewport.SOUTH_WEST, 
              lut = Lut.COLOR, cell_to_point = False, outline = True)

      s.render(image_name = "diffusion_%2d.jpg" % i)

1	# $Id: diffusion.py 575 2006-03-03 03:33:07Z lkettle $
2	from esys.escript import *
3	from esys.escript.linearPDEs import LinearPDE
4	from esys.finley import Rectangle
5	from esys.pyvisi import Scene, DataCollector, Map
6	from esys.pyvisi.constant import *
7
8	#... set some parameters ...
9	xc=[0.02,0.002]
10	r=0.001
11	qc=50.e6
12	Tref=0.
13	rhocp=2.6e6
14	eta=75.
15	kappa=240.
16	tend=5.
17	# ... time, time step size and counter ...
18	t=0
19	h=0.1
20	i=0
21	#... generate domain ...
22	mydomain = Rectangle(l0=0.05,l1=0.01,n0=250, n1=50)
23	#... open PDE ...
24	mypde=LinearPDE(mydomain)
25	mypde.setSymmetryOn()
26	mypde.setValue(A=kappakronecker(mydomain),D=rhocp/h,d=eta,y=etaTref)
27	# ... set heat source: ....
28	x=mydomain.getX()
29	qH=qc*whereNegative(length(x-xc)-r)
30	# ... set initial temperature ....
31	T=Tref
32
33	s = Scene(renderer = Renderer.DISPLAY, x_size = 800, y_size = 600)
34	dc = DataCollector(source = Source.ESCRIPT)
35
36	# ... start iteration:
37	while t<tend:
38	i+=1
39	t+=h
40	print "time step :",t
41	mypde.setValue(Y=qH+rhocp/h*T)
42	T=mypde.getSolution()
43	#saveVTK("T.%d.xml"%i,temp=T)
44
45	dc.setData(temp = T)
46	Map(scene = s, data_collector = dc, viewport = Viewport.SOUTH_WEST,
47	lut = Lut.COLOR, cell_to_point = False, outline = True)
48
49	s.render(image_name = "diffusion_%2d.jpg" % i)
50