1 # -*- coding: latin-1 -*-
2 # Copyright (C) 2009-2013 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.
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 : 2011-2013 : CEA-Saclay, DEN, DM2S, SFME, LGLS, F-91191 Gif-sur-Yvette, France
22 # =============================================================================================
29 # Construire le modele de bloc
30 # ============================
32 doc = hexablock.addDocument("bielle_v5")
34 # Construire les 2 grilles cylindriques
35 # -------------------------------------
37 centre_pb = doc.addVertex(-0.107, 0, -0.02)
38 centre_gb = doc.addVertex(0.119, 0, -0.02)
40 angle_px = math.pi / 3
41 vecteur_px = doc.addVector(math.cos(angle_px), math.sin(angle_px), 0)
42 vecteur_gx = doc.addVector(1, 0, 0)
44 vecteur_z = doc.addVector(0, 0, 1)
46 grille_p = doc.makeCylindrical(centre_pb, vecteur_px, vecteur_z, 0.012, 360, 0.04, 1, 3, 1, False)
47 grille_g = doc.makeCylindrical(centre_gb, vecteur_gx, vecteur_z, 0.025, 360, 0.04, 1, 3, 1, False)
49 # Relier les 2 grilles
50 # --------------------
52 quad_p = grille_p.getQuadJK(1, 2, 0)
53 quad_g = grille_g.getQuadJK(1, 1, 0)
55 point_p1 = grille_p.getVertexIJK(1, 0, 0)
56 point_p2 = grille_p.getVertexIJK(1, 2, 0)
58 point_g1 = grille_g.getVertexIJK(1, 1, 0)
59 point_g2 = grille_g.getVertexIJK(1, 2, 0)
61 prisme = doc.joinQuad(quad_p, quad_g, point_p1, point_g1, point_p2, point_g2, 3)
63 # Charger la geometrie
64 # ====================
66 bielle = geompy.ImportSTEP("bielle.stp")
68 # Selectionner des sous-parties de la geometrie
69 # ---------------------------------------------
71 sommets = geompy.SubShapeAllSortedCentres (bielle, geompy.ShapeType["VERTEX"])
72 nodes_id = geompy.SubShapeAllSortedCentresIDs (bielle, geompy.ShapeType["VERTEX"])
73 ## 0 1 2 3 4 5 6 7 8 9
74 ## 23, 70, 15, 52, 17, 57, 13, 47, 19, 62,
75 ## 11, 42, 6, 30, 9, 37, 7, 32, 26, 75]
77 sommets_petit = [ 6, 8, 7, 9 ]
78 sommets_grand = [ 10, 12, 11, 13 ]
79 node_little_subid = [ 13, 19, 47, 62 ]
80 node_big_subid = [ 11, 6, 42, 30 ]
82 aretes = geompy.SubShapeAllSortedCentres (bielle, geompy.ShapeType["EDGE"])
83 edge_id = geompy.SubShapeAllSortedCentresIDs (bielle, geompy.ShapeType["EDGE"])
84 ## 0 1 2 3 4 5 6 7 8 9
85 ## 16, 56, 22, 71, 69, 53, 58, 14, 51, 18,
86 ## 61, 48, 63, 12, 46, 20, 66, 43, 29, 10,
87 ## 41, 5, 31, 38, 33, 25, 76, 8, 36, 74
90 aretes_petit = [ 7, 9, 8, 10 ]
91 aretes_grand = [ 19, 21, 20, 22 ]
92 arc_little_subid = [ 14, 18, 51, 61 ]
93 arc_big_subid = [ 10, 5, 41, 31 ]
119 # Associer le modele de bloc avec la geometrie
120 # ============================================
122 geometry = doc.addShape(bielle, "bielle")
125 # Nettoyer les associations implicites du document
126 # ------------------------------------------------
128 doc.clearAssociation()
130 # Associer les cercles exterieurs
131 # -------------------------------
133 def cercle(grille, k, subid, p):
134 ms = grille.getVertexIJK(0, 0, k)
136 ma1 = grille.getEdgeJ (0, 2, k)
137 ma2 = grille.getEdgeJ (0, 1, k)
138 ma3 = grille.getEdgeJ (0, 0, k)
140 doc.associateClosedLine (ms, [ma1, ma2, ma3 ], [geometry], [subid], p, False)
142 cercle (grille_p, 0, subid_pbe, 5.0/6)
143 cercle (grille_p, 1, subid_phe, 5.0/6)
145 cercle (grille_g, 0, subid_gbe, 0)
146 cercle (grille_g, 1, subid_ghe, 0)
148 # Associer les arcs exterieurs (excentres)
149 # ----------------------------------------
151 def arc(grille, i1, i2, k, subid):
152 ma1 = grille.getEdgeJ(1, i1, k)
153 ma2 = grille.getEdgeJ(1, i2, k)
155 doc.associateOpenedLine ([ma1, ma2 ], [geometry], [subid], 0, 1)
157 arc(grille_p, 1, 0, 0, subid_pbcd)
158 arc(grille_p, 1, 0, 1, subid_phcd)
160 arc(grille_g, 0, 2, 0, subid_gbcd)
161 arc(grille_g, 0, 2, 1, subid_ghcd)
163 # Associer les sommets des arcs de cercle de raccord
164 # --------------------------------------------------
166 hm = prisme.getHexa(1)
167 for i in xrange(0, 4):
169 subid = node_little_subid [i]
170 vm.setAssociation (geometry, subid)
172 hm = prisme.getHexa(2)
173 for i in xrange(0, 4):
175 subid = node_big_subid [i]
176 vm.setAssociation (geometry, subid)
178 # Associer les arcs de cercle de raccord
179 # --------------------------------------
181 hm = prisme.getHexa(0)
182 for i in xrange(0, 4):
184 subid = arc_little_subid [i]
185 em.addAssociation (geometry, subid, 0, 1)
187 hm = prisme.getHexa(2)
188 for i in xrange(0, 4):
190 subid = arc_big_subid [i]
191 em.addAssociation (geometry, subid, 0, 1)
193 # Arrondir des associations implicites cylindriques
194 # -------------------------------------------------
197 for h, i, ech in [ [0, 0, 0.95], [0, 1, 0.95], [2, 2, 0.85], [2, 3, 0.85] ]:
198 hm = prisme.getHexa(h)
204 vax = v1.getAssoX ();
205 vay = v1.getAssoY ();
206 vaz = v1.getAssoZ ();
208 vbx = v2.getAssoX ();
209 vby = v2.getAssoY ();
210 vbz = v2.getAssoZ ();
212 vmx = ( vax + vbx ) / 2.0 * ech
213 vmy = ( vay + vby ) / 2.0
214 vmz = ( vaz + vbz ) / 2.0
216 va = geompy.MakeVertex(vax, vay, vaz)
217 vb = geompy.MakeVertex(vbx, vby, vbz)
218 vm = geompy.MakeVertex(vmx, vmy, vmz)
219 eg = geompy.MakeArc(va, vm, vb)
222 arc = geompy.MakeCompound(larc)
223 shape = doc.addShape (arc, "arc")
225 # Mailler le modele de bloc
226 for h, i in [ [0, 0], [0, 1], [2, 2], [2, 3] ]:
227 hm = prisme.getHexa(h)
230 em.clearAssociation()
232 em.addAssociation (shape, shape_subid, 0.0, 1.0)
234 # Mailler le modele de bloc
235 # =========================
237 # Definir 5 groupes de faces
238 # --------------------------
240 groupe_petit = doc.addQuadGroup("Petit")
241 groupe_grand = doc.addQuadGroup("Grand")
242 groupe_bas = doc.addQuadGroup("Bas")
243 groupe_haut = doc.addQuadGroup("Haut")
244 groupe_contour = doc.addQuadGroup("Contour")
246 # Constituer les groupes petit et grand
247 # -------------------------------------
250 groupe_petit.addElement( grille_p.getQuadJK(0, i, 0) )
251 groupe_grand.addElement( grille_g.getQuadJK(0, i, 0) )
253 # Constituer les groupes bas et haut
254 # ----------------------------------
257 groupe_bas.addElement( grille_p.getQuadIJ(0, i, 0) )
258 groupe_bas.addElement( grille_g.getQuadIJ(0, i, 0) )
260 groupe_haut.addElement( grille_p.getQuadIJ(0, i, 1) )
261 groupe_haut.addElement( grille_g.getQuadIJ(0, i, 1) )
264 h = prisme.getHexa(i)
266 groupe_bas.addElement( h.getQuad(2) )
267 groupe_haut.addElement( h.getQuad(3) )
269 # Constituer le groupe contour
270 # ----------------------------
273 groupe_contour.addElement( grille_p.getQuadJK(1, i, 0) )
276 groupe_contour.addElement( grille_g.getQuadJK(1, i, 0) )
279 h = prisme.getHexa(i)
281 groupe_contour.addElement( h.getQuad(4) )
282 groupe_contour.addElement( h.getQuad(5) )
284 # Definir 3 groupes de volumes
285 # ----------------------------
287 groupe_petit = doc.addHexaGroup("Petit")
288 groupe_grand = doc.addHexaGroup("Grand")
289 groupe_prisme = doc.addHexaGroup("Prisme")
292 groupe_petit.addElement( grille_p.getHexa(i) )
293 groupe_grand.addElement( grille_g.getHexa(i) )
296 groupe_prisme.addElement( prisme.getHexa(i) )
298 # Mailler le modele de bloc avec association
299 # ------------------------------------------
301 ########## hexablock.addLaws(doc, 0.003, True) Ne marche plus !!!!
303 law = doc.addLaw("Uniform", 4)
305 for j in range(doc.countPropagation()):
306 propa = doc.getPropagation(j)
309 doc.save("bielle_v5")
310 doc.setName("bielle_v5")
311 hexablock.addToStudy(doc)
313 blocs = hexablock.mesh(doc)
315 muv, mue, muq, muh = hexablock.dump(doc, blocs)