src/RENUMBER/RENUMBER_METISRenumbering.cxx

   1 // Copyright (C) 2007-2020  CEA/DEN, EDF R&D
   2 //
   3 // This library is free software; you can redistribute it and/or
   4 // modify it under the terms of the GNU Lesser General Public
   5 // License as published by the Free Software Foundation; either
   6 // version 2.1 of the License, or (at your option) any later version.
   7 //
   8 // This library is distributed in the hope that it will be useful,
   9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 // Lesser General Public License for more details.
  12 //
  13 // You should have received a copy of the GNU Lesser General Public
  14 // License along with this library; if not, write to the Free Software
  15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  16 //
  17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  18 //
  19
  20 #ifdef MED_ENABLE_PARMETIS
  21 // include parmetis.h even if it is not needed here
  22 // to avoid inclusion of c++ definitions within extern "C"
  23 // from metis.h from parmetis.h from mpi.h(openmpi) from mpicxx.h
  24 #include <parmetis.h>
  25 #endif
  26 extern "C"
  27 {
  28 #include "metis.h"
  29 }
  30
  31 #include "MEDCouplingMemArray.hxx"
  32 #include "MCAuto.hxx"
  33
  34 #include "RENUMBER_METISRenumbering.hxx"
  35
  36 #ifdef MEDCOUPLING_USE_64BIT_IDS
  37 #define ID_TYPE_SIZE 64
  38 #else
  39 #define ID_TYPE_SIZE 32
  40 #endif
  41
  42 void METISRenumbering::renumber(const mcIdType *graph, const mcIdType *index_graph, mcIdType nbCell, MEDCoupling::DataArrayIdType *&iperm, MEDCoupling::DataArrayIdType *&perm)
  43 {
  44   MEDCoupling::MCAuto<MEDCoupling::DataArrayIdType> out0(MEDCoupling::DataArrayIdType::New()),out1(MEDCoupling::DataArrayIdType::New());
  45   out0->alloc(nbCell,1); out1->alloc(nbCell,1);
  46   out0->fillWithZero(); out1->fillWithZero();
  47   int num_flag=1;
  48   int options=0;
  49
  50 #if ID_TYPE_SIZE == IDXTYPEWIDTH
  51
  52   METIS_NodeND(&nbCell,(idx_t*)index_graph,(idx_t*)graph,&num_flag,&options,out0->getPointer(),out1->getPointer());
  53
  54 #else
  55
  56   mcIdType indexSize = nbCell + 1, graphSize = index_graph[indexSize];
  57   std::vector<idx_t> indexVec( index_graph, index_graph + indexSize );
  58   std::vector<idx_t> graphVec( graph, graph + graphSize );
  59   std::vector<idx_t> out0Vec( nbCell ), out1Vec( nbCell );
  60   idx_t nb = static_cast<idx_t>( nbCell );
  61   METIS_NodeND(&nb,indexVec.data(),graphVec.data(),&num_flag,&options,out0Vec.data(),out1Vec.data());
  62   std::copy( out0Vec.begin(),out0Vec.end(),out0->getPointer() );
  63   std::copy( out1Vec.begin(),out1Vec.end(),out1->getPointer() );
  64
  65 #endif
  66
  67   iperm=out0.retn(); perm=out1.retn();
  68 }