tests/test_periodicity_2D.py

   1 # -*- coding: iso-8859-1 -*-
   2
   3 import salome
   4
   5 import GEOM
   6 from salome.geom import geomBuilder
   7 geompy = geomBuilder.New(salome.myStudy)
   8
   9 import math
  10
  11 O = geompy.MakeVertex(0, 0, 0)
  12 OX = geompy.MakeVectorDXDYDZ(1, 0, 0)
  13 OY = geompy.MakeVectorDXDYDZ(0, 1, 0)
  14 OZ = geompy.MakeVectorDXDYDZ(0, 0, 1)
  15
  16 Face_1 = geompy.MakeFaceHW(10, 10, 1)
  17 geomObj_1 = geompy.MakeMarker(0, 0, 0, 1, 0, 0, 0, 1, 0)
  18 Sketch_1 = geompy.MakeSketcherOnPlane("Sketcher:F -2.5209155082703 -1.4416453838348:TT -1.1105282306671 -2.9872753620148:TT 0.76354801654816 -2.3303825855255:TT 1.9614112377167 -3.0838770866394:TT 3.8354876041412 -1.2677619457245:TT 4.2218952178955 0.644955098629:TT 3.2751967906952 2.5576722621918:TT 0.58966463804245 3.5430111885071:TT -3.7380990982056 3.2338852882385:TT -4.433632850647 0.85747921466827:WW", Face_1 )
  19 vertices = geompy.ExtractShapes(Sketch_1, geompy.ShapeType["VERTEX"], True)
  20 Curve_1 = geompy.MakeInterpol(vertices, True, True)
  21
  22 part = geompy.MakePartition([Face_1], [Curve_1], Limit=geompy.ShapeType["FACE"])
  23 geompy.addToStudy(part, "part")
  24
  25 Vx = geompy.MakeVectorDXDYDZ(1, 0, 0)
  26 Vy = geompy.MakeVectorDXDYDZ(0, 1, 0)
  27 Vz = geompy.MakeVectorDXDYDZ(0, 0, 1)
  28
  29 p1 = geompy.MakeVertex(-5, -5, 0)
  30 p2 = geompy.MakeVertex(5, 5, 0)
  31 left_edges = geompy.GetShapesOnPlaneWithLocation(part, geompy.ShapeType["EDGE"], Vx, p1, GEOM.ST_ON)
  32 left = geompy.CreateGroup(part, geompy.ShapeType["EDGE"])
  33 geompy.UnionList(left, left_edges)
  34 geompy.addToStudyInFather(part, left, "left")
  35
  36 right_edges = geompy.GetShapesOnPlaneWithLocation(part, geompy.ShapeType["EDGE"], Vx, p2, GEOM.ST_ON)
  37 right = geompy.CreateGroup(part, geompy.ShapeType["EDGE"])
  38 geompy.UnionList(right, right_edges)
  39 geompy.addToStudyInFather(part, right, "right")
  40
  41 bottom_edges = geompy.GetShapesOnPlaneWithLocation(part, geompy.ShapeType["EDGE"], Vy, p1, GEOM.ST_ON)
  42 bottom = geompy.CreateGroup(part, geompy.ShapeType["EDGE"])
  43 geompy.UnionList(bottom, bottom_edges)
  44 geompy.addToStudyInFather(part, bottom, "bottom")
  45
  46 top_edges = geompy.GetShapesOnPlaneWithLocation(part, geompy.ShapeType["EDGE"], Vy, p2, GEOM.ST_ON)
  47 top = geompy.CreateGroup(part, geompy.ShapeType["EDGE"])
  48 geompy.UnionList(top, top_edges)
  49 geompy.addToStudyInFather(part, top, "top")
  50
  51
  52 # Mesh
  53 # ====
  54
  55 import SMESH
  56 from salome.smesh import smeshBuilder
  57 smesh = smeshBuilder.New(salome.myStudy)
  58
  59 from salome.BLSURFPlugin import BLSURFPluginBuilder
  60
  61 Mesh = smesh.Mesh(part, "Mesh")
  62
  63 algo2d = Mesh.Triangle(algo=smeshBuilder.MG_CADSurf)
  64 algo2d.SetVerbosity(10)
  65 #algo2d.SetTopology(smesh.PreCAD)
  66 algo2d.SetGeometricMesh( 1 )
  67 algo2d.SetAngleMesh( 4 )
  68 algo2d.SetPhySize( 8 )
  69 #algo2d.SetGradation(1.05)
  70 algo2d.SetOptionValue( 'periodic_tolerance', '1e-2' )
  71 # Deactivate PreCAD
  72 algo2d.SetTopology(BLSURFPluginBuilder.FromCAD)
  73
  74 # Periodicity
  75
  76 def proj_x(shape1):
  77     shape2 = geompy.MakeTranslation(shape1, 10, 0., 0)
  78     return shape2
  79
  80 def proj_y(shape1):
  81     shape2 = geompy.MakeTranslation(shape1, 0, 10., 0)
  82     return shape2
  83
  84 algo2d.AddAdvancedEdgesPeriodicity(left, right, proj_x)
  85 algo2d.AddAdvancedEdgesPeriodicity(bottom, top, proj_y)
  86
  87 Mesh.Compute()
  88
  89 gr_left = Mesh.Group(left)
  90 gr_right = Mesh.Group(right)
  91 gr_bottom = Mesh.Group(bottom)
  92 gr_top = Mesh.Group(top)
  93
  94 Mesh.Compute()
  95
  96 left_translated = Mesh.TranslateObjectMakeMesh( gr_left, SMESH.DirStruct( SMESH.PointStruct ( 10, 0, 0 )), 0, 'left_translated' )
  97 bottom_translated = Mesh.TranslateObjectMakeMesh( gr_bottom, SMESH.DirStruct( SMESH.PointStruct ( 0, 10, 0 )), 0, 'bottom_translated' )
  98
  99 def checkProjection(gr, mesh_translated, tol=1e-7):
 100     name = gr.GetName() + "_" + mesh_translated.GetName().split("_")[0]
 101     mesh_source = smesh.CopyMesh(gr, gr.GetName())
 102     mesh_check = smesh.Concatenate([mesh_source.GetMesh(), mesh_translated.GetMesh()], 0, name=name)
 103     ll_coincident_nodes = mesh_check.FindCoincidentNodes(tol)
 104     coincident_nodes = [item for sublist in ll_coincident_nodes for item in sublist]
 105     mesh_check.MakeGroupByIds("coincident_nodes", SMESH.NODE, coincident_nodes)
 106     mesh_nodes = mesh_check.GetNodesId()
 107     if len(ll_coincident_nodes) != mesh_translated.NbNodes():
 108         non_coincident_nodes = list(set(mesh_nodes) - set(coincident_nodes))
 109         mesh_check.MakeGroupByIds("non_coincident_nodes", SMESH.NODE, non_coincident_nodes)
 110         raise Exception("Projection failed for %s"%name)
 111
 112 checkProjection(gr_right, left_translated)
 113 checkProjection(gr_top, bottom_translated)
 114
 115 if salome.sg.hasDesktop():
 116   salome.sg.updateObjBrowser(1)