Teko_InterlacedEpetra.hpp

// @HEADER
// *****************************************************************************
//      Teko: A package for block and physics based preconditioning
//
// Copyright 2010 NTESS and the Teko contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER

#ifndef __Teko_InterlacedEpetra_hpp__
#define __Teko_InterlacedEpetra_hpp__

// Teuchos includes
#include "Teuchos_RCP.hpp"

// include basic Epetra information
#ifdef HAVE_MPI
#include "Epetra_MpiComm.h"
#else
#include "Epetra_SerialComm.h"
#endif
#include "Epetra_Map.h"
#include "Epetra_CrsMatrix.h"
#include "Epetra_MultiVector.h"
#include "Epetra_Vector.h"
#include "Epetra_Export.h"
#include "Epetra_Import.h"

#include <vector>

namespace Teko {
namespace Epetra {
namespace Strided {

void buildSubMaps(int numGlobals, int numVars, const Epetra_Comm& comm,
                  std::vector<std::pair<int, Teuchos::RCP<Epetra_Map> > >& subMaps);

void buildSubMaps(int numGlobals, const std::vector<int>& vars, const Epetra_Comm& comm,
                  std::vector<std::pair<int, Teuchos::RCP<Epetra_Map> > >& subMaps);

void buildSubMaps(int numGlobals, int numMyElements, int minMyGID, const std::vector<int>& vars,
                  const Epetra_Comm& comm,
                  std::vector<std::pair<int, Teuchos::RCP<Epetra_Map> > >& subMaps);

void buildSubMaps(const Epetra_Map& globalMap, const std::vector<int>& vars,
                  const Epetra_Comm& comm,
                  std::vector<std::pair<int, Teuchos::RCP<Epetra_Map> > >& subMaps);

// build conversion import and export operators
void buildExportImport(const Epetra_Map& baseMap,
                       const std::vector<std::pair<int, Teuchos::RCP<Epetra_Map> > >& subMaps,
                       std::vector<Teuchos::RCP<Epetra_Export> >& subExport,
                       std::vector<Teuchos::RCP<Epetra_Import> >& subImport);

// build a vector of subVectors
void buildSubVectors(const std::vector<std::pair<int, Teuchos::RCP<Epetra_Map> > >& subMaps,
                     std::vector<Teuchos::RCP<Epetra_MultiVector> >& subVectors, int count);

void associateSubVectors(const std::vector<std::pair<int, Teuchos::RCP<Epetra_Map> > >& subMaps,
                         std::vector<Teuchos::RCP<const Epetra_MultiVector> >& subVectors);

// build a single subblock Epetra_CrsMatrix
Teuchos::RCP<Epetra_CrsMatrix> buildSubBlock(
    int i, int j, const Epetra_CrsMatrix& A,
    const std::vector<std::pair<int, Teuchos::RCP<Epetra_Map> > >& subMaps);

// Rebuild a single subblock of a matrix
void rebuildSubBlock(int i, int j, const Epetra_CrsMatrix& A,
                     const std::vector<std::pair<int, Teuchos::RCP<Epetra_Map> > >& subMaps,
                     Epetra_CrsMatrix& mat);

// copy contents of many subvectors to a single vector
void many2one(Epetra_MultiVector& one,
              const std::vector<Teuchos::RCP<const Epetra_MultiVector> >& many,
              const std::vector<Teuchos::RCP<Epetra_Export> >& subExport);

// copy contents of a single vector to many subvectors
void one2many(std::vector<Teuchos::RCP<Epetra_MultiVector> >& many,
              const Epetra_MultiVector& single,
              const std::vector<Teuchos::RCP<Epetra_Import> >& subImport);

}  // namespace Strided
}  // end namespace Epetra
}  // end namespace Teko

#endif