/usr/include/dolfin/graph/ParMETIS.h is in libdolfin-dev 2016.2.0-2.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 | // Copyright (C) 2008-2009 Niclas Jansson, Ola Skavhaug and Anders Logg
//
// This file is part of DOLFIN.
//
// DOLFIN 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 3 of the License, or
// (at your option) any later version.
//
// DOLFIN 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 DOLFIN. If not, see <http://www.gnu.org/licenses/>.
//
// Modified by Garth N. Wells, 2010
//
// First added: 2010-02-10
// Last changed:
#ifndef __PARMETIS_PARTITIONER_H
#define __PARMETIS_PARTITIONER_H
#include <cstdint>
#include <cstddef>
#include <vector>
#include <dolfin/common/MPI.h>
#include <dolfin/common/Set.h>
#ifdef HAS_PARMETIS
#include <parmetis.h>
#endif
#include "CSRGraph.h"
namespace dolfin
{
class CellType;
/// This class provides an interface to ParMETIS
class ParMETIS
{
public:
/// Compute cell partition from local mesh data. The output
/// vector cell_partition contains the desired destination process
/// numbers for each cell. Cells shared on multiple processes
/// have an entry in ghost_procs pointing to the set of sharing
/// process numbers. The mode argument determines which ParMETIS
/// function is called. It can be one of "partition",
/// "adaptive_repartition" or "refine". For meshes that have
/// already been partitioned or are already well partitioned, it
/// can be advantageous to use "adaptive_repartition" or "refine".
static void compute_partition(const MPI_Comm mpi_comm,
std::vector<int>& cell_partition,
std::map<std::int64_t, std::vector<int>>& ghost_procs,
const boost::multi_array<std::int64_t, 2>& cell_vertices,
const std::size_t num_global_vertices,
const CellType& cell_type,
const std::string mode="partition");
private:
#ifdef HAS_PARMETIS
// Create a dual graph from the cell-vertex topology using
// ParMETIS built in ParMETIS_V3_Mesh2Dual
static CSRGraph<idx_t>
dual_graph(MPI_Comm mpi_comm,
const boost::multi_array<std::int64_t, 2>& cell_vertices,
const int num_vertices_per_cell);
// Standard ParMETIS partition. CSRGraph should be const, but
// ParMETIS accesses it non-const, so has to be non-const here
template <typename T>
static void partition(MPI_Comm mpi_comm,
CSRGraph<T>& csr_graph,
std::vector<int>& cell_partition,
std::map<std::int64_t, std::vector<int>>& ghost_procs);
// ParMETIS adaptive repartition. CSRGraph should be const, but
// ParMETIS accesses it non-const, so has to be non-const here
template <typename T>
static void adaptive_repartition(MPI_Comm mpi_comm,
CSRGraph<T>& csr_graph,
std::vector<int>& cell_partition);
// ParMETIS refine repartition. CSRGraph should be const, but
// ParMETIS accesses it non-const, so has to be non-const here
template <typename T>
static void refine(MPI_Comm mpi_comm, CSRGraph<T>& csr_graph,
std::vector<int>& cell_partition);
#endif
};
}
#endif
|