examples/command_line/dataset.py

   1 def create_geometry(out_filename=None):
   2   import GEOM
   3   import salome
   4   from salome.geom import geomBuilder
   5   geompy = geomBuilder.New()
   6
   7   # create a cloud of points
   8   points = [
   9     geompy.MakeVertex(0,0,0),
  10     geompy.MakeVertex(9,0,0),
  11     geompy.MakeVertex(0,9,0),
  12     geompy.MakeVertex(9,9,0),
  13     geompy.MakeVertex(3,3,1),
  14     geompy.MakeVertex(6,6,2)]
  15
  16   # create SmoothingSurface object
  17   smoothingsurface = geompy.MakeSmoothingSurface(points)
  18   if out_filename:
  19     geompy.ExportBREP(smoothingsurface, out_filename)
  20
  21   # add object in the study
  22   id_smoothingsurface = geompy.addToStudy(smoothingsurface,"SmoothingSurface")
  23
  24   return smoothingsurface
  25 #
  26
  27 def create_meshes(geometry, out_filename=None):
  28   import SMESH
  29   import salome
  30   from salome.smesh import smeshBuilder
  31
  32   smesh = smeshBuilder.New()
  33   Mesh_tri = smesh.Mesh(geometry)
  34
  35   Regular_1D = Mesh_tri.Segment()
  36   Max_Size_1 = Regular_1D.MaxSize(1.28993)
  37
  38   MEFISTO_2D = Mesh_tri.Triangle(algo=smeshBuilder.MEFISTO)
  39   Nb_Segments_1 = smesh.CreateHypothesis('NumberOfSegments')
  40   Nb_Segments_1.SetNumberOfSegments( 15 )
  41   Nb_Segments_1.SetDistrType( 0 )
  42
  43   Mesh_quad = smesh.Mesh(geometry)
  44   status = Mesh_quad.AddHypothesis(Nb_Segments_1)
  45   status = Mesh_quad.AddHypothesis(Regular_1D)
  46
  47   Quadrangle_2D = Mesh_quad.Quadrangle(algo=smeshBuilder.QUADRANGLE)
  48   isDone = Mesh_tri.Compute()
  49   isDone = Mesh_quad.Compute()
  50   smesh.SetName(Mesh_tri, 'Mesh_tri')
  51   if out_filename:
  52     Mesh_tri.ExportMED(out_filename, 0, SMESH.MED_V2_2, 1, None ,1)
  53   smesh.SetName(Mesh_quad, 'Mesh_quad')
  54   if out_filename:
  55     Mesh_quad.ExportMED(out_filename, 0, SMESH.MED_V2_2, 0, None ,1)
  56
  57   # Set names of Mesh objects
  58   smesh.SetName(Regular_1D.GetAlgorithm(), 'Regular_1D')
  59   smesh.SetName(Quadrangle_2D.GetAlgorithm(), 'Quadrangle_2D')
  60   smesh.SetName(MEFISTO_2D.GetAlgorithm(), 'MEFISTO_2D')
  61   smesh.SetName(Nb_Segments_1, 'Nb. Segments_1')
  62   smesh.SetName(Max_Size_1, 'Max Size_1')
  63   smesh.SetName(Mesh_tri.GetMesh(), 'Mesh_tri')
  64   smesh.SetName(Mesh_quad.GetMesh(), 'Mesh_quad')
  65
  66   return Mesh_tri, Mesh_quad
  67 #
  68
  69 from MEDLoader import *
  70
  71 def create_field(mesh_filename, mesh_name, field_name, out_filename=None):
  72   mesh = MEDLoader.ReadUMeshFromFile(mesh_filename, mesh_name)
  73   src_field = mesh.fillFromAnalytic(ON_CELLS, 1, "7-sqrt((x-5.)*(x-5.)+(y-5.)*(y-5.)+(z-5.)*(z-5.))")
  74   src_field.setName(field_name)
  75   if out_filename:
  76     MEDLoader.WriteField(out_filename, src_field, True)
  77   return src_field
  78 #