src/GMSHPlugin/GMSHPluginBuilder.py

   1 # Copyright (C) 2012-2015  ALNEOS
   2 # Copyright (C) 2016-2020  EDF R&D
   3 #
   4 # This library is free software; you can redistribute it and/or
   5 # modify it under the terms of the GNU Lesser General Public
   6 # License as published by the Free Software Foundation; either
   7 # version 2.1 of the License, or (at your option) any later version.
   8 #
   9 # This library is distributed in the hope that it will be useful,
  10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12 # Lesser General Public License for more details.
  13 #
  14 # You should have received a copy of the GNU Lesser General Public
  15 # License along with this library; if not, write to the Free Software
  16 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  17 #
  18 # See http://www.alneos.com/ or email : contact@alneos.fr
  19 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  20 #
  21
  22 ##
  23 # @package GMSHPluginBuilder
  24 # Python API for the GMSH meshing plug-in module.
  25
  26 from salome.smesh.smesh_algorithm import Mesh_Algorithm
  27
  28 # import GMSHPlugin module if possible
  29 noGMSHPlugin = 0
  30 try:
  31     import GMSHPlugin
  32 except ImportError:
  33     noGMSHPlugin = 1
  34     pass
  35
  36 # Types of algorithms
  37 GMSH     = "GMSH"
  38 GMSH_2D  = "GMSH_2D"
  39
  40 ## Base of all GMSH algorithms.
  41 #
  42 class GMSH_Algorithm(Mesh_Algorithm):
  43
  44     meshMethod = "Tetrahedron"
  45     algoType   = GMSH
  46
  47     def __init__(self, mesh, geom=0):
  48         Mesh_Algorithm.__init__(self)
  49         if noGMSHPlugin: print("Warning: GMSHPlugin module unavailable")
  50         self.Create(mesh, geom, self.algoType, "libGMSHEngine.so")
  51         self.params = None
  52
  53     ## Defines hypothesis having several parameters
  54     #
  55     def Parameters(self):
  56         if self.algoType == GMSH_2D:
  57             hypType = "GMSH_Parameters_2D"
  58         elif self.algoType == GMSH:
  59             hypType = "GMSH_Parameters"
  60
  61         if self.params and self.params.GetName() != hypType:
  62             self.mesh.RemoveHypothesis( self.params, self.geom )
  63             self.params = None
  64         if not self.params:
  65             self.params = self.Hypothesis(hypType, [],"libGMSHEngine.so",UseExisting=0)
  66         return self.params
  67
  68 class GMSH_2D_Algorithm(GMSH_Algorithm):
  69
  70     meshMethod = "Triangle"
  71     algoType   = GMSH_2D
  72
  73     ## Private constructor.
  74     def __init__(self, mesh, geom=0):
  75         GMSH_Algorithm.__init__(self, mesh, geom)