test/SMESH_test1_AndDisplay.py

   1 #  -*- coding: iso-8859-1 -*-
   2 # Copyright (C) 2007-2024  CEA, EDF, OPEN CASCADE
   3 #
   4 # Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
   5 # CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
   6 #
   7 # This library is free software; you can redistribute it and/or
   8 # modify it under the terms of the GNU Lesser General Public
   9 # License as published by the Free Software Foundation; either
  10 # version 2.1 of the License, or (at your option) any later version.
  11 #
  12 # This library is distributed in the hope that it will be useful,
  13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15 # Lesser General Public License for more details.
  16 #
  17 # You should have received a copy of the GNU Lesser General Public
  18 # License along with this library; if not, write to the Free Software
  19 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  20 #
  21 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  22 #
  23
  24 #  File   : SMESH_test1.py
  25 #  Module : SMESH
  26 #
  27 import salome
  28 salome.salome_init()
  29 import GEOM
  30 from salome.geom import geomBuilder
  31 geompy = geomBuilder.New()
  32
  33 import SMESH, SALOMEDS
  34 from salome.smesh import smeshBuilder
  35 smesh =  smeshBuilder.New()
  36
  37 # ---- define a box
  38
  39 box = geompy.MakeBox(0., 0., 0., 100., 200., 300.)
  40 idbox = geompy.addToStudy(box, "box")
  41
  42 # ---- add first face of box in study
  43
  44 subShapeList = geompy.SubShapeAll(box, geompy.ShapeType["FACE"])
  45 face = subShapeList[0]
  46 name = geompy.SubShapeName(face, box)
  47 print(name)
  48 idface = geompy.addToStudyInFather(box, face, name)
  49
  50 # ---- add shell from box  in study
  51
  52 subShellList = geompy.SubShapeAll(box, geompy.ShapeType["SHELL"])
  53 shell = subShellList[0]
  54 name = geompy.SubShapeName(shell, box)
  55 print(name)
  56 idshell = geompy.addToStudyInFather(box, shell, name)
  57
  58 # ---- add first edge of face in study
  59
  60 edgeList = geompy.SubShapeAll(face, geompy.ShapeType["EDGE"])
  61 edge = edgeList[0]
  62 name = geompy.SubShapeName(edge, face)
  63 print(name)
  64 idedge = geompy.addToStudyInFather(face, edge, name)
  65
  66
  67 # ---- SMESH
  68
  69 # ---- Init a Mesh with the box
  70
  71 mesh = smesh.Mesh(box, "Meshbox")
  72
  73 print("-------------------------- add hypothesis to box")
  74 algoReg1 = mesh.Segment()
  75 hypNbSeg1 = algoReg1.NumberOfSegments(7)
  76 print(hypNbSeg1.GetName())
  77 print(hypNbSeg1.GetId())
  78 print(hypNbSeg1.GetNumberOfSegments())
  79 smesh.SetName(hypNbSeg1, "NumberOfSegments_7")
  80
  81 algoMef1 = mesh.Triangle()
  82 hypArea1 = algoMef1.MaxElementArea(2500)
  83 print(hypArea1.GetName())
  84 print(hypArea1.GetId())
  85 print(hypArea1.GetMaxElementArea())
  86 smesh.SetName(hypArea1, "MaxElementArea_2500")
  87
  88 # ---- add hypothesis to edge
  89 print("-------------------------- add hypothesis to edge")
  90 edge = salome.IDToObject(idedge)
  91
  92 algoReg2 = mesh.Segment(edge)
  93 hypLen1 = algoReg2.LocalLength(100)
  94 smesh.SetName(algoReg2.GetSubMesh(), "SubMeshEdge")
  95 print(hypLen1.GetName())
  96 print(hypLen1.GetId())
  97 print(hypLen1.GetLength())
  98 smesh.SetName(hypLen1, "Local_Length_100")
  99
 100 # ---- add hypothesis to face
 101 print("-------------------------- add hypothesis to face")
 102 face = salome.IDToObject(idface)
 103
 104 algoMef2 = mesh.Triangle(face)
 105 hypArea2 = algoMef2.MaxElementArea(500)
 106 smesh.SetName(algoMef2.GetSubMesh(), "SubMeshFace")
 107 print(hypArea2.GetName())
 108 print(hypArea2.GetId())
 109 print(hypArea2.GetMaxElementArea())
 110 smesh.SetName(hypArea2, "MaxElementArea_500")
 111
 112 isDone = mesh.Compute()
 113 if not isDone:
 114     raise Exception("Error when computing Mesh")
 115
 116 salome.sg.updateObjBrowser()
 117
 118 sg = salome.ImportComponentGUI('SMESH')
 119 if not isinstance(sg, type(salome.salome_ComponentGUI)):
 120     sg.CreateAndDisplayActor('0:1:2:3')