46#ifndef XPETRA_REORDEREDBLOCKEDMULTIVECTOR_HPP
47#define XPETRA_REORDEREDBLOCKEDMULTIVECTOR_HPP
49#include <KokkosCompat_DefaultNode.hpp>
57#include "Xpetra_BlockedMap.hpp"
58#include "Xpetra_BlockedMultiVector.hpp"
69 template <
class Scalar,
82#undef XPETRA_REORDEREDBLOCKEDMULTIVECTOR_SHORT
111 brm_ = Teuchos::null;
122 size_t numBlocks = brm->GetNumBlocks();
131 map = bmap->getMap(Teuchos::as<size_t>(leaf->GetIndex()),
false);
134 std::vector<Teuchos::RCP<const Map> > subMaps (numBlocks, Teuchos::null);
136 for(
size_t i = 0; i < numBlocks; i++) {
153 std::string
description()
const {
return "ReorderedBlockedMultiVector"; }
171template<
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
179 size_t numBlocks = brm->GetNumBlocks();
187 map = bmap->getMap(Teuchos::as<size_t>(leaf->GetIndex()), bThyraMode);
190 std::vector<Teuchos::RCP<const Xpetra::Map<LocalOrdinal,GlobalOrdinal,Node> > > subMaps (numBlocks, Teuchos::null);
192 for(
size_t i = 0; i < numBlocks; i++) {
216template<
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
227 size_t rowSz = rowMgr->GetNumBlocks();
243 if(subBVec == Teuchos::null) {
259 std::vector<Teuchos::RCP<const Map> > rowSubMaps (1, submap);
263 rbvec->setMultiVector(0,Teuchos::rcp_const_cast<MultiVector>(vec),
false);
274 std::vector<Teuchos::RCP<const Map> > rowSubMaps (rowSz, Teuchos::null);
275 for(
size_t i = 0; i < rowSz; i++) {
284 for(
size_t i = 0; i < rowSz; i++) {
287 rbvec->setMultiVector(i,Teuchos::rcp_const_cast<MultiVector>(subvec),
false);
294template<
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
305 size_t rowSz = rowMgr->GetNumBlocks();
320 if(bbvec == Teuchos::null) {
327 std::vector<Teuchos::RCP<const Map> > rowXpSubMaps (1, xpsubmap);
328 std::vector<Teuchos::RCP<const Map> > rowTySubMaps (1, thysubmap);
335 rbvec->setMultiVector(0,vec,
true);
338 rbvec = Teuchos::rcp_dynamic_cast<BlockedMultiVector>(vec);
344 std::vector<Teuchos::RCP<const Map> > rowXpSubMaps (rowSz, Teuchos::null);
345 std::vector<Teuchos::RCP<const Map> > rowTySubMaps (rowSz, Teuchos::null);
346 for(
size_t i = 0; i < rowSz; i++) {
359 for(
size_t i = 0; i < rowSz; i++) {
362 rbvec->setMultiVector(i,Teuchos::rcp_const_cast<MultiVector>(subvec),
true);
369template<
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
373 if(bvec->getBlockedMap()->getThyraMode() ==
false) {
382template<
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
388 if(bvec->getBlockedMap()->getThyraMode() ==
false) {
389 rbvec =
mergeSubBlocks(brm,Teuchos::rcp_const_cast<const BlockedMultiVector>(bvec));
394 return Teuchos::rcp_const_cast<MultiVector>(rbvec);
400#define XPETRA_REORDEREDBLOCKEDMULTIVECTOR_SHORT
static const EVerbosityLevel verbLevel_default
virtual size_t getNumVectors() const
Number of columns in the multivector.
Xpetra utility class for common map-related routines.
static Teuchos::RCP< const Xpetra::Map< LocalOrdinal, GlobalOrdinal, Node > > concatenateMaps(const std::vector< Teuchos::RCP< const Xpetra::Map< LocalOrdinal, GlobalOrdinal, Node > > > &subMaps)
Helper function to concatenate several maps.
std::string description() const
Return a simple one-line description of this object.
GlobalOrdinal global_ordinal_type
Teuchos::RCP< const Xpetra::BlockedMultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > > fullVec_
LocalOrdinal local_ordinal_type
virtual ~ReorderedBlockedMultiVector()
Destructor.
void describe(Teuchos::FancyOStream &out, const Teuchos::EVerbosityLevel verbLevel=Teuchos::Describable::verbLevel_default) const
Print the object with some verbosity level to an FancyOStream object.
ReorderedBlockedMultiVector(Teuchos::RCP< const BlockedMap > &rangeMap, Teuchos::RCP< const Xpetra::BlockReorderManager > brm, Teuchos::RCP< const Xpetra::BlockedMultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > > bvec)
Constructor.
Teuchos::RCP< const Xpetra::BlockReorderManager > brm_
Teuchos::RCP< const Xpetra::Map< LocalOrdinal, GlobalOrdinal, Node > > mergeSubBlockMaps(Teuchos::RCP< const Xpetra::BlockReorderManager > brm)
#define TEUCHOS_ASSERT(assertion_test)
TEUCHOS_DEPRECATED RCP< T > rcp(T *p, Dealloc_T dealloc, bool owns_mem)
Teuchos::RCP< const Xpetra::Map< LocalOrdinal, GlobalOrdinal, Node > > mergeSubBlockMaps(Teuchos::RCP< const Xpetra::BlockReorderManager > brm, Teuchos::RCP< const Xpetra::BlockedCrsMatrix< Scalar, LocalOrdinal, GlobalOrdinal, Node > > bmat, bool bThyraMode)
Teuchos::RCP< const Xpetra::Matrix< Scalar, LocalOrdinal, GlobalOrdinal, Node > > mergeSubBlocks(Teuchos::RCP< const Xpetra::BlockReorderManager > rowMgr, Teuchos::RCP< const Xpetra::BlockReorderManager > colMgr, Teuchos::RCP< const Xpetra::BlockedCrsMatrix< Scalar, LocalOrdinal, GlobalOrdinal, Node > > bmat)
Teuchos::RCP< const Xpetra::Matrix< Scalar, LocalOrdinal, GlobalOrdinal, Node > > mergeSubBlocksThyra(Teuchos::RCP< const Xpetra::BlockReorderManager > rowMgr, Teuchos::RCP< const Xpetra::BlockReorderManager > colMgr, Teuchos::RCP< const Xpetra::BlockedCrsMatrix< Scalar, LocalOrdinal, GlobalOrdinal, Node > > bmat)
Teuchos::RCP< const Xpetra::MultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > > buildReorderedBlockedMultiVector(Teuchos::RCP< const Xpetra::BlockReorderManager > brm, Teuchos::RCP< const Xpetra::BlockedMultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > > bvec)