Ifpack2_Zoltan2Partitioner_decl.hpp

// @HEADER
// *****************************************************************************
//       Ifpack2: Templated Object-Oriented Algebraic Preconditioner Package
//
// Copyright 2009 NTESS and the Ifpack2 contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER
 
#ifndef IFPACK2_ZOLTAN2PARTITIONER_DECL_HPP
#define IFPACK2_ZOLTAN2PARTITIONER_DECL_HPP
 
#if defined(HAVE_IFPACK2_ZOLTAN2)
#include "Ifpack2_ConfigDefs.hpp"
#include "Ifpack2_OverlappingPartitioner_decl.hpp"
 
#include "Zoltan2_PartitioningProblem.hpp"
#include "Zoltan2_TpetraRowGraphAdapter.hpp"
 
namespace Ifpack2 {


template <class GraphType>
class Zoltan2Partitioner : public OverlappingPartitioner<GraphType> {
 public:
  typedef typename GraphType::local_ordinal_type local_ordinal_type;
  typedef typename GraphType::global_ordinal_type global_ordinal_type;
  typedef typename GraphType::node_type node_type;
  typedef Tpetra::RowGraph<local_ordinal_type, global_ordinal_type, node_type>
      row_graph_type;
  typedef Zoltan2::TpetraRowGraphAdapter<row_graph_type>
      Zoltan2GraphAdapterType;

  Zoltan2Partitioner(const Teuchos::RCP<const row_graph_type>& graph);

  virtual ~Zoltan2Partitioner();

  void setPartitionParameters(Teuchos::ParameterList& List);

  void computePartitions();
 
 private:
  std::string zoltan2AlgoName_;
};
 
}  // namespace Ifpack2
 
#endif  // HAVE_IFPACK2_ZOLTAN2
#endif  // IFPACK2_ZOLTAN2PARTITIONER_DECL_HPP