46 #ifndef MUELU_RAPSHIFTFACTORY_DEF_HPP
47 #define MUELU_RAPSHIFTFACTORY_DEF_HPP
60 #include "MueLu_PerfUtils.hpp"
65 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
67 : implicitTranspose_(false) {}
70 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
74 #define SET_VALID_ENTRY(name) validParamList->setEntry(name, MasterList::getEntry(name))
83 #undef SET_VALID_ENTRY
85 validParamList->
set<
RCP<const FactoryBase> >(
"A", Teuchos::null,
"Generating factory of the matrix A used during the prolongator smoothing process");
86 validParamList->
set<
RCP<const FactoryBase> >(
"M", Teuchos::null,
"Generating factory of the matrix M used during the non-Galerkin RAP");
87 validParamList->
set<
RCP<const FactoryBase> >(
"Mdiag", Teuchos::null,
"Generating factory of the matrix Mdiag used during the non-Galerkin RAP");
88 validParamList->
set<
RCP<const FactoryBase> >(
"K", Teuchos::null,
"Generating factory of the matrix K used during the non-Galerkin RAP");
92 validParamList->
set<
bool>(
"CheckMainDiagonal",
false,
"Check main diagonal for zeros");
93 validParamList->
set<
bool>(
"RepairMainDiagonal",
false,
"Repair zeros on main diagonal");
97 validParamList->
set<
RCP<const FactoryBase> >(
"cfl-based shift array", Teuchos::null,
"MueLu-generated shift array for CFL-based shifting");
101 norecurse.disableRecursiveValidation();
102 validParamList->
set<
ParameterList>(
"matrixmatrix: kernel params", norecurse,
"MatrixMatrix kernel parameters");
104 return validParamList;
108 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
112 bool use_mdiag =
false;
114 use_mdiag = pL.
get<
bool>(
"rap: shift diagonal M");
117 bool use_low_storage =
false;
119 use_low_storage = pL.
get<
bool>(
"rap: shift low storage");
120 use_mdiag = use_low_storage ?
true : use_mdiag;
123 if (implicitTranspose_ ==
false) {
124 Input(coarseLevel,
"R");
127 if (!use_low_storage)
128 Input(fineLevel,
"K");
130 Input(fineLevel,
"A");
131 Input(coarseLevel,
"P");
134 Input(fineLevel,
"M");
136 Input(fineLevel,
"Mdiag");
146 "deltaT was not provided by the user on level0!");
154 "cfl was not provided by the user on level0!");
157 Input(fineLevel,
"cfl-based shift array");
162 for (std::vector<
RCP<const FactoryBase> >::const_iterator it = transferFacts_.begin(); it != transferFacts_.end(); ++it) {
163 (*it)->CallDeclareInput(coarseLevel);
167 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
173 bool M_is_diagonal =
false;
175 M_is_diagonal = pL.
get<
bool>(
"rap: shift diagonal M");
178 bool use_low_storage =
false;
180 use_low_storage = pL.
get<
bool>(
"rap: shift low storage");
181 M_is_diagonal = use_low_storage ?
true : M_is_diagonal;
194 double dt = Get<double>(fineLevel,
"deltaT");
195 double cfl = Get<double>(fineLevel,
"cfl");
196 double ts_at_cfl1 = dt / cfl;
202 for (
int i = 1; i < (int)CFLs.
size(); i++)
203 myCFLs[i] = (CFLs[i] > cfl) ? cfl : CFLs[i];
206 std::ostringstream ofs;
207 ofs <<
"RAPShiftFactory: CFL schedule = ";
208 for (
int i = 0; i < (int)CFLs.
size(); i++)
209 ofs <<
" " << myCFLs[i];
212 GetOStream(
Statistics0) <<
"RAPShiftFactory: Timestep at CFL=1 is " << ts_at_cfl1 <<
" " << std::endl;
215 for (
int i = 0; i < (int)myshifts.
size(); i++)
216 myshifts[i] = 1.0 / (ts_at_cfl1 * myCFLs[i]);
217 doubleShifts = myshifts();
220 std::ostringstream ofs;
221 ofs <<
"RAPShiftFactory: shift schedule = ";
222 for (
int i = 0; i < (int)doubleShifts.
size(); i++)
223 ofs <<
" " << doubleShifts[i];
226 Set(coarseLevel,
"cfl-based shift array", myshifts);
228 myshifts = Get<Teuchos::ArrayRCP<double> >(fineLevel,
"cfl-based shift array");
229 doubleShifts = myshifts();
230 Set(coarseLevel,
"cfl-based shift array", myshifts);
242 K = Get<RCP<Matrix> >(fineLevel,
"A");
244 K = Get<RCP<Matrix> >(fineLevel,
"K");
246 M = Get<RCP<Matrix> >(fineLevel,
"M");
248 Mdiag = Get<RCP<Vector> >(fineLevel,
"Mdiag");
250 RCP<Matrix> P = Get<RCP<Matrix> >(coarseLevel,
"P");
265 if (!M_is_diagonal) {
269 MP->leftScale(*Mdiag);
272 Set(coarseLevel,
"AP Pattern", KP);
275 bool doOptimizedStorage =
true;
283 bool doFillComplete =
true;
284 if (implicitTranspose_) {
286 Kc =
Xpetra::MatrixMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::Multiply(*P,
true, *KP,
false, Kc, GetOStream(
Statistics2), doFillComplete, doOptimizedStorage);
287 Mc =
Xpetra::MatrixMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::Multiply(*P,
true, *MP,
false, Mc, GetOStream(
Statistics2), doFillComplete, doOptimizedStorage);
289 RCP<Matrix> R = Get<RCP<Matrix> >(coarseLevel,
"R");
291 Kc =
Xpetra::MatrixMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::Multiply(*R,
false, *KP,
false, Kc, GetOStream(
Statistics2), doFillComplete, doOptimizedStorage);
292 Mc =
Xpetra::MatrixMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::Multiply(*R,
false, *MP,
false, Mc, GetOStream(
Statistics2), doFillComplete, doOptimizedStorage);
299 int level = coarseLevel.GetLevelID();
301 if (!use_low_storage) {
303 if (level < (
int)shifts_.size())
304 shift = shifts_[level];
306 shift = Teuchos::as<Scalar>(pL.
get<
double>(
"rap: shift"));
309 if (level < (
int)shifts_.size()) {
311 shift = shifts_[level];
314 for (
int i = 1; i < level - 1; i++) {
317 shift = (prod1 * shifts_[level] - prod1);
319 }
else if (doubleShifts.
size() != 0) {
320 double d_shift = 0.0;
321 if (level < doubleShifts.
size())
322 d_shift = doubleShifts[level] - doubleShifts[level - 1];
325 GetOStream(
Warnings1) <<
"WARNING: RAPShiftFactory has detected a negative shift... This implies a less stable coarse grid." << std::endl;
326 shift = Teuchos::as<Scalar>(d_shift);
328 double base_shift = pL.
get<
double>(
"rap: shift");
330 shift = Teuchos::as<Scalar>(base_shift);
332 shift = Teuchos::as<Scalar>(pow(base_shift, level) - pow(base_shift, level - 1));
335 GetOStream(
Runtime0) <<
"RAPShiftFactory: Using shift " << shift << std::endl;
340 Xpetra::MatrixMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::TwoMatrixAdd(*Kc,
false,
Teuchos::ScalarTraits<Scalar>::one(), *Mc,
false, shift, Ac, GetOStream(
Statistics2));
345 if (relativeFloor.
size() > 0)
348 bool repairZeroDiagonals = pL.
get<
bool>(
"RepairMainDiagonal") || pL.
get<
bool>(
"rap: fix zero diagonals");
349 bool checkAc = pL.
get<
bool>(
"CheckMainDiagonal") || pL.
get<
bool>(
"rap: fix zero diagonals");
351 if (checkAc || repairZeroDiagonals)
356 params->
set(
"printLoadBalancingInfo",
true);
359 Set(coarseLevel,
"A", Ac);
361 if (!use_low_storage)
362 Set(coarseLevel,
"K", Kc);
364 if (!M_is_diagonal) {
365 Set(coarseLevel,
"M", Mc);
370 Mc->getLocalDiagCopy(*Mcv);
371 Set(coarseLevel,
"Mdiag", Mcv);
377 if (transferFacts_.begin() != transferFacts_.end()) {
381 for (std::vector<
RCP<const FactoryBase> >::const_iterator it = transferFacts_.begin(); it != transferFacts_.end(); ++it) {
383 GetOStream(
Runtime0) <<
"RAPShiftFactory: call transfer factory: " << fac->
description() << std::endl;
392 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
395 TEUCHOS_TEST_FOR_EXCEPTION(Teuchos::rcp_dynamic_cast<const TwoLevelFactoryBase>(factory) == Teuchos::null,
Exceptions::BadCast,
"MueLu::RAPShiftFactory::AddTransferFactory: Transfer factory is not derived from TwoLevelFactoryBase. This is very strange. (Note: you can remove this exception if there's a good reason for)");
396 transferFacts_.push_back(factory);
401 #define MUELU_RAPSHIFTFACTORY_SHORT
402 #endif // MUELU_RAPSHIFTFACTORY_DEF_HPP
Exception indicating invalid cast attempted.
void Build(Level &fineLevel, Level &coarseLevel) const
Build an object with this factory.
virtual void CallBuild(Level &requestedLevel) const =0
void Release(const FactoryBase &factory)
Decrement the storage counter for all the inputs of a 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)
One-liner description of what is happening.
RCP< const ParameterList > GetValidParameterList() const
Return a const parameter list of valid parameters that setParameterList() will accept.
static void CheckRepairMainDiagonal(RCP< Xpetra::Matrix< Scalar, LocalOrdinal, GlobalOrdinal, Node >> &Ac, bool const &repairZeroDiagonals, Teuchos::FancyOStream &fos, const typename Teuchos::ScalarTraits< Scalar >::magnitudeType threshold=Teuchos::ScalarTraits< typename Teuchos::ScalarTraits< Scalar >::magnitudeType >::zero())
static const NoFactory * get()
Print even more statistics.
void resize(const size_type n, const T &val=T())
bool isParameter(const std::string &name) const
void DeclareInput(Level &fineLevel, Level &coarseLevel) const
Input.
Print statistics that do not involve significant additional computation.
TEUCHOS_DEPRECATED RCP< T > rcp(T *p, Dealloc_T dealloc, bool owns_mem)
MueLu::DefaultScalar Scalar
Class that holds all level-specific information.
Timer to be used in factories. Similar to SubMonitor but adds a timer level by level.
static RCP< Xpetra::Matrix< Scalar, LocalOrdinal, GlobalOrdinal, Node > > BuildCopy(const RCP< const Xpetra::Matrix< Scalar, LocalOrdinal, GlobalOrdinal, Node > > A)
static std::string PrintMatrixInfo(const Matrix &A, const std::string &msgTag, RCP< const Teuchos::ParameterList > params=Teuchos::null)
static RCP< Vector > Build(const Teuchos::RCP< const Map > &map, bool zeroOut=true)
static void TwoMatrixAdd(const Matrix &A, bool transposeA, SC alpha, Matrix &B, SC beta)
void AddTransferFactory(const RCP< const FactoryBase > &factory)
Add transfer factory in the end of list of transfer factories in RepartitionAcFactory.
static void Multiply(const Matrix &A, bool transposeA, const Matrix &B, bool transposeB, Matrix &C, bool call_FillComplete_on_result=true, bool doOptimizeStorage=true, const std::string &label=std::string(), const RCP< ParameterList > ¶ms=null)
int GetLevelID() const
Return level number.
Exception throws to report errors in the internal logical of the program.
void DeclareInput(const std::string &ename, const FactoryBase *factory, const FactoryBase *requestedBy=NoFactory::get())
Callback from FactoryBase::CallDeclareInput() and FactoryBase::DeclareInput()
virtual std::string description() const
Return a simple one-line description of this object.
bool IsAvailable(const std::string &ename, const FactoryBase *factory=NoFactory::get()) const
Test whether a need's value has been saved.
#define SET_VALID_ENTRY(name)