Loading...
Searching...
No Matches
ZoltanInterface.h
// Copyright (C) 2010 Garth N. Wells
//
// 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/>.
//
// First added: 2010-11-16
// Last changed:
#ifndef __DOLFIN_ZOLTAN_GRAPH_COLORING_H
#define __DOLFIN_ZOLTAN_GRAPH_COLORING_H
#include <vector>
#ifdef HAS_TRILINOS
#include <zoltan_cpp.h>
#endif
#include "Graph.h"
namespace dolfin
{
class Mesh;
class ZoltanInterface
{
public:
static std::size_t
compute_local_vertex_coloring(const Graph& graph,
std::vector<std::size_t>& colors);
private:
#ifdef HAS_TRILINOS
class ZoltanGraphInterface
{
public:
ZoltanGraphInterface(const Graph& graph);
private:
// Graph object
const Graph& _graph;
public:
void num_vertex_edges(unsigned int* num_edges) const;
// Zoltan call-back functions
static int get_number_of_objects(void* data, int* ierr);
static void get_object_list(void* data, int sizeGID, int sizeLID,
ZOLTAN_ID_PTR globalID,
ZOLTAN_ID_PTR localID, int wgt_dim,
float* obj_wgts, int* ierr);
static void get_number_edges(void* data, int num_gid_entries,
int num_lid_entries,
int num_obj, ZOLTAN_ID_PTR global_ids,
ZOLTAN_ID_PTR local_ids, int* num_edges,
int* ierr);
static void get_all_edges(void* data, int num_gid_entries,
int num_lid_entries, int num_obj,
ZOLTAN_ID_PTR global_ids,
ZOLTAN_ID_PTR local_ids,
int* num_edges,
ZOLTAN_ID_PTR nbor_global_id,
int* nbor_procs, int wgt_dim,
float* ewgts, int* ierr);
};
#endif
};
}
#endif
dolfin::Mesh
Definition Mesh.h:84
dolfin::ZoltanInterface
Definition ZoltanInterface.h:41
dolfin::ZoltanInterface::compute_local_vertex_coloring
static std::size_t compute_local_vertex_coloring(const Graph &graph, std::vector< std::size_t > &colors)
Compute vertex colors.
Definition ZoltanInterface.cpp:38
dolfin
Definition adapt.h:30
dolfin::Graph
std::vector< graph_set_type > Graph
Vector of unordered Sets.
Definition Graph.h:39