47 #include "Epetra/Teko_StridedMappingStrategy.hpp"
48 #include "Epetra/Teko_InterlacedEpetra.hpp"
49 #include "Epetra/Teko_EpetraHelpers.hpp"
51 #include "Thyra_EpetraThyraWrappers.hpp"
52 #include "Thyra_EpetraLinearOp.hpp"
53 #include "Thyra_DefaultProductMultiVector.hpp"
54 #include "Thyra_DefaultProductVectorSpace.hpp"
55 #include "Thyra_DefaultSpmdMultiVector.hpp"
56 #include "Thyra_DefaultBlockedLinearOp.hpp"
57 #include "Thyra_get_Epetra_Operator.hpp"
61 using Teuchos::rcp_dynamic_cast;
76 StridedMappingStrategy::StridedMappingStrategy(
const std::vector<int>& vars,
77 const RCP<const Epetra_Map>& map,
78 const Epetra_Comm& comm) {
81 buildBlockTransferData(vars, rangeMap_, comm);
92 void StridedMappingStrategy::copyEpetraIntoThyra(
93 const Epetra_MultiVector& X,
94 const Teuchos::Ptr<Thyra::MultiVectorBase<double> >& thyra_X)
const {
95 int count = X.NumVectors();
97 std::vector<RCP<Epetra_MultiVector> > subX;
100 Strided::buildSubVectors(blockMaps_, subX, count);
103 Strided::one2many(subX, X, blockImport_);
106 Teuchos::Array<RCP<Thyra::MultiVectorBase<double> > > thyra_subX;
107 Teuchos::Ptr<Thyra::ProductMultiVectorBase<double> > prod_X =
108 Teuchos::ptr_dynamic_cast<Thyra::ProductMultiVectorBase<double> >(thyra_X);
109 for (
unsigned int i = 0; i < blockMaps_.size(); i++) {
110 RCP<Thyra::DefaultSpmdMultiVector<double> > vec =
111 rcp_dynamic_cast<Thyra::DefaultSpmdMultiVector<double> >(
112 prod_X->getNonconstMultiVectorBlock(i));
113 fillDefaultSpmdMultiVector(vec, subX[i]);
125 void StridedMappingStrategy::copyThyraIntoEpetra(
126 const RCP<
const Thyra::MultiVectorBase<double> >& thyra_Y, Epetra_MultiVector& Y)
const {
127 std::vector<RCP<const Epetra_MultiVector> > subY;
128 RCP<const Thyra::DefaultProductMultiVector<double> > prod_Y =
129 rcp_dynamic_cast<
const Thyra::DefaultProductMultiVector<double> >(thyra_Y);
132 for (
unsigned int i = 0; i < blockMaps_.size(); i++)
134 Thyra::get_Epetra_MultiVector(*blockMaps_[i].second, prod_Y->getMultiVectorBlock(i)));
137 Strided::associateSubVectors(blockMaps_, subY);
140 Strided::many2one(Y, subY, blockExport_);
155 void StridedMappingStrategy::buildBlockTransferData(
const std::vector<int>& vars,
156 const Teuchos::RCP<const Epetra_Map>& baseMap,
157 const Epetra_Comm& comm) {
159 Strided::buildSubMaps(*baseMap, vars, comm, blockMaps_);
160 Strided::buildExportImport(*baseMap, blockMaps_, blockExport_, blockImport_);
173 const Teuchos::RCP<Thyra::BlockedLinearOpBase<double> > StridedMappingStrategy::buildBlockedThyraOp(
174 const RCP<const Epetra_CrsMatrix>& crsContent,
const std::string& label)
const {
175 int dim = blockMaps_.size();
177 RCP<Thyra::DefaultBlockedLinearOp<double> > A = Thyra::defaultBlockedLinearOp<double>();
179 A->beginBlockFill(dim, dim);
180 for (
int i = 0; i < dim; i++) {
181 for (
int j = 0; j < dim; j++) {
183 std::stringstream ss;
184 ss << label <<
"_" << i <<
"," << j;
187 A->setNonconstBlock(i, j,
188 Thyra::nonconstEpetraLinearOp(
189 Strided::buildSubBlock(i, j, *crsContent, blockMaps_), ss.str()));
207 void StridedMappingStrategy::rebuildBlockedThyraOp(
208 const RCP<const Epetra_CrsMatrix>& crsContent,
209 const RCP<Thyra::BlockedLinearOpBase<double> >& A)
const {
210 int dim = blockMaps_.size();
212 for (
int i = 0; i < dim; i++) {
213 for (
int j = 0; j < dim; j++) {
215 RCP<Thyra::LinearOpBase<double> > Aij = A->getNonconstBlock(i, j);
216 RCP<Epetra_CrsMatrix> eAij =
217 rcp_dynamic_cast<Epetra_CrsMatrix>(Thyra::get_Epetra_Operator(*Aij),
true);
220 Strided::rebuildSubBlock(i, j, *crsContent, blockMaps_, *eAij);