/[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/examples/cookbook/onedheatdiffbase.py revision 2904 by gross, Tue Feb 2 04:21:52 2010 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) 2009 by University of Queensland  # Copyright (c) 2009 by University of Queensland
# 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 *  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
37    
38  ##ESTABLISHING VARIABLES  #################################################ESTABLISHING VARIABLES
39  #Domain related.  #Domain related.
40  mx = 500*m #meters - model length  mx = 500*m #meters - model length
41  my = 100*m #meters - model width  my = 100*m #meters - model width
# Line 37  ndx = 100 # mesh steps in x direction Line 43  ndx = 100 # mesh steps in x direction
43  ndy = 1 # mesh steps in y direction - one dimension means one element  ndy = 1 # mesh steps in y direction - one dimension means one element
44  boundloc = mx/2 # location of boundary between the two blocks  boundloc = mx/2 # location of boundary between the two blocks
45  #PDE related  #PDE related
46  rho = 7874. *kg/m**3 #kg/m^{3} density of iron  rho = 2750. *kg/m**3 #kg/m{3} density of iron
47  cp = 449.*J/(kg*K) # J/Kg.K thermal capacity  cp = 790.*J/(kg*K) # J/Kg.K thermal capacity
48  rhocp = rho*cp  rhocp = rho*cp
49  kappa = 80.*W/m/K   # watts/m.Kthermal conductivity  kappa = 2.2*W/m/K # watts/m.Kthermal conductivity
50  qH=0 * J/(sec*m**3) # J/(sec.m^{3}) no heat source  qH=0 * J/(sec*m**3) # J/(sec.m{3}) no heat source
51  T1=20 * Celsius # initial temperature at Block 1  T1=20 * Celsius # initial temperature at Block 1
52  T2=2273. * Celsius # initial temperature at Block 2  T2=2273. * Celsius # base temperature at Block 2
53    
54    
55    ################################################ESTABLISHING PARAMETERS
56  t=0 * day  # our start time, usually zero  t=0 * day  # our start time, usually zero
57  tend=50 * yr # - time to end simulation  tend=50 * yr # - time to end simulation
58  outputs = 200 # number of time steps required.  outputs = 200 # number of time steps required.
59  h=(tend-t)/outputs #size of time step  h=(tend-t)/outputs #size of time step
60  #user warning statement  #user warning statement
61  print "Expected Number of time outputs is: ", (tend-t)/h  print("Expected Number of time outputs is: ", (tend-t)/h)
62  i=0 #loop counter  i=0 #loop counter
63  #the folder to put our outputs in, leave blank "" for script path  #the folder to put our outputs in, leave blank "" for script path
64  save_path= os.path.join("data","onedheatdiff001")  save_path= os.path.join("data","example01")
65  #ensure the dir exists  #ensure the dir exists
66  mkDir(save_path, os.path.join(save_path,"tempT"))  mkDir(save_path, os.path.join(save_path,"tempT"))
67    
68  #... generate domain ...  ####################################################DOMAIN CONSTRUCTION
69  blocks = Rectangle(l0=mx,l1=my,n0=ndx, n1=ndy)  blocks = Rectangle(l0=mx,l1=my,n0=ndx, n1=ndy)
70    
71    ###############################################ESCRIPT PDE CONSTRUCTION
72  #... open PDE and set coefficients ...  #... open PDE and set coefficients ...
73  mypde=LinearPDE(blocks)  mypde=LinearPDE(blocks)
74  mypde.setSymmetryOn()  mypde.setSymmetryOn()
# Line 69  mypde.setValue(A=A,D=rhocp/h) Line 79  mypde.setValue(A=A,D=rhocp/h)
79  x=Solution(blocks).getX()  x=Solution(blocks).getX()
80  T= T1*whereNegative(x[0]-boundloc)+T2*(1-whereNegative(x[0]-boundloc))  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+rhocp/h*T)        mypde.setValue(Y=qH+rhocp/h*T)
87        T=mypde.getSolution()        T=mypde.getSolution()
88        totE=integrate(rhocp*T)        totE=integrate(rhocp*T)
89        print "time step %s at t=%e days completed. total energy = %e."%(i,t/day,totE)        print("time step %s at t=%e days completed. total energy = %e."%(i,t/day,totE))

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

  ViewVC Help
Powered by ViewVC 1.1.26