47 #ifndef __Teko_TpetraStridedMappingStrategy_hpp__
48 #define __Teko_TpetraStridedMappingStrategy_hpp__
54 #include "Teuchos_RCP.hpp"
57 #include "Thyra_MultiVectorBase.hpp"
58 #include "Thyra_LinearOpBase.hpp"
59 #include "Thyra_BlockedLinearOpBase.hpp"
62 #include "Tpetra_Map.hpp"
63 #include "Tpetra_MultiVector.hpp"
64 #include "Tpetra_Import.hpp"
65 #include "Tpetra_Export.hpp"
66 #include "Tpetra_CrsMatrix.hpp"
69 #include "Teko_TpetraOperatorWrapper.hpp"
70 #include "Teko_ConfigDefs.hpp"
73 namespace TpetraHelpers {
75 class TpetraStridedMappingStrategy :
public MappingStrategy {
89 TpetraStridedMappingStrategy(
const std::vector<int>& vars,
90 const Teuchos::RCP<
const Tpetra::Map<LO, GO, NT> >& map,
91 const Teuchos::Comm<int>& comm);
104 virtual void copyTpetraIntoThyra(
const Tpetra::MultiVector<ST, LO, GO, NT>& tpetra_X,
105 const Teuchos::Ptr<Thyra::MultiVectorBase<ST> >& thyra_X)
const;
114 virtual void copyThyraIntoTpetra(
const Teuchos::RCP<
const Thyra::MultiVectorBase<ST> >& thyra_Y,
115 Tpetra::MultiVector<ST, LO, GO, NT>& tpetra_Y)
const;
124 virtual const Teuchos::RCP<const Tpetra::Map<LO, GO, NT> > domainMap()
const {
135 virtual const Teuchos::RCP<const Tpetra::Map<LO, GO, NT> > rangeMap()
const {
return rangeMap_; }
141 virtual std::string toString()
const {
return std::string(
"TpetraStridedMappingStrategy"); }
162 void buildBlockTransferData(
const std::vector<int>& vars,
163 const Teuchos::RCP<
const Tpetra::Map<LO, GO, NT> >& baseMap,
164 const Teuchos::Comm<int>& comm);
177 const std::vector<std::pair<int, Teuchos::RCP<Tpetra::Map<LO, GO, NT> > > >& getMaps()
const {
192 const Teuchos::RCP<Thyra::BlockedLinearOpBase<ST> > buildBlockedThyraOp(
193 const Teuchos::RCP<
const Tpetra::CrsMatrix<ST, LO, GO, NT> >& mat,
194 const std::string& label =
"<ANYM>")
const;
205 void rebuildBlockedThyraOp(
const RCP<
const Tpetra::CrsMatrix<ST, LO, GO, NT> >& mat,
206 const RCP<Thyra::BlockedLinearOpBase<ST> >& A)
const;
215 Teuchos::RCP<const Tpetra::Map<LO, GO, NT> > domainMap_;
216 Teuchos::RCP<const Tpetra::Map<LO, GO, NT> > rangeMap_;
221 std::vector<std::pair<int, Teuchos::RCP<Tpetra::Map<LO, GO, NT> > > > blockMaps_;
222 std::vector<Teuchos::RCP<Tpetra::Export<LO, GO, NT> > > blockExport_;
223 std::vector<Teuchos::RCP<Tpetra::Import<LO, GO, NT> > > blockImport_;
virtual void copyThyraIntoTpetra(const RCP< const Thyra::MultiVectorBase< ST > > &thyraX, Tpetra::MultiVector< ST, LO, GO, NT > &tpetraX) const =0
Copy an Thyra::MultiVectorBase into a Epetra_MultiVector.