46#ifndef XPETRA_MAPEXTRACTOR_DEF_HPP_
47#define XPETRA_MAPEXTRACTOR_DEF_HPP_
49#include <Xpetra_MultiVectorFactory.hpp>
50#include <Xpetra_VectorFactory.hpp>
51#include <Xpetra_BlockedMultiVector.hpp>
58 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
65 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
72 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
79 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
87 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
95 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
101 "ExtractVector: Error, block = " << block <<
" is too big. The MapExtractor only contains " << map_->getNumMaps() <<
" partial blocks.");
103 "ExtractVector: map_->getMap(" << block <<
",false) is null");
109 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
116 "ExtractVector: Error, block = " << block <<
" is too big. The MapExtractor only contains " << map_->getNumMaps()
117 <<
" partial blocks.");
125 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
130 ExtractVector(*full, block, *partial);
134 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
143 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
148 ExtractVector(*full, block, *partial);
152 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
157 ExtractVector(*full, block, *partial);
161 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
168 "ExtractVector: Error, block = " << block <<
" is too big. The MapExtractor only contains " << map_->getNumMaps()
169 <<
" partial blocks.");
174 ExtractVector(*full, block, *vv);
175 if(bThyraMode ==
false)
179 "MapExtractor::ExtractVector: ExtractVector in Thyra-style numbering only possible if MapExtractor has been "
180 "created using Thyra-style numbered submaps.");
181 vv->replaceMap(getMap(block,
true));
186 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
193 "ExtractVector: Error, block = " << block <<
" is too big. The MapExtractor only contains " << map_->getNumMaps()
194 <<
" partial blocks.");
200 ExtractVector(*full, block, *vv);
201 if(bThyraMode ==
false)
205 "MapExtractor::ExtractVector: ExtractVector in Thyra-style numbering only possible if MapExtractor has been "
206 "created using Thyra-style numbered submaps.");
207 vv->replaceMap(getMap(block,
true));
212 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
219 "ExtractVector: Error, block = " << block <<
" is too big. The MapExtractor only contains " << map_->getNumMaps()
220 <<
" partial blocks.");
234 RCP<MultiVector> rcpNonConstFull = Teuchos::rcp_const_cast<MultiVector>(full);
235 rcpNonConstFull->replaceMap(map_->getImporter(block)->getSourceMap());
236 ExtractVector(*rcpNonConstFull, block, *vv);
239 "MapExtractor::ExtractVector: ExtractVector in Thyra-style numbering only possible if MapExtractor has been "
240 "created using Thyra-style numbered submaps.");
241 if(bThyraMode ==
true)
242 vv->replaceMap(getMap(block,
true));
243 rcpNonConstFull->replaceMap(oldThyMapFull);
252 "ExtractVector: Number of blocks in map extractor is " << map_->getNumMaps() <<
" but should be "
253 << bfull->getBlockedMap()->getNumMaps()
254 <<
" (number of blocks in BlockedMultiVector)");
255 return bfull->getMultiVector(block, bThyraMode);
260 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
267 "ExtractVector: Error, block = " << block <<
" is too big. The MapExtractor only contains " << map_->getNumMaps()
268 <<
" partial blocks.");
282 full->replaceMap(map_->getImporter(block)->getSourceMap());
283 ExtractVector(*full, block, *vv);
286 "MapExtractor::ExtractVector: ExtractVector in Thyra-style numbering only possible if MapExtractor has been "
287 "created using Thyra-style numbered submaps.");
288 if(bThyraMode ==
true)
289 vv->replaceMap(getMap(block,
true));
290 full->replaceMap(oldThyMapFull);
299 "ExtractVector: Number of blocks in map extractor is " << map_->getNumMaps() <<
" but should be "
300 << bfull->getBlockedMap()->getNumMaps()
301 <<
" (number of blocks in BlockedMultiVector)");
302 return bfull->getMultiVector(block, bThyraMode);
307 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
314 "ExtractVector: Error, block = " << block <<
" is too big. The MapExtractor only contains " << map_->getNumMaps()
315 <<
" partial blocks.");
320 "ExtractVector: Number of blocks in map extractor is " << map_->getNumMaps() <<
" but should be "
321 << full->getBlockedMap()->getNumMaps()
322 <<
" (number of blocks in BlockedMultiVector)");
328 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
335 "ExtractVector: Error, block = " << block <<
" is too big. The MapExtractor only contains " << map_->getNumMaps()
336 <<
" partial blocks.");
341 "ExtractVector: Number of blocks in map extractor is " << map_->getNumMaps() <<
" but should be "
342 << full->getBlockedMap()->getNumMaps()
343 <<
" (number of blocks in BlockedMultiVector)");
349 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
356 "ExtractVector: Error, block = " << block <<
" is too big. The MapExtractor only contains " << map_->getNumMaps()
357 <<
" partial blocks.");
362 "MapExtractor::InsertVector: InsertVector in Thyra-style numbering only possible if MapExtractor has been created "
363 "using Thyra-style numbered submaps.");
374 RCP<MultiVector> rcpNonConstPartial = Teuchos::rcp_const_cast<MultiVector>(rcpPartial);
381 "MapExtractor::InsertVector: InsertVector in Thyra-style mode: Xpetra GIDs of partial vector are not identical "
382 "to target Map of Importer. This should not be.");
389 rcpNonConstPartial->replaceMap(getMap(block,
false));
390 full.
replaceMap(map_->getImporter(block)->getSourceMap());
397 rcpNonConstPartial->replaceMap(oldThyMapPartial);
407 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
414 "ExtractVector: Error, block = " << block <<
" is too big. The MapExtractor only contains " << map_->getNumMaps()
415 <<
" partial blocks.");
420 "MapExtractor::InsertVector: InsertVector in Thyra-style numbering only possible if MapExtractor has been created "
421 "using Thyra-style numbered submaps.");
432 RCP<MultiVector> rcpNonConstPartial = Teuchos::rcp_const_cast<MultiVector>(rcpPartial);
439 "MapExtractor::InsertVector: InsertVector in Thyra-style mode: Xpetra GIDs of partial vector are not identical "
440 "to target Map of Importer. This should not be.");
447 rcpNonConstPartial->replaceMap(getMap(block,
false));
448 full.
replaceMap(map_->getImporter(block)->getSourceMap());
455 rcpNonConstPartial->replaceMap(oldThyMapPartial);
466 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
471 InsertVector(*partial, block, *full, bThyraMode);
475 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
480 InsertVector(*partial, block, *full, bThyraMode);
484 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
490 bool bThyraMode)
const
494 InsertVector(*partial, block, *full, bThyraMode);
503 full->setMultiVector(block, partial, bThyraMode);
505 throw std::runtime_error(
"Xpetra::MultiVector::setMultiVector() doesn't exist in "+std::string(__FILE__)+
":"+std::to_string(__LINE__));
511 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
517 bool bThyraMode)
const
521 InsertVector(*partial, block, *full, bThyraMode);
527 bfull->setMultiVector(block, partial, bThyraMode);
532 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
538 bool bThyraMode)
const
543 full->setMultiVector(block, partial, bThyraMode);
547 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
553 bool bThyraMode )
const
557 full->setMultiVector(block, partial, bThyraMode);
561 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
562 RCP<Xpetra::Vector<Scalar, LocalOrdinal, GlobalOrdinal, Node>>
564 getVector(
size_t i,
bool bThyraMode,
bool bZero)
const
568 "MapExtractor::getVector: getVector in Thyra-style numbering only possible if MapExtractor has been created using "
569 "Thyra-style numbered submaps.");
575 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
578 getVector(
size_t i,
size_t numvec,
bool bThyraMode,
bool bZero)
const
582 "MapExtractor::getVector: getVector in Thyra-style numbering only possible if MapExtractor has been created using "
583 "Thyra-style numbered submaps.");
589 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
594 return map_->getThyraMode();
598 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
603 return map_->getNumMaps();
607 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
610 getMap(
size_t i,
bool bThyraMode)
const
612 return map_->getMap(i, bThyraMode);
616 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
625 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
634 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
639 return map_->getFullMap();
643 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
648 return map_->getMapIndexForGID(gid);
#define XPETRA_TEST_FOR_EXCEPTION(throw_exception_test, Exception, msg)
virtual void doExport(const DistObject< Packet, LocalOrdinal, GlobalOrdinal, Node > &source, const Export< LocalOrdinal, GlobalOrdinal, Node > &exporter, CombineMode CM)=0
Export data into this object using an Export object ("forward mode").
virtual Teuchos::RCP< const Map< LocalOrdinal, GlobalOrdinal, Node > > getMap() const =0
The Map describing the parallel distribution of this object.
virtual void doImport(const DistObject< Packet, LocalOrdinal, GlobalOrdinal, Node > &source, const Import< LocalOrdinal, GlobalOrdinal, Node > &importer, CombineMode CM)=0
Import data into this object using an Import object ("forward mode").
Exception throws to report errors in the internal logical of the program.
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.
virtual void replaceMap(const RCP< const Map< LocalOrdinal, GlobalOrdinal, Node > > &map)=0
static Teuchos::RCP< Xpetra::Vector< Scalar, LocalOrdinal, GlobalOrdinal, Node > > Build(const Teuchos::RCP< const Xpetra::Map< LocalOrdinal, GlobalOrdinal, Node > > &map, bool zeroOut=true)
Constructor specifying the number of non-zeros for all rows.
#define TEUCHOS_TEST_FOR_EXCEPTION(throw_exception_test, Exception, msg)
TEUCHOS_DEPRECATED RCP< T > rcp(T *p, Dealloc_T dealloc, bool owns_mem)