TEUCHOS_UNIT_TEST_TEMPLATE_4_DECL(TransPFactory, Correctness, Scalar, LocalOrdinal, GlobalOrdinal, Node)
{
# include <MueLu_UseShortNames.hpp>
MUELU_TESTING_SET_OSTREAM;
MUELU_TESTING_LIMIT_SCOPE(Scalar,GlobalOrdinal,Node);
out << "version: " << MueLu::Version() << std::endl;
RCP<const Teuchos::Comm<int> > comm = TestHelpers::Parameters::getDefaultComm();
Level fineLevel, coarseLevel;
TestHelpers::TestFactory<SC, LO, GO, NO>::createTwoLevelHierarchy(fineLevel, coarseLevel);
// Test of createTwoLevelHierarchy: to be moved...
TEST_EQUALITY(fineLevel.GetLevelID(), 0);
TEST_EQUALITY(coarseLevel.GetLevelID(), 1);
//compilation warning TEST_EQUALITY(fineLevel.GetPreviousLevel().get(), NULL);
//TEST_EQUALITY(coarseLevel.GetPreviousLevel().get(), &fineLevel);
// --
RCP<Matrix> Op = TestHelpers::TestFactory<SC, LO, GO, NO>::Build1DPoisson(27*comm->getSize());
fineLevel.Set("A",Op);
SaPFactory sapFactory;
TransPFactory transPFact;
transPFact.SetFactory("P", rcpFromRef(sapFactory));
coarseLevel.Request(sapFactory);
coarseLevel.Request(transPFact);
coarseLevel.Request("P", &sapFactory);
coarseLevel.Request("R", &transPFact);
sapFactory.BuildP(fineLevel,coarseLevel);
transPFact.Build(fineLevel,coarseLevel);
RCP<Matrix> P = coarseLevel.Get< RCP<Matrix> >("P", &sapFactory);
RCP<Matrix> R = coarseLevel.Get< RCP<Matrix> >("R", &transPFact);
RCP<MultiVector> result1 = MultiVectorFactory::Build(P->getDomainMap(),1);
RCP<MultiVector> result2 = MultiVectorFactory::Build(R->getRangeMap(),1);
RCP<MultiVector> X = MultiVectorFactory::Build(P->getRangeMap(),1);
X->randomize();
//Calculate P^T * X
P->apply(*X,*result1,Teuchos::TRANS,(SC)1.0,(SC)0.0);
//Calculate R * X
R->apply(*X,*result2,Teuchos::NO_TRANS,(SC)1.0,(SC)0.0);
Teuchos::Array<typename Teuchos::ScalarTraits<SC>::magnitudeType> normX(1), normResult1(1),normResult2(1);
X->norm2(normX);
out << "This test checks the correctness of the R created by TransPFactory." << std::endl;
out << "||X||_2 = " << normX << std::endl;
result1->norm2(normResult1);
result2->norm2(normResult2);
TEST_FLOATING_EQUALITY(normResult1[0], normResult2[0], 1e-12);
} //Correctness test
TEUCHOS_UNIT_TEST(RAPFactory, Correctness)
{
out << "version: " << MueLu::Version() << std::endl;
RCP<const Teuchos::Comm<int> > comm = Parameters::getDefaultComm();
Level fineLevel, coarseLevel; TestHelpers::TestFactory<SC, LO, GO, NO, LMO>::createTwoLevelHierarchy(fineLevel, coarseLevel);
RCP<Matrix> Op = TestHelpers::TestFactory<SC, LO, GO, NO, LMO>::Build1DPoisson(27*comm->getSize());
fineLevel.Set("A",Op);
TentativePFactory tentpFactory;
SaPFactory sapFactory;
sapFactory.SetFactory("P",rcpFromRef(tentpFactory));
TransPFactory transPFactory;
transPFactory.SetFactory("P", rcpFromRef(sapFactory)); //todo:rcpFromRef
coarseLevel.Request("P",&sapFactory);
coarseLevel.Request("R",&transPFactory);
coarseLevel.Request(sapFactory);
coarseLevel.Request(transPFactory);
sapFactory.Build(fineLevel,coarseLevel);
transPFactory.Build(fineLevel,coarseLevel);
RAPFactory rap;
rap.SetFactory("P", rcpFromRef(sapFactory));
rap.SetFactory("R", rcpFromRef(transPFactory));
coarseLevel.Request(rap);
coarseLevel.Request("A",&rap);
rap.Build(fineLevel,coarseLevel);
RCP<Matrix> A = fineLevel.Get< RCP<Matrix> >("A");
RCP<Matrix> P = coarseLevel.Get< RCP<Matrix> >("P", &sapFactory);
RCP<Matrix> R = coarseLevel.Get< RCP<Matrix> >("R", &transPFactory);
RCP<MultiVector> workVec1 = MultiVectorFactory::Build(P->getRangeMap(),1);
RCP<MultiVector> workVec2 = MultiVectorFactory::Build(Op->getRangeMap(),1);
RCP<MultiVector> result1 = MultiVectorFactory::Build(R->getRangeMap(),1);
RCP<MultiVector> X = MultiVectorFactory::Build(P->getDomainMap(),1);
X->randomize();
//Calculate result1 = R*(A*(P*X))
P->apply(*X,*workVec1,Teuchos::NO_TRANS,(SC)1.0,(SC)0.0);
Op->apply(*workVec1,*workVec2,Teuchos::NO_TRANS,(SC)1.0,(SC)0.0);
R->apply(*workVec2,*result1,Teuchos::NO_TRANS,(SC)1.0,(SC)0.0);
RCP<Matrix> coarseOp = coarseLevel.Get< RCP<Matrix> >("A", &rap);
//Calculate result2 = (R*A*P)*X
RCP<MultiVector> result2 = MultiVectorFactory::Build(R->getRangeMap(),1);
coarseOp->apply(*X,*result2,Teuchos::NO_TRANS,(SC)1.0,(SC)0.0);
Teuchos::Array<Teuchos::ScalarTraits<SC>::magnitudeType> normX(1), normResult1(1),normResult2(1);
X->norm2(normX);
out << "This test checks the correctness of the Galerkin triple "
<< "matrix product by comparing (RAP)*X to R(A(P*X))." << std::endl;
out << "||X||_2 = " << normX << std::endl;
result1->norm2(normResult1);
result2->norm2(normResult2);
TEST_FLOATING_EQUALITY(normResult1[0], normResult2[0], 1e-12);
} // Correctness test
TEUCHOS_UNIT_TEST(RAPFactory, ImplicitTranspose)
{
out << "version: " << MueLu::Version() << std::endl;
RCP<const Teuchos::Comm<int> > comm = Parameters::getDefaultComm();
if (comm->getSize() > 1 && TestHelpers::Parameters::getLib() == Xpetra::UseEpetra ) {
out << "Skipping ImplicitTranspose test for Epetra and #proc>1" << std::endl;
return;
}
// build test-specific default factory manager
RCP<FactoryManager> defManager = rcp(new FactoryManager());
defManager->SetFactory("A", rcp(MueLu::NoFactory::get(),false)); // dummy factory for A
defManager->SetFactory("Nullspace", rcp(new NullspaceFactory())); // real null space factory for Ptent
defManager->SetFactory("Graph", rcp(new CoalesceDropFactory())); // real graph factory for Ptent
defManager->SetFactory("Aggregates", rcp(new CoupledAggregationFactory())); // real aggregation factory for Ptent
Level fineLevel, coarseLevel;
TestHelpers::TestFactory<SC, LO, GO, NO, LMO>::createTwoLevelHierarchy(fineLevel, coarseLevel);
// overwrite default factory manager
fineLevel.SetFactoryManager(defManager);
coarseLevel.SetFactoryManager(defManager);
RCP<Matrix> Op = TestHelpers::TestFactory<SC, LO, GO, NO, LMO>::Build1DPoisson(19*comm->getSize());
fineLevel.Set("A",Op);
TentativePFactory tentpFactory;
SaPFactory sapFactory;
sapFactory.SetFactory("P",rcpFromRef(tentpFactory));
TransPFactory transPFactory;
transPFactory.SetFactory("P", rcpFromRef(sapFactory));
coarseLevel.Request("P", &sapFactory);
coarseLevel.Request("R", &transPFactory);
coarseLevel.Request(sapFactory);
coarseLevel.Request(transPFactory);
sapFactory.Build(fineLevel, coarseLevel);
transPFactory.Build(fineLevel,coarseLevel);
RAPFactory rap;
ParameterList rapList = *(rap.GetValidParameterList());
rapList.set("transpose: use implicit", true);
rap.SetParameterList(rapList);
rap.SetFactory("P", rcpFromRef(sapFactory));
rap.SetFactory("R", rcpFromRef(transPFactory));
coarseLevel.Request("A", &rap);
coarseLevel.Request(rap);
rap.Build(fineLevel,coarseLevel);
RCP<Matrix> A = fineLevel.Get< RCP<Matrix> >("A");
RCP<Matrix> P = coarseLevel.Get< RCP<Matrix> >("P", &sapFactory);
RCP<Matrix> R = coarseLevel.Get< RCP<Matrix> >("R", &transPFactory);
//std::string filename = "A.dat";
//Utils::Write(filename,Op);
//filename = "P.dat";
//Utils::Write(filename,P);
RCP<MultiVector> workVec1 = MultiVectorFactory::Build(P->getRangeMap(),1);
RCP<MultiVector> workVec2 = MultiVectorFactory::Build(Op->getRangeMap(),1);
RCP<MultiVector> result1 = MultiVectorFactory::Build(P->getDomainMap(),1);
RCP<MultiVector> X = MultiVectorFactory::Build(P->getDomainMap(),1);
X->randomize();
//out.precision(12);
//out.setOutputToRootOnly(-1);
//X->describe(out,Teuchos::VERB_EXTREME);
//Calculate result1 = P^T*(A*(P*X))
P->apply(*X,*workVec1,Teuchos::NO_TRANS,(SC)1.0,(SC)0.0);
Op->apply(*workVec1,*workVec2,Teuchos::NO_TRANS,(SC)1.0,(SC)0.0);
P->apply(*workVec2,*result1,Teuchos::TRANS,(SC)1.0,(SC)0.0);
RCP<Matrix> coarseOp = coarseLevel.Get< RCP<Matrix> >("A", &rap);
//Calculate result2 = (R*A*P)*X
RCP<MultiVector> result2 = MultiVectorFactory::Build(P->getDomainMap(),1);
coarseOp->apply(*X,*result2,Teuchos::NO_TRANS,(SC)1.0,(SC)0.0);
Teuchos::Array<Teuchos::ScalarTraits<SC>::magnitudeType> normX(1), normResult1(1),normResult2(1);
X->norm2(normX);
out << "This test checks the correctness of the Galerkin triple "
<< "matrix product by comparing (RAP)*X to R(A(P*X)), where R is the implicit transpose of P." << std::endl;
out << "||X||_2 = " << normX << std::endl;
result1->norm2(normResult1);
result2->norm2(normResult2);
TEST_FLOATING_EQUALITY(normResult1[0], normResult2[0], 1e-12);
} // Correctness test
