src/TEST_PY/recettes/tuyau.py

   1 # -*- coding: latin-1 -*-
   2 # Copyright (C) 2009-2013  CEA/DEN, 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.
   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.salome-platform.org/ or email : webmaster.salome@opencascade.com
  19 #
  20
  21 # Francis KLOSS : 2012 : CEA-Saclay, DEN, DM2S, SFME, LGLS, F-91191 Gif-sur-Yvette, France
  22 # ========================================================================================
  23
  24 import geompy
  25 import hexablock
  26
  27 # Charger la geometrie
  28 # ====================
  29
  30 nom = "tuyau"
  31
  32 geometrie = geompy.ImportBREP(nom+".brep")
  33
  34 # Selectionner des sous-parties de la geometrie
  35 # ---------------------------------------------
  36
  37 aretes  = geompy.SubShapeAllSortedCentres    (geometrie, geompy.ShapeType["EDGE"])
  38 edge_id = geompy.SubShapeAllSortedCentresIDs (geometrie, geompy.ShapeType["EDGE"])
  39 faces   = geompy.SubShapeAllSortedCentres    (geometrie, geompy.ShapeType["FACE"])
  40 face_id = geompy.SubShapeAllSortedCentresIDs (geometrie, geompy.ShapeType["FACE"])
  41
  42 subid_arc_int_bas  = 30
  43 subid_arc_int_haut = 34
  44 subid_arc_ext_bas  = 12
  45 subid_arc_ext_haut = 16
  46
  47 arc_int_bas  = aretes[12]
  48 arc_int_haut = aretes[13]
  49 arc_ext_bas  = aretes[11]
  50 arc_ext_haut = aretes[14]
  51
  52 subid_ell_int_bas  = 21
  53 subid_ell_int_haut = 33
  54 subid_ell_ext_bas  =  8
  55 subid_ell_ext_haut = 15
  56
  57 ell_int_bas  = aretes[2]
  58 ell_int_haut = aretes[3]
  59 ell_ext_bas  = aretes[1]
  60 ell_ext_haut = aretes[4]
  61
  62 subid_gen_int_avan = 26
  63 subid_gen_int_arri = 28
  64 subid_gen_ext_avan =  5
  65 subid_gen_ext_arri = 10
  66
  67 gen_int_avan = aretes[8]
  68 gen_int_arri = aretes[7]
  69 gen_ext_avan = aretes[9]
  70 gen_ext_arri = aretes[6]
  71
  72 subid_face_int_bas  = 24
  73 subid_face_int_haut = 31
  74 subid_face_ext_bas  =  3
  75 subid_face_ext_haut = 13
  76
  77 face_int_bas  = faces[4]
  78 face_int_haut = faces[5]
  79 face_ext_bas  = faces[2]
  80 face_ext_haut = faces[3]
  81
  82 # Construire le modele de bloc
  83 # ============================
  84
  85 doc = hexablock.addDocument(nom)
  86
  87 # Definir le tuyau
  88 # ----------------
  89
  90 origin = doc.addVertex(0, 0, 0)
  91
  92 direction = doc.addVector(1, 0, 0)
  93
  94 # Construire le modele de blocs du tuyau
  95 # --------------------------------------
  96
  97 base = doc.addVector(0, 1, 0)
  98
  99 ###  tuyau = doc.addPipe(origin, direction, 1, 2, 10)
 100 ###  modele = doc.makePipe(tuyau, base, 1, 4, 1)
 101
 102 modele = doc.makePipeUni (origin, base, direction, 1.0, 2.0, 360.0, 10.0, 1, 4, 1)
 103
 104 # Associer le modele de bloc a la geometrie
 105 # =========================================
 106
 107 sh_tuyau = hexablock.addShape (doc, geometrie,nom)
 108 doc.clearAssociation()
 109
 110 # Associer les cercles
 111 # --------------------
 112
 113 cer_int_dep = modele.getVertexIJK(0, 0, 1)
 114 cer_ext_dep = modele.getVertexIJK(1, 0, 1)
 115
 116 cer_int = []
 117 cer_ext = []
 118 for j in xrange(4):
 119     a = modele.getEdgeJ(0, j, 1)
 120     cer_int.append(a)
 121
 122     a = modele.getEdgeJ(1, j, 1)
 123     cer_ext.append(a)
 124
 125 doc.associateClosedLine(cer_int_dep, cer_int, [sh_tuyau], [34,30], 1, True)
 126 doc.associateClosedLine(cer_ext_dep, cer_ext, [sh_tuyau], [16,12], 1, True)
 127
 128 # Associer les ellipses
 129 # ---------------------
 130
 131 ell_int_dep = modele.getVertexIJK(0, 0, 0)
 132 ell_ext_dep = modele.getVertexIJK(1, 0, 0)
 133
 134 ell_int = []
 135 ell_ext = []
 136 for j in xrange(4):
 137     a = modele.getEdgeJ(0, j, 0)
 138     ell_int.append(a)
 139
 140     a = modele.getEdgeJ(1, j, 0)
 141     ell_ext.append(a)
 142
 143 doc.associateClosedLine(ell_int_dep, ell_int, [sh_tuyau], [33,21], 1, True)
 144 doc.associateClosedLine(ell_ext_dep, ell_ext, [sh_tuyau], [15, 8], 1, True)
 145
 146 # Associer les generatrices
 147 # -------------------------
 148
 149 def generatrice(face):
 150     n = 10
 151     l = []
 152     for i in xrange(0, n+1):
 153         v = float(i) / n
 154         s = geompy.MakeVertexOnSurface(face, 0.5, v)
 155         l.append(s)
 156
 157     return geompy.MakeInterpol(l)
 158
 159 gen = []
 160 gen_int_haut = generatrice(face_int_haut)
 161 gen_ext_haut = generatrice(face_ext_haut)
 162 gen_int_bas  = generatrice(face_int_bas )
 163 gen_ext_bas  = generatrice(face_ext_bas )
 164 gen.append(gen_int_haut)
 165 gen.append(gen_ext_haut)
 166 gen.append(gen_int_bas)
 167 gen.append(gen_ext_bas)
 168
 169 generatrices = geompy.MakeCompound(gen)
 170 sh_gen = hexablock.addShape (doc, generatrices, "generatrices")
 171
 172 modele.getEdgeK(0, 0, 0).addAssociation(sh_tuyau, 28, 0, 1)
 173 modele.getEdgeK(1, 0, 0).addAssociation(sh_tuyau, 10, 0, 1)
 174
 175 modele.getEdgeK(0, 1, 0).addAssociation(sh_gen  ,  2, 0, 1)
 176 modele.getEdgeK(1, 1, 0).addAssociation(sh_gen  ,  5, 0, 1)
 177
 178 modele.getEdgeK(0, 2, 0).addAssociation(sh_tuyau, 26, 0, 1)
 179 modele.getEdgeK(1, 2, 0).addAssociation(sh_tuyau,  5, 0, 1)
 180
 181 modele.getEdgeK(0, 3, 0).addAssociation(sh_gen  ,  8, 0, 1)
 182 modele.getEdgeK(1, 3, 0).addAssociation(sh_gen  , 11, 0, 1)
 183
 184 # Associer les faces courbees
 185 # ---------------------------
 186
 187 modele.getQuadJK(0, 0, 0).addAssociation(sh_tuyau, 31)
 188 modele.getQuadJK(0, 1, 0).addAssociation(sh_tuyau, 31)
 189 modele.getQuadJK(0, 2, 0).addAssociation(sh_tuyau, 24)
 190 modele.getQuadJK(0, 3, 0).addAssociation(sh_tuyau, 24)
 191
 192 modele.getQuadJK(1, 0, 0).addAssociation(sh_tuyau, 13)
 193 modele.getQuadJK(1, 1, 0).addAssociation(sh_tuyau, 13)
 194 modele.getQuadJK(1, 2, 0).addAssociation(sh_tuyau,  3)
 195 modele.getQuadJK(1, 3, 0).addAssociation(sh_tuyau,  3)
 196
 197 # Mailler le modele de bloc
 198 # =========================
 199
 200 # Definir 3 groupes d'aretes
 201 # --------------------------
 202
 203 groupe_cercles      = doc.addEdgeGroup("Cercles")
 204 groupe_ellipses     = doc.addEdgeGroup("Ellipses")
 205 groupe_generatrices = doc.addEdgeGroup("Generatrices")
 206
 207 # Definir 4 groupes de faces
 208 # --------------------------
 209
 210 groupe_couronne  = doc.addQuadGroup("Couronne")
 211 groupe_ovale     = doc.addQuadGroup("Ovale")
 212 groupe_interieur = doc.addQuadGroup("Interieur")
 213 groupe_exterieur = doc.addQuadGroup("Exterieur")
 214
 215 # Constituer les groupes d'aretes
 216 # -------------------------------
 217
 218 for i in xrange(2):
 219     for j in xrange(4):
 220         arete = modele.getEdgeJ(i, j, 1)
 221         groupe_cercles.addElement(arete)
 222
 223         arete = modele.getEdgeJ(i, j, 0)
 224         groupe_ellipses.addElement(arete)
 225
 226         arete = modele.getEdgeK(i, j, 0)
 227         groupe_generatrices.addElement(arete)
 228
 229 # Constituer les groupes de faces
 230 # -------------------------------
 231
 232 for j in xrange(4):
 233     quad = modele.getQuadIJ(0, j, 1)
 234     groupe_couronne.addElement(quad)
 235
 236     quad = modele.getQuadIJ(0, j, 0)
 237     groupe_ovale.addElement(quad)
 238
 239     quad = modele.getQuadJK(0, j, 0)
 240     groupe_interieur.addElement(quad)
 241
 242     quad = modele.getQuadJK(1, j, 0)
 243     groupe_exterieur.addElement(quad)
 244
 245 # Mailler le modele de bloc avec ses associations
 246 # -----------------------------------------------
 247
 248 ##########  hexablock.addLaws(doc, 0.7, True) Ne marche pas
 249
 250 law = doc.addLaw("Uniform", 4)
 251 for j in range(doc.countPropagation()):
 252     propa = doc.getPropagation(j)
 253     propa.setLaw(law)
 254
 255 doc.save("tuyau_v5")
 256 doc.setName("tuyau_v5")
 257 hexablock.addToStudy(doc)
 258
 259 blocs = hexablock.mesh(doc)
 260
 261 muv, mue, muq, muh = hexablock.dump(doc, blocs)