pycad/py_src/gmsh.py

# $Id:$

"""
mesh generation using gmsh

@var __author__: name of author
@var __copyright__: copyrights
@var __license__: licence agreement
@var __url__: url entry point on documentation
@var __version__: version
@var __date__: date of the version
"""


__author__="Lutz Gross, l.gross@uq.edu.au"
__copyright__="""  Copyright (c) 2007 by ACcESS MNRF
                    http://www.access.edu.au
                Primary Business: Queensland, Australia"""
__license__="""Licensed under the Open Software License version 3.0
             http://www.opensource.org/licenses/osl-3.0.php"""
__url__="http://www.iservo.edu.au/esys/escript"
__version__="$Revision:$"
__date__="$Date:$"

import design
import tempfile
import os

class Design(design.Design):
    """
    design fo gmsh
    """
    ISO="iso"
    TRI="tri"
    ANISO="aniso"
    NETGEN="netgen"
    TETGEN="tetgen"
    def __init__(self,dim=3,element_size=1.,order=1,keep_files=False):
       """
       initializes the gmsh design

       @param dim: patial dimension
       @element_size: global element size
       @order: element order
       @keep_files: flag to keep work files.
       """ 
       design.Design.__init__(self,dim=dim,element_size=element_size,order=order,keep_files=keep_files)
       self.setScriptFileName()
       self.setMeshFileName()
       self.setOptions()
    def setScriptFileName(self,name=None):
       """
       set the filename for the gmsh input script. if no name is given a name with extension geo is generated.
       """
       if name == None:
           self.__scriptname=tempfile.mkstemp(suffix=".geo")[1]
       else:
           self.__scriptname=name
           self.setKeepFilesOn()
    def getScriptFileName(self):
       """
       returns the name of the file for the gmsh script
       """
       return self.__scriptname
    def setMeshFileName(self, name=None):
       """
       sets the name for the gmsh mesh file. if no name is given a name with extension msh is generated.
       """
       if name == None:
           self.__mshname=tempfile.mkstemp(suffix=".msh")[1]
       else:
           self.__mshname=name
           self.setKeepFilesOn()
    def getMeshFileName(self):
       """
       returns the name of the file for the gmsh msh
       """
       return self.__mshname
    def setOptions(self,algorithm=None,optimize_quality=True,smoothing=3):
        """
        sets options for the mesh generator
        """ 
        if algorithm==None: algorithm=self.ISO
        self.__algo=algorithm
        self.__optimize_quality=optimize_quality
        self.__smoothing=smoothing
    def __del__(self):
        """
        clean up
        """
        if not self.keepFiles():
               os.unlink(self.getScriptFileName())
               os.unlink(self.getMeshFileName())
    def getScriptString(self):
        """
        returns the gmsh script to generate the mesh
        """
        prim=self.getAllPrimitives()
        out="// generated by esys.pycad\n"
        for p in prim:
           out+=p.getGmshCommand(self.getElementSize())+"\n"
        return out
    def getCommandString(self):
        """
        returns the gmsh comand
        """
        if self.__optimize_quality:
              opt="-optimize "
        else:
              opt=""
        exe="gmsh -%s -algo %s -smooth %s %s -v 0 -order %s -o %s %s"%(self.getDim(),
                                                                       self.__algo,
                                                                       self.__smoothing,
                                                                       opt,
                                                                       self.getElementOrder(),
                                                                       self.getMeshFileName(),
                                                                       self.getScriptFileName())
        return exe
    def getMeshHandler(self):
        """
        returns a handle to a mesh meshing the design. In the current implementation 
        a mesh file name in gmsh format is returned.
        """
        open(self.getScriptFileName(),"w").write(self.getScriptString())
        os.system(self.getCommandString())
        return self.getMeshFileName()
1	gross	932	# $Id:$
2
3			"""
4			mesh generation using gmsh
5
6			@var __author__: name of author
7			@var __copyright__: copyrights
8			@var __license__: licence agreement
9			@var __url__: url entry point on documentation
10			@var __version__: version
11			@var __date__: date of the version
12			"""
13
14
15			__author__="Lutz Gross, l.gross@uq.edu.au"
16			__copyright__=""" Copyright (c) 2007 by ACcESS MNRF
17			http://www.access.edu.au
18			Primary Business: Queensland, Australia"""
19			__license__="""Licensed under the Open Software License version 3.0
20			http://www.opensource.org/licenses/osl-3.0.php"""
21			__url__="http://www.iservo.edu.au/esys/escript"
22			__version__="$Revision:$"
23			__date__="$Date:$"
24
25			import design
26			import tempfile
27			import os
28
29			class Design(design.Design):
30			"""
31			design fo gmsh
32			"""
33			ISO="iso"
34			TRI="tri"
35			ANISO="aniso"
36			NETGEN="netgen"
37			TETGEN="tetgen"
38			def __init__(self,dim=3,element_size=1.,order=1,keep_files=False):
39			"""
40			initializes the gmsh design
41
42			@param dim: patial dimension
43			@element_size: global element size
44			@order: element order
45			@keep_files: flag to keep work files.
46			"""
47			design.Design.__init__(self,dim=dim,element_size=element_size,order=order,keep_files=keep_files)
48			self.setScriptFileName()
49			self.setMeshFileName()
50			self.setOptions()
51			def setScriptFileName(self,name=None):
52			"""
53			set the filename for the gmsh input script. if no name is given a name with extension geo is generated.
54			"""
55			if name == None:
56			self.__scriptname=tempfile.mkstemp(suffix=".geo")[1]
57			else:
58			self.__scriptname=name
59			self.setKeepFilesOn()
60			def getScriptFileName(self):
61			"""
62			returns the name of the file for the gmsh script
63			"""
64			return self.__scriptname
65			def setMeshFileName(self, name=None):
66			"""
67			sets the name for the gmsh mesh file. if no name is given a name with extension msh is generated.
68			"""
69			if name == None:
70			self.__mshname=tempfile.mkstemp(suffix=".msh")[1]
71			else:
72			self.__mshname=name
73			self.setKeepFilesOn()
74			def getMeshFileName(self):
75			"""
76			returns the name of the file for the gmsh msh
77			"""
78			return self.__mshname
79			def setOptions(self,algorithm=None,optimize_quality=True,smoothing=3):
80			"""
81			sets options for the mesh generator
82			"""
83			if algorithm==None: algorithm=self.ISO
84			self.__algo=algorithm
85			self.__optimize_quality=optimize_quality
86			self.__smoothing=smoothing
87			def __del__(self):
88			"""
89			clean up
90			"""
91			if not self.keepFiles():
92			os.unlink(self.getScriptFileName())
93			os.unlink(self.getMeshFileName())
94			def getScriptString(self):
95			"""
96			returns the gmsh script to generate the mesh
97			"""
98			prim=self.getAllPrimitives()
99			out="// generated by esys.pycad\n"
100			for p in prim:
101			out+=p.getGmshCommand(self.getElementSize())+"\n"
102			return out
103			def getCommandString(self):
104			"""
105			returns the gmsh comand
106			"""
107			if self.__optimize_quality:
108			opt="-optimize "
109			else:
110			opt=""
111			exe="gmsh -%s -algo %s -smooth %s %s -v 0 -order %s -o %s %s"%(self.getDim(),
112			self.__algo,
113			self.__smoothing,
114			opt,
115			self.getElementOrder(),
116			self.getMeshFileName(),
117			self.getScriptFileName())
118			return exe
119			def getMeshHandler(self):
120			"""
121			returns a handle to a mesh meshing the design. In the current implementation
122			a mesh file name in gmsh format is returned.
123			"""
124			open(self.getScriptFileName(),"w").write(self.getScriptString())
125			os.system(self.getCommandString())
126			return self.getMeshFileName()