43 #ifndef IFPACK2_SINGLETONFILTER_DEF_HPP
44 #define IFPACK2_SINGLETONFILTER_DEF_HPP
45 #include "Ifpack2_SingletonFilter_decl.hpp"
47 #include "Tpetra_ConfigDefs.hpp"
48 #include "Tpetra_RowMatrix.hpp"
49 #include "Tpetra_Map.hpp"
50 #include "Tpetra_MultiVector.hpp"
51 #include "Tpetra_Vector.hpp"
55 template<
class MatrixType>
66 if (A_->getComm()->getSize() != 1 || A_->getNodeNumRows() != A_->getGlobalNumRows()) {
67 throw std::runtime_error(
"Ifpack2::SingeltonFilter can be used with Comm().getSize() == 1 only. This class is a tool for Ifpack2_AdditiveSchwarz, and it is not meant to be used otherwise.");
71 size_t NumRowsA_ = A_->getNodeNumRows();
75 MaxNumEntriesA_ = A_->getNodeMaxNumRowEntries();
78 Indices_.resize(MaxNumEntriesA_);
79 Values_.resize(MaxNumEntriesA_);
82 Reorder_.resize(NumRowsA_);
83 Reorder_.assign(Reorder_.size(),-1);
87 for (
size_t i = 0 ; i < NumRowsA_ ; ++i) {
89 A_->getLocalRowCopy(i,Indices_,Values_,Nnz);
91 Reorder_[i] = NumRows_++;
99 InvReorder_.resize(NumRows_);
100 for (
size_t i = 0 ; i < NumRowsA_ ; ++i) {
103 InvReorder_[Reorder_[i]] = i;
105 NumEntries_.resize(NumRows_);
106 SingletonIndex_.resize(NumSingletons_);
111 for (
size_t i = 0 ; i < NumRowsA_ ; ++i) {
113 A_->getLocalRowCopy(i,Indices_,Values_,Nnz);
114 LocalOrdinal ii = Reorder_[i];
116 NumEntries_[ii] = Nnz;
118 if (Nnz > MaxNumEntries_)
119 MaxNumEntries_ = Nnz;
122 SingletonIndex_[count] = i;
128 ReducedMap_ = Teuchos::rcp(
new Tpetra::Map<LocalOrdinal,GlobalOrdinal,Node>(NumRows_,0,A_->getComm()) );
131 Diagonal_ = Teuchos::rcp(
new Tpetra::Vector<Scalar,LocalOrdinal,GlobalOrdinal,Node>(ReducedMap_) );
133 Tpetra::Vector<Scalar,LocalOrdinal,GlobalOrdinal,Node> DiagonalA(A_->getRowMap());
134 A_->getLocalDiagCopy(DiagonalA);
135 const Teuchos::ArrayRCP<const Scalar> & DiagonalAview = DiagonalA.get1dView();
136 for (
size_t i = 0 ; i < NumRows_ ; ++i) {
137 LocalOrdinal ii = InvReorder_[i];
138 Diagonal_->replaceLocalValue((LocalOrdinal)i,DiagonalAview[ii]);
142 template<
class MatrixType>
145 template<
class MatrixType>
146 Teuchos::RCP<const Teuchos::Comm<int> >
149 return A_->getComm();
152 #ifdef TPETRA_ENABLE_DEPRECATED_CODE
153 template<
class MatrixType>
155 Teuchos::RCP<typename MatrixType::node_type>
158 return Teuchos::null;
160 #endif // TPETRA_ENABLE_DEPRECATED_CODE
162 template<
class MatrixType>
163 Teuchos::RCP<
const Tpetra::Map<
typename MatrixType::local_ordinal_type,
164 typename MatrixType::global_ordinal_type,
165 typename MatrixType::node_type> >
171 template<
class MatrixType>
172 Teuchos::RCP<
const Tpetra::Map<
typename MatrixType::local_ordinal_type,
173 typename MatrixType::global_ordinal_type,
174 typename MatrixType::node_type> >
180 template<
class MatrixType>
181 Teuchos::RCP<
const Tpetra::Map<
typename MatrixType::local_ordinal_type,
182 typename MatrixType::global_ordinal_type,
183 typename MatrixType::node_type> >
189 template<
class MatrixType>
190 Teuchos::RCP<
const Tpetra::Map<
typename MatrixType::local_ordinal_type,
191 typename MatrixType::global_ordinal_type,
192 typename MatrixType::node_type> >
198 template<
class MatrixType>
199 Teuchos::RCP<
const Tpetra::RowGraph<
typename MatrixType::local_ordinal_type,
200 typename MatrixType::global_ordinal_type,
201 typename MatrixType::node_type> >
204 throw std::runtime_error(
"Ifpack2::SingletonFilter: does not support getGraph.");
207 template<
class MatrixType>
213 template<
class MatrixType>
219 template<
class MatrixType>
225 template<
class MatrixType>
231 template<
class MatrixType>
234 return A_->getIndexBase();
237 template<
class MatrixType>
243 template<
class MatrixType>
249 template<
class MatrixType>
252 throw std::runtime_error(
"Ifpack2::SingletonFilter does not implement getNumEntriesInGlobalRow.");
255 template<
class MatrixType>
258 return NumEntries_[localRow];
261 template<
class MatrixType>
264 return MaxNumEntries_;
267 template<
class MatrixType>
270 return MaxNumEntries_;
273 template<
class MatrixType>
279 template<
class MatrixType>
282 return A_->isLocallyIndexed();
285 template<
class MatrixType>
288 return A_->isGloballyIndexed();
291 template<
class MatrixType>
294 return A_->isFillComplete();
297 template<
class MatrixType>
299 const Teuchos::ArrayView<GlobalOrdinal> &,
300 const Teuchos::ArrayView<Scalar> &,
303 throw std::runtime_error(
"Ifpack2::SingletonFilter does not implement getGlobalRowCopy.");
306 template<
class MatrixType>
308 const Teuchos::ArrayView<LocalOrdinal> &Indices,
309 const Teuchos::ArrayView<Scalar> &Values,
310 size_t &NumEntries)
const
312 TEUCHOS_TEST_FOR_EXCEPTION((LocalRow < 0 || (
size_t) LocalRow >= NumRows_ || (
size_t) Indices.size() < NumEntries_[LocalRow]), std::runtime_error,
"Ifpack2::SingletonFilter::getLocalRowCopy invalid row or array size.");
315 LocalOrdinal ARow = InvReorder_[LocalRow];
316 A_->getLocalRowCopy(ARow,Indices_(),Values_(),Nnz);
320 for (
size_t i = 0 ; i < Nnz ; ++i) {
321 LocalOrdinal ii = Reorder_[Indices_[i]];
323 Indices[NumEntries] = ii;
324 Values[NumEntries] = Values_[i];
331 template<
class MatrixType>
333 Teuchos::ArrayView<const GlobalOrdinal> &,
334 Teuchos::ArrayView<const Scalar> &)
const
336 throw std::runtime_error(
"Ifpack2::SingletonFilter: does not support getGlobalRowView.");
339 template<
class MatrixType>
341 Teuchos::ArrayView<const LocalOrdinal> &,
342 Teuchos::ArrayView<const Scalar> &)
const
344 throw std::runtime_error(
"Ifpack2::SingletonFilter: does not support getLocalRowView.");
347 template<
class MatrixType>
351 return A_->getLocalDiagCopy(diag);
354 template<
class MatrixType>
357 throw std::runtime_error(
"Ifpack2::SingletonFilter does not support leftScale.");
360 template<
class MatrixType>
363 throw std::runtime_error(
"Ifpack2::SingletonFilter does not support rightScale.");
366 template<
class MatrixType>
368 Tpetra::MultiVector<Scalar,LocalOrdinal,GlobalOrdinal,Node> &Y,
369 Teuchos::ETransp mode,
373 typedef Scalar DomainScalar;
374 typedef Scalar RangeScalar;
378 TEUCHOS_TEST_FOR_EXCEPTION(X.getNumVectors() != Y.getNumVectors(), std::runtime_error,
379 "Ifpack2::SingletonFilter::apply ERROR: X.getNumVectors() != Y.getNumVectors().");
381 RangeScalar zero = Teuchos::ScalarTraits<RangeScalar>::zero();
382 Teuchos::ArrayRCP<Teuchos::ArrayRCP<const DomainScalar> > x_ptr = X.get2dView();
383 Teuchos::ArrayRCP<Teuchos::ArrayRCP<RangeScalar> > y_ptr = Y.get2dViewNonConst();
386 size_t NumVectors = Y.getNumVectors();
389 for (
size_t i = 0 ; i < NumRows_ ; ++i) {
392 getLocalRowCopy(i,Indices_(),Values_(),Nnz);
393 if (mode==Teuchos::NO_TRANS){
394 for (
size_t j = 0 ; j < Nnz ; ++j)
395 for (
size_t k = 0 ; k < NumVectors ; ++k)
396 y_ptr[k][i] += (RangeScalar)Values_[j] * (RangeScalar)x_ptr[k][Indices_[j]];
398 else if (mode==Teuchos::TRANS){
399 for (
size_t j = 0 ; j < Nnz ; ++j)
400 for (
size_t k = 0 ; k < NumVectors ; ++k)
401 y_ptr[k][Indices_[j]] += (RangeScalar)Values_[j] * (RangeScalar)x_ptr[k][i];
404 for (
size_t j = 0 ; j < Nnz ; ++j)
405 for (
size_t k = 0 ; k < NumVectors ; ++k)
406 y_ptr[k][Indices_[j]] += Teuchos::ScalarTraits<RangeScalar>::conjugate((RangeScalar)Values_[j]) * (RangeScalar)x_ptr[k][i];
411 template<
class MatrixType>
417 template<
class MatrixType>
423 template<
class MatrixType>
425 Tpetra::MultiVector<Scalar,LocalOrdinal,GlobalOrdinal,Node>& LHS)
427 this->
template SolveSingletonsTempl<Scalar,Scalar>(RHS, LHS);
430 template<
class MatrixType>
431 template<
class DomainScalar,
class RangeScalar>
433 Tpetra::MultiVector<RangeScalar,LocalOrdinal,GlobalOrdinal,Node>& LHS)
435 Teuchos::ArrayRCP<Teuchos::ArrayRCP<const DomainScalar> > RHS_ptr = RHS.get2dView();
436 Teuchos::ArrayRCP<Teuchos::ArrayRCP<RangeScalar> > LHS_ptr = LHS.get2dViewNonConst();
438 for (
size_t i = 0 ; i < NumSingletons_ ; ++i) {
439 LocalOrdinal ii = SingletonIndex_[i];
443 for (
size_t j = 0 ; j < Nnz ; ++j) {
444 if (Indices_[j] == ii) {
445 for (
size_t k = 0 ; k < LHS.getNumVectors() ; ++k)
446 LHS_ptr[k][ii] = (RangeScalar)RHS_ptr[k][ii] / (RangeScalar)Values_[j];
452 template<
class MatrixType>
454 const Tpetra::MultiVector<Scalar,LocalOrdinal,GlobalOrdinal,Node>& RHS,
455 Tpetra::MultiVector<Scalar,LocalOrdinal,GlobalOrdinal,Node>& ReducedRHS)
457 this->
template CreateReducedRHSTempl<Scalar,Scalar>(LHS, RHS, ReducedRHS);
460 template<
class MatrixType>
461 template<
class DomainScalar,
class RangeScalar>
463 const Tpetra::MultiVector<RangeScalar,LocalOrdinal,GlobalOrdinal,Node>& RHS,
464 Tpetra::MultiVector<RangeScalar,LocalOrdinal,GlobalOrdinal,Node>& ReducedRHS)
466 Teuchos::ArrayRCP<Teuchos::ArrayRCP<const RangeScalar > > RHS_ptr = RHS.get2dView();
467 Teuchos::ArrayRCP<Teuchos::ArrayRCP<const DomainScalar> > LHS_ptr = LHS.get2dView();
468 Teuchos::ArrayRCP<Teuchos::ArrayRCP<RangeScalar> > ReducedRHS_ptr = ReducedRHS.get2dViewNonConst();
470 size_t NumVectors = LHS.getNumVectors();
472 for (
size_t i = 0 ; i < NumRows_ ; ++i)
473 for (
size_t k = 0 ; k < NumVectors ; ++k)
474 ReducedRHS_ptr[k][i] = RHS_ptr[k][InvReorder_[i]];
476 for (
size_t i = 0 ; i < NumRows_ ; ++i) {
477 LocalOrdinal ii = InvReorder_[i];
481 for (
size_t j = 0 ; j < Nnz ; ++j) {
482 if (Reorder_[Indices_[j]] == -1) {
483 for (
size_t k = 0 ; k < NumVectors ; ++k)
484 ReducedRHS_ptr[k][i] -= (RangeScalar)Values_[j] * (RangeScalar)LHS_ptr[k][Indices_[j]];
490 template<
class MatrixType>
492 Tpetra::MultiVector<Scalar,LocalOrdinal,GlobalOrdinal,Node>& LHS)
494 this->
template UpdateLHSTempl<Scalar,Scalar>(ReducedLHS, LHS);
497 template<
class MatrixType>
498 template<
class DomainScalar,
class RangeScalar>
500 Tpetra::MultiVector<RangeScalar,LocalOrdinal,GlobalOrdinal,Node>& LHS)
503 Teuchos::ArrayRCP<Teuchos::ArrayRCP<RangeScalar> > LHS_ptr = LHS.get2dViewNonConst();
504 Teuchos::ArrayRCP<Teuchos::ArrayRCP<const DomainScalar> > ReducedLHS_ptr = ReducedLHS.get2dView();
506 for (
size_t i = 0 ; i < NumRows_ ; ++i)
507 for (
size_t k = 0 ; k < LHS.getNumVectors() ; ++k)
508 LHS_ptr[k][InvReorder_[i]] = (RangeScalar)ReducedLHS_ptr[k][i];
511 template<
class MatrixType>
514 throw std::runtime_error(
"Ifpack2::SingletonFilter does not implement getFrobeniusNorm.");
519 #define IFPACK2_SINGLETONFILTER_INSTANT(S,LO,GO,N) \
520 template class Ifpack2::SingletonFilter< Tpetra::RowMatrix<S, LO, GO, N> >;