parmetis_partitioner.h

Go to the documentation of this file.

// The libMesh Finite Element Library.
// Copyright (C) 2002-2014 Benjamin S. Kirk, John W. Peterson, Roy H. Stogner

// 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, or (at your option) any later version.

// 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



#ifndef LIBMESH_PARMETIS_PARTITIONER_H
#define LIBMESH_PARMETIS_PARTITIONER_H

// Local Includes -----------------------------------
#include "libmesh/id_types.h"
#include "libmesh/partitioner.h"
#include "libmesh/vectormap.h"

typedef int idxtype;

// C++ Includes   -----------------------------------
#include <cstddef>
#include <vector>

namespace libMesh
{



// ------------------------------------------------------------
// ParmetisPartitioner class definition
class ParmetisPartitioner : public Partitioner
{
public:

  ParmetisPartitioner () {}

  virtual UniquePtr<Partitioner> clone () const
  {
    return UniquePtr<Partitioner>(new ParmetisPartitioner());
  }


protected:

  virtual void _do_repartition (MeshBase& mesh,
                                const unsigned int n);

  virtual void _do_partition (MeshBase& mesh,
                              const unsigned int n);

private:

  // These methods & data only need to be available if the
  // ParMETIS library is available.
#ifdef LIBMESH_HAVE_PARMETIS

  void initialize (const MeshBase& mesh, const unsigned int n_sbdmns);

  void build_graph (const MeshBase& mesh);

  void assign_partitioning (MeshBase& mesh);

  std::vector<dof_id_type> _n_active_elem_on_proc;

  vectormap<dof_id_type, dof_id_type> _global_index_by_pid_map;

  std::vector<int>    _vtxdist;
  std::vector<int>    _xadj;
  std::vector<int>    _adjncy;
  std::vector<int>    _part;
  std::vector<float>  _tpwgts;
  std::vector<float>  _ubvec;
  std::vector<int>    _options;
  std::vector<int>    _vwgt;

  int _wgtflag;
  int _ncon;
  int _numflag;
  int _nparts;
  int _edgecut;

#endif
};


} // namespace libMesh



#endif // LIBMESH_PARMETIS_PARTITIONER_H