Ifpack2_LinePartitioner_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_LINEPARTITIONER_DECL_HPP
#define IFPACK2_LINEPARTITIONER_DECL_HPP
 
#include "Ifpack2_ConfigDefs.hpp"
#include "Ifpack2_OverlappingPartitioner.hpp"
#include "Teuchos_ScalarTraits.hpp"
#include "Tpetra_MultiVector.hpp"
 
namespace Ifpack2 {


 
template <class GraphType, class Scalar>
class LinePartitioner : 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 typename Teuchos::ScalarTraits<Scalar>::magnitudeType magnitude_type;
  typedef Tpetra::MultiVector<magnitude_type, local_ordinal_type, global_ordinal_type, node_type> multivector_type;
 
  typedef typename row_graph_type::nonconst_global_inds_host_view_type nonconst_global_inds_host_view_type;
  typedef typename row_graph_type::nonconst_local_inds_host_view_type nonconst_local_inds_host_view_type;

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

  virtual ~LinePartitioner();

  void setPartitionParameters(Teuchos::ParameterList& List);

  void computePartitions();
 
 private:
  // Useful functions
  int Compute_Blocks_AutoLine(Teuchos::ArrayView<local_ordinal_type> blockIndices) const;
  void local_automatic_line_search(int NumEqns, Teuchos::ArrayView<local_ordinal_type> blockIndices, local_ordinal_type last, local_ordinal_type next, local_ordinal_type LineID, double tol, Teuchos::Array<local_ordinal_type> itemp, Teuchos::Array<magnitude_type> dtemp) const;
 
  // User data
  int NumEqns_;
  Teuchos::RCP<multivector_type> coord_;
  double threshold_;
};
 
}  // namespace Ifpack2
 
#endif  // IFPACK2_LINEPARTITIONER_DECL_HPP