47 #ifndef __Teko_InterlacedTpetra_hpp__
48 #define __Teko_InterlacedTpetra_hpp__
51 #include "Teuchos_RCP.hpp"
54 #include "Tpetra_Map.hpp"
55 #include "Tpetra_CrsMatrix.hpp"
56 #include "Tpetra_MultiVector.hpp"
57 #include "Tpetra_Vector.hpp"
58 #include "Tpetra_Export.hpp"
59 #include "Tpetra_Import.hpp"
61 #include "Teko_ConfigDefs.hpp"
66 namespace TpetraHelpers {
88 void buildSubMaps(GO numGlobals,
int numVars,
const Teuchos::Comm<int>& comm,
89 std::vector<std::pair<
int, Teuchos::RCP<Tpetra::Map<LO, GO, NT> > > >& subMaps);
110 void buildSubMaps(GO numGlobals,
const std::vector<int>& vars,
const Teuchos::Comm<int>& comm,
111 std::vector<std::pair<
int, Teuchos::RCP<Tpetra::Map<LO, GO, NT> > > >& subMaps);
134 void buildSubMaps(GO numGlobals, LO numMyElements, GO minMyGID,
const std::vector<int>& vars,
135 const Teuchos::Comm<int>& comm,
136 std::vector<std::pair<
int, Teuchos::RCP<Tpetra::Map<LO, GO, NT> > > >& subMaps);
158 void buildSubMaps(
const Tpetra::Map<LO, GO, NT>& globalMap,
const std::vector<int>& vars,
159 const Teuchos::Comm<int>& comm,
160 std::vector<std::pair<
int, Teuchos::RCP<Tpetra::Map<LO, GO, NT> > > >& subMaps);
163 void buildExportImport(
164 const Tpetra::Map<LO, GO, NT>& baseMap,
165 const std::vector<std::pair<
int, Teuchos::RCP<Tpetra::Map<LO, GO, NT> > > >& subMaps,
166 std::vector<Teuchos::RCP<Tpetra::Export<LO, GO, NT> > >& subExport,
167 std::vector<Teuchos::RCP<Tpetra::Import<LO, GO, NT> > >& subImport);
170 void buildSubVectors(
171 const std::vector<std::pair<
int, Teuchos::RCP<Tpetra::Map<LO, GO, NT> > > >& subMaps,
172 std::vector<Teuchos::RCP<Tpetra::MultiVector<ST, LO, GO, NT> > >& subVectors, LO count);
177 void associateSubVectors(
178 const std::vector<std::pair<
int, Teuchos::RCP<Tpetra::Map<LO, GO, NT> > > >& subMaps,
179 std::vector<Teuchos::RCP<
const Tpetra::MultiVector<ST, LO, GO, NT> > >& subVectors);
182 Teuchos::RCP<Tpetra::CrsMatrix<ST, LO, GO, NT> > buildSubBlock(
183 int i,
int j,
const Teuchos::RCP<
const Tpetra::CrsMatrix<ST, LO, GO, NT> >& A,
184 const std::vector<std::pair<
int, Teuchos::RCP<Tpetra::Map<LO, GO, NT> > > >& subMaps);
187 void rebuildSubBlock(
188 int i,
int j,
const Teuchos::RCP<
const Tpetra::CrsMatrix<ST, LO, GO, NT> >& A,
189 const std::vector<std::pair<
int, Teuchos::RCP<Tpetra::Map<LO, GO, NT> > > >& subMaps,
190 Tpetra::CrsMatrix<ST, LO, GO, NT>& mat);
193 void many2one(Tpetra::MultiVector<ST, LO, GO, NT>& one,
194 const std::vector<Teuchos::RCP<
const Tpetra::MultiVector<ST, LO, GO, NT> > >& many,
195 const std::vector<Teuchos::RCP<Tpetra::Export<LO, GO, NT> > >& subExport);
198 void one2many(std::vector<Teuchos::RCP<Tpetra::MultiVector<ST, LO, GO, NT> > >& many,
199 const Tpetra::MultiVector<ST, LO, GO, NT>& single,
200 const std::vector<Teuchos::RCP<Tpetra::Import<LO, GO, NT> > >& subImport);