1 # -*- coding: latin-1 -*-
2 # Copyright (C) 2009-2014 CEA/DEN, EDF R&D
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, or (at your option) any later version.
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.
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
18 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
21 # Francis KLOSS : 2012 : CEA-Saclay, DEN, DM2S, SFME, LGLS, F-91191 Gif-sur-Yvette, France
22 # ========================================================================================
25 geompy = hexablock.geompy
27 # Charger la geometrie
28 # ====================
30 nom = "troisCylindres"
32 piece = geompy.ImportBREP(nom+".brep")
34 # Selectionner des sous-parties de la geometrie
35 # ---------------------------------------------
37 sommets = geompy.SubShapeAllSortedCentres(piece, geompy.ShapeType["VERTEX"])
38 aretes = geompy.SubShapeAllSortedCentres(piece, geompy.ShapeType["EDGE" ])
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])
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])
47 moyen0, moyen_x , moyen_y , moyen_z , moyen1, moyen2, moyen3, moyen_rayon = geompy.KindOfShape(aretes[20])
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])
52 # Construire le modele de bloc
53 # ============================
55 doc = hexablock.addDocument(nom)
57 doc.addShape(piece,nom)
59 # Construire le grand cylindre
60 # ----------------------------
62 grand_base = doc.addVertex(grand_xb, grand_yb, grand_zb)
63 grand_oppo = doc.addVertex(grand_xh, grand_yh, grand_zh)
65 grand_dir = doc.addVectorVertices(grand_base, grand_oppo)
67 grand_hauteur = grand_dir.getNorm()
69 grand_cylindre = doc.addCylinder(grand_base, grand_dir, grand_rayon, grand_hauteur)
71 # Construire le moyen cylindre cote grand
72 # ---------------------------------------
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
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 )
84 moy_g_dir = doc.addVectorVertices(moy_g_base, moy_g_oppo)
86 moy_g_hauteur = moy_g_dir.getNorm()
88 moy_g_cylindre = doc.addCylinder(moy_g_base, moy_g_dir, moyen_rayon, moy_g_hauteur)
90 # Construire le moyen cylindre cote petit
91 # ---------------------------------------
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 )
99 moy_p_dir = doc.addVectorVertices(moy_b_base, moy_p_oppo)
101 moy_p_hauteur = moy_p_dir.getNorm()
103 moy_p_cylindre = doc.addCylinder(moy_p_base, moy_p_dir, moyen_rayon, moy_p_hauteur)
105 # Construire le petit cylindre
106 # ----------------------------
108 petit_base = doc.addVertex(petit_xb, petit_yb, petit_zb)
109 petit_oppo = doc.addVertex(petit_xh, petit_yh, petit_zh)
111 petit_dir = doc.addVectorVertices(petit_base, petit_oppo)
113 petit_hauteur = petit_dir.getNorm()
115 petit_cylindre = doc.addCylinder(petit_base, petit_dir, petit_rayon, petit_hauteur)
117 # Construire le T et la X
118 # -----------------------
120 en_te = doc.makeCylinders(grand_cylindre, moy_g_cylindre)
121 croix = doc.makeCylinders(moy_p_cylindre, petit_cylindre)
123 # Relier le T et la X
124 # -------------------
128 for i in xrange(1, -1, -1):
129 for j in xrange( [hexablock.CV_MAXI_INT, hexablock.CV_MAXI_EXT][i] ):
130 quad = croix.getQuadIJ(hexablock.CYL_BIG, i, j, hexablock.CYL_BIG_SLICES)
131 croix_quads.append(quad)
133 en_te_q = en_te.getQuadIJ(hexablock.CYL_SMALL, 1, hexablock.V_N, 0)
135 en_te_v0 = en_te.getVertexIJK(hexablock.CYL_SMALL, 2, hexablock.V_N , 0)
136 en_te_v1 = en_te.getVertexIJK(hexablock.CYL_SMALL, 2, hexablock.V_NW, 0)
138 croix_v0 = croix.getVertexIJK(hexablock.CYL_BIG, 2, hexablock.V_E , hexablock.CYL_BIG_SLICES)
139 croix_v1 = croix.getVertexIJK(hexablock.CYL_BIG, 2, hexablock.V_NE, hexablock.CYL_BIG_SLICES)
141 prisme = doc.joinQuads(croix_quads, en_te_q, croix_v0, en_te_v0, croix_v1, en_te_v1, 1)
146 # fkl: comme pour la tuyauterie: associer pour le prisme (sauf si item 142 et 146 de Taches.ods)
148 # Mailler le modele de bloc
149 # =========================
151 # Definir 5 groupes de faces
152 # --------------------------
154 groupe_grand_a = doc.addQuadGroup("Grand_A")
155 groupe_grand_b = doc.addQuadGroup("Grand_B")
156 groupe_moyen_f = doc.addQuadGroup("Moyen")
157 groupe_petit_a = doc.addQuadGroup("Petit_A")
158 groupe_petit_b = doc.addQuadGroup("Petit_B")
161 for j in xrange( [hexablock.CV_MAXI_INT, hexablock.CV_MAXI_EXT][i] ):
162 quad = en_te.getQuadIJ(hexablock.CYL_BIG , i, j, 0)
163 groupe_grand_a.addElement(quad)
165 quad = en_te.getQuadIJ(hexablock.CYL_BIG , i, j, hexablock.CYL_BIG_SLICES)
166 groupe_grand_b.addElement(quad)
168 quad = croix.getQuadIJ(hexablock.CYL_BIG , i, j, 0)
169 groupe_moyen_f.addElement(quad)
171 quad = croix.getQuadIJ(hexablock.CYL_SMALL, i, j, 0)
172 groupe_petit_a.addElement(quad)
174 quad = croix.getQuadIJ(hexablock.CYL_SMALL, i, j, hexablock.CYL_SMALL_SLICES)
175 groupe_petit_b.addElement(quad)
177 # Definir 3 groupes de volumes
178 # ----------------------------
180 groupe_grand = doc.addHexaGroup("Grand")
181 groupe_moyen = doc.addHexaGroup("Moyen")
182 groupe_petit = doc.addHexaGroup("Petit")
184 for i in xrange( doc.countUsedHexa() ):
185 h = doc.getUsedHexa(i)
186 groupe_moyen.addElement(h)
189 for j in xrange( [hexablock.CV_MAXI_INT, hexablock.CV_MAXI_EXT][i] ):
190 for k in [ 0, hexablock.CYL_BIG_SLICES-1 ]:
191 h = en_te.getHexaIJK(hexablock.CYL_BIG, i, j, k)
192 groupe_grand.addElement(h)
193 groupe_moyen.removeElement(h)
195 for k in [0, hexablock.CYL_SMALL_SLICES-1 ]:
196 h = croix.getHexaIJK(hexablock.CYL_SMALL, i, j, k)
197 groupe_petit.addElement(h)
198 groupe_moyen.removeElement(h)
200 # Mailler le modele de bloc avec association
201 # ------------------------------------------
203 ########## hexablock.addLaws(doc, 10, True) Ne marche pas
205 law = doc.addLaw("Uniform", 4)
206 for j in range(doc.countPropagation()):
207 propa = doc.getPropagation(j)
210 doc.save("troisCylindres_v5")
211 doc.setName("troisCylindres_v5")
212 hexablock.addToStudy(doc)
214 blocs = hexablock.mesh(doc, dim=3)
216 muv, mue, muq, muh = hexablock.dump(doc, blocs)