src/MEDPartitioner/MEDPARTITIONER_metis.c

   1 // Copyright (C) 2007-2016  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 // Creation of this C code is forced by the following.
  21 //
  22 // In case if Metis is a part of Parmetis V3, extern "C" {#include "metis.h"} causes
  23 // inclusion of C++ code of MPI via parmetis.h <- mpi.h <- mpicxx.h
  24 // that breaks compilation. To workaround this problem we create a wrapping C
  25 // function, inclusion of whose declaration causes no problem.
  26
  27 #include "MEDPARTITIONER_metis.h"
  28
  29 #if defined(MED_ENABLE_METIS) & !defined(MED_ENABLE_PARMETIS)
  30   #ifndef MED_ENABLE_METIS_V5
  31     #include "defs.h"
  32   #endif // MED_ENABLE_METIS_V5
  33   #include "metis.h"
  34   #ifdef MED_ENABLE_METIS_V5
  35     #define idxtype idx_t
  36   #endif // MED_ENABLE_METIS_V5
  37 #else
  38   typedef int idxtype;
  39 #endif // defined(MED_ENABLE_METIS) & !defined(MED_ENABLE_PARMETIS)
  40
  41 void MEDPARTITIONER_METIS_PartGraphRecursive(int *nvtxs, idxtype *xadj, idxtype *adjncy, idxtype *vwgt,
  42                                              idxtype *adjwgt, int *wgtflag, int *numflag, int *nparts,
  43                                              int *options, int *edgecut, idxtype *part)
  44 {
  45 #if defined(MED_ENABLE_METIS)
  46   #ifndef MED_ENABLE_METIS_V5
  47   METIS_PartGraphRecursive(nvtxs, xadj, adjncy, vwgt,
  48     adjwgt, wgtflag, numflag, nparts,
  49                            options, edgecut, part);
  50   #else
  51   int ncon=1;
  52   options[METIS_OPTION_NCUTS]=1;
  53   options[METIS_OPTION_NITER]=1;
  54   options[METIS_OPTION_UFACTOR]=1;
  55   METIS_PartGraphRecursive(nvtxs, &ncon, xadj, adjncy, vwgt, 0 /* vsize*/,
  56                            adjwgt, nparts,/* tpwgts*/ 0,/* ubvec */ 0,
  57                            options, edgecut, part);
  58   #endif
  59 #endif
  60 }
  61
  62 void MEDPARTITIONER_METIS_PartGraphKway(int *nvtxs, idxtype *xadj, idxtype *adjncy, idxtype *vwgt,
  63                                         idxtype *adjwgt, int *wgtflag, int *numflag, int *nparts,
  64                                         int *options, int *edgecut, idxtype *part)
  65 {
  66 #if defined(MED_ENABLE_METIS)
  67   #ifndef MED_ENABLE_METIS_V5
  68   METIS_PartGraphKway(nvtxs, xadj, adjncy, vwgt,
  69     adjwgt, wgtflag, numflag, nparts,
  70     options, edgecut, part);
  71   #else
  72   int ncon=1;
  73   options[METIS_OPTION_NCUTS]=1;
  74   options[METIS_OPTION_NITER]=1;
  75   options[METIS_OPTION_UFACTOR]=1;
  76   METIS_PartGraphKway(nvtxs, &ncon, xadj, adjncy, vwgt, 0 /* vsize*/,
  77                       adjwgt, nparts, 0 , 0 /* ubvec */,
  78                       options, edgecut, part);
  79   #endif
  80 #endif
  81 }