tests/periodicity_reflexion_prepro.py

   1 #!/usr/bin/env python
   2 # -*- coding: utf-8 -*-
   3
   4 import salome
   5 import math
   6
   7 import GEOM
   8 from salome.geom import geomBuilder
   9 geompy = geomBuilder.New(salome.myStudy)
  10
  11 simple = False
  12
  13 r = 10
  14 dist = 10
  15
  16 p1 = geompy.MakeVertex(0., 0., 0.)
  17 p2 = geompy.MakeVertex(100., 100., 100.)
  18 box = geompy.MakeBoxTwoPnt(p1, p2)
  19 geompy.addToStudy(box, "box")
  20
  21 p3 = geompy.MakeVertex(50., 5., 25.)
  22 sphere1 = geompy.MakeSpherePntR(p3, 15.)
  23 geompy.addToStudy(sphere1, "sphere1")
  24
  25 Vx = geompy.MakeVectorDXDYDZ(1, 0, 0)
  26 Vy = geompy.MakeVectorDXDYDZ(0, 1, 0)
  27 Vz = geompy.MakeVectorDXDYDZ(0, 0, 1)
  28
  29 p4 = geompy.MakeVertex(100., 0., 0.)
  30 axe = geompy.MakePrismVecH(p4, Vz, 1)
  31 geompy.addToStudy(axe, "axe")
  32
  33 sphere1_rota = geompy.MakeRotation(sphere1, axe, -math.pi/2.)
  34 geompy.addToStudy(sphere1_rota, "sphere1_rota")
  35
  36 part = geompy.MakePartition([box], [sphere1, sphere1_rota])
  37 geompy.addToStudy(part, "part")
  38
  39 left_faces = geompy.GetShapesOnPlane(part, geompy.ShapeType["FACE"], Vy, GEOM.ST_ON)
  40 left = geompy.CreateGroup(part, geompy.ShapeType["FACE"])
  41 geompy.UnionList(left, left_faces)
  42 geompy.addToStudyInFather(part, left, "left")
  43
  44 right_faces = geompy.GetShapesOnPlaneWithLocation(part, geompy.ShapeType["FACE"], Vy, p2, GEOM.ST_ON)
  45 right = geompy.CreateGroup(part, geompy.ShapeType["FACE"])
  46 geompy.UnionList(right, right_faces)
  47 geompy.addToStudyInFather(part, right, "right")
  48
  49 back_faces = geompy.GetShapesOnPlane(part, geompy.ShapeType["FACE"], Vx, GEOM.ST_ON)
  50 back = geompy.CreateGroup(part, geompy.ShapeType["FACE"])
  51 geompy.UnionList(back, back_faces)
  52 geompy.addToStudyInFather(part, back, "back")
  53
  54 front_faces = geompy.GetShapesOnPlaneWithLocation(part, geompy.ShapeType["FACE"], Vx, p2, GEOM.ST_ON)
  55 front = geompy.CreateGroup(part, geompy.ShapeType["FACE"])
  56 geompy.UnionList(front, front_faces)
  57 geompy.addToStudyInFather(part, front, "front")
  58
  59 bottom_faces = geompy.GetShapesOnPlane(part, geompy.ShapeType["FACE"], Vz, GEOM.ST_ON)
  60 bottom = geompy.CreateGroup(part, geompy.ShapeType["FACE"])
  61 geompy.UnionList(bottom, bottom_faces)
  62 geompy.addToStudyInFather(part, bottom, "bottom")
  63
  64 top_faces = geompy.GetShapesOnPlaneWithLocation(part, geompy.ShapeType["FACE"], Vz, p2, GEOM.ST_ON)
  65 top = geompy.CreateGroup(part, geompy.ShapeType["FACE"])
  66 geompy.UnionList(top, top_faces)
  67 geompy.addToStudyInFather(part, top, "top")
  68
  69 p1_rota = geompy.MakeRotation(p1, axe, -math.pi/2.)
  70 geompy.addToStudy(sphere1_rota, "sphere1_rota")
  71
  72 p5 = geompy.MakeVertex(100, 0, 100)
  73 geompy.addToStudy(p5, "p5")
  74
  75 # Mesh
  76 # ====
  77
  78 import SMESH
  79 from salome.smesh import smeshBuilder
  80 smesh = smeshBuilder.New(salome.myStudy)
  81
  82 Mesh = smesh.Mesh(part, "Mesh")
  83
  84 algo2d = Mesh.Triangle(algo=smeshBuilder.MG_CADSurf)
  85 algo2d.SetGeometricMesh( 1 )
  86 algo2d.SetAngleMesh( 4 )
  87 algo2d.SetPhySize( 8 )
  88
  89 # Periodicity
  90 #algo2d.SetVerbosity(10)
  91 #algo2d.SetPreCADOptionValue("periodic_tolerance", "1e-2")
  92 algo2d.AddPreCadFacesPeriodicity(left, front, [p1, p4, p5], [p1_rota, p4, p5])
  93
  94
  95 gr_left = Mesh.Group(left)
  96 gr_right = Mesh.Group(right)
  97 gr_front = Mesh.Group(front)
  98 gr_back = Mesh.Group(back)
  99 gr_bottom = Mesh.Group(bottom)
 100 gr_top = Mesh.Group(top)
 101
 102 Mesh.Compute()
 103
 104 left_rotated = Mesh.RotateObjectMakeMesh( gr_left, axe, -math.pi/2, NewMeshName='left_rotated' )
 105
 106 def checkProjection(gr, mesh_translated, tol=1e-7):
 107     name = gr.GetName() + "_" + mesh_translated.GetName().split("_")[0]
 108     mesh_source = smesh.CopyMesh(gr, gr.GetName())
 109     mesh_check = smesh.Concatenate([mesh_source.GetMesh(), mesh_translated.GetMesh()], 0, name=name)
 110     ll_coincident_nodes = mesh_check.FindCoincidentNodes(tol)
 111     coincident_nodes = [item for sublist in ll_coincident_nodes for item in sublist]
 112     mesh_check.MakeGroupByIds("coincident_nodes", SMESH.NODE, coincident_nodes)
 113     mesh_nodes = mesh_check.GetNodesId()
 114     if len(ll_coincident_nodes) != mesh_translated.NbNodes():
 115         non_coincident_nodes = list(set(mesh_nodes) - set(coincident_nodes))
 116         mesh_check.MakeGroupByIds("non_coincident_nodes", SMESH.NODE, non_coincident_nodes)
 117         raise Exception("Projection failed for %s"%name)
 118
 119 checkProjection(gr_front, left_rotated)
 120
 121 salome.sg.updateObjBrowser(True)
 122