src/MEDSPLITTER/MEDSPLITTER_Graph.hxx

   1 // Copyright (C) 2007-2012  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.
   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 #ifndef GRAPH_HXX_
  21 #define GRAPH_HXX_
  22
  23 #include "MEDSPLITTER.hxx"
  24
  25 #include "MEDMEM_SkyLineArray.hxx"
  26
  27 namespace MEDSPLITTER {
  28
  29 class ParaDomainSelector;
  30
  31 class MEDSPLITTER_EXPORT Graph
  32 {
  33   public:
  34     typedef enum {METIS,SCOTCH} splitter_type;
  35
  36     Graph(){}
  37
  38     //creates a graph from a SKYLINEARRAY
  39     Graph(MEDMEM::MEDSKYLINEARRAY* graph, int* edgeweight=0);
  40
  41     virtual ~Graph();
  42
  43     void setEdgesWeights(int* edgeweight){m_edgeweight=edgeweight;}
  44     void setVerticesWeights(int* cellweight){m_cellweight=cellweight;}
  45
  46     //computes partitioning of the graph
  47     virtual void partGraph(int ndomain, const string&, ParaDomainSelector* sel=0)=0;
  48
  49     //! returns the partitioning
  50     const int* getPart() const {return m_partition->getValue();}
  51
  52     //! returns the number of graph vertices (which can correspond to the cells in the mesh!)
  53     int nbVertices() const {return m_graph->getNumberOf();}
  54
  55     const MEDMEM::MEDSKYLINEARRAY* getGraph() const {return m_graph;}
  56
  57   protected:
  58
  59     MEDMEM::MEDSKYLINEARRAY* m_graph;
  60
  61     MEDMEM::MEDSKYLINEARRAY* m_partition;
  62
  63     int* m_edgeweight;
  64
  65     int* m_cellweight;
  66   };
  67
  68 }
  69 #endif /*GRAPH_HXX_*/