47 #ifndef XPETRA_BLOCKEDMULTIVECTOR_DEF_HPP
48 #define XPETRA_BLOCKEDMULTIVECTOR_DEF_HPP
52 #include "Xpetra_MultiVectorFactory.hpp"
53 #include "Xpetra_BlockedVector.hpp"
54 #include "Xpetra_MapExtractor.hpp"
58 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
64 numVectors_ = NumVectors;
66 vv_.reserve(map->getNumMaps());
69 for (
size_t r = 0; r < map->getNumMaps(); ++r)
85 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
86 BlockedMultiVector<Scalar, LocalOrdinal, GlobalOrdinal, Node>::
88 Teuchos::RCP<const MultiVector> v) {
91 "BlockedMultiVector: inconsistent number of local elements of MultiVector and BlockedMap. The BlockedMap has "
92 << bmap->getMap()->getLocalNumElements() <<
" local elements. The vector has " << v->getMap()->getLocalNumElements()
96 "BlockedMultiVector: inconsistent number of global elements of MultiVector and BlockedMap. The BlockedMap has "
97 << bmap->getMap()->getGlobalNumElements() <<
" local elements. The vector has " << v->getMap()->getGlobalNumElements()
106 numVectors_ = v->getNumVectors();
111 vv_.reserve(bmap->getNumMaps());
114 for (
size_t r = 0; r < bmap->getNumMaps(); ++r)
115 vv_.push_back(this->ExtractVector(v, r, bmap->getThyraMode()));
129 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
132 Teuchos::RCP<const MultiVector> v) {
136 std::vector<RCP<const Map>> maps;
137 maps.reserve(mapExtractor->NumMaps());
138 for (
size_t r = 0; r < mapExtractor->NumMaps(); ++r)
139 maps.push_back(mapExtractor->getMap(r, mapExtractor->getThyraMode()));
143 vv_.reserve(mapExtractor->NumMaps());
146 for (
size_t r = 0; r < mapExtractor->NumMaps(); ++r)
147 vv_.push_back(this->ExtractVector(v, r, mapExtractor->getThyraMode()));
150 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
153 std::vector<Teuchos::RCP<MultiVector>>& vin) {
156 vv_.resize(vin.size());
157 for (
size_t i = 0; i < vv_.size(); i++)
161 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
164 for (
size_t r = 0; r < vv_.size(); ++r) {
165 vv_[r] = Teuchos::null;
168 map_ = Teuchos::null;
172 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
180 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
186 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
192 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
198 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
204 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
208 for (
size_t r = 0; r < map_->getNumMaps(); r++) {
209 getMultiVector(r)->putScalar(value);
213 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
214 Teuchos::RCP<const Xpetra::Vector<Scalar, LocalOrdinal, GlobalOrdinal, Node>>
217 RCP<Xpetra::BlockedVector<Scalar, LocalOrdinal, GlobalOrdinal, Node>> ret =
220 for (
size_t r = 0; r < getBlockedMap()->getNumMaps(); r++) {
221 RCP<const Xpetra::Vector<Scalar, LocalOrdinal, GlobalOrdinal, Node>> subvec =
222 this->getMultiVector(r, this->getBlockedMap()->getThyraMode())->getVector(j);
223 ret->setMultiVector(r, subvec, this->getBlockedMap()->getThyraMode());
228 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
229 Teuchos::RCP<Xpetra::Vector<Scalar, LocalOrdinal, GlobalOrdinal, Node>>
232 RCP<Xpetra::Vector<Scalar, LocalOrdinal, GlobalOrdinal, Node>> ret =
237 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
238 Teuchos::ArrayRCP<const Scalar>
241 if (map_->getNumMaps() == 1) {
245 TEUCHOS_UNREACHABLE_RETURN(Teuchos::null);
248 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
249 Teuchos::ArrayRCP<Scalar>
252 if (map_->getNumMaps() == 1) {
256 TEUCHOS_UNREACHABLE_RETURN(Teuchos::null);
259 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
265 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
271 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
277 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
281 for (
size_t r = 0; r < map_->getNumMaps(); ++r) {
282 if (getMultiVector(r) != Teuchos::null) {
283 getMultiVector(r)->scale(alpha);
288 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
290 scale(Teuchos::ArrayView<const Scalar> alpha) {
292 for (
size_t r = 0; r < map_->getNumMaps(); ++r) {
293 if (getMultiVector(r) != Teuchos::null) {
294 getMultiVector(r)->scale(alpha);
299 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
303 Teuchos::RCP<const MultiVector> rcpA = Teuchos::rcpFromRef(A);
304 Teuchos::RCP<const BlockedMultiVector> bA = Teuchos::rcp_dynamic_cast<
const BlockedMultiVector>(rcpA);
305 TEUCHOS_TEST_FOR_EXCEPTION(numVectors_ != rcpA->getNumVectors(),
307 "BlockedMultiVector::update: update with incompatible vector (different number of vectors in multivector).");
308 if (bA != Teuchos::null) {
311 TEUCHOS_TEST_FOR_EXCEPTION(map_->getThyraMode() != bA->getBlockedMap()->getThyraMode(),
313 "BlockedMultiVector::update: update with incompatible vector (different thyra mode).");
314 TEUCHOS_TEST_FOR_EXCEPTION(map_->getNumMaps() != bA->getBlockedMap()->getNumMaps(),
316 "BlockedMultiVector::update: update with incompatible vector (different number of partial vectors).");
317 for (
size_t r = 0; r < map_->getNumMaps(); r++) {
318 XPETRA_TEST_FOR_EXCEPTION(getMultiVector(r)->getMap()->getLocalNumElements() != bA->getMultiVector(r)->getMap()->getLocalNumElements(),
320 "BlockedMultiVector::update: in subvector "
321 << r <<
": Cannot add a vector of (local) length " << bA->getMultiVector(r)->getMap()->getLocalNumElements()
322 <<
" to the existing vector with " << getMultiVector(r)->getMap()->getLocalNumElements() <<
" entries.");
323 XPETRA_TEST_FOR_EXCEPTION(getMultiVector(r)->getMap()->getGlobalNumElements() != bA->getMultiVector(r)->getMap()->getGlobalNumElements(),
325 "BlockedMultiVector::update: in subvector "
326 << r <<
": Cannot add a vector of length " << bA->getMultiVector(r)->getMap()->getGlobalNumElements()
327 <<
" to the existing vector with " << getMultiVector(r)->getMap()->getGlobalNumElements() <<
" entries.");
335 Teuchos::RCP<MultiVector> lmv = getMultiVector(r);
336 Teuchos::RCP<MultiVector> rmv = bA->getMultiVector(r);
339 Teuchos::RCP<BlockedMultiVector> blmv = Teuchos::rcp_dynamic_cast<
BlockedMultiVector>(lmv);
340 Teuchos::RCP<BlockedMultiVector> brmv = Teuchos::rcp_dynamic_cast<
BlockedMultiVector>(rmv);
342 if (blmv.is_null() ==
true && brmv.is_null() ==
false) {
344 TEUCHOS_TEST_FOR_EXCEPTION(brmv->getBlockedMap()->getNumMaps() > 1,
346 "BlockedMultiVector::update: Standard MultiVector object does not accept BlockedMultVector object as "
347 "parameter in update call.");
348 lmv->
update(alpha, *(brmv->getMultiVector(0)), beta);
350 lmv->update(alpha, *rmv, beta);
357 if (getBlockedMap()->getNumMaps() == 1) {
361 getMultiVector(0)->getMap()->isSameAs(*(rcpA->getMap())) ==
false,
363 "BlockedMultiVector::update: update with incompatible vector (maps of full vector do not match with map in MapExtractor).");
364 getMultiVector(0)->update(alpha, *rcpA, beta);
370 for (
size_t r = 0; r < map_->getNumMaps(); r++) {
373 Teuchos::RCP<const MultiVector> part = this->ExtractVector(rcpA, r, map_->getThyraMode());
376 "BlockedMultiVector::update: in subvector "
377 << r <<
": Cannot add a vector of (local) length " << part->getMap()->getLocalNumElements()
378 <<
" to the existing vector with " << getMultiVector(r)->getMap()->getLocalNumElements() <<
" entries.");
381 "BlockedMultiVector::update: in subvector "
382 << r <<
": Cannot add a vector of length " << part->getMap()->getGlobalNumElements()
383 <<
" to the existing vector with " << getMultiVector(r)->getMap()->getGlobalNumElements() <<
" entries.");
384 getMultiVector(r)->update(alpha, *part, beta);
390 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
394 Teuchos::RCP<const MultiVector> rcpA = Teuchos::rcpFromRef(A);
395 Teuchos::RCP<const MultiVector> rcpB = Teuchos::rcpFromRef(B);
396 Teuchos::RCP<const BlockedMultiVector> bA = Teuchos::rcp_dynamic_cast<
const BlockedMultiVector>(rcpA);
397 Teuchos::RCP<const BlockedMultiVector> bB = Teuchos::rcp_dynamic_cast<
const BlockedMultiVector>(rcpB);
398 if (bA != Teuchos::null && bB != Teuchos::null) {
399 TEUCHOS_TEST_FOR_EXCEPTION(map_->getThyraMode() != bA->getBlockedMap()->getThyraMode(),
401 "BlockedMultiVector::update: update with incompatible vector (different thyra mode in vector A).");
402 TEUCHOS_TEST_FOR_EXCEPTION(map_->getNumMaps() != bA->getBlockedMap()->getNumMaps(),
404 "BlockedMultiVector::update: update with incompatible vector (different number of partial vectors in vector A).");
405 TEUCHOS_TEST_FOR_EXCEPTION(
406 numVectors_ != bA->getNumVectors(),
408 "BlockedMultiVector::update: update with incompatible vector (different number of vectors in multivector in vector A).");
409 TEUCHOS_TEST_FOR_EXCEPTION(map_->getThyraMode() != bB->getBlockedMap()->getThyraMode(),
411 "BlockedMultiVector::update: update with incompatible vector (different thyra mode in vector B).");
412 TEUCHOS_TEST_FOR_EXCEPTION(map_->getNumMaps() != bB->getBlockedMap()->getNumMaps(),
414 "BlockedMultiVector::update: update with incompatible vector (different number of partial vectors in vector B).");
415 TEUCHOS_TEST_FOR_EXCEPTION(
416 numVectors_ != bB->getNumVectors(),
418 "BlockedMultiVector::update: update with incompatible vector (different number of vectors in multivector in vector B).");
420 for (
size_t r = 0; r < map_->getNumMaps(); r++) {
423 "BlockedMultiVector::update: update with incompatible vector (different maps in partial vector " << r <<
").");
424 getMultiVector(r)->update(alpha, *(bA->getMultiVector(r)), beta, *(bB->getMultiVector(r)), gamma);
431 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
433 norm1(
const Teuchos::ArrayView<
typename Teuchos::ScalarTraits<Scalar>::magnitudeType>& norms)
const {
435 typedef typename ScalarTraits<Scalar>::magnitudeType Magnitude;
436 Array<Magnitude> temp_norms(getNumVectors());
437 std::fill(norms.begin(), norms.end(), ScalarTraits<Magnitude>::zero());
438 std::fill(temp_norms.begin(), temp_norms.end(), ScalarTraits<Magnitude>::zero());
439 for (
size_t r = 0; r < map_->getNumMaps(); ++r) {
440 if (getMultiVector(r) != Teuchos::null) {
441 getMultiVector(r)->norm1(temp_norms);
442 for (
size_t c = 0; c < getNumVectors(); ++c)
443 norms[c] += temp_norms[c];
448 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
450 norm2(
const Teuchos::ArrayView<
typename Teuchos::ScalarTraits<Scalar>::magnitudeType>& norms)
const {
452 typedef typename ScalarTraits<Scalar>::magnitudeType Magnitude;
453 Array<Magnitude> results(getNumVectors());
454 Array<Magnitude> temp_norms(getNumVectors());
455 std::fill(norms.begin(), norms.end(), ScalarTraits<Magnitude>::zero());
456 std::fill(results.begin(), results.end(), ScalarTraits<Magnitude>::zero());
457 std::fill(temp_norms.begin(), temp_norms.end(), ScalarTraits<Magnitude>::zero());
458 for (
size_t r = 0; r < map_->getNumMaps(); ++r) {
459 if (getMultiVector(r) != Teuchos::null) {
460 getMultiVector(r)->norm2(temp_norms);
461 for (
size_t c = 0; c < getNumVectors(); ++c)
462 results[c] += temp_norms[c] * temp_norms[c];
465 for (
size_t c = 0; c < getNumVectors(); ++c)
466 norms[c] = Teuchos::ScalarTraits<Magnitude>::squareroot(results[c]);
469 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
471 normInf(
const Teuchos::ArrayView<
typename Teuchos::ScalarTraits<Scalar>::magnitudeType>& norms)
const {
473 typedef typename ScalarTraits<Scalar>::magnitudeType Magnitude;
474 Array<Magnitude> temp_norms(getNumVectors());
475 std::fill(norms.begin(), norms.end(), ScalarTraits<Magnitude>::zero());
476 std::fill(temp_norms.begin(), temp_norms.end(), ScalarTraits<Magnitude>::zero());
477 for (
size_t r = 0; r < map_->getNumMaps(); ++r) {
478 if (getMultiVector(r) != Teuchos::null) {
479 getMultiVector(r)->normInf(temp_norms);
480 for (
size_t c = 0; c < getNumVectors(); ++c)
481 norms[c] = std::max(norms[c], temp_norms[c]);
486 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
492 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
503 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
512 "BlockedMultiVector::elementWiseMultipy: B must have same blocked map than this.");
515 TEUCHOS_TEST_FOR_EXCEPTION(A.
getMap()->getLocalNumElements() != B.
getMap()->getLocalNumElements(),
517 "BlockedMultiVector::elementWiseMultipy: A has " << A.
getMap()->getLocalNumElements() <<
" elements, B has "
518 << B.
getMap()->getLocalNumElements() <<
".");
519 TEUCHOS_TEST_FOR_EXCEPTION(A.
getMap()->getGlobalNumElements() != B.
getMap()->getGlobalNumElements(),
521 "BlockedMultiVector::elementWiseMultipy: A has " << A.
getMap()->getGlobalNumElements() <<
" elements, B has "
522 << B.
getMap()->getGlobalNumElements() <<
".");
524 RCP<const BlockedMap> bmap = getBlockedMap();
525 RCP<const Xpetra::Vector<Scalar, LocalOrdinal, GlobalOrdinal, Node>> rcpA = Teuchos::rcpFromRef(A);
526 RCP<const Xpetra::BlockedVector<Scalar, LocalOrdinal, GlobalOrdinal, Node>> bA =
529 RCP<const MultiVector> rcpB = Teuchos::rcpFromRef(B);
530 RCP<const BlockedMultiVector> bB = Teuchos::rcp_dynamic_cast<
const BlockedMultiVector>(rcpB);
531 TEUCHOS_TEST_FOR_EXCEPTION(
537 for (
size_t m = 0; m < bmap->getNumMaps(); m++) {
538 RCP<const Xpetra::Vector<Scalar, LocalOrdinal, GlobalOrdinal, Node>> partA = bA->
getMultiVector(m, bmap->getThyraMode())->getVector(0);
539 RCP<const Xpetra::MultiVector<Scalar, LocalOrdinal, GlobalOrdinal, Node>> partB = bB->getMultiVector(m, bmap->getThyraMode());
540 RCP<Xpetra::MultiVector<Scalar, LocalOrdinal, GlobalOrdinal, Node>> thisPart = this->getMultiVector(m, bmap->getThyraMode());
542 thisPart->elementWiseMultiply(scalarAB, *partA, *partB, scalarThis);
546 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
553 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
558 return map_->getFullMap()->getLocalNumElements();
561 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
566 return map_->getFullMap()->getGlobalNumElements();
569 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
575 for (
size_t r = 0; r < map_->getNumMaps(); ++r) {
576 RCP<const MultiVector> a = getMultiVector(r);
578 if ((a == Teuchos::null && b != Teuchos::null) || (a != Teuchos::null && b == Teuchos::null))
580 if (a != Teuchos::null && b != Teuchos::null && !a->isSameSize(*b))
586 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
590 return std::string(
"BlockedMultiVector");
593 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
595 describe(Teuchos::FancyOStream& out,
const Teuchos::EVerbosityLevel verbLevel)
const {
596 out << description() << std::endl;
597 for (
size_t r = 0; r < map_->getNumMaps(); r++)
598 getMultiVector(r)->
describe(out, verbLevel);
601 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
605 RCP<const BlockedMap> bmap = Teuchos::rcp_dynamic_cast<
const BlockedMap>(map);
606 if (bmap.is_null() ==
true) {
608 if (this->getBlockedMap()->getNumMaps() > 1) {
610 "BlockedMultiVector::replaceMap: map is not of type BlockedMap. General implementation not available, yet.");
611 TEUCHOS_UNREACHABLE_RETURN();
615 std::vector<Teuchos::RCP<const Map>> subMaps(1, map);
616 map_ = Teuchos::rcp(
new BlockedMap(map, subMaps, this->getBlockedMap()->getThyraMode()));
617 this->getMultiVector(0)->replaceMap(map);
620 RCP<const BlockedMap> mybmap = Teuchos::rcp_dynamic_cast<
const BlockedMap>(map_);
625 for (
size_t r = 0; r < map_->getNumMaps(); r++)
626 getMultiVector(r)->replaceMap(bmap->getMap(r, map_->getThyraMode()));
629 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
637 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
645 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
653 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
661 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
664 for (
size_t r = 0; r < map_->getNumMaps(); ++r) {
665 getMultiVector(r)->
setSeed(seed);
669 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
672 for (
size_t r = 0; r < map_->getNumMaps(); ++r) {
673 getMultiVector(r)->
randomize(bUseXpetraImplementation);
677 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
679 randomize(
const Scalar& minVal,
const Scalar& maxVal,
bool bUseXpetraImplementation) {
680 for (
size_t r = 0; r < map_->getNumMaps(); ++r) {
681 getMultiVector(r)->
randomize(minVal, maxVal, bUseXpetraImplementation);
685 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
691 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
697 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
698 Teuchos::RCP<const Xpetra::Map<LocalOrdinal, GlobalOrdinal, Node>>
705 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
706 Teuchos::RCP<const Xpetra::BlockedMap<LocalOrdinal, GlobalOrdinal, Node>>
713 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
714 Teuchos::RCP<Xpetra::MultiVector<Scalar, LocalOrdinal, GlobalOrdinal, Node>>
718 TEUCHOS_TEST_FOR_EXCEPTION(r > map_->getNumMaps(),
720 "Error, r = " << r <<
" is too big. The BlockedMultiVector only contains " << map_->getNumMaps()
721 <<
" partial blocks.");
725 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
726 Teuchos::RCP<Xpetra::MultiVector<Scalar, LocalOrdinal, GlobalOrdinal, Node>>
729 XPETRA_MONITOR(
"BlockedMultiVector::getMultiVector(r,bThyraMode)");
730 TEUCHOS_TEST_FOR_EXCEPTION(r > map_->getNumMaps(),
732 "Error, r = " << r <<
" is too big. The BlockedMultiVector only contains " << map_->getNumMaps()
733 <<
" partial blocks.");
740 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
752 "Error, r = " << r <<
" is too big. The BlockedMultiVector only contains " << map_->getNumMaps() <<
" partial blocks.");
755 "The BlockedMultiVectors expects " << getNumVectors() <<
" vectors. The provided partial multivector has "
756 << v->getNumVectors() <<
" vectors.");
760 Teuchos::RCP<MultiVector> vv = Teuchos::rcp_const_cast<
MultiVector>(v);
761 TEUCHOS_TEST_FOR_EXCEPTION(vv == Teuchos::null, Xpetra::Exceptions::RuntimeError,
"Partial vector must not be Teuchos::null");
765 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
766 Teuchos::RCP<Xpetra::MultiVector<Scalar, LocalOrdinal, GlobalOrdinal, Node>>
773 for (
size_t r = 0; r < map_->getNumMaps(); ++r) {
774 RCP<MultiVector> vi = getMultiVector(r);
776 if (bvi.is_null() ==
true) {
777 this->InsertVector(vi, r, v, map_->getThyraMode());
779 RCP<MultiVector> mvi = bvi->
Merge();
780 this->InsertVector(mvi, r, v, map_->getThyraMode());
789 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
792 Teuchos::RCP<const MultiVector> rcpRhs = Teuchos::rcpFromRef(rhs);
793 Teuchos::RCP<const BlockedMultiVector> bRhs = Teuchos::rcp_dynamic_cast<
const BlockedMultiVector>(rcpRhs);
794 if (bRhs == Teuchos::null)
797 map_ = Teuchos::rcp(
new BlockedMap(*(bRhs->getBlockedMap())));
798 vv_ = std::vector<Teuchos::RCP<MultiVector>>(map_->getNumMaps());
799 for (
size_t r = 0; r < map_->getNumMaps(); ++r) {
801 RCP<MultiVector> src = bRhs->getMultiVector(r, map_->getThyraMode());
805 map_->getMap(r, bRhs->getBlockedMap()->getThyraMode()), rcpRhs->getNumVectors(),
true);
810 if (bsrc.is_null() ==
true && bvv.is_null() ==
true) {
812 }
else if (bsrc.is_null() ==
true && bvv.is_null() ==
false) {
815 }
else if (bsrc.is_null() ==
false && bvv.is_null() ==
true) {
818 if (bsrc->getBlockedMap()->getNumMaps() > 1) {
820 "BlockedMultiVector::assign: source vector is of type BlockedMultiVector (with more than "
821 "1 blocks) and target is a MultiVector.");
822 TEUCHOS_UNREACHABLE_RETURN();
824 RCP<MultiVector> ssrc = bsrc->getMultiVector(0, map_->getThyraMode());
828 "BlockedMultiVector::assign: sub block must not be of type BlockedMultiVector.");
834 "BlockedMultiVector::assign: source and target are BlockedMultiVectors with a different number of submaps.");
839 numVectors_ = rcpRhs->getNumVectors();
842 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
847 ExtractVector(*full, block, *partial);
850 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
851 void BlockedMultiVector<Scalar, LocalOrdinal, GlobalOrdinal, Node>::
855 ExtractVector(*full, block, *partial);
858 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
859 RCP<Xpetra::MultiVector<Scalar, LocalOrdinal, GlobalOrdinal, Node>>
860 BlockedMultiVector<Scalar, LocalOrdinal, GlobalOrdinal, Node>::
863 bool bThyraMode)
const {
866 "ExtractVector: Error, block = " << block <<
" is too big. The MapExtractor only contains " << map_->getNumMaps()
867 <<
" partial blocks.");
870 RCP<const BlockedMultiVector> bfull = Teuchos::rcp_dynamic_cast<
const BlockedMultiVector>(full);
871 if (bfull.is_null() ==
true) {
874 const RCP<MultiVector> vv =
880 RCP<const Map> oldThyMapFull = full->getMap();
881 RCP<MultiVector> rcpNonConstFull = Teuchos::rcp_const_cast<MultiVector>(full);
882 rcpNonConstFull->replaceMap(map_->getImporter(block)->getSourceMap());
883 ExtractVector(*rcpNonConstFull, block, *vv);
884 TEUCHOS_TEST_FOR_EXCEPTION(map_->getThyraMode() ==
false && bThyraMode ==
true,
885 Xpetra::Exceptions::RuntimeError,
886 "MapExtractor::ExtractVector: ExtractVector in Thyra-style numbering only possible if MapExtractor has been "
887 "created using Thyra-style numbered submaps.");
888 if (bThyraMode ==
true)
889 vv->replaceMap(map_->getMap(block,
true));
890 rcpNonConstFull->replaceMap(oldThyMapFull);
896 Xpetra::Exceptions::RuntimeError,
897 "ExtractVector: Number of blocks in map extractor is " << map_->getNumMaps() <<
" but should be "
898 << bfull->getBlockedMap()->getNumMaps()
899 <<
" (number of blocks in BlockedMultiVector)");
900 return bfull->getMultiVector(block, bThyraMode);
904 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
905 RCP<Xpetra::MultiVector<Scalar, LocalOrdinal, GlobalOrdinal, Node>>
906 BlockedMultiVector<Scalar, LocalOrdinal, GlobalOrdinal, Node>::
909 bool bThyraMode)
const {
912 "ExtractVector: Error, block = " << block <<
" is too big. The MapExtractor only contains " << map_->getNumMaps()
913 <<
" partial blocks.");
915 map_->getMap(block,
false) == null, Xpetra::Exceptions::RuntimeError,
"ExtractVector: map_->getmap(" << block <<
",false) is null");
916 RCP<BlockedMultiVector> bfull = Teuchos::rcp_dynamic_cast<BlockedMultiVector>(full);
917 if (bfull.is_null() ==
true) {
920 const RCP<MultiVector> vv =
926 RCP<const Map> oldThyMapFull = full->getMap();
927 full->replaceMap(map_->getImporter(block)->getSourceMap());
928 ExtractVector(*full, block, *vv);
929 TEUCHOS_TEST_FOR_EXCEPTION(map_->getThyraMode() ==
false && bThyraMode ==
true,
930 Xpetra::Exceptions::RuntimeError,
931 "MapExtractor::ExtractVector: ExtractVector in Thyra-style numbering only possible if MapExtractor has been "
932 "created using Thyra-style numbered submaps.");
933 if (bThyraMode ==
true)
934 vv->replaceMap(map_->getMap(block,
true));
935 full->replaceMap(oldThyMapFull);
941 Xpetra::Exceptions::RuntimeError,
942 "ExtractVector: Number of blocks in map extractor is " << map_->getNumMaps() <<
" but should be "
943 << bfull->getBlockedMap()->getNumMaps()
944 <<
" (number of blocks in BlockedMultiVector)");
945 return bfull->getMultiVector(block, bThyraMode);
949 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
950 void BlockedMultiVector<Scalar, LocalOrdinal, GlobalOrdinal, Node>::
956 "ExtractVector: Error, block = " << block <<
" is too big. The BlockedMultiVector only contains " << map_->getNumMaps()
957 <<
" partial blocks.");
958 XPETRA_TEST_FOR_EXCEPTION(map_->getMap(block) == null, Xpetra::Exceptions::RuntimeError,
"ExtractVector: maps_[" << block <<
"] is null");
962 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
967 bool bThyraMode)
const {
970 "ExtractVector: Error, block = " << block <<
" is too big. The MapExtractor only contains " << map_->getNumMaps()
971 <<
" partial blocks.");
973 map_->getMap(block,
false) == null, Xpetra::Exceptions::RuntimeError,
"ExtractVector: map_->getmap(" << block <<
",false) is null");
975 Xpetra::Exceptions::RuntimeError,
976 "MapExtractor::InsertVector: InsertVector in Thyra-style numbering only possible if MapExtractor has been created "
977 "using Thyra-style numbered submaps.");
986 RCP<const MultiVector> rcpPartial = Teuchos::rcpFromRef(partial);
987 RCP<MultiVector> rcpNonConstPartial = Teuchos::rcp_const_cast<
MultiVector>(rcpPartial);
988 RCP<const Map> oldThyMapPartial = rcpNonConstPartial->
getMap();
989 RCP<const Map> oldThyMapFull = full.
getMap();
1002 rcpNonConstPartial->replaceMap(map_->getMap(block,
false));
1003 full.
replaceMap(map_->getImporter(block)->getSourceMap());
1010 rcpNonConstPartial->replaceMap(oldThyMapPartial);
1017 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
1022 bool bThyraMode)
const {
1023 RCP<Xpetra::BlockedMultiVector<Scalar, LocalOrdinal, GlobalOrdinal, Node>> bfull =
1025 if (bfull.is_null() ==
true)
1026 InsertVector(*partial, block, *full, bThyraMode);
1028 XPETRA_TEST_FOR_EXCEPTION(map_->getMap(block) == null, Xpetra::Exceptions::RuntimeError,
"InsertVector: maps_[" << block <<
"] is null");
1033 full->setMultiVector(block, partial, bThyraMode);
1035 throw Xpetra::Exceptions::RuntimeError(
"MultiVector::setMultiVector() is not implemented.");
1040 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
1041 void BlockedMultiVector<Scalar, LocalOrdinal, GlobalOrdinal, Node>::
1045 bool bThyraMode)
const {
1046 RCP<Xpetra::BlockedMultiVector<Scalar, LocalOrdinal, GlobalOrdinal, Node>> bfull =
1048 if (bfull.is_null() ==
true)
1049 InsertVector(*partial, block, *full, bThyraMode);
1051 XPETRA_TEST_FOR_EXCEPTION(map_->getMap(block) == null, Xpetra::Exceptions::RuntimeError,
"InsertVector: maps_[" << block <<
"] is null");
1052 bfull->setMultiVector(block, partial, bThyraMode);
1058 #endif // XPETRA_BLOCKEDMULTIVECTOR_DEF_HPP
virtual void replaceMap(const RCP< const Map< LocalOrdinal, GlobalOrdinal, Node >> &map)=0
virtual void doExport(const DistObject< Scalar, LocalOrdinal, GlobalOrdinal, Node > &source, const Export< LocalOrdinal, GlobalOrdinal, Node > &exporter, CombineMode CM)=0
Export data into this object using an Export object ("forward mode").
static Teuchos::RCP< MultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > > Build(const Teuchos::RCP< const Map< LocalOrdinal, GlobalOrdinal, Node >> &map, size_t NumVectors, bool zeroOut=true)
Constructor specifying the number of non-zeros for all rows.
BlockedMultiVector(const Teuchos::RCP< const BlockedMap > &map, size_t NumVectors, bool zeroOut=true)
Constructor.
Exception throws to report errors in the internal logical of the program.
virtual void Xpetra_randomize()
Set multi-vector values to random numbers. XPetra implementation.
Factory for any type of Xpetra::MultiVector and its derived classes.
virtual void update(const Scalar &alpha, const MultiVector &A, const Scalar &beta)
Update multi-vector values with scaled values of A, this = beta*this + alpha*A.
virtual void doImport(const DistObject< Scalar, LocalOrdinal, GlobalOrdinal, Node > &source, const Import< LocalOrdinal, GlobalOrdinal, Node > &importer, CombineMode CM)=0
Import data into this object using an Import object ("forward mode").
Teuchos::RCP< MultiVector > Merge() const
merge BlockedMultiVector blocks to a single MultiVector
virtual Teuchos::RCP< const Map< LocalOrdinal, GlobalOrdinal, Node > > getMap() const =0
The Map describing the parallel distribution of this object.
virtual void describe(Teuchos::FancyOStream &out, const Teuchos::EVerbosityLevel verbLevel=Teuchos::Describable::verbLevel_default) const
Print the object with the given verbosity level to a FancyOStream.
virtual void setSeed(unsigned int seed)
Set seed for Random function.
size_t global_size_t
Global size_t object.
virtual Teuchos::ArrayRCP< const Scalar > getData(size_t j) const
Const view of the local values in a particular vector of this multivector.
Teuchos::RCP< MultiVector > getMultiVector(size_t r) const
return partial multivector associated with block row r
virtual Teuchos::ArrayRCP< Scalar > getDataNonConst(size_t j)
View of the local values in a particular vector of this multivector.
#define XPETRA_TEST_FOR_EXCEPTION(throw_exception_test, Exception, msg)
virtual size_t getNumVectors() const
Number of columns in the multivector.
CombineMode
Xpetra::Combine Mode enumerable type.
#define XPETRA_MONITOR(funcName)
virtual void randomize(bool bUseXpetraImplementation=false)
Set multi-vector values to random numbers.