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

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

Parent Directory Parent Directory | Revision Log Revision Log


Revision 2648 - (show annotations)
Mon Sep 7 00:06:15 2009 UTC (11 years, 2 months ago) by jfenwick
File MIME type: text/x-python
File size: 3499 byte(s)
ecording some changes related to Anthony's cookbook.
These tests still do not run.

test_heatref.py is the current problem.
Current problems include:
undefined symbols
a dependence on X somehow.

I've already addressed the matplotlib X thing somewhere else.
Find and apply.


1
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 from esys.pycad import * #domain constructor
27 from esys.pycad.gmsh import Design #Finite Element meshing package
28 from esys.finley import MakeDomain #Converter for escript
29 import os #file path tool
30 import numpy as np #numerial python for arrays
31 from math import * # math package
32 #used to construct polygons for plotting from pycad
33 from cblib import getLoopCoords, needdirs
34
35 #set modal to 1 for a syncline or -1 for an anticline structural
36 #configuration
37 modal=1
38
39 # the folder to put our outputs in, leave blank "" for script path -
40 # note this folder path must exist to work
41 save_path = "data/heatrefrac001"
42 needdirs([save_path])
43
44 #Model Parameters
45 width=5000.0 #width of model
46 depth=-6000.0 #depth of model
47 sspl=51 #number of discrete points in spline
48 dsp=width/(sspl-1) #dx of spline steps for width
49 dep_sp=2500.0 #avg depth of spline
50 h_sp=1500.0 #heigh of spline
51 orit=-1.0 #orientation of spline 1.0=>up -1.0=>down
52
53 # Overall Domain
54 p0=Point(0.0, 0.0, 0.0)
55 p1=Point(0.0, depth, 0.0)
56 p2=Point(width, depth, 0.0)
57 p3=Point(width, 0.0, 0.0)
58
59 l01=Line(p0, p1)
60 l12=Line(p1, p2)
61 l23=Line(p2, p3)
62 l30=Line(p3, p0)
63
64 c=CurveLoop(l01, l12, l23, l30)
65
66 # Material Boundary
67 x=[ Point(i*dsp\
68 ,-dep_sp+modal*orit*h_sp*cos(pi*i*dsp/dep_sp+pi))\
69 for i in range(0,sspl)\
70 ]
71
72 mysp = Spline(*tuple(x))
73 x1=Spline.getStartPoint(mysp)
74 x2=Spline.getEndPoint(mysp)
75
76 # TOP BLOCK
77 tbl1=Line(p0,x1)
78 tbl2=mysp
79 tbl3=Line(x2,p3)
80 tblockloop = CurveLoop(tbl1,tbl2,tbl3,l30)
81 tblock = PlaneSurface(tblockloop)
82
83 tpg = getLoopCoords(tblockloop)
84 np.savetxt(os.path.join(save_path,"toppg"),tpg,delimiter=" ")
85
86 # BOTTOM BLOCK
87 bbl1=Line(x1,p1)
88 bbl3=Line(p2,x2)
89 bbl4=-mysp
90 bblockloop = CurveLoop(bbl1,l12,bbl3,bbl4)
91 bblock = PlaneSurface(bblockloop)
92
93 #clockwise check as splines must be set as polygons in the point order
94 #they were created. Otherwise get a line across plot.
95 bblockloop2=CurveLoop(mysp,Line(x2,p2),Line(p2,p1),Line(p1,x1))
96 bpg = getLoopCoords(bblockloop2)
97 np.savetxt(os.path.join(save_path,"botpg"),bpg,delimiter=" ")
98
99 # Create a Design which can make the mesh
100 d=Design(dim=2, element_size=200)
101 # Add the subdomains and flux boundaries.
102 d.addItems(PropertySet("top",tblock),PropertySet("bottom",bblock),PropertySet("linebottom",l12))
103 # Create the geometry, mesh and Escript domain
104 d.setScriptFileName(os.path.join(save_path,"heatrefraction_mesh001.geo"))
105
106 d.setMeshFileName(os.path.join(save_path,"heatrefraction_mesh001.msh"))
107 domain=MakeDomain(d, integrationOrder=-1, reducedIntegrationOrder=-1, optimizeLabeling=True)
108 # Create a file that can be read back in to python with mesh=ReadMesh(fileName)
109 domain.write(os.path.join(save_path,"heatrefraction_mesh001.fly"))
110
111

  ViewVC Help
Powered by ViewVC 1.1.26