Xpetra_BlockedMultiVector_decl.hpp

Go to the documentation of this file.

// @HEADER
// *****************************************************************************
//             Xpetra: A linear algebra interface package
//
// Copyright 2012 NTESS and the Xpetra contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER

#ifndef XPETRA_BLOCKEDMULTIVECTOR_DECL_HPP
#define XPETRA_BLOCKEDMULTIVECTOR_DECL_HPP

/* this file is automatically generated - do not edit (see script/interfaces.py) */

#include "Xpetra_ConfigDefs.hpp"
#include "Xpetra_Map_decl.hpp"
#include "Xpetra_MultiVector_decl.hpp"

#include "Xpetra_BlockedMap_decl.hpp"

namespace Xpetra {

#ifndef DOXYGEN_SHOULD_SKIP_THIS
// forward declaration of Vector, needed to prevent circular inclusions
template <class S, class LO, class GO, class N>
class Vector;

// forward declaration of MapExtractor, we just need the class sig here.
template <class S, class LO, class GO, class N>
class MapExtractor;
#endif

template <class Scalar,
          class LocalOrdinal,
          class GlobalOrdinal,
          class Node = Tpetra::KokkosClassic::DefaultNode::DefaultNodeType>
class BlockedMultiVector
  : public MultiVector<Scalar, LocalOrdinal, GlobalOrdinal, Node> {
 public:
  typedef Scalar scalar_type;
  typedef LocalOrdinal local_ordinal_type;
  typedef GlobalOrdinal global_ordinal_type;
  typedef Node node_type;

 private:
#undef XPETRA_BLOCKEDMULTIVECTOR_SHORT
#include "Xpetra_UseShortNames.hpp"

 public:



  BlockedMultiVector(const Teuchos::RCP<const BlockedMap>& map, size_t NumVectors, bool zeroOut = true);

  BlockedMultiVector(Teuchos::RCP<const Xpetra::BlockedMap<LocalOrdinal, GlobalOrdinal, Node>> bmap,
                     Teuchos::RCP<const MultiVector> v);

  BlockedMultiVector(Teuchos::RCP<const Xpetra::MapExtractor<Scalar, LocalOrdinal, GlobalOrdinal, Node>> mapExtractor,
                     Teuchos::RCP<const MultiVector> v);

  BlockedMultiVector(const Teuchos::RCP<const BlockedMap>& map, std::vector<Teuchos::RCP<MultiVector>>& vin);

  virtual ~BlockedMultiVector();

  BlockedMultiVector<Scalar, LocalOrdinal, GlobalOrdinal, Node>& operator=(const MultiVector& rhs);



  virtual void replaceGlobalValue(GlobalOrdinal /* globalRow */, size_t /* vectorIndex */, const Scalar& /* value */);

  virtual void sumIntoGlobalValue(GlobalOrdinal /* globalRow */, size_t /* vectorIndex */, const Scalar& /* value */);

  virtual void replaceLocalValue(LocalOrdinal /* myRow */, size_t /* vectorIndex */, const Scalar& /* value */);

  virtual void sumIntoLocalValue(LocalOrdinal /* myRow */, size_t /* vectorIndex */, const Scalar& /* value */);

  virtual void putScalar(const Scalar& value);



  virtual Teuchos::RCP<const Xpetra::Vector<Scalar, LocalOrdinal, GlobalOrdinal, Node>> getVector(size_t j) const;

  virtual Teuchos::RCP<Xpetra::Vector<Scalar, LocalOrdinal, GlobalOrdinal, Node>> getVectorNonConst(size_t j);

  virtual Teuchos::ArrayRCP<const Scalar> getData(size_t j) const;

  virtual Teuchos::ArrayRCP<Scalar> getDataNonConst(size_t j);



  virtual void dot(const MultiVector& /* A */, const Teuchos::ArrayView<Scalar>& /* dots */) const;

  virtual void abs(const MultiVector& /* A */);

  virtual void reciprocal(const MultiVector& /* A */);

  virtual void scale(const Scalar& alpha);

  virtual void scale(Teuchos::ArrayView<const Scalar> alpha);

  virtual void update(const Scalar& alpha, const MultiVector& A, const Scalar& beta);

  virtual void update(const Scalar& alpha, const MultiVector& A, const Scalar& beta, const MultiVector& B, const Scalar& gamma);

  virtual void norm1(const Teuchos::ArrayView<typename Teuchos::ScalarTraits<Scalar>::magnitudeType>& norms) const;

  virtual void norm2(const Teuchos::ArrayView<typename Teuchos::ScalarTraits<Scalar>::magnitudeType>& norms) const;

  virtual void normInf(const Teuchos::ArrayView<typename Teuchos::ScalarTraits<Scalar>::magnitudeType>& norms) const;

  virtual void meanValue(const Teuchos::ArrayView<Scalar>& /* means */) const;

  virtual void multiply(Teuchos::ETransp /* transA */, Teuchos::ETransp /* transB */, const Scalar& /* alpha */, const MultiVector& /* A */, const MultiVector& /* B */, const Scalar& /* beta */);

  virtual void elementWiseMultiply(Scalar scalarAB, const Xpetra::Vector<Scalar, LocalOrdinal, GlobalOrdinal, Node>& A, const MultiVector& B, Scalar scalarThis);



  virtual size_t getNumVectors() const;

  virtual size_t getLocalLength() const;

  virtual global_size_t getGlobalLength() const;

  virtual bool isSameSize(const Xpetra::MultiVector<Scalar, LocalOrdinal, GlobalOrdinal, Node>& vec) const;



  virtual std::string description() const;

  virtual void describe(Teuchos::FancyOStream& out, const Teuchos::EVerbosityLevel verbLevel = Teuchos::Describable::verbLevel_default) const;

  virtual void replaceMap(const RCP<const Map>& map);

  virtual void doImport(const DistObject<Scalar, LocalOrdinal, GlobalOrdinal, Node>& /* source */, const Import& /* importer */, CombineMode /* CM */);

  virtual void doExport(const DistObject<Scalar, LocalOrdinal, GlobalOrdinal, Node>& /* dest */, const Import& /* importer */, CombineMode /* CM */);

  virtual void doImport(const DistObject<Scalar, LocalOrdinal, GlobalOrdinal, Node>& /* source */, const Export& /* exporter */, CombineMode /* CM */);

  virtual void doExport(const DistObject<Scalar, LocalOrdinal, GlobalOrdinal, Node>& /* dest */, const Export& /* exporter */, CombineMode /* CM */);



  virtual void setSeed(unsigned int seed);

  virtual void randomize(bool bUseXpetraImplementation = false);

  virtual void randomize(const Scalar& minVal, const Scalar& maxVal, bool bUseXpetraImplementation = false);

  virtual void Xpetra_randomize();

  virtual void Xpetra_randomize(const Scalar& minVal, const Scalar& maxVal);


  Teuchos::RCP<const Map> getMap() const;

  Teuchos::RCP<const Xpetra::BlockedMap<LocalOrdinal, GlobalOrdinal, Node>> getBlockedMap() const;

  Teuchos::RCP<MultiVector> getMultiVector(size_t r) const;

  Teuchos::RCP<MultiVector> getMultiVector(size_t r, bool bThyraMode) const;

  void setMultiVector(size_t r, Teuchos::RCP<const MultiVector> v, bool bThyraMode);

  Teuchos::RCP<MultiVector> Merge() const;

 protected:
  virtual void assign(const MultiVector& rhs);

 private:
  // helper routines for interaction of MultiVector and BlockedMultiVectors

  void ExtractVector(RCP<const MultiVector>& full, size_t block, RCP<MultiVector>& partial) const;
  void ExtractVector(RCP<MultiVector>& full, size_t block, RCP<MultiVector>& partial) const;

  RCP<MultiVector> ExtractVector(RCP<const MultiVector>& full, size_t block, bool bThyraMode = false) const;
  RCP<MultiVector> ExtractVector(RCP<MultiVector>& full, size_t block, bool bThyraMode = false) const;

  void ExtractVector(const MultiVector& full, size_t block, MultiVector& partial) const;

  void InsertVector(const MultiVector& partial, size_t block, MultiVector& full, bool bThyraMode = false) const;          // 1019
  void InsertVector(RCP<const MultiVector> partial, size_t block, RCP<MultiVector> full, bool bThyraMode = false) const;  // 1059
  void InsertVector(RCP<MultiVector> partial, size_t block, RCP<MultiVector> full, bool bThyraMode = false) const;        // 1069

 private:
  Teuchos::RCP<const BlockedMap> map_;         
  std::vector<Teuchos::RCP<MultiVector>> vv_;  
  size_t numVectors_;                          

};  // BlockedMultiVector class

}  // namespace Xpetra

#define XPETRA_BLOCKEDMULTIVECTOR_SHORT
#endif  // XPETRA_BLOCKEDMULTIVECTOR_DECL_HPP