/[escript]/trunk/doc/examples/cookbook/example01a.py
ViewVC logotype

Diff of /trunk/doc/examples/cookbook/example01a.py

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

trunk/doc/cookbook/onedheatdiff.py revision 2597 by ahallam, Thu Aug 6 03:30:09 2009 UTC trunk/doc/examples/cookbook/example01a.py revision 3892 by jfenwick, Tue Apr 10 08:57:23 2012 UTC
# Line 1  Line 1 
1    
2    from __future__ import print_function
3  ########################################################  ########################################################
4  #  #
5  # Copyright (c) 2003-2009 by University of Queensland  # Copyright (c) 2009 by University of Queensland
6  # Earth Systems Science Computational Center (ESSCC)  # Earth Systems Science Computational Center (ESSCC)
7  # http://www.uq.edu.au/esscc  # http://www.uq.edu.au/esscc
8  #  #
# Line 11  Line 12 
12  #  #
13  ########################################################  ########################################################
14    
15  __copyright__="""Copyright (c) 2003-2009 by University of Queensland  __copyright__="""Copyright (c) 2009 by University of Queensland
16  Earth Systems Science Computational Center (ESSCC)  Earth Systems Science Computational Center (ESSCC)
17  http://www.uq.edu.au/esscc  http://www.uq.edu.au/esscc
18  Primary Business: Queensland, Australia"""  Primary Business: Queensland, Australia"""
# Line 22  __url__="https://launchpad.net/escript-f Line 23  __url__="https://launchpad.net/escript-f
23  """  """
24  Author: Antony Hallam antony.hallam@uqconnect.edu.au  Author: Antony Hallam antony.hallam@uqconnect.edu.au
25  """  """
26    ############################################################FILE HEADER
27    # example01a.py
28    # Model temperature diffusion between two granite blocks of unequal
29    # initial temperature. Solve for total energy in the system.
30    
31    #######################################################EXTERNAL MODULES
32  # To solve the problem it is necessary to import the modules we require.  # To solve the problem it is necessary to import the modules we require.
33  from esys.escript import * # This imports everything from the escript library  from esys.escript import * # This imports everything from the escript library
34    from esys.escript.unitsSI import *
35  from esys.escript.linearPDEs import LinearPDE # This defines LinearPDE as LinearPDE  from esys.escript.linearPDEs import LinearPDE # This defines LinearPDE as LinearPDE
36  from esys.finley import Rectangle # This imports the rectangle domain function from finley  from esys.finley import Rectangle # This imports the rectangle domain function
 import os #This package is necessary to handle saving our data.  
37    
38    #################################################ESTABLISHING VARIABLES
39    #Domain related.
40  ##ESTABLISHING VARIABLES  mx = 500*m #meters - model length
41    my = 100*m #meters - model width
42    ndx = 100 # mesh steps in x direction
43    ndy = 1 # mesh steps in y direction - one dimension means one element
44    boundloc = mx/2 # location of boundary between the two blocks
45  #PDE related  #PDE related
46  q=50.e6 #our heat source temperature  rho = 2750. *kg/m**3 #kg/m{3} density of iron
47  Tref=0. #the starting temperature of our iron bar  cp = 790.*J/(kg*K) # J/Kg.K thermal capacity
48  rho=2.6e6  rhocp = rho*cp
49  eta=0#75.  kappa = 2.2*W/m/K # watts/m.Kthermal conductivity
50  kappa=240.  qH=0 * J/(sec*m**3) # J/(sec.m{3}) no heat source
51  #Script/Iteration Related  T1=20 * Celsius # initial temperature at Block 1
52  t=0 #our start time, usually zero  T2=2273. * Celsius # base temperature at Block 2
53  tend=5.#the time we want to end the simulation  
54  h=0.05 #size of time step  
55    ################################################ESTABLISHING PARAMETERS
56  print "Expected Number of Output Files is: ", (tend-t)/h  t=0 * day  # our start time, usually zero
57    tend=50 * yr # - time to end simulation
58  i=0 #loop counter  outputs = 200 # number of time steps required.
59  save_path = "data/onedheatdiff" #the folder to put our outputs in, leave blank "" for script path - note this folder path must exist to work  h=(tend-t)/outputs #size of time step
60    #user warning statement
61  #... generate domain ...  print("Expected Number of time outputs is: ", (tend-t)/h)
62  rod = Rectangle(l0=0.05,l1=.01,n0=500, n1=1)  i=0 #loop counter
63  # extract finite points  #the folder to put our outputs in, leave blank "" for script path
64  x=rod.getX()  save_path= os.path.join("data","example01")
65  #... open PDE ...  #ensure the dir exists
66  mypde=LinearPDE(rod)  mkDir(save_path, os.path.join(save_path,"tempT"))
67    
68    ####################################################DOMAIN CONSTRUCTION
69    blocks = Rectangle(l0=mx,l1=my,n0=ndx, n1=ndy)
70    
71    ###############################################ESCRIPT PDE CONSTRUCTION
72    #... open PDE and set coefficients ...
73    mypde=LinearPDE(blocks)
74  mypde.setSymmetryOn()  mypde.setSymmetryOn()
75  mypde.setValue(A=kappa*kronecker(rod),D=rho/h,d=eta,y=eta*Tref)  A=zeros((2,2))
76  # ... set heat source: ....  A[0,0]=kappa
77    mypde.setValue(A=A,D=rhocp/h)
 qH=q*whereZero(x[0])  
78  # ... set initial temperature ....  # ... set initial temperature ....
79  T=Tref  x=Solution(blocks).getX()
80    T= T1*whereNegative(x[0]-boundloc)+T2*(1-whereNegative(x[0]-boundloc))
81    
82  # ... start iteration:  ########################################################START ITERATION
83  while t<=tend:  while t<tend:
84        i+=1        i+=1
85        t+=h        t+=h
86        mypde.setValue(Y=qH+rho/h*T)        mypde.setValue(Y=qH+rhocp/h*T)
87        T=mypde.getSolution()        T=mypde.getSolution()
88        print T        totE=integrate(rhocp*T)
89        saveVTK(os.path.join(save_path,"data%03d.vtu") %i,sol=T)        print("time step %s at t=%e days completed. total energy = %e."%(i,t/day,totE))
         
 #~ command = ('mencoder',  
            #~ 'mf://*.png',  
            #~ '-mf',  
            #~ 'type=png:w=800:h=600:fps=25',  
            #~ '-ovc',  
            #~ 'lavc',  
            #~ '-lavcopts',  
            #~ 'vcodec=mpeg4',  
            #~ '-oac',  
            #~ 'copy',  
            #~ '-o',  
            #~ 'output.avi')  
   

Legend:
Removed from v.2597  
changed lines
  Added in v.3892

  ViewVC Help
Powered by ViewVC 1.1.26