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