doc/salome/examples/modifying_meshes_ex23.py

   1 # Extrusion along a Path
   2
   3 import math
   4
   5 import salome
   6 salome.salome_init()
   7 import GEOM
   8 from salome.geom import geomBuilder
   9 geompy = geomBuilder.New()
  10
  11 import SMESH, SALOMEDS
  12 from salome.smesh import smeshBuilder
  13 smesh =  smeshBuilder.New()
  14
  15 # 1. Create points
  16 points = [[0, 0], [50, 30], [50, 110], [0, 150], [-80, 150], [-130, 70], [-130, -20]]
  17
  18 iv = 1
  19 vertices = []
  20 for point in points:
  21     vert = geompy.MakeVertex(point[0], point[1], 0)
  22     #geompy.addToStudy(vert, "Vertex_" + repr(iv))
  23     vertices.append(vert)
  24     iv += 1
  25     pass
  26
  27 # 2. Create edges and wires
  28 Edge_straight = geompy.MakeEdge(vertices[0], vertices[4])
  29 Edge_bezierrr = geompy.MakeBezier(vertices)
  30 Wire_polyline = geompy.MakePolyline(vertices)
  31 Edge_Circle   = geompy.MakeCircleThreePnt(vertices[0], vertices[1], vertices[2])
  32
  33 geompy.addToStudy(Edge_straight, "Edge_straight")
  34 geompy.addToStudy(Edge_bezierrr, "Edge_bezierrr")
  35 geompy.addToStudy(Wire_polyline, "Wire_polyline")
  36 geompy.addToStudy(Edge_Circle  , "Edge_Circle")
  37
  38 # 3. Explode wire on edges, as they will be used for mesh extrusion
  39 Wire_polyline_edges = geompy.SubShapeAll(Wire_polyline, geompy.ShapeType["EDGE"])
  40 for ii in range(len(Wire_polyline_edges)):
  41     geompy.addToStudyInFather(Wire_polyline, Wire_polyline_edges[ii], "Edge_" + repr(ii + 1))
  42     pass
  43
  44 # Mesh
  45
  46 # Mesh the given shape with the given 1d hypothesis
  47 def Mesh1D(shape1d, nbSeg, name, smesh_builder):
  48   mesh1d_tool = smesh_builder.Mesh(shape1d, name)
  49   algo = mesh1d_tool.Segment()
  50   hyp  = algo.NumberOfSegments(nbSeg)
  51   isDone = mesh1d_tool.Compute()
  52   if not isDone: print('Mesh ', name, ': computation failed')
  53   return mesh1d_tool
  54
  55 # Create a mesh with six nodes, seven edges and two quadrangle faces
  56 def MakeQuadMesh2(mesh_name, smesh_builder):
  57   quad_1 = smesh_builder.Mesh(name = mesh_name)
  58
  59   # six nodes
  60   n1 = quad_1.AddNode(0, 20, 10)
  61   n2 = quad_1.AddNode(0, 40, 10)
  62   n3 = quad_1.AddNode(0, 40, 30)
  63   n4 = quad_1.AddNode(0, 20, 30)
  64   n5 = quad_1.AddNode(0,  0, 30)
  65   n6 = quad_1.AddNode(0,  0, 10)
  66
  67   # seven edges
  68   quad_1.AddEdge([n1, n2]) # 1
  69   quad_1.AddEdge([n2, n3]) # 2
  70   quad_1.AddEdge([n3, n4]) # 3
  71   quad_1.AddEdge([n4, n1]) # 4
  72   quad_1.AddEdge([n4, n5]) # 5
  73   quad_1.AddEdge([n5, n6]) # 6
  74   quad_1.AddEdge([n6, n1]) # 7
  75
  76   # two quadrangle faces
  77   quad_1.AddFace([n1, n2, n3, n4]) # 8
  78   quad_1.AddFace([n1, n4, n5, n6]) # 9
  79   return [quad_1, [1,2,3,4,5,6,7], [8,9]]
  80
  81 # Path meshes
  82 Edge_straight_mesh = Mesh1D(Edge_straight, 7, "Edge_straight", smesh_builder=smesh)
  83 Edge_bezierrr_mesh = Mesh1D(Edge_bezierrr, 7, "Edge_bezierrr", smesh_builder=smesh)
  84 Wire_polyline_mesh = Mesh1D(Wire_polyline, 3, "Wire_polyline", smesh_builder=smesh)
  85 Edge_Circle_mesh   = Mesh1D(Edge_Circle  , 8, "Edge_Circle"  , smesh_builder=smesh)
  86
  87 # Initial meshes (to be extruded)
  88 [quad_1, ee_1, ff_1] = MakeQuadMesh2("quad_1", smesh_builder=smesh)
  89 [quad_2, ee_2, ff_2] = MakeQuadMesh2("quad_2", smesh_builder=smesh)
  90 [quad_3, ee_3, ff_3] = MakeQuadMesh2("quad_3", smesh_builder=smesh)
  91 [quad_4, ee_4, ff_4] = MakeQuadMesh2("quad_4", smesh_builder=smesh)
  92 [quad_5, ee_5, ff_5] = MakeQuadMesh2("quad_5", smesh_builder=smesh)
  93 [quad_6, ee_6, ff_6] = MakeQuadMesh2("quad_6", smesh_builder=smesh)
  94 [quad_7, ee_7, ff_7] = MakeQuadMesh2("quad_7", smesh_builder=smesh)
  95
  96 # ExtrusionAlongPath
  97 # IDsOfElements, PathMesh, PathShape, NodeStart,
  98 # HasAngles, Angles, HasRefPoint, RefPoint
  99 refPoint = SMESH.PointStruct(0, 0, 0)
 100 a10 = math.radians( 10.0 )
 101 a45 = math.radians( 45.0 )
 102
 103 # 1. Extrusion of two mesh edges along a straight path
 104 error = quad_1.ExtrusionAlongPath([1,2], Edge_straight_mesh, Edge_straight, 1,
 105                                   0, [], 0, refPoint)
 106
 107 # 2. Extrusion of one mesh edge along a curved path
 108 error = quad_2.ExtrusionAlongPath([2], Edge_bezierrr_mesh, Edge_bezierrr, 1,
 109                                   0, [], 0, refPoint)
 110
 111 # 3. Extrusion of one mesh edge along a curved path with usage of angles
 112 error = quad_3.ExtrusionAlongPath([2], Edge_bezierrr_mesh, Edge_bezierrr, 1,
 113                                   1, [a45, a45, a45, 0, -a45, -a45, -a45], 0, refPoint)
 114
 115 # 4. Extrusion of one mesh edge along the path, which is a part of a meshed wire
 116 error = quad_4.ExtrusionAlongPath([4], Wire_polyline_mesh, Wire_polyline_edges[0], 1,
 117                                   1, [a10, a10, a10], 0, refPoint)
 118
 119 # 5. Extrusion of two mesh faces along the path, which is a part of a meshed wire
 120 error = quad_5.ExtrusionAlongPath(ff_5 , Wire_polyline_mesh, Wire_polyline_edges[2], 4,
 121                                   0, [], 0, refPoint)
 122
 123 # 6. Extrusion of two mesh faces along a closed path
 124 error = quad_6.ExtrusionAlongPath(ff_6 , Edge_Circle_mesh, Edge_Circle, 1,
 125                                   0, [], 0, refPoint)
 126
 127 # 7. Extrusion of two mesh faces along a closed path with usage of angles
 128 error = quad_7.ExtrusionAlongPath(ff_7, Edge_Circle_mesh, Edge_Circle, 1,
 129                                   1, [a45, -a45, a45, -a45, a45, -a45, a45, -a45], 0, refPoint)
 130
 131
 132
 133 # Make the same meshes using a fully functional method ExtrusionAlongPathObjects() having
 134 # the following arguments:
 135 #   Nodes, Edges, Faces, PathObject, PathShape=None,
 136 #   NodeStart=1, HasAngles=False, Angles=[], LinearVariation=False,
 137 #   HasRefPoint=False, RefPoint=[0,0,0], MakeGroups=False,
 138 #   ScaleFactors=[], ScalesVariation=False
 139
 140 quad_1 = MakeQuadMesh2("quad_1", smesh_builder=smesh)[0]
 141 quad_2 = MakeQuadMesh2("quad_2", smesh_builder=smesh)[0]
 142 quad_3 = MakeQuadMesh2("quad_3", smesh_builder=smesh)[0]
 143 quad_4 = MakeQuadMesh2("quad_4", smesh_builder=smesh)[0]
 144 quad_5 = MakeQuadMesh2("quad_5", smesh_builder=smesh)[0]
 145 quad_6 = MakeQuadMesh2("quad_6", smesh_builder=smesh)[0]
 146 quad_7 = MakeQuadMesh2("quad_7", smesh_builder=smesh)[0]
 147
 148 # 1. Extrusion of two mesh edges along a straight path
 149 nn, ee, ff = [], [1,2], []
 150 error = quad_1.ExtrusionAlongPathObjects( nn, ee, ff, Edge_straight_mesh )
 151
 152 # 2. Extrusion of one mesh edge along a curved path
 153 nn, ee, ff = [], [2], []
 154 error = quad_2.ExtrusionAlongPathObjects( nn, ee, ff, Edge_bezierrr_mesh )
 155
 156 # 3. Extrusion of one mesh edge along a curved path with usage of angles
 157 error = quad_3.ExtrusionAlongPathObjects( nn, ee, ff, Edge_bezierrr_mesh,
 158                                           Angles=[a45, a45, a45, 0, -a45, -a45, -a45])
 159
 160 # 4. Extrusion of one mesh edge along the path, which is a part of a meshed wire
 161 nn, ee, ff = [], [4], []
 162 error = quad_4.ExtrusionAlongPathObjects( nn, ee, ff, Wire_polyline_mesh, Wire_polyline_edges[0],
 163                                           Angles=[a10, a10, a10])
 164
 165 # 5. Extrusion of two mesh faces along the path, which is a part of a meshed wire
 166 nn, ee, ff = [], [], quad_5
 167 error = quad_5.ExtrusionAlongPathObjects( nn, ee, ff, Wire_polyline_mesh, Wire_polyline_edges[2],
 168                                           NodeStart=4 )
 169
 170 # 6. Extrusion of two mesh faces along a closed path
 171 nn, ee, ff = [], [], quad_6
 172 error = quad_6.ExtrusionAlongPathObjects( nn, ee, ff, Edge_Circle_mesh )
 173
 174 # 7. Extrusion of two mesh faces along a closed path with usage of angles
 175 nn, ee, ff = [], [], quad_7
 176 error = quad_7.ExtrusionAlongPathObjects( nn, ee, ff, Edge_Circle_mesh, Edge_Circle,
 177                                           Angles=[a45, -a45, a45, -a45, a45, -a45, a45, -a45])
 178
 179
 180 salome.sg.updateObjBrowser()