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

Annotation of /trunk/doc/examples/cookbook/heatrefraction_mesher001.py

Parent Directory Parent Directory | Revision Log Revision Log


Revision 2658 - (hide annotations)
Thu Sep 10 02:58:44 2009 UTC (9 years, 8 months ago) by ahallam
File MIME type: text/x-python
File size: 4377 byte(s)
Updates to all files scripts to support MPI testing proceedure. Updates to cookbook, new section on functino spaces/domains (needs work). Finalising first 3 chapters for editing.
1 ahallam 2588
2     ########################################################
3     #
4     # Copyright (c) 2003-2009 by University of Queensland
5     # Earth Systems Science Computational Center (ESSCC)
6     # http://www.uq.edu.au/esscc
7     #
8     # Primary Business: Queensland, Australia
9     # Licensed under the Open Software License version 3.0
10     # http://www.opensource.org/licenses/osl-3.0.php
11     #
12     ########################################################
13    
14     __copyright__="""Copyright (c) 2003-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     http://www.opensource.org/licenses/osl-3.0.php"""
20     __url__="https://launchpad.net/escript-finley"
21    
22     """
23     Author: Antony Hallam antony.hallam@uqconnect.edu.au
24     """
25    
26 ahallam 2658 ############################################################FILE HEADER
27     # heatrefraction_mesher001.py
28     # Create either a 2D syncline or anticline model using pycad meshing
29     # tools.
30    
31     #######################################################EXTERNAL MODULES
32 ahallam 2597 from esys.pycad import * #domain constructor
33     from esys.pycad.gmsh import Design #Finite Element meshing package
34     from esys.finley import MakeDomain #Converter for escript
35     import os #file path tool
36     import numpy as np #numerial python for arrays
37     from math import * # math package
38     #used to construct polygons for plotting from pycad
39 jfenwick 2648 from cblib import getLoopCoords, needdirs
40 ahallam 2588
41 ahallam 2658 #################################################ESTABLISHING VARIABLES
42 ahallam 2597 #set modal to 1 for a syncline or -1 for an anticline structural
43     #configuration
44     modal=1
45 ahallam 2588
46 ahallam 2597 # the folder to put our outputs in, leave blank "" for script path -
47     # note this folder path must exist to work
48 ahallam 2658 save_path= os.path.join("data","heatrefrac001")
49 jfenwick 2648 needdirs([save_path])
50 ahallam 2597
51 ahallam 2658 ################################################ESTABLISHING PARAMETERS
52 ahallam 2634 #Model Parameters
53     width=5000.0 #width of model
54     depth=-6000.0 #depth of model
55     sspl=51 #number of discrete points in spline
56     dsp=width/(sspl-1) #dx of spline steps for width
57     dep_sp=2500.0 #avg depth of spline
58     h_sp=1500.0 #heigh of spline
59     orit=-1.0 #orientation of spline 1.0=>up -1.0=>down
60    
61 ahallam 2658 ####################################################DOMAIN CONSTRUCTION
62     # Domain Corners
63 ahallam 2634 p0=Point(0.0, 0.0, 0.0)
64     p1=Point(0.0, depth, 0.0)
65     p2=Point(width, depth, 0.0)
66     p3=Point(width, 0.0, 0.0)
67 ahallam 2658 # Join corners in anti-clockwise manner.
68 ahallam 2588 l01=Line(p0, p1)
69     l12=Line(p1, p2)
70     l23=Line(p2, p3)
71     l30=Line(p3, p0)
72 ahallam 2658 # Join line segments to create domain boundary.
73 ahallam 2588 c=CurveLoop(l01, l12, l23, l30)
74 ahallam 2658 # Generate Material Boundary
75 ahallam 2634 x=[ Point(i*dsp\
76     ,-dep_sp+modal*orit*h_sp*cos(pi*i*dsp/dep_sp+pi))\
77     for i in range(0,sspl)\
78     ]
79 ahallam 2588 mysp = Spline(*tuple(x))
80 ahallam 2658 # Start and end of material boundary.
81 ahallam 2588 x1=Spline.getStartPoint(mysp)
82     x2=Spline.getEndPoint(mysp)
83    
84 ahallam 2658 # Create TOP BLOCK
85     # lines
86 ahallam 2588 tbl1=Line(p0,x1)
87     tbl2=mysp
88     tbl3=Line(x2,p3)
89 ahallam 2658 # curve
90 ahallam 2588 tblockloop = CurveLoop(tbl1,tbl2,tbl3,l30)
91 ahallam 2658 # surface
92 ahallam 2588 tblock = PlaneSurface(tblockloop)
93 ahallam 2658 # polygon for python plotting
94 ahallam 2597 tpg = getLoopCoords(tblockloop)
95     np.savetxt(os.path.join(save_path,"toppg"),tpg,delimiter=" ")
96 ahallam 2588
97 ahallam 2658 # Create BOTTOM BLOCK
98     # lines
99 ahallam 2588 bbl1=Line(x1,p1)
100     bbl3=Line(p2,x2)
101     bbl4=-mysp
102 ahallam 2658 # curve
103 ahallam 2588 bblockloop = CurveLoop(bbl1,l12,bbl3,bbl4)
104 ahallam 2658 # surface
105 ahallam 2588 bblock = PlaneSurface(bblockloop)
106    
107 ahallam 2597 #clockwise check as splines must be set as polygons in the point order
108     #they were created. Otherwise get a line across plot.
109 ahallam 2588 bblockloop2=CurveLoop(mysp,Line(x2,p2),Line(p2,p1),Line(p1,x1))
110 ahallam 2597 bpg = getLoopCoords(bblockloop2)
111     np.savetxt(os.path.join(save_path,"botpg"),bpg,delimiter=" ")
112 ahallam 2588
113 ahallam 2658 #############################################EXPORTING MESH FOR ESCRIPT
114 ahallam 2588 # Create a Design which can make the mesh
115     d=Design(dim=2, element_size=200)
116 ahallam 2597 # Add the subdomains and flux boundaries.
117 ahallam 2658 d.addItems(PropertySet("top",tblock),PropertySet("bottom",bblock),\
118     PropertySet("linebottom",l12))
119 ahallam 2588 # Create the geometry, mesh and Escript domain
120 ahallam 2658 d.setScriptFileName(os.path.join(save_path,\
121     "heatrefraction_mesh001.geo"))
122 ahallam 2588
123 ahallam 2597 d.setMeshFileName(os.path.join(save_path,"heatrefraction_mesh001.msh"))
124 ahallam 2658 domain=MakeDomain(d, integrationOrder=-1, reducedIntegrationOrder=-1,\
125     optimizeLabeling=True)
126     # Create a file that can be read back in to python with
127     # mesh=ReadMesh(fileName)
128 ahallam 2597 domain.write(os.path.join(save_path,"heatrefraction_mesh001.fly"))
129 ahallam 2588
130    

  ViewVC Help
Powered by ViewVC 1.1.26