doc/salome/examples/modifying_meshes_ex23.py

   1 # Extrusion along a Path
   2
   3 import math
   4 import salome
   5
   6 # Geometry
   7 import geompy
   8
   9 # 1. Create points
  10 points = [[0, 0], [50, 30], [50, 110], [0, 150], [-80, 150], [-130, 70], [-130, -20]]
  11
  12 iv = 1
  13 vertices = []
  14 for point in points:
  15     vert = geompy.MakeVertex(point[0], point[1], 0)
  16     geompy.addToStudy(vert, "Vertex_" + `iv`)
  17     vertices.append(vert)
  18     iv += 1
  19     pass
  20
  21 # 2. Create edges and wires
  22 Edge_straight = geompy.MakeEdge(vertices[0], vertices[4])
  23 Edge_bezierrr = geompy.MakeBezier(vertices)
  24 Wire_polyline = geompy.MakePolyline(vertices)
  25 Edge_Circle   = geompy.MakeCircleThreePnt(vertices[0], vertices[1], vertices[2])
  26
  27 geompy.addToStudy(Edge_straight, "Edge_straight")
  28 geompy.addToStudy(Edge_bezierrr, "Edge_bezierrr")
  29 geompy.addToStudy(Wire_polyline, "Wire_polyline")
  30 geompy.addToStudy(Edge_Circle  , "Edge_Circle")
  31
  32 # 3. Explode wire on edges, as they will be used for mesh extrusion
  33 Wire_polyline_edges = geompy.SubShapeAll(Wire_polyline, geompy.ShapeType["EDGE"])
  34 for ii in range(len(Wire_polyline_edges)):
  35     geompy.addToStudyInFather(Wire_polyline, Wire_polyline_edges[ii], "Edge_" + `ii + 1`)
  36     pass
  37
  38 # Mesh
  39 import smesh
  40
  41 # Mesh the given shape with the given 1d hypothesis
  42 def Mesh1D(shape1d, nbSeg, name):
  43   mesh1d_tool = smesh.Mesh(shape1d, name)
  44   algo = mesh1d_tool.Segment()
  45   hyp  = algo.NumberOfSegments(nbSeg)
  46   isDone = mesh1d_tool.Compute()
  47   if not isDone: print 'Mesh ', name, ': computation failed'
  48   return mesh1d_tool
  49
  50 # Create a mesh with six nodes, seven edges and two quadrangle faces
  51 def MakeQuadMesh2(mesh_name):
  52   quad_1 = smesh.Mesh(name = mesh_name)
  53
  54   # six nodes
  55   n1 = quad_1.AddNode(0, 20, 10)
  56   n2 = quad_1.AddNode(0, 40, 10)
  57   n3 = quad_1.AddNode(0, 40, 30)
  58   n4 = quad_1.AddNode(0, 20, 30)
  59   n5 = quad_1.AddNode(0,  0, 30)
  60   n6 = quad_1.AddNode(0,  0, 10)
  61
  62   # seven edges
  63   quad_1.AddEdge([n1, n2]) # 1
  64   quad_1.AddEdge([n2, n3]) # 2
  65   quad_1.AddEdge([n3, n4]) # 3
  66   quad_1.AddEdge([n4, n1]) # 4
  67   quad_1.AddEdge([n4, n5]) # 5
  68   quad_1.AddEdge([n5, n6]) # 6
  69   quad_1.AddEdge([n6, n1]) # 7
  70
  71   # two quadrangle faces
  72   quad_1.AddFace([n1, n2, n3, n4]) # 8
  73   quad_1.AddFace([n1, n4, n5, n6]) # 9
  74   return [quad_1, [1,2,3,4,5,6,7], [8,9]]
  75
  76 # Path meshes
  77 Edge_straight_mesh = Mesh1D(Edge_straight, 7, "Edge_straight")
  78 Edge_bezierrr_mesh = Mesh1D(Edge_bezierrr, 7, "Edge_bezierrr")
  79 Wire_polyline_mesh = Mesh1D(Wire_polyline, 3, "Wire_polyline")
  80 Edge_Circle_mesh   = Mesh1D(Edge_Circle  , 8, "Edge_Circle")
  81
  82 # Initial meshes (to be extruded)
  83 [quad_1, ee_1, ff_1] = MakeQuadMesh2("quad_1")
  84 [quad_2, ee_2, ff_2] = MakeQuadMesh2("quad_2")
  85 [quad_3, ee_3, ff_3] = MakeQuadMesh2("quad_3")
  86 [quad_4, ee_4, ff_4] = MakeQuadMesh2("quad_4")
  87 [quad_5, ee_5, ff_5] = MakeQuadMesh2("quad_5")
  88 [quad_6, ee_6, ff_6] = MakeQuadMesh2("quad_6")
  89 [quad_7, ee_7, ff_7] = MakeQuadMesh2("quad_7")
  90
  91 # ExtrusionAlongPath
  92 # IDsOfElements, PathMesh, PathShape, NodeStart,
  93 # HasAngles, Angles, HasRefPoint, RefPoint
  94 refPoint = smesh.PointStruct(0, 0, 0)
  95 a10 = 10.0*math.pi/180.0
  96 a45 = 45.0*math.pi/180.0
  97
  98 # 1. Extrusion of two mesh edges along a straight path
  99 error = quad_1.ExtrusionAlongPath([1,2], Edge_straight_mesh, Edge_straight, 1,
 100                                   0, [], 0, refPoint)
 101
 102 # 2. Extrusion of one mesh edge along a curved path
 103 error = quad_2.ExtrusionAlongPath([2], Edge_bezierrr_mesh, Edge_bezierrr, 1,
 104                                   0, [], 0, refPoint)
 105
 106 # 3. Extrusion of one mesh edge along a curved path with usage of angles
 107 error = quad_3.ExtrusionAlongPath([2], Edge_bezierrr_mesh, Edge_bezierrr, 1,
 108                                   1, [a45, a45, a45, 0, -a45, -a45, -a45], 0, refPoint)
 109
 110 # 4. Extrusion of one mesh edge along the path, which is a part of a meshed wire
 111 error = quad_4.ExtrusionAlongPath([4], Wire_polyline_mesh, Wire_polyline_edges[0], 1,
 112                                   1, [a10, a10, a10], 0, refPoint)
 113
 114 # 5. Extrusion of two mesh faces along the path, which is a part of a meshed wire
 115 error = quad_5.ExtrusionAlongPath(ff_5 , Wire_polyline_mesh, Wire_polyline_edges[2], 4,
 116                                   0, [], 0, refPoint)
 117
 118 # 6. Extrusion of two mesh faces along a closed path
 119 error = quad_6.ExtrusionAlongPath(ff_6 , Edge_Circle_mesh, Edge_Circle, 1,
 120                                   0, [], 0, refPoint)
 121
 122 # 7. Extrusion of two mesh faces along a closed path with usage of angles
 123 error = quad_7.ExtrusionAlongPath(ff_7, Edge_Circle_mesh, Edge_Circle, 1,
 124                                   1, [a45, -a45, a45, -a45, a45, -a45, a45, -a45], 0, refPoint)
 125
 126 salome.sg.updateObjBrowser(1)