src/SMESH/SMESH_ParallelMesh.cxx

   1 // Copyright (C) 2007-2023  CEA, EDF, OPEN CASCADE
   2 //
   3 // Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
   4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
   5 //
   6 // This library is free software; you can redistribute it and/or
   7 // modify it under the terms of the GNU Lesser General Public
   8 // License as published by the Free Software Foundation; either
   9 // version 2.1 of the License, or (at your option) any later version.
  10 //
  11 // This library is distributed in the hope that it will be useful,
  12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14 // Lesser General Public License for more details.
  15 //
  16 // You should have received a copy of the GNU Lesser General Public
  17 // License along with this library; if not, write to the Free Software
  18 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  19 //
  20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  21 //
  22
  23 //  File   : SMESH_ParallelMesh.cxx
  24 //  Author : Yoann AUDOUIN, EDF
  25 //  Module : SMESH
  26 //
  27 #include "SMESH_ParallelMesh.hxx"
  28
  29 #include "SMESH_Gen.hxx"
  30
  31 #ifdef WIN32
  32   #include <windows.h>
  33 #endif
  34
  35 #ifndef WIN32
  36 #include <boost/filesystem.hpp>
  37 namespace fs=boost::filesystem;
  38 #endif
  39
  40 #ifndef WIN32
  41 #include <boost/asio.hpp>
  42 #endif
  43
  44 #include <utilities.h>
  45
  46 #ifdef _DEBUG_
  47 static int MYDEBUG = 1;
  48 #else
  49 static int MYDEBUG = 0;
  50 #endif
  51
  52 SMESH_ParallelMesh::SMESH_ParallelMesh(int               theLocalId,
  53                        SMESH_Gen*        theGen,
  54                        bool              theIsEmbeddedMode,
  55                        SMESHDS_Document* theDocument) :SMESH_Mesh(theLocalId,
  56                                                                   theGen,
  57                                                                   theIsEmbeddedMode,
  58                                                                   theDocument)
  59 {
  60   MESSAGE("SMESH_ParallelMesh::SMESH_ParallelMesh(int localId)");
  61   CreateTmpFolder();
  62 };
  63
  64 SMESH_ParallelMesh::~SMESH_ParallelMesh()
  65 {
  66   DeletePoolThreads();
  67   if(!MYDEBUG)
  68     DeleteTmpFolder();
  69 };
  70
  71
  72
  73 //=============================================================================
  74 /*!
  75  * \brief Build folder for parallel computation
  76  */
  77 //=============================================================================
  78 void SMESH_ParallelMesh::CreateTmpFolder()
  79 {
  80 #ifndef WIN32
  81   // Temporary folder that will be used by parallel computation
  82   tmp_folder = fs::temp_directory_path()/fs::unique_path(fs::path("SMESH_%%%%-%%%%"));
  83   fs::create_directories(tmp_folder);
  84 #endif
  85 }
  86 //
  87 //=============================================================================
  88 /*!
  89  * \brief Delete temporary folder used for parallel computation
  90  */
  91 //=============================================================================
  92 void SMESH_ParallelMesh::DeleteTmpFolder()
  93 {
  94 #ifndef WIN32
  95     fs::remove_all(tmp_folder);
  96 #endif
  97 }
  98
  99 //=============================================================================
 100 /*!
 101  * \brief Get the number of Threads to be used for the pool of Threads
 102  */
 103 //=============================================================================
 104 int SMESH_ParallelMesh::GetPoolNbThreads()
 105 {
 106   int nbThreads = -1;
 107
 108   if(_method == ParallelismMethod::MultiThread){
 109     nbThreads = _NbThreads;
 110   }else if( _method == ParallelismMethod::MultiNode){
 111     //TODO: Check of that is the right way
 112     nbThreads = std::max(_nbProc, _nbNode*_nbProcPerNode);
 113   } else {
 114     throw SALOME_Exception("Unknown method "+std::to_string(_method));
 115   }
 116
 117   return nbThreads;
 118 }
 119
 120 //=============================================================================
 121 /*!
 122  * \brief Set Number of thread for multithread run
 123  */
 124 //=============================================================================
 125 void SMESH_ParallelMesh::SetNbThreads(long nbThreads)
 126 {
 127   if(nbThreads < 1)
 128     throw SALOME_Exception("Number of threads should be higher than 1");
 129   _NbThreads=nbThreads;
 130 };
 131
 132 bool SMESH_ParallelMesh::ComputeSubMeshes(
 133           SMESH_Gen* gen,
 134           SMESH_Mesh & aMesh,
 135           const TopoDS_Shape & aShape,
 136           const ::MeshDimension       aDim,
 137           TSetOfInt*                  aShapesId /*=0*/,
 138           TopTools_IndexedMapOfShape* allowedSubShapes,
 139           SMESH_subMesh::compute_event &computeEvent,
 140           const bool includeSelf,
 141           const bool complexShapeFirst,
 142           const bool   aShapeOnly)
 143 {
 144   InitPoolThreads();
 145   return gen->parallelComputeSubMeshes(
 146             aMesh, aShape, aDim,
 147             aShapesId, allowedSubShapes,
 148             computeEvent,
 149             includeSelf,
 150             complexShapeFirst,
 151             aShapeOnly);
 152 }