147 FactoryMonitor m(*
this,
"Setup blocked SIMPLE Smoother", currentLevel);
150 this->
GetOStream(
Warnings0) <<
"MueLu::SimpleSmoother::Setup(): Setup() has already been called";
155 RCP<BlockedCrsMatrix> bA = Teuchos::rcp_dynamic_cast<BlockedCrsMatrix>(
A_);
156 TEUCHOS_TEST_FOR_EXCEPTION(bA == Teuchos::null,
Exceptions::BadCast,
"MueLu::SimpleSmoother::Setup: input matrix A is not of type BlockedCrsMatrix! error.");
163 F_ = bA->getMatrix(0, 0);
164 G_ = bA->getMatrix(0, 1);
165 D_ = bA->getMatrix(1, 0);
166 Z_ = bA->getMatrix(1, 1);
169 bool bSIMPLEC = pL.get<
bool>(
"UseSIMPLEC");
173 RCP<Vector> diagFVector = VectorFactory::Build(
F_->getRowMap());
175 F_->getLocalDiagCopy(*diagFVector);
196 RCP<BlockedVector> bdiagFinv = Teuchos::rcp_dynamic_cast<BlockedVector>(
diagFinv_);
197 if(bdiagFinv.is_null() ==
false && bdiagFinv->getBlockedMap()->getNumMaps() == 1) {
198 RCP<Vector> nestedVec = bdiagFinv->getMultiVector(0,bdiagFinv->getBlockedMap()->getThyraMode())->getVectorNonConst(0);
205 RCP<const FactoryManagerBase> velpredictFactManager =
FactManager_.at(0);
207 velPredictSmoo_ = currentLevel.
Get< RCP<SmootherBase> >(
"PreSmoother",velpredictFactManager->GetFactory(
"Smoother").get());
210 RCP<const FactoryManagerBase> schurFactManager =
FactManager_.at(1);
212 schurCompSmoo_ = currentLevel.
Get< RCP<SmootherBase> >(
"PreSmoother", schurFactManager->GetFactory(
"Smoother").get());
222 "MueLu::SimpleSmoother::Apply(): Setup() has not been called");
225 RCP<MultiVector> rcpDebugX = Teuchos::rcpFromRef(X);
226 RCP<const MultiVector> rcpDebugB = Teuchos::rcpFromRef(B);
227 RCP<BlockedMultiVector> rcpBDebugX = Teuchos::rcp_dynamic_cast<BlockedMultiVector>(rcpDebugX);
228 RCP<const BlockedMultiVector> rcpBDebugB = Teuchos::rcp_dynamic_cast<const BlockedMultiVector>(rcpDebugB);
229 if(rcpBDebugB.is_null() ==
false) {
234 if(rcpBDebugX.is_null() ==
false) {
241 const SC zero = Teuchos::ScalarTraits<SC>::zero();
242 const SC one = Teuchos::ScalarTraits<SC>::one();
256 RCP<MultiVector> rcpX = Teuchos::rcpFromRef(X);
257 RCP<const MultiVector> rcpB = Teuchos::rcpFromRef(B);
260 bool bCopyResultX =
false;
261 bool bReorderX =
false;
262 bool bReorderB =
false;
263 RCP<BlockedCrsMatrix> bA = Teuchos::rcp_dynamic_cast<BlockedCrsMatrix>(
A_);
265 RCP<BlockedMultiVector> bX = Teuchos::rcp_dynamic_cast<BlockedMultiVector>(rcpX);
266 RCP<const BlockedMultiVector> bB = Teuchos::rcp_dynamic_cast<const BlockedMultiVector>(rcpB);
269 RCP<Xpetra::ReorderedBlockedCrsMatrix<Scalar,LocalOrdinal,GlobalOrdinal,Node> > rbA =
270 Teuchos::rcp_dynamic_cast<Xpetra::ReorderedBlockedCrsMatrix<Scalar,LocalOrdinal,GlobalOrdinal,Node> >(bA);
272 if(rbA.is_null() ==
false) {
277 if (bX.is_null() ==
true) {
278 RCP<MultiVector> vectorWithBlockedMap = Teuchos::rcp(
new BlockedMultiVector(rbA->getBlockedCrsMatrix()->getBlockedDomainMap(), rcpX));
279 rcpX.swap(vectorWithBlockedMap);
285 if (bB.is_null() ==
true) {
286 RCP<const MultiVector> vectorWithBlockedMap = Teuchos::rcp(
new BlockedMultiVector(rbA->getBlockedCrsMatrix()->getBlockedRangeMap(), rcpB));
287 rcpB.swap(vectorWithBlockedMap);
293 if (bX.is_null() ==
true) {
294 RCP<MultiVector> vectorWithBlockedMap = Teuchos::rcp(
new BlockedMultiVector(bA->getBlockedDomainMap(), rcpX));
295 rcpX.swap(vectorWithBlockedMap);
299 if(bB.is_null() ==
true) {
300 RCP<const MultiVector> vectorWithBlockedMap = Teuchos::rcp(
new BlockedMultiVector(bA->getBlockedRangeMap(),rcpB));
301 rcpB.swap(vectorWithBlockedMap);
306 bX = Teuchos::rcp_dynamic_cast<BlockedMultiVector>(rcpX);
307 bB = Teuchos::rcp_dynamic_cast<const BlockedMultiVector>(rcpB);
310 if(rbA.is_null() ==
false) {
312 Teuchos::RCP<const Xpetra::BlockReorderManager > brm = rbA->getBlockReorderManager();
315 if(bX->getBlockedMap()->getNumMaps() != bA->getDomainMapExtractor()->NumMaps() || bReorderX) {
317 Teuchos::RCP<MultiVector> reorderedBlockedVector = buildReorderedBlockedMultiVector(brm, bX);
318 rcpX.swap(reorderedBlockedVector);
321 if(bB->getBlockedMap()->getNumMaps() != bA->getRangeMapExtractor()->NumMaps() || bReorderB) {
323 Teuchos::RCP<const MultiVector> reorderedBlockedVector = buildReorderedBlockedMultiVector(brm, bB);
324 rcpB.swap(reorderedBlockedVector);
332 RCP<MultiVector> residual = MultiVectorFactory::Build(rcpB->getMap(), rcpB->getNumVectors());
333 RCP<BlockedMultiVector> bresidual = Teuchos::rcp_dynamic_cast<BlockedMultiVector>(residual);
334 RCP<MultiVector> r1 = bresidual->getMultiVector(0,bRangeThyraMode);
335 RCP<MultiVector> r2 = bresidual->getMultiVector(1,bRangeThyraMode);
338 RCP<MultiVector> xtilde = MultiVectorFactory::Build(rcpX->getMap(), rcpX->getNumVectors());
339 RCP<BlockedMultiVector> bxtilde = Teuchos::rcp_dynamic_cast<BlockedMultiVector>(xtilde);
340 RCP<MultiVector> xtilde1 = bxtilde->getMultiVector(0, bDomainThyraMode);
341 RCP<MultiVector> xtilde2 = bxtilde->getMultiVector(1, bDomainThyraMode);
344 RCP<MultiVector> xhat = MultiVectorFactory::Build(rcpX->getMap(), rcpX->getNumVectors());
345 RCP<BlockedMultiVector> bxhat = Teuchos::rcp_dynamic_cast<BlockedMultiVector>(xhat);
346 RCP<MultiVector> xhat1 = bxhat->getMultiVector(0, bDomainThyraMode);
347 RCP<MultiVector> xhat2 = bxhat->getMultiVector(1, bDomainThyraMode);
353 residual->update(one, *rcpB, zero);
354 if(InitialGuessIsZero ==
false || run > 0)
355 A_->apply(*rcpX, *residual, Teuchos::NO_TRANS, -one, one);
359 xtilde1->putScalar(zero);
360 xtilde2->putScalar(zero);
365 RCP<MultiVector> schurCompRHS =
rangeMapExtractor_->getVector(1, rcpB->getNumVectors(), bRangeThyraMode);
366 D_->apply(*xtilde1, *schurCompRHS);
368 schurCompRHS->update(one, *r2, -one);
376 xhat2->update(omega, *xtilde2, zero);
379 RCP<MultiVector> xhat1_temp =
domainMapExtractor_->getVector(0, rcpX->getNumVectors(), bDomainThyraMode);
380 G_->apply(*xhat2, *xhat1_temp);
382 xhat1->elementWiseMultiply(one, *
diagFinv_, *xhat1_temp, zero);
383 xhat1->update(one, *xtilde1, -one);
385 rcpX->update(one, *bxhat, one);
388 if (bCopyResultX ==
true) {
389 RCP<const MultiVector> Xmerged = bX->Merge();
390 X.update(one, *Xmerged, zero);