1 |
# $Id$ |
2 |
from esys.escript import * |
3 |
from esys.escript.linearPDEs import LinearPDE |
4 |
from esys.finley import Brick |
5 |
from esys.pyvisi import Scene, DataCollector, Map |
6 |
from esys.pyvisi.constant import * |
7 |
#... set some parameters ... |
8 |
lam=1. |
9 |
mu=0.1 |
10 |
alpha=1.e-6 |
11 |
xc=[0.3,0.3,1.] |
12 |
beta=8. |
13 |
T_ref=0. |
14 |
T_0=1. |
15 |
#... generate domain ... |
16 |
mydomain = Brick(l0=1.,l1=1., l2=1.,n0=10, n1=10, n2=10) |
17 |
x=mydomain.getX() |
18 |
#... set temperature ... |
19 |
T=T_0*exp(-beta*length(x-xc)) |
20 |
#... open symmetric PDE ... |
21 |
mypde=LinearPDE(mydomain) |
22 |
mypde.setSymmetryOn() |
23 |
#... set coefficients ... |
24 |
C=Tensor4(0.,Function(mydomain)) |
25 |
for i in range(mydomain.getDim()): |
26 |
for j in range(mydomain.getDim()): |
27 |
C[i,i,j,j]+=lam |
28 |
C[j,i,j,i]+=mu |
29 |
C[j,i,i,j]+=mu |
30 |
msk=whereZero(x[0])*[1.,0.,0.] \ |
31 |
+whereZero(x[1])*[0.,1.,0.] \ |
32 |
+whereZero(x[2])*[0.,0.,1.] |
33 |
sigma0=(lam+2./3.*mu)*alpha*(T-T_ref)*kronecker(mydomain) |
34 |
mypde.setValue(A=C,X=sigma0,q=msk) |
35 |
#... solve pde ... |
36 |
u=mypde.getSolution() |
37 |
#... calculate von-Misses |
38 |
g=grad(u) |
39 |
sigma=mu*(g+transpose(g))+lam*trace(g)*kronecker(mydomain)-sigma0 |
40 |
sigma_mises=sqrt(((sigma[0,0]-sigma[1,1])**2+(sigma[1,1]-sigma[2,2])**2+ \ |
41 |
(sigma[2,2]-sigma[0,0])**2)/6. \ |
42 |
+sigma[0,1]**2 + sigma[1,2]**2 + sigma[2,0]**2) |
43 |
#... output ... |
44 |
#saveVTK("deform.xml",disp=u,stress=sigma_mises) |
45 |
|
46 |
s = Scene(renderer = Renderer.ONLINE, x_size = 800, y_size = 600) |
47 |
dc = DataCollector(source = Source.ESCRIPT) |
48 |
|
49 |
dc.setData(disp = u,stress = sigma_mises) |
50 |
Map(scene = s, data_collector = dc, viewport = Viewport.SOUTH_WEST, |
51 |
lut = Lut.COLOR, cell_to_point = True, outline = True) |
52 |
|
53 |
s.render(image_name = "heatedblock.jpg") |