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