Panzer_TpetraVector_ReadOnly_GlobalEvaluationData.hpp

Go to the documentation of this file.

// @HEADER
// *****************************************************************************
//           Panzer: A partial differential equation assembly
//       engine for strongly coupled complex multiphysics systems
//
// Copyright 2011 NTESS and the Panzer contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER

#ifndef __Panzer_TpetraVector_ReadOnly_GlobalEvaluationData_hpp__
#define __Panzer_TpetraVector_ReadOnly_GlobalEvaluationData_hpp__

#include "Tpetra_Import.hpp"
#include "Tpetra_Vector.hpp"
#include "Tpetra_Map.hpp"

#include "Teuchos_RCP.hpp"

#include "Thyra_VectorSpaceBase.hpp"
#include "Thyra_VectorBase.hpp"

#include "Panzer_NodeType.hpp"
#include "Panzer_ReadOnlyVector_GlobalEvaluationData.hpp"

namespace panzer {

template <typename ScalarT,typename LocalOrdinalT,typename GlobalOrdinalT,
          typename NodeT=panzer::TpetraNodeType>
class TpetraVector_ReadOnly_GlobalEvaluationData : public ReadOnlyVector_GlobalEvaluationData {
public:
   typedef Tpetra::Vector<ScalarT,LocalOrdinalT,GlobalOrdinalT,NodeT> VectorType;
   typedef Tpetra::Map<LocalOrdinalT,GlobalOrdinalT,NodeT> MapType;
   typedef Tpetra::Import<LocalOrdinalT,GlobalOrdinalT,NodeT> ImportType;

   TpetraVector_ReadOnly_GlobalEvaluationData()
      : isInitialized_(false) { }

   TpetraVector_ReadOnly_GlobalEvaluationData(const TpetraVector_ReadOnly_GlobalEvaluationData & src)
      : isInitialized_(false) { initialize(src.importer_, src.ghostedMap_, src.ownedMap_); }

   TpetraVector_ReadOnly_GlobalEvaluationData(const Teuchos::RCP<const ImportType>& importer,
                                              const Teuchos::RCP<const MapType>&    ghostedMap,
                                              const Teuchos::RCP<const MapType>&    ownedMap)
      : isInitialized_(false) { initialize(importer, ghostedMap, ownedMap); }

   void useConstantValues(const std::vector<GlobalOrdinalT> & indices,double value);

   void initialize(const Teuchos::RCP<const ImportType>& importer,
                   const Teuchos::RCP<const MapType>&    ghostedMap,
                   const Teuchos::RCP<const MapType>&    ownedMap);

   virtual void globalToGhost(int mem);

   virtual void initializeData(); 
  
   virtual void ghostToGlobal(int /* mem */) {} 

   virtual bool requiresDirichletAdjustment() const { return false; }

   void setOwnedVector_Tpetra(const Teuchos::RCP<const VectorType>& ownedVector);

   Teuchos::RCP<const VectorType> getOwnedVector_Tpetra() const;

   Teuchos::RCP<VectorType> getGhostedVector_Tpetra() const;

   void setOwnedVector(const Teuchos::RCP<const Thyra::VectorBase<double> >& ownedVector);

   Teuchos::RCP<const Thyra::VectorBase<double> > getOwnedVector() const;

   Teuchos::RCP<Thyra::VectorBase<double> > getGhostedVector() const;

   bool isInitialized() const { return isInitialized_; }

   void print(std::ostream & os) const;

private:
   bool isInitialized_;

   Teuchos::RCP<const MapType> ghostedMap_;
   Teuchos::RCP<const MapType> ownedMap_;

   Teuchos::RCP<const Thyra::VectorSpaceBase<double> > ghostedSpace_;
   Teuchos::RCP<const Thyra::VectorSpaceBase<double> > ownedSpace_;

   Teuchos::RCP<const ImportType> importer_;
   Teuchos::RCP<VectorType>       ghostedVector_;
   Teuchos::RCP<const VectorType> ownedVector_;

   typedef std::pair<std::vector<GlobalOrdinalT>,double> FilteredGlobalPair;
   typedef std::pair<std::vector<LocalOrdinalT>,double> FilteredLocalPair;
   std::vector<FilteredGlobalPair> globalFilteredPairs_;
   std::vector<FilteredLocalPair> filteredPairs_;
};

}

#endif