tests/multithread.py

   1 # -*- coding: utf-8 -*-
   2 # Copyright (C) 2017-2019  CEA/DEN, EDF R&D
   3 #
   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.
   8 #
   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.
  13 #
  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
  17 #
  18 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  19 #
  20
  21 import os
  22 import sys
  23 import salome
  24 import time
  25 import multiprocessing
  26
  27 salome.salome_init()
  28
  29 cpu_count = multiprocessing.cpu_count()
  30 divider = min(4, cpu_count)/2.3
  31
  32 print ("Running test on workstation with %d available cores" % cpu_count)
  33
  34 ###
  35 ### GEOM component
  36 ###
  37
  38 import GEOM
  39 from salome.geom import geomBuilder
  40 import math
  41 import SALOMEDS
  42
  43
  44 geompy = geomBuilder.New()
  45
  46 O = geompy.MakeVertex(0, 0, 0)
  47 OX = geompy.MakeVectorDXDYDZ(1, 0, 0)
  48 OY = geompy.MakeVectorDXDYDZ(0, 1, 0)
  49 OZ = geompy.MakeVectorDXDYDZ(0, 0, 1)
  50 flight_solid_brep_1 = geompy.ImportBREP(os.path.join(os.getenv("BLSURFPLUGIN_ROOT_DIR"),"share/salome/resources/blsurfplugin/flight_solid.brep"))
  51 geompy.addToStudy( O, 'O' )
  52 geompy.addToStudy( OX, 'OX' )
  53 geompy.addToStudy( OY, 'OY' )
  54 geompy.addToStudy( OZ, 'OZ' )
  55 geompy.addToStudy( flight_solid_brep_1, 'flight_solid.brep_1' )
  56
  57 ###
  58 ### SMESH component
  59 ###
  60
  61 import  SMESH, SALOMEDS
  62 from salome.smesh import smeshBuilder
  63
  64 smesh = smeshBuilder.New()
  65
  66 Mesh_1 = smesh.Mesh(flight_solid_brep_1)
  67 MG_CADSurf = Mesh_1.Triangle(algo=smeshBuilder.MG_CADSurf)
  68 MG_CADSurf_Parameters_1 = MG_CADSurf.Parameters()
  69 MG_CADSurf_Parameters_1.SetPhySize( 1 )
  70 MG_CADSurf_Parameters_1.SetMaxSize( 1 )
  71 MG_CADSurf_Parameters_1.SetGradation( 1.05 )
  72 MG_CADSurf_Parameters_1.SetAngleMesh( 1 )
  73 MG_CADSurf_Parameters_1.SetChordalError( 2.40018 )
  74 # 4 procs are used by default
  75 # => No need to set an option
  76
  77 time0 = time.time()
  78 isDone = Mesh_1.Compute()
  79 time1 = time.time()
  80
  81 time_multithread = time1-time0
  82
  83 print ("Time in multi thread (4 procs): %.3f s"%(time_multithread))
  84
  85 Mesh_2 = smesh.Mesh(flight_solid_brep_1)
  86 MG_CADSurf = Mesh_2.Triangle(algo=smeshBuilder.MG_CADSurf)
  87 MG_CADSurf_Parameters_2 = MG_CADSurf.Parameters()
  88 MG_CADSurf_Parameters_2.SetPhySize( 1 )
  89 MG_CADSurf_Parameters_2.SetMaxSize( 1 )
  90 MG_CADSurf_Parameters_2.SetGradation( 1.05 )
  91 MG_CADSurf_Parameters_2.SetAngleMesh( 1 )
  92 MG_CADSurf_Parameters_2.SetChordalError( 2.40018 )
  93 # Use only one thread
  94 MG_CADSurf_Parameters_2.SetMaxNumberOfThreads( 1 )
  95
  96 time2 = time.time()
  97 isDone = Mesh_2.Compute()
  98 time3 = time.time()
  99
 100 time_singlethread = time3-time2
 101 print ("Time in single thread (1 proc): %.3f s"%(time_singlethread))
 102
 103 if cpu_count == 1:
 104     print ("Warning: cannot validate test - only 1 cpu core is available")
 105 else:
 106     assert time_multithread < time_singlethread/divider
 107
 108 if salome.sg.hasDesktop():
 109   salome.sg.updateObjBrowser()