53 #ifndef MUELU_SUBBLOCKAFACTORY_DEF_HPP_
54 #define MUELU_SUBBLOCKAFACTORY_DEF_HPP_
69 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
74 validParamList->
set<
int > (
"block row", 0,
"Block row of subblock matrix A");
75 validParamList->
set<
int > (
"block col", 0,
"Block column of subblock matrix A");
77 validParamList->
set< std::string > (
"Range map: Striding info",
"{}",
"Striding information for range map");
78 validParamList->
set< LocalOrdinal > (
"Range map: Strided block id", -1,
"Strided block id for range map" );
79 validParamList->
set< std::string > (
"Domain map: Striding info",
"{}",
"Striding information for domain map");
80 validParamList->
set< LocalOrdinal > (
"Domain map: Strided block id", -1,
"Strided block id for domain map" );
82 return validParamList;
85 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
87 Input(currentLevel,
"A");
90 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
95 size_t row = Teuchos::as<size_t>(pL.
get<
int>(
"block row"));
96 size_t col = Teuchos::as<size_t>(pL.
get<
int>(
"block col"));
114 if (bOp != Teuchos::null) {
116 if (bOp->Rows() == 1 && bOp->Cols() == 1) {
118 Op = bOp->getCrsMatrix();
119 TEUCHOS_TEST_FOR_EXCEPTION(Teuchos::rcp_dynamic_cast<CrsMatrixWrap>(Op) == Teuchos::null,
Exceptions::BadCast,
"SubBlockAFactory::Build: sub block A[" << row <<
"," << col <<
"] must be a single block CrsMatrixWrap object!");
124 GetOStream(
Statistics1) <<
"A(" << row <<
"," << col <<
") is a " << bOp->Rows() <<
"x" << bOp->Cols() <<
" block matrix" << std::endl;
125 GetOStream(
Statistics2) <<
"with altogether " << bOp->getGlobalNumRows() <<
"x" << bOp->getGlobalNumCols() <<
" rows and columns." << std::endl;
126 currentLevel.
Set(
"A", Op,
this);
142 TEUCHOS_TEST_FOR_EXCEPTION(Teuchos::rcp_dynamic_cast<CrsMatrixWrap>(Op) == Teuchos::null,
Exceptions::BadCast,
"SubBlockAFactory::Build: sub block A[" << row <<
"," << col <<
"] is NOT a BlockedCrsMatrix but also NOT a CrsMatrixWrap object? This cannot be.");
150 std::vector<size_t> rangeStridingInfo;
151 std::vector<size_t> domainStridingInfo;
152 LocalOrdinal rangeStridedBlockId = 0;
153 LocalOrdinal domainStridedBlockId = 0;
154 bool bRangeUserSpecified = CheckForUserSpecifiedBlockInfo(
true, rangeStridingInfo, rangeStridedBlockId);
155 bool bDomainUserSpecified = CheckForUserSpecifiedBlockInfo(
false, domainStridingInfo, domainStridedBlockId);
166 if(bRangeUserSpecified) srangeMap =
Teuchos::rcp(
new StridedMap(rangeMap,rangeStridingInfo,rangeMap->getIndexBase(),rangeStridedBlockId,0));
167 else srangeMap = rcp_dynamic_cast<
const StridedMap>(rangeMap);
169 if(bDomainUserSpecified) sdomainMap =
Teuchos::rcp(
new StridedMap(domainMap,domainStridingInfo,domainMap->getIndexBase(),domainStridedBlockId,0));
170 else sdomainMap = rcp_dynamic_cast<
const StridedMap>(domainMap);
180 std::vector<size_t> stridedData = sFullRangeMap->getStridingData();
181 if (stridedData.size() == 1 && row > 0) {
183 srangeMap = StridedMapFactory::Build(rangeMap, stridedData, 0, sFullRangeMap->getOffset());
187 srangeMap = StridedMapFactory::Build(rangeMap, stridedData, row, sFullRangeMap->getOffset());
196 std::vector<size_t> stridedData = sFullDomainMap->getStridingData();
197 if (stridedData.size() == 1 && col > 0) {
199 sdomainMap = StridedMapFactory::Build(domainMap, stridedData, 0, sFullDomainMap->getOffset());
203 sdomainMap = StridedMapFactory::Build(domainMap, stridedData, col, sFullDomainMap->getOffset());
210 GetOStream(
Statistics1) <<
"A(" << row <<
"," << col <<
") is a single block and has strided maps:"
211 <<
"\n range map fixed block size = " << srangeMap ->getFixedBlockSize() <<
", strided block id = " << srangeMap ->getStridedBlockId()
212 <<
"\n domain map fixed block size = " << sdomainMap->getFixedBlockSize() <<
", strided block id = " << sdomainMap->getStridedBlockId() << std::endl;
213 GetOStream(
Statistics2) <<
"A(" << row <<
"," << col <<
") has " << Op->getGlobalNumRows() <<
"x" << Op->getGlobalNumCols() <<
" rows and columns." << std::endl;
216 if (Op->IsView(
"stridedMaps") ==
true)
217 Op->RemoveView(
"stridedMaps");
218 Op->CreateView(
"stridedMaps", srangeMap, sdomainMap);
222 currentLevel.
Set(
"A", Op,
this);
225 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
230 stridedBlockId = pL.
get<LocalOrdinal>(
"Range map: Strided block id");
232 stridedBlockId = pL.
get<LocalOrdinal>(
"Domain map: Strided block id");
237 if(bRange ==
true) str = std::string(
"Range map: Striding info");
238 else str = std::string(
"Domain map: Striding info");
240 std::string strStridingInfo = pL.
get<std::string>(str);
241 if(strStridingInfo.empty() ==
false) {
243 stridingInfo = Teuchos::createVector(arrayVal);
247 if(stridingInfo.size() > 0)
return true;
Exception indicating invalid cast attempted.
T & Get(const std::string &ename, const FactoryBase *factory=NoFactory::get())
Get data without decrementing associated storage counter (i.e., read-only access). Usage: Level->Get< RCP<Matrix> >("A", factory) if factory == NULL => use default factory.
T & get(const std::string &name, T def_value)
ParameterList & set(std::string const &name, T const &value, std::string const &docString="", RCP< const ParameterEntryValidator > const &validator=null)
Timer to be used in factories. Similar to Monitor but with additional timers.
#define TEUCHOS_TEST_FOR_EXCEPTION(throw_exception_test, Exception, msg)
bool CheckForUserSpecifiedBlockInfo(bool bRange, std::vector< size_t > &stridingInfo, LocalOrdinal &stridedBlockId) const
void DeclareInput(Level ¤tLevel) const
Specifies the data that this class needs, and the factories that generate that data.
void Build(Level ¤tLevel) const
Build an object with this factory.
Print even more statistics.
bool isParameter(const std::string &name) const
TEUCHOS_DEPRECATED RCP< T > rcp(T *p, Dealloc_T dealloc, bool owns_mem)
Class that holds all level-specific information.
RCP< const ParameterList > GetValidParameterList() const
Input.
void Set(const std::string &ename, const T &entry, const FactoryBase *factory=NoFactory::get())
Exception throws to report errors in the internal logical of the program.
static const RCP< const NoFactory > getRCP()
Static Get() functions.