43 #ifndef IFPACK2_SPARSITYFILTER_DEF_HPP 44 #define IFPACK2_SPARSITYFILTER_DEF_HPP 46 #include "Tpetra_Map.hpp" 47 #include "Tpetra_MultiVector.hpp" 48 #include "Tpetra_Vector.hpp" 56 template<
class MatrixType>
58 size_t AllowedNumEntries,
59 LocalOrdinal AllowedBandwidth):
61 AllowedNumEntries_(AllowedNumEntries),
62 AllowedBandwidth_(AllowedBandwidth),
71 A_->getComm()->getSize() != 1 || A_->getNodeNumRows() != A_->getGlobalNumRows(),
72 std::runtime_error,
"Ifpack2::SparsityFilter: " 73 "This class may only be used when A.getComm()->getSize() == 1.");
76 NumRows_ = A_->getNodeNumRows();
81 NumEntries_.resize(NumRows_);
85 MaxNumEntries_ = A_->getNodeMaxNumRowEntries();
86 MaxNumEntriesA_ = A_->getNodeMaxNumRowEntries();
89 Indices_.
resize(MaxNumEntries_);
90 Values_.
resize(MaxNumEntries_);
92 size_t ActualMaxNumEntries = 0;
93 for (
size_t i = 0 ; i < NumRows_ ; ++i) {
94 NumEntries_[i] = MaxNumEntriesA_;
96 A_->getLocalRowCopy(i,Indices_,Values_,Nnz);
100 if (Nnz > ActualMaxNumEntries)
101 ActualMaxNumEntries = Nnz;
104 MaxNumEntries_ = ActualMaxNumEntries;
108 template<
class MatrixType>
112 template<
class MatrixType>
115 return A_->getComm();
119 template<
class MatrixType>
122 return A_->getNode();
126 template<
class MatrixType>
127 Teuchos::RCP<
const Tpetra::Map<
typename MatrixType::local_ordinal_type,
128 typename MatrixType::global_ordinal_type,
129 typename MatrixType::node_type> >
132 return A_->getRowMap();
136 template<
class MatrixType>
137 Teuchos::RCP<
const Tpetra::Map<
typename MatrixType::local_ordinal_type,
138 typename MatrixType::global_ordinal_type,
139 typename MatrixType::node_type> >
142 return A_->getColMap();
146 template<
class MatrixType>
147 Teuchos::RCP<
const Tpetra::Map<
typename MatrixType::local_ordinal_type,
148 typename MatrixType::global_ordinal_type,
149 typename MatrixType::node_type> >
152 return A_->getDomainMap();
156 template<
class MatrixType>
157 Teuchos::RCP<
const Tpetra::Map<
typename MatrixType::local_ordinal_type,
158 typename MatrixType::global_ordinal_type,
159 typename MatrixType::node_type> >
162 return A_->getRangeMap();
166 template<
class MatrixType>
167 Teuchos::RCP<
const Tpetra::RowGraph<
typename MatrixType::local_ordinal_type,
168 typename MatrixType::global_ordinal_type,
169 typename MatrixType::node_type> >
172 throw std::runtime_error(
"Ifpack2::SparsityFilter: does not support getGraph.");
176 template<
class MatrixType>
183 template<
class MatrixType>
190 template<
class MatrixType>
198 template<
class MatrixType>
205 template<
class MatrixType>
208 return A_->getIndexBase();
212 template<
class MatrixType>
219 template<
class MatrixType>
226 template<
class MatrixType>
229 throw std::runtime_error(
"Ifpack2::SparsityFilter does not implement getNumEntriesInGlobalRow.");
233 template<
class MatrixType>
236 return NumEntries_[localRow];
240 template<
class MatrixType>
243 return A_->getGlobalNumDiags();
247 template<
class MatrixType>
250 return A_->getNodeNumDiags();
254 template<
class MatrixType>
257 return MaxNumEntries_;
261 template<
class MatrixType>
264 return MaxNumEntries_;
268 template<
class MatrixType>
275 template<
class MatrixType>
278 return A_->isLowerTriangular();
282 template<
class MatrixType>
285 return A_->isUpperTriangular();
289 template<
class MatrixType>
292 return A_->isLocallyIndexed();
296 template<
class MatrixType>
299 return A_->isGloballyIndexed();
303 template<
class MatrixType>
306 return A_->isFillComplete();
310 template<
class MatrixType>
314 size_t &NumEntries)
const 316 throw std::runtime_error(
"Ifpack2::SparsityFilter does not implement getGlobalRowCopy.");
320 template<
class MatrixType>
324 size_t &NumEntries)
const 326 TEUCHOS_TEST_FOR_EXCEPTION((LocalRow < 0 || (
size_t) LocalRow >= NumRows_ || (
size_t) Indices.
size() < NumEntries_[LocalRow]), std::runtime_error,
"Ifpack2::SparsityFilter::getLocalRowCopy invalid row or array size.");
334 size_t A_NumEntries=0;
335 A_->getLocalRowCopy(LocalRow,Indices_(),Values_(),A_NumEntries);
339 if (A_NumEntries > AllowedNumEntries_) {
341 for (
size_t i = 0 ; i < A_NumEntries ; ++i) {
343 if (Indices_[i] == LocalRow)
351 std::sort(Values2.rbegin(),Values2.rend());
353 Threshold = Values2[AllowedNumEntries_ - 1];
361 for (
size_t i = 0 ; i < A_NumEntries ; ++i) {
362 if (std::abs(Indices_[i] - LocalRow) > AllowedBandwidth_)
368 Values[NumEntries] = Values_[i];
369 Indices[NumEntries] = Indices_[i];
372 if (NumEntries > AllowedNumEntries_)
379 template<
class MatrixType>
384 throw std::runtime_error(
"Ifpack2::SparsityFilter: does not support getGlobalRowView.");
388 template<
class MatrixType>
393 throw std::runtime_error(
"Ifpack2::SparsityFilter: does not support getLocalRowView.");
397 template<
class MatrixType>
401 return A_->getLocalDiagCopy(diag);
405 template<
class MatrixType>
408 throw std::runtime_error(
"Ifpack2::SparsityFilter does not support leftScale.");
412 template<
class MatrixType>
415 throw std::runtime_error(
"Ifpack2::SparsityFilter does not support rightScale.");
419 template<
class MatrixType>
421 Tpetra::MultiVector<Scalar,LocalOrdinal,GlobalOrdinal,Node> &Y,
429 "Ifpack2::SparsityFilter::apply ERROR: X.getNumVectors() != Y.getNumVectors().");
436 size_t NumVectors = Y.getNumVectors();
438 for (
size_t i = 0 ; i < NumRows_ ; ++i) {
441 getLocalRowCopy(i,Indices_(),Values_(),Nnz);
442 if (mode==Teuchos::NO_TRANS){
443 for (
size_t j = 0 ; j < Nnz ; ++j)
444 for (
size_t k = 0 ; k < NumVectors ; ++k)
445 y_ptr[k][i] += Values_[j] * x_ptr[k][Indices_[j]];
447 else if (mode==Teuchos::TRANS){
448 for (
size_t j = 0 ; j < Nnz ; ++j)
449 for (
size_t k = 0 ; k < NumVectors ; ++k)
450 y_ptr[k][Indices_[j]] += Values_[j] * x_ptr[k][i];
453 for (
size_t j = 0 ; j < Nnz ; ++j)
454 for (
size_t k = 0 ; k < NumVectors ; ++k)
462 template<
class MatrixType>
469 template<
class MatrixType>
476 template<
class MatrixType>
479 throw std::runtime_error(
"Ifpack2::SparsityFilter does not implement getFrobeniusNorm.");
484 #define IFPACK2_SPARSITYFILTER_INSTANT(S,LO,GO,N) \ 485 template class Ifpack2::SparsityFilter< Tpetra::RowMatrix<S, LO, GO, N> >; virtual void getLocalDiagCopy(Tpetra::Vector< Scalar, LocalOrdinal, GlobalOrdinal, Node > &diag) const
Get a copy of the diagonal entries owned by this node, with local row indices.
Definition: Ifpack2_SparsityFilter_def.hpp:398
virtual size_t getNodeNumCols() const
Returns the number of columns needed to apply the forward operator on this node, i.e., the number of elements listed in the column map.
Definition: Ifpack2_SparsityFilter_def.hpp:199
virtual size_t getGlobalMaxNumRowEntries() const
Returns the maximum number of entries across all rows/columns on all nodes.
Definition: Ifpack2_SparsityFilter_def.hpp:255
virtual global_size_t getGlobalNumRows() const
Returns the number of global rows in this matrix.
Definition: Ifpack2_SparsityFilter_def.hpp:177
virtual void getGlobalRowCopy(GlobalOrdinal GlobalRow, const Teuchos::ArrayView< GlobalOrdinal > &Indices, const Teuchos::ArrayView< Scalar > &Values, size_t &NumEntries) const
Extract a list of entries in a specified global row of this matrix. Put into pre-allocated storage...
Definition: Ifpack2_SparsityFilter_def.hpp:311
virtual global_size_t getGlobalNumDiags() const
Returns the number of global diagonal entries, based on global row/column index comparisons.
Definition: Ifpack2_SparsityFilter_def.hpp:241
#define TEUCHOS_TEST_FOR_EXCEPTION(throw_exception_test, Exception, msg)
virtual bool supportsRowViews() const
Returns true if RowViews are supported.
Definition: Ifpack2_SparsityFilter_def.hpp:470
virtual size_t getNodeNumDiags() const
Returns the number of local diagonal entries, based on global row/column index comparisons.
Definition: Ifpack2_SparsityFilter_def.hpp:248
virtual ~SparsityFilter()
Destructor.
Definition: Ifpack2_SparsityFilter_def.hpp:109
virtual bool hasColMap() const
Indicates whether this matrix has a well-defined column map.
Definition: Ifpack2_SparsityFilter_def.hpp:269
virtual Teuchos::RCP< const Teuchos::Comm< int > > getComm() const
Returns the communicator.
Definition: Ifpack2_SparsityFilter_def.hpp:113
SparsityFilter(const Teuchos::RCP< const row_matrix_type > &Matrix, size_t AllowedNumEntries, LocalOrdinal AllowedBandwidth=-Teuchos::ScalarTraits< LocalOrdinal >::one())
Constructor.
Definition: Ifpack2_SparsityFilter_def.hpp:57
virtual size_t getNumEntriesInGlobalRow(GlobalOrdinal globalRow) const
Returns the current number of entries on this node in the specified global row.
Definition: Ifpack2_SparsityFilter_def.hpp:227
virtual bool isGloballyIndexed() const
If matrix indices are in the global range, this function returns true. Otherwise, this function retur...
Definition: Ifpack2_SparsityFilter_def.hpp:297
virtual bool isLocallyIndexed() const
If matrix indices are in the local range, this function returns true. Otherwise, this function return...
Definition: Ifpack2_SparsityFilter_def.hpp:290
LinearOp zero(const VectorSpace &vs)
virtual void getLocalRowView(LocalOrdinal DropRow, Teuchos::ArrayView< const LocalOrdinal > &indices, Teuchos::ArrayView< const Scalar > &values) const
Extract a const, non-persisting view of local indices in a specified row of the matrix.
Definition: Ifpack2_SparsityFilter_def.hpp:389
virtual size_t getNodeMaxNumRowEntries() const
Returns the maximum number of entries across all rows/columns on this node.
Definition: Ifpack2_SparsityFilter_def.hpp:262
virtual bool hasTransposeApply() const
Indicates whether this operator supports applying the adjoint operator.
Definition: Ifpack2_SparsityFilter_def.hpp:463
virtual bool isFillComplete() const
Returns true if fillComplete() has been called.
Definition: Ifpack2_SparsityFilter_def.hpp:304
virtual mag_type getFrobeniusNorm() const
Returns the Frobenius norm of the matrix.
Definition: Ifpack2_SparsityFilter_def.hpp:477
virtual GlobalOrdinal getIndexBase() const
Returns the index base for global indices for this matrix.
Definition: Ifpack2_SparsityFilter_def.hpp:206
virtual global_size_t getGlobalNumEntries() const
Returns the global number of entries in this matrix.
Definition: Ifpack2_SparsityFilter_def.hpp:213
virtual size_t getNumEntriesInLocalRow(LocalOrdinal localRow) const
Returns the current number of entries on this node in the specified local row.
Definition: Ifpack2_SparsityFilter_def.hpp:234
virtual Teuchos::RCP< const Tpetra::Map< LocalOrdinal, GlobalOrdinal, Node > > getRangeMap() const
Returns the Map that describes the range distribution in this matrix.
Definition: Ifpack2_SparsityFilter_def.hpp:160
virtual void apply(const Tpetra::MultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > &X, Tpetra::MultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > &Y, Teuchos::ETransp mode=Teuchos::NO_TRANS, Scalar alpha=Teuchos::ScalarTraits< Scalar >::one(), Scalar beta=Teuchos::ScalarTraits< Scalar >::zero()) const
Computes the operator-multivector application.
Definition: Ifpack2_SparsityFilter_def.hpp:420
void resize(size_type new_size, const value_type &x=value_type())
virtual void getGlobalRowView(GlobalOrdinal GlobalRow, Teuchos::ArrayView< const GlobalOrdinal > &indices, Teuchos::ArrayView< const Scalar > &values) const
Extract a const, non-persisting view of global indices in a specified row of the matrix.
Definition: Ifpack2_SparsityFilter_def.hpp:380
virtual Teuchos::RCP< const Tpetra::Map< LocalOrdinal, GlobalOrdinal, Node > > getRowMap() const
Returns the Map that describes the row distribution in this matrix.
Definition: Ifpack2_SparsityFilter_def.hpp:130
virtual void getLocalRowCopy(LocalOrdinal DropRow, const Teuchos::ArrayView< LocalOrdinal > &Indices, const Teuchos::ArrayView< Scalar > &Values, size_t &NumEntries) const
Extract a list of entries in a specified local row of the graph. Put into storage allocated by callin...
Definition: Ifpack2_SparsityFilter_def.hpp:321
static magnitudeType magnitude(T a)
virtual bool isUpperTriangular() const
Indicates whether this matrix is upper triangular.
Definition: Ifpack2_SparsityFilter_def.hpp:283
virtual void rightScale(const Tpetra::Vector< Scalar, LocalOrdinal, GlobalOrdinal, Node > &x)
Scales the RowMatrix on the right with the Vector x.
Definition: Ifpack2_SparsityFilter_def.hpp:413
virtual void leftScale(const Tpetra::Vector< Scalar, LocalOrdinal, GlobalOrdinal, Node > &x)
Scales the RowMatrix on the left with the Vector x.
Definition: Ifpack2_SparsityFilter_def.hpp:406
virtual global_size_t getGlobalNumCols() const
Returns the number of global columns in this matrix.
Definition: Ifpack2_SparsityFilter_def.hpp:184
virtual bool isLowerTriangular() const
Indicates whether this matrix is lower triangular.
Definition: Ifpack2_SparsityFilter_def.hpp:276
Preconditioners and smoothers for Tpetra sparse matrices.
Definition: Ifpack2_AdditiveSchwarz_decl.hpp:72
virtual Teuchos::RCP< const Tpetra::Map< LocalOrdinal, GlobalOrdinal, Node > > getDomainMap() const
Returns the Map that describes the domain distribution in this matrix.
Definition: Ifpack2_SparsityFilter_def.hpp:150
virtual Teuchos::RCP< const Tpetra::RowGraph< LocalOrdinal, GlobalOrdinal, Node > > getGraph() const
Returns the RowGraph associated with this matrix.
Definition: Ifpack2_SparsityFilter_def.hpp:170
virtual size_t getNodeNumEntries() const
Returns the local number of entries in this matrix.
Definition: Ifpack2_SparsityFilter_def.hpp:220
virtual size_t getNodeNumRows() const
Returns the number of rows owned on the calling node.
Definition: Ifpack2_SparsityFilter_def.hpp:191
virtual Teuchos::RCP< const Tpetra::Map< LocalOrdinal, GlobalOrdinal, Node > > getColMap() const
Returns the Map that describes the column distribution in this matrix.
Definition: Ifpack2_SparsityFilter_def.hpp:140
virtual Teuchos::RCP< Node > getNode() const
Returns the underlying node.
Definition: Ifpack2_SparsityFilter_def.hpp:120