135 FactoryMonitor m(*
this,
"Setup blocked Uzawa Smoother", currentLevel);
138 this->GetOStream(
Warnings0) <<
"MueLu::UzawaSmoother::Setup(): Setup() has already been called";
141 A_ = Factory::Get<RCP<Matrix> > (currentLevel,
"A");
143 RCP<BlockedCrsMatrix> bA = Teuchos::rcp_dynamic_cast<BlockedCrsMatrix>(A_);
144 TEUCHOS_TEST_FOR_EXCEPTION(bA == Teuchos::null,
Exceptions::BadCast,
"MueLu::UzawaSmoother::Setup: input matrix A is not of type BlockedCrsMatrix! error.");
147 rangeMapExtractor_ = bA->getRangeMapExtractor();
148 domainMapExtractor_ = bA->getDomainMapExtractor();
151 F_ = bA->getMatrix(0, 0);
152 G_ = bA->getMatrix(0, 1);
153 D_ = bA->getMatrix(1, 0);
154 Z_ = bA->getMatrix(1, 1);
159 RCP<const FactoryManagerBase> velpredictFactManager = FactManager_.at(0);
161 velPredictSmoo_ = currentLevel.
Get< RCP<SmootherBase> >(
"PreSmoother",velpredictFactManager->GetFactory(
"Smoother").get());
164 RCP<const FactoryManagerBase> schurFactManager = FactManager_.at(1);
166 schurCompSmoo_ = currentLevel.
Get< RCP<SmootherBase> >(
"PreSmoother", schurFactManager->GetFactory(
"Smoother").get());
177 Teuchos::RCP<Teuchos::FancyOStream> fos = Teuchos::getFancyOStream(Teuchos::rcpFromRef(std::cout));
179 SC zero = Teuchos::ScalarTraits<SC>::zero(), one = Teuchos::ScalarTraits<SC>::one();
181 bool bRangeThyraMode = rangeMapExtractor_->getThyraMode();
182 bool bDomainThyraMode = domainMapExtractor_->getThyraMode();
190 RCP<MultiVector> rcpX = Teuchos::rcpFromRef(X);
191 RCP<const MultiVector> rcpB = Teuchos::rcpFromRef(B);
194 bool bCopyResultX =
false;
195 RCP<BlockedCrsMatrix> bA = Teuchos::rcp_dynamic_cast<BlockedCrsMatrix>(A_);
197 RCP<BlockedMultiVector> bX = Teuchos::rcp_dynamic_cast<BlockedMultiVector>(rcpX);
198 RCP<const BlockedMultiVector> bB = Teuchos::rcp_dynamic_cast<const BlockedMultiVector>(rcpB);
200 if(bX.is_null() ==
true) {
201 RCP<MultiVector> test = Teuchos::rcp(
new BlockedMultiVector(bA->getBlockedDomainMap(),rcpX));
206 if(bB.is_null() ==
true) {
207 RCP<const MultiVector> test = Teuchos::rcp(
new BlockedMultiVector(bA->getBlockedRangeMap(),rcpB));
212 bX = Teuchos::rcp_dynamic_cast<BlockedMultiVector>(rcpX);
213 bB = Teuchos::rcp_dynamic_cast<const BlockedMultiVector>(rcpB);
216 RCP<Xpetra::ReorderedBlockedCrsMatrix<Scalar,LocalOrdinal,GlobalOrdinal,Node> > rbA = Teuchos::rcp_dynamic_cast<Xpetra::ReorderedBlockedCrsMatrix<Scalar,LocalOrdinal,GlobalOrdinal,Node> >(bA);
217 if(rbA.is_null() ==
false) {
219 Teuchos::RCP<const Xpetra::BlockReorderManager > brm = rbA->getBlockReorderManager();
222 if(bX->getBlockedMap()->getNumMaps() != bA->getDomainMapExtractor()->NumMaps()) {
224 Teuchos::RCP<MultiVector> test =
225 buildReorderedBlockedMultiVector(brm, bX);
228 if(bB->getBlockedMap()->getNumMaps() != bA->getRangeMapExtractor()->NumMaps()) {
230 Teuchos::RCP<const MultiVector> test =
231 buildReorderedBlockedMultiVector(brm, bB);
240 RCP<MultiVector> residual = MultiVectorFactory::Build(rcpB->getMap(), rcpB->getNumVectors());
241 RCP<BlockedMultiVector> bresidual = Teuchos::rcp_dynamic_cast<BlockedMultiVector>(residual);
242 Teuchos::RCP<MultiVector> r1 = bresidual->getMultiVector(0,bRangeThyraMode);
243 Teuchos::RCP<MultiVector> r2 = bresidual->getMultiVector(1,bRangeThyraMode);
246 RCP<MultiVector> xtilde = MultiVectorFactory::Build(rcpX->getMap(), rcpX->getNumVectors());
247 RCP<BlockedMultiVector> bxtilde = Teuchos::rcp_dynamic_cast<BlockedMultiVector>(xtilde);
248 RCP<MultiVector> xtilde1 = bxtilde->getMultiVector(0,bDomainThyraMode);
249 RCP<MultiVector> xtilde2 = bxtilde->getMultiVector(1,bDomainThyraMode);
254 residual->update(one,*rcpB,zero);
255 A_->apply(*rcpX, *residual, Teuchos::NO_TRANS, -one, one);
259 bxtilde->putScalar(zero);
260 velPredictSmoo_->Apply(*xtilde1,*r1);
264 RCP<MultiVector> schurCompRHS = MultiVectorFactory::Build(r2->getMap(),rcpB->getNumVectors());
265 D_->apply(*xtilde1,*schurCompRHS);
266 schurCompRHS->update(one,*r2,-one);
270 schurCompSmoo_->Apply(*xtilde2,*schurCompRHS);
272 rcpX->update(omega,*bxtilde,one);
275 if (bCopyResultX ==
true) {
276 RCP<MultiVector> Xmerged = bX->Merge();
277 X.update(one, *Xmerged, zero);