# Contents of /trunk/doc/examples/cookbook/onedheatdiff.py

Revision 2697 - (show annotations)
Wed Sep 30 01:29:22 2009 UTC (11 years, 7 months ago) by jfenwick
File MIME type: text/x-python
File size: 2636 byte(s)
```Trying to get cookbook scripts passing.
Split cblib in two to avoid importing pylab in scripts which
don't otheriwse need it.

```
 1 2 ######################################################## 3 # 4 # Copyright (c) 2009 by University of Queensland 5 # Earth Systems Science Computational Center (ESSCC) 6 7 # 8 # Primary Business: Queensland, Australia 9 # Licensed under the Open Software License version 3.0 10 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 20 __url__= 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=kappa*kronecker(rod),D=rho/h,d=eta,y=eta*Tref) 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