Back to index

salome-med  6.5.0
MEDPARTITIONER_Graph.hxx
Go to the documentation of this file.
00001 // Copyright (C) 2007-2012  CEA/DEN, EDF R&D
00002 //
00003 // This library is free software; you can redistribute it and/or
00004 // modify it under the terms of the GNU Lesser General Public
00005 // License as published by the Free Software Foundation; either
00006 // version 2.1 of the License.
00007 //
00008 // This library is distributed in the hope that it will be useful,
00009 // but WITHOUT ANY WARRANTY; without even the implied warranty of
00010 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
00011 // Lesser General Public License for more details.
00012 //
00013 // You should have received a copy of the GNU Lesser General Public
00014 // License along with this library; if not, write to the Free Software
00015 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
00016 //
00017 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
00018 //
00019 
00020 #ifndef __MEDPARTITIONER_GRAPH_HXX__
00021 #define __MEDPARTITIONER_GRAPH_HXX__
00022 
00023 #include "MEDPARTITIONER.hxx"
00024 #include "MEDPARTITIONER_SkyLineArray.hxx"
00025 
00026 #include <string>
00027 
00028 namespace MEDPARTITIONER 
00029 {
00030   class ParaDomainSelector;
00031   class MEDPARTITIONER_EXPORT Graph
00032   {
00033   public:
00034     typedef enum {METIS,SCOTCH} splitter_type;
00035 
00036     Graph() { }
00037     //creates a graph from a SKYLINEARRAY
00038     Graph(MEDPARTITIONER::SkyLineArray* graph, int* edgeweight=0);
00039     virtual ~Graph();
00040 
00041     void setEdgesWeights(int *edgeweight) { _edge_weight=edgeweight; }
00042     void setVerticesWeights(int *cellweight) { _cell_weight=cellweight; }
00043     
00044     //computes partitioning of the graph
00045     virtual void partGraph(int ndomain, const std::string&, ParaDomainSelector *sel=0) = 0;
00046     
00047     //returns the partitioning
00048     const int *getPart() const { return _partition->getValue(); }
00049     
00050     //returns the number of graph vertices (which can correspond to the cells in the mesh!)
00051     int nbVertices() const { return _graph->getNumberOf(); }
00052     
00053     const SkyLineArray *getGraph() const { return _graph; }
00054 
00055   protected:
00056     SkyLineArray* _graph;
00057     SkyLineArray* _partition;
00058     int* _edge_weight;  
00059     int* _cell_weight;
00060   };
00061 }
00062 #endif