src/TEST_PY/recettes/troisCylindres.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 = "troisCylindres"
  31
  32 piece = geompy.ImportBREP(nom+".brep")
  33
  34 # Selectionner des sous-parties de la geometrie
  35 # ---------------------------------------------
  36
  37 sommets = geompy.SubShapeAllSortedCentres(piece, geompy.ShapeType["VERTEX"])
  38 aretes  = geompy.SubShapeAllSortedCentres(piece, geompy.ShapeType["EDGE"  ])
  39
  40 moy_b_x, moy_b_y, moy_b_z = geompy.PointCoordinates(sommets[14])
  41 moy_p_x, moy_p_y, moy_p_z = geompy.PointCoordinates(sommets[ 6])
  42 moy_g_x, moy_g_y, moy_g_z = geompy.PointCoordinates(sommets[ 3])
  43
  44 grand0, grand_xb, grand_yb, grand_zb,  grand1, grand2, grand3,  grand_rayon = geompy.KindOfShape(aretes[ 0])
  45 grand4, grand_xh, grand_yh, grand_zh,  grand5, grand6, grand7,  grand8      = geompy.KindOfShape(aretes[ 1])
  46
  47 moyen0, moyen_x , moyen_y , moyen_z ,  moyen1, moyen2, moyen3,  moyen_rayon = geompy.KindOfShape(aretes[20])
  48
  49 petit0, petit_xb, petit_yb, petit_zb,  petit1, petit2, petit3,  petit_rayon = geompy.KindOfShape(aretes[12])
  50 petit4, petit_xh, petit_yh, petit_zh,  petit5, petit6, petit7,  petit8      = geompy.KindOfShape(aretes[13])
  51
  52 # Construire le modele de bloc
  53 # ============================
  54
  55 doc = hexablock.addDocument(nom)
  56
  57 ### doc.addShape(piece,nom)
  58
  59 # Construire le grand cylindre
  60 # ----------------------------
  61
  62 grand_base = doc.addVertex(grand_xb, grand_yb, grand_zb)
  63 grand_oppo = doc.addVertex(grand_xh, grand_yh, grand_zh)
  64
  65 grand_dir = doc.addVectorVertices(grand_base, grand_oppo)
  66
  67 grand_hauteur = grand_dir.getNorm()
  68
  69 ###  grand_cylindre = doc.addCylinder(grand_base, grand_dir, grand_rayon, grand_hauteur)
  70
  71 # Construire le moyen cylindre cote grand
  72 # ---------------------------------------
  73
  74 moy_g_r1 = 0.8
  75 moy_g_r2 = 0.5
  76
  77 moy_g_base_x = moy_p_x + (moy_g_x-moy_p_x)*moy_g_r1 + moyen_x - moy_b_x
  78 moy_g_base_y = moy_p_y + (moy_g_y-moy_p_y)*moy_g_r1 + moyen_y - moy_b_y
  79 moy_g_base_z = moy_p_z + (moy_g_z-moy_p_z)*moy_g_r1 + moyen_z - moy_b_z
  80
  81 moy_g_base = doc.addVertex(moy_g_base_x, moy_g_base_y, moy_g_base_z)
  82 moy_g_oppo = doc.addVertex( (grand_xh+grand_xb) * moy_g_r2, (grand_yh+grand_yb) * moy_g_r2, (grand_zh+grand_zb) * moy_g_r2 )
  83
  84 moy_g_dir = doc.addVectorVertices(moy_g_base, moy_g_oppo)
  85
  86 moy_g_hauteur = moy_g_dir.getNorm()
  87
  88 ###  moy_g_cylindre = doc.addCylinder(moy_g_base, moy_g_dir, moyen_rayon, moy_g_hauteur)
  89
  90 # Construire le moyen cylindre cote petit
  91 # ---------------------------------------
  92
  93 moy_p_r = 0.18
  94
  95 moy_p_base = doc.addVertex(moyen_x, moyen_y, moyen_z)
  96 moy_b_base = doc.addVertex(moy_b_x, moy_b_y, moy_b_z)
  97 moy_p_oppo = doc.addVertex( moy_p_x+(moy_g_x-moy_p_x) * moy_p_r, moy_p_y+(moy_g_y-moy_p_y) * moy_p_r, moy_p_z+(moy_g_z-moy_p_z) * moy_p_r )
  98
  99 moy_p_dir = doc.addVectorVertices(moy_b_base, moy_p_oppo)
 100
 101 moy_p_hauteur = moy_p_dir.getNorm()*1.2
 102
 103 ###  moy_p_cylindre = doc.addCylinder(moy_p_base, moy_p_dir, moyen_rayon, moy_p_hauteur)
 104
 105 # Construire le petit cylindre
 106 # ----------------------------
 107
 108 petit_base = doc.addVertex(petit_xb, petit_yb, petit_zb)
 109 petit_oppo = doc.addVertex(petit_xh, petit_yh, petit_zh)
 110
 111 petit_dir  = doc.addVectorVertices(petit_base, petit_oppo)
 112
 113 petit_hauteur = petit_dir.getNorm()
 114
 115 ###  petit_cylindre = doc.addCylinder(petit_base, petit_dir, petit_rayon, petit_hauteur)
 116
 117 # Construire le T et la X
 118 # -----------------------
 119
 120 en_te = doc.makeCylinders (grand_base, grand_dir, grand_rayon, grand_hauteur,
 121                            moy_g_base, moy_g_dir, moyen_rayon, moy_g_hauteur)
 122
 123 hexablock.what ()
 124
 125 croix = doc.makeCylinders (moy_p_base, moy_p_dir, moyen_rayon, moy_p_hauteur,
 126                            petit_base, petit_dir, petit_rayon, petit_hauteur)
 127
 128 hexablock.what ()
 129
 130 # Relier le T et la X
 131 # -------------------
 132
 133 CYL_SMALL  = 0
 134 CYL_BIG    = 1
 135 CYL_KMAX   = 3
 136 CYL_JMAX   = 4
 137
 138
 139 qstart = croix.getQuadIJ (CYL_BIG, 0,0,CYL_KMAX)
 140 ##  qstart.setColor(5)
 141 croix_quads = [qstart]
 142
 143 for nj in range (CYL_JMAX) :
 144     quad = croix.getQuadIJ (CYL_BIG, 1,nj, CYL_KMAX)
 145     quad.setColor(5)
 146     croix_quads.append (quad)
 147
 148 qdest = en_te.getQuadIJ (CYL_SMALL, 0,0,0)
 149 qdest.setColor(5)
 150
 151 va1 = qstart.getVertex (0)
 152 va2 = qstart.getVertex (1)
 153
 154 vb1 = qdest.nearestVertex (va1)
 155 vb2 = qdest.nearestVertex (va2)
 156
 157 prisme = doc.joinQuadsUni (croix_quads, qdest, va1,vb1,va2,vb2, 1)
 158
 159 # Associer
 160 # ========
 161
 162
 163 # Mailler le modele de bloc avec association
 164 # ------------------------------------------
 165
 166 doc.addLaws(10, True)
 167
 168 law = doc.addLaw("Uniform", 4)
 169 for j in range(doc.countPropagation()):
 170     propa = doc.getPropagation(j)
 171     propa.setLaw(law)
 172
 173 doc.save("troisCylindres")
 174 doc.setName("troisCylindres")
 175 doc.saveVtk("troisCylindres1.vtk")
 176 hexablock.addToStudy(doc)
 177
 178 blocs = hexablock.mesh(doc, dim=3)
 179
 180 muv, mue, muq, muh = hexablock.dump(doc, blocs)