From: cvw <cvw>
Date: Tue, 27 Mar 2012 13:18:20 +0000 (+0000)
Subject: ParMetisGraph or exclusive MetisGraph
X-Git-Tag: V6_main_FINAL~714
X-Git-Url: http://git.salome-platform.org/gitweb/?a=commitdiff_plain;h=efb2caf1e46dda5124cdd9bda9877b7172db9c15;p=tools%2Fmedcoupling.git

ParMetisGraph or exclusive MetisGraph
---

diff --git a/src/MEDPartitioner/MEDPARTITIONER_ParMetisGraph.cxx b/src/MEDPartitioner/MEDPARTITIONER_ParMetisGraph.cxx
new file mode 100644
index 000000000..719690af4
--- /dev/null
+++ b/src/MEDPartitioner/MEDPARTITIONER_ParMetisGraph.cxx
@@ -0,0 +1,141 @@
+// Copyright (C) 2007-2011  CEA/DEN, EDF R&D
+//
+// This library is free software; you can redistribute it and/or
+// modify it under the terms of the GNU Lesser General Public
+// License as published by the Free Software Foundation; either
+// version 2.1 of the License.
+//
+// This library is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+// Lesser General Public License for more details.
+//
+// You should have received a copy of the GNU Lesser General Public
+// License along with this library; if not, write to the Free Software
+// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
+//
+// See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
+//
+
+#include "MEDPARTITIONER_MetisGraph.hxx"
+#include "MEDPARTITIONER_ParaDomainSelector.hxx"
+#include "MEDPARTITIONER_Utils.hxx"
+
+#include "InterpKernelException.hxx"
+
+#include <iostream>
+
+#ifdef MED_ENABLE_PARMETIS
+#include <mpi.h>
+#include "parmetis.h"
+#endif
+
+using namespace MEDPARTITIONER;
+
+METISGraph::METISGraph():Graph()
+{
+}
+
+METISGraph::METISGraph(MEDPARTITIONER::SkyLineArray* graph, int* edgeweight)
+  :Graph(graph,edgeweight)
+{
+}
+
+METISGraph::~METISGraph()
+{
+}
+
+void METISGraph::partGraph(int ndomain,
+                           const std::string& options_string,
+                           ParaDomainSelector *parallelizer)
+{
+  using std::vector;
+  vector<int> ran,vx,va; //for randomize
+  
+  if (MyGlobals::_Verbose>10)
+    std::cout << "proc " << MyGlobals::_Rank << " : METISGraph::partGraph" << std::endl;
+  
+  // number of graph vertices
+  int n=_graph->getNumberOf();
+  //graph
+  int * xadj=const_cast<int*>(_graph->getIndex());
+  int * adjncy=const_cast<int*>(_graph->getValue());
+  //constraints
+  int * vwgt=_cell_weight;
+  int * adjwgt=_edge_weight;
+  int wgtflag=(_edge_weight!=0)?1:0+(_cell_weight!=0)?2:0;
+  //base 0 or 1
+  int base=0;
+  //ndomain
+  int nparts=ndomain;
+  //options
+  /*
+    (0=default_option,option,random_seed) see defs.h
+    #define PMV3_OPTION_DBGLVL 1
+    #define PMV3_OPTION_SEED 2
+    #define PMV3_OPTION_IPART 3
+    #define PMV3_OPTION_PSR 3
+    seems no changes int options[4]={1,0,33,0}; //test for a random seed of 33
+  */
+  int options[4]={0,0,0,0};
+  // output parameters
+  int edgecut;
+  int* partition=new int[n];
+
+#if !defined(MED_ENABLE_PARMETIS)
+  throw INTERP_KERNEL::Exception("METISGraph::partGraph : PARMETIS is not available. Check your products, please.");
+#else
+
+  if (MyGlobals::_Verbose>10) 
+    std::cout << "proc " << MyGlobals::_Rank << " : METISGraph::partGraph ParMETIS_PartKway new" << std::endl;
+  int * vtxdist=parallelizer->getProcVtxdist();
+  MPI_Comm comm=MPI_COMM_WORLD;
+  ParMETIS_PartKway(vtxdist, xadj, adjncy, vwgt, 
+                                    adjwgt, &wgtflag, &base, &nparts, options, 
+                                    &edgecut, partition, &comm );
+
+
+  /*doc from parmetis.h
+    void __cdecl ParMETIS_PartKway(
+    idxtype *vtxdist, idxtype *xadj, idxtype *adjncy, idxtype *vwgt, 
+    idxtype *adjwgt, int *wgtflag, int *numflag, int *nparts, int *options, 
+    int *edgecut, idxtype *part, MPI_Comm *comm);
+
+    void __cdecl ParMETIS_V3_PartKway(
+    idxtype *vtxdist, idxtype *xadj, idxtype *adjncy, idxtype *vwgt, 
+    idxtype *adjwgt, int *wgtflag, int *numflag, int *ncon, int *nparts, 
+    float *tpwgts, float *ubvec, int *options, int *edgecut, idxtype *part, 
+    MPI_Comm *comm);
+  */
+
+  vector<int> index(n+1);
+  vector<int> value(n);
+  index[0]=0;
+  if (ran.size()>0 && MyGlobals::_Atomize==0) //there is randomize
+    {
+      if (MyGlobals::_Is0verbose>100)
+        std::cout << "randomize" << std::endl;
+      for (int i=0; i<n; i++)
+        {
+          index[i+1]=index[i]+1;
+          value[ran[i]]=partition[i];
+        }
+    }
+  else
+    {
+      for (int i=0; i<n; i++)
+        {
+          index[i+1]=index[i]+1;
+          value[i]=partition[i];
+        }
+    }
+  delete [] partition;
+
+  //creating a skylinearray with no copy of the index and partition array
+  //the fifth argument true specifies that only the pointers are passed 
+  //to the object
+  
+  _partition = new MEDPARTITIONER::SkyLineArray(index,value);
+#endif
+}
+