examples/pyvisi/diffusion_with_pyvisi.py

# Import the necessary modules.
from esys.escript import *
from esys.escript.linearPDEs import LinearPDE
from esys.finley import Rectangle
from esys.pyvisi import Scene, DataCollector, Map, Camera
from esys.pyvisi.constant import *

PYVISI_EXAMPLE_IMAGES_PATH = "data_sample_images/"
X_SIZE = 400
Y_SIZE = 400
JPG_RENDERER = Renderer.DISPLAY

#... 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

# Create a Scene.
s = Scene(renderer = JPG_RENDERER, x_size = X_SIZE, y_size = Y_SIZE)
# Create a DataCollector reading directly from escript objects.
dc = DataCollector(source = Source.ESCRIPT)

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

# ... start iteration:
while t<0.4*100:
      i+=1
      t+=h
      print "time step :",t
      mypde.setValue(Y=qH+rhocp/h*T)
      T=mypde.getSolution()

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

      # Create a Camera.
      #c= Camera(scene = s, data_collector = dc, 
      #        viewport = Viewport.SOUTH_WEST)
      
      # Render the object.
      s.render(image_name = PYVISI_EXAMPLE_IMAGES_PATH +  
              "diffusion%02d.jpg" % i)

1	ksteube	1147	# Import the necessary modules.
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, Camera
6			from esys.pyvisi.constant import *
7
8			PYVISI_EXAMPLE_IMAGES_PATH = "data_sample_images/"
9			X_SIZE = 400
10			Y_SIZE = 400
11			JPG_RENDERER = Renderer.DISPLAY
12
13			#... set some parameters ...
14			xc=[0.02,0.002]
15			r=0.001
16			qc=50.e6
17			Tref=0.
18			rhocp=2.6e6
19			eta=75.
20			kappa=240.
21			tend=5.
22			# ... time, time step size and counter ...
23			t=0
24			h=0.1
25			i=0
26			#... generate domain ...
27			mydomain = Rectangle(l0=0.05,l1=0.01,n0=250, n1=50)
28			#... open PDE ...
29			mypde=LinearPDE(mydomain)
30			mypde.setSymmetryOn()
31			mypde.setValue(A=kappakronecker(mydomain),D=rhocp/h,d=eta,y=etaTref)
32			# ... set heat source: ....
33			x=mydomain.getX()
34			qH=qc*whereNegative(length(x-xc)-r)
35			# ... set initial temperature ....
36			T=Tref
37
38			# Create a Scene.
39			s = Scene(renderer = JPG_RENDERER, x_size = X_SIZE, y_size = Y_SIZE)
40			# Create a DataCollector reading directly from escript objects.
41			dc = DataCollector(source = Source.ESCRIPT)
42
43			Map(scene = s, data_collector = dc,
44			viewport = Viewport.SOUTH_WEST, lut = Lut.COLOR,
45			cell_to_point = False, outline = True)
46
47			# ... start iteration:
48			while t<0.4*100:
49			i+=1
50			t+=h
51			print "time step :",t
52			mypde.setValue(Y=qH+rhocp/h*T)
53			T=mypde.getSolution()
54
55			dc.setData(temp = T)
56			# Create a Map.
57			#Map(scene = s, data_collector = dc,
58			#viewport = Viewport.SOUTH_WEST, lut = Lut.COLOR,
59			#cell_to_point = False, outline = True)
60
61			# Create a Camera.
62			#c= Camera(scene = s, data_collector = dc,
63			# viewport = Viewport.SOUTH_WEST)
64
65			# Render the object.
66			s.render(image_name = PYVISI_EXAMPLE_IMAGES_PATH +
67			"diffusion%02d.jpg" % i)
68