src/SMESH_SWIG/SMESH_AdvancedEditor.py

   1 #! /usr/bin/python
   2 #  -*- coding: iso-8859-1 -*-
   3 #  Copyright (C) 2007-2008  CEA/DEN, EDF R&D, OPEN CASCADE
   4 #
   5 #  Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
   6 #  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
   7 #
   8 #  This library is free software; you can redistribute it and/or
   9 #  modify it under the terms of the GNU Lesser General Public
  10 #  License as published by the Free Software Foundation; either
  11 #  version 2.1 of the License.
  12 #
  13 #  This library is distributed in the hope that it will be useful,
  14 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
  15 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  16 #  Lesser General Public License for more details.
  17 #
  18 #  You should have received a copy of the GNU Lesser General Public
  19 #  License along with this library; if not, write to the Free Software
  20 #  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  21 #
  22 #  See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  23 #
  24 import smesh
  25 import math
  26
  27 def GetNewNodes(mesh,Elems,OldNodes):
  28     """
  29     Auxilary function, which return list of nodes from
  30     given Elems avoided nodes from OldNodes
  31     """
  32     newnodes = []
  33     for i in Elems:
  34         nbn = mesh.GetElemNbNodes(i)
  35         for j in range(0,nbn):
  36             nn = mesh.GetElemNode(i,j)
  37             isold = 0
  38             for k in range(0,len(newnodes)):
  39                 if nn==newnodes[k]:
  40                     isold = 1
  41                     break
  42                 pass
  43             if isold: continue
  44             for k in range(0,len(OldNodes)):
  45                 if nn==OldNodes[k]:
  46                     isold = 1
  47                     break
  48                 pass
  49             if isold: continue
  50             newnodes.append(nn)
  51             pass
  52         pass
  53     return newnodes
  54
  55
  56 # create empty mesh
  57 mesh = smesh.Mesh()
  58
  59 tol = 0.001
  60
  61 # create a cross from quadrangle faces
  62 # 1. create first edge and make extrusion along 0x
  63 n1 = mesh.AddNode(55,-5,0)
  64 n2 = mesh.AddNode(55,5,0)
  65 e1 = mesh.AddEdge([n1,n2])
  66 dir1 = smesh.DirStruct(smesh.PointStruct(-10,0,0))
  67 mesh.ExtrusionSweep([e1],dir1,11)
  68 # 2. create second edge and make extrusion along 0y
  69 n3 = mesh.AddNode(-5,-55,0)
  70 n4 = mesh.AddNode(5,-55,0)
  71 e2 = mesh.AddEdge([n3,n4])
  72 dir2 = smesh.DirStruct(smesh.PointStruct(0,10,0))
  73 mesh.ExtrusionSweep([e2],dir2,11)
  74
  75 # since result has coincident nodes and faces
  76 # we have to make merge
  77 nodes = mesh.FindCoincidentNodes(0.001)
  78 mesh.MergeNodes(nodes)
  79 mesh.MergeEqualElements()
  80
  81 # make extrusion faces along 0z
  82 faces = mesh.GetElementsByType(smesh.FACE)
  83 nbf = len(faces)
  84 maxang = 2.0
  85 zstep = 5
  86 nbzsteps = 50
  87 dir3 = smesh.DirStruct(smesh.PointStruct(0,0,zstep))
  88 newfaces = [] # list for keeping created top faces
  89               # during extrusion
  90
  91 for i in range(0,nbzsteps):
  92     mesh.ExtrusionSweep(faces,dir3,1)
  93     # find top faces after each extrusion and keep them
  94     res = mesh.GetLastCreatedElems()
  95     nbr = len(res)
  96     nfaces = []
  97     for j in res:
  98         nbn = mesh.GetElemNbNodes(j)
  99         if nbn!=4: continue
 100         nn1 = mesh.GetElemNode(j,0)
 101         xyz1 = mesh.GetNodeXYZ(nn1)
 102         nn2 = mesh.GetElemNode(j,1)
 103         xyz2 = mesh.GetNodeXYZ(nn2)
 104         nn3 = mesh.GetElemNode(j,2)
 105         xyz3 = mesh.GetNodeXYZ(nn3)
 106         if abs(xyz1[2]-xyz2[2])<tol and abs(xyz1[2]-xyz3[2])<tol :
 107             # this face is a top face
 108             nfaces.append(j)
 109             pass
 110         pass
 111     if len(nfaces)!=nbf:
 112         print "len(nfaces)!=nbf"
 113         break
 114     newfaces.append(nfaces)
 115     # update faces for before next step of extrusion
 116     faces = nfaces
 117     pass
 118
 119 # rotate faces from newfaces
 120 axisr1 = smesh.AxisStruct(0,0,0,0,0,1)
 121 for i in range(0,nbzsteps):
 122     ang = maxang*(1-math.cos((i+1)*math.pi/nbzsteps))
 123     mesh.Rotate(newfaces[i],axisr1,ang,0)
 124
 125
 126 # create circles
 127 # create two edges and rotate them for creation
 128 # full circle
 129 n5 = mesh.AddNode(65,0,0)
 130 n6 = mesh.AddNode(67.5,0,0)
 131 n7 = mesh.AddNode(70,0,0)
 132 e56 = mesh.AddEdge([n5,n6])
 133 e67 = mesh.AddEdge([n6,n7])
 134 axisr2 = smesh.AxisStruct(65,0,0,0,1,0)
 135 mesh.RotationSweep([e56,e67],axisr2, math.pi/6, 12, tol)
 136 res = mesh.GetLastCreatedElems()
 137 faces1 = []
 138 for i in res:
 139     nbn = mesh.GetElemNbNodes(i)
 140     if nbn>2: faces1.append(i)
 141     pass
 142 nbf1 = len(faces1)
 143
 144 # create other two edges and rotate them for creation
 145 # other full circle
 146 n8 = mesh.AddNode(-65,0,0)
 147 n9 = mesh.AddNode(-67.5,0,0)
 148 n10 = mesh.AddNode(-70,0,0)
 149 e8 = mesh.AddEdge([n8,n9])
 150 e9 = mesh.AddEdge([n9,n10])
 151 axisr3 = smesh.AxisStruct(-65,0,0,0,-1,0)
 152 mesh.RotationSweep([e8,e9],axisr3, math.pi/6, 12, tol)
 153 res = mesh.GetLastCreatedElems()
 154 faces2 = []
 155 for i in res:
 156     nbn = mesh.GetElemNbNodes(i)
 157     if nbn>2: faces2.append(i)
 158     pass
 159 nbf2 = len(faces2)
 160
 161 # there are coincident nodes after rotation
 162 # therefore we have to merge nodes
 163 nodes = mesh.FindCoincidentNodes(0.001)
 164 mesh.MergeNodes(nodes)
 165
 166 nbcircs = 2
 167 nbrsteps = 24
 168 nbrs = nbcircs*nbrsteps
 169 dz = nbzsteps*zstep/nbrs
 170
 171 # create first spiral
 172 oldnodes = []
 173 newnodes = GetNewNodes(mesh,faces1,oldnodes)
 174 oldnodes = newnodes
 175
 176 nodes = []
 177 mesh.RotationSweep(faces1,axisr1, math.pi*2/nbrsteps, nbrs, tol)
 178 res = mesh.GetLastCreatedElems()
 179
 180 for i in range(0,nbrs):
 181     volumes = []
 182     for j in range(0,nbf1): volumes.append(res[i+j*nbrs])
 183     newnodes = GetNewNodes(mesh,volumes,oldnodes)
 184     for j in newnodes:
 185         xyz = mesh.GetNodeXYZ(j)
 186         mesh.MoveNode(j,xyz[0],xyz[1],xyz[2]+dz*(i+1))
 187         pass
 188     oldnodes = newnodes
 189     pass
 190
 191 # create second spiral
 192 oldnodes = []
 193 newnodes = GetNewNodes(mesh,faces2,oldnodes)
 194 oldnodes = newnodes
 195
 196 nodes = []
 197 mesh.RotationSweep(faces2,axisr1, math.pi*2/nbrsteps, nbrs, tol)
 198 res = mesh.GetLastCreatedElems()
 199
 200 for i in range(0,nbrs):
 201     volumes = []
 202     for j in range(0,nbf2): volumes.append(res[i+j*nbrs])
 203     newnodes = GetNewNodes(mesh,volumes,oldnodes)
 204     for j in newnodes:
 205         xyz = mesh.GetNodeXYZ(j)
 206         mesh.MoveNode(j,xyz[0],xyz[1],xyz[2]+dz*(i+1))
 207         pass
 208     oldnodes = newnodes
 209     pass
 210
 211 smesh.salome.sg.updateObjBrowser(1)