43 #ifndef __Panzer_TpetraVector_ReadOnly_GlobalEvaluationData_impl_hpp__ 44 #define __Panzer_TpetraVector_ReadOnly_GlobalEvaluationData_impl_hpp__ 46 #include "Thyra_TpetraThyraWrappers.hpp" 52 template <
typename ScalarT,
typename LocalOrdinalT,
typename GlobalOrdinalT,
typename NodeT>
57 TEUCHOS_TEST_FOR_EXCEPTION(isInitialized_,std::logic_error,
58 "TpetraVector_ReadOnly_GED has been initialized, cannot call \"useConstantValues\"!");
64 globalFilteredPairs_.push_back(pair);
67 template <
typename ScalarT,
typename LocalOrdinalT,
typename GlobalOrdinalT,
typename NodeT>
71 const RCP<const MapType>& ghostedMap,
72 const RCP<const MapType>& ownedMap)
75 ghostedMap_ = ghostedMap;
79 ghostedVector_ = Teuchos::rcp(
new VectorType(ghostedMap_));
82 ghostedSpace_ = Thyra::tpetraVectorSpace<ScalarT, LocalOrdinalT,
83 GlobalOrdinalT, NodeT>(ghostedMap_);
84 ownedSpace_ = Thyra::tpetraVectorSpace<ScalarT, LocalOrdinalT,
85 GlobalOrdinalT, NodeT>(ownedMap_);
90 filteredPairs_.resize(globalFilteredPairs_.size());
91 for(std::size_t i=0;i<globalFilteredPairs_.size();i++) {
92 std::vector<LocalOrdinalT>
lids;
93 const std::vector<GlobalOrdinalT> & gids = globalFilteredPairs_[i].first;
94 for(std::size_t j=0;j<gids.size();j++) {
95 LocalOrdinalT lid = ghostedMap->getLocalElement(gids[j]);
103 filteredPairs_[i].first =
lids;
104 filteredPairs_[i].second = globalFilteredPairs_[i].second;
107 isInitialized_ =
true;
110 template <
typename ScalarT,
typename LocalOrdinalT,
typename GlobalOrdinalT,
typename NodeT>
115 TEUCHOS_TEST_FOR_EXCEPTION(ownedVector_ == Teuchos::null, std::logic_error,
116 "Owned vector has not been set, can't perform the halo exchange!");
123 ghostedVector_->doImport(*ownedVector_, *importer_, Tpetra::INSERT);
124 PHX::ExecSpace().fence();
127 template <
typename ScalarT,
typename LocalOrdinalT,
typename GlobalOrdinalT,
typename NodeT>
132 TEUCHOS_TEST_FOR_EXCEPTION(!isInitialized_,std::logic_error,
133 "TpetraVector_ReadOnly_GED has not been initialized, cannot call \"initializeData\"!");
135 ghostedVector_->putScalar(0.0);
136 PHX::ExecSpace().fence();
138 typedef typename VectorType::dual_view_type::t_dev::memory_space DMS;
139 auto values = ghostedVector_->getLocalViewHost(Tpetra::Access::OverwriteAll);
142 for(std::size_t i=0;i<filteredPairs_.size();i++) {
143 const std::vector<int> &
lids = filteredPairs_[i].first;
144 double value = filteredPairs_[i].second;
145 for(std::size_t j=0;j<
lids.size();j++)
146 values(
lids[j], 0) = value;
150 template <
typename ScalarT,
typename LocalOrdinalT,
typename GlobalOrdinalT,
typename NodeT>
155 TEUCHOS_ASSERT(isInitialized_);
156 ownedVector_ = ownedVector;
159 template <
typename ScalarT,
typename LocalOrdinalT,
typename GlobalOrdinalT,
typename NodeT>
160 Teuchos::RCP<const typename TpetraVector_ReadOnly_GlobalEvaluationData<ScalarT,LocalOrdinalT,GlobalOrdinalT,NodeT>::VectorType>
164 TEUCHOS_ASSERT(isInitialized_);
168 template <
typename ScalarT,
typename LocalOrdinalT,
typename GlobalOrdinalT,
typename NodeT>
169 Teuchos::RCP<typename TpetraVector_ReadOnly_GlobalEvaluationData<ScalarT,LocalOrdinalT,GlobalOrdinalT,NodeT>::VectorType>
173 TEUCHOS_ASSERT(isInitialized_);
174 TEUCHOS_ASSERT(ghostedVector_!=Teuchos::null);
176 return ghostedVector_;
179 template <
typename ScalarT,
typename LocalOrdinalT,
typename GlobalOrdinalT,
typename NodeT>
185 typedef Thyra::TpetraOperatorVectorExtraction<ScalarT,LocalOrdinalT,GlobalOrdinalT,NodeT> TOE;
186 TEUCHOS_ASSERT(isInitialized_);
187 ownedVector_ = TOE::getConstTpetraVector(ownedVector);
190 template <
typename ScalarT,
typename LocalOrdinalT,
typename GlobalOrdinalT,
typename NodeT>
191 Teuchos::RCP<const Thyra::VectorBase<double> >
195 TEUCHOS_ASSERT(isInitialized_);
196 return (ownedVector_ == Teuchos::null) ? Teuchos::null :
197 Thyra::createConstVector(ownedVector_, ownedSpace_);
200 template <
typename ScalarT,
typename LocalOrdinalT,
typename GlobalOrdinalT,
typename NodeT>
201 Teuchos::RCP<Thyra::VectorBase<double> >
205 TEUCHOS_ASSERT(isInitialized_);
206 TEUCHOS_ASSERT(ghostedVector_!=Teuchos::null);
208 return Thyra::createVector(ghostedVector_,ghostedSpace_);
211 template <
typename ScalarT,
typename LocalOrdinalT,
typename GlobalOrdinalT,
typename NodeT>
216 const std::string tab =
" ";
218 os << tab <<
"TpetraVector_ReadOnly_GlobalEvaluationData\n" 219 << tab <<
" init = " << isInitialized_ <<
"\n" 220 << tab <<
" owned = " << ownedVector_ <<
"\n" 221 << tab <<
" ghosted = " << ghostedVector_ <<
"\n";
Tpetra::Vector< ScalarT, LocalOrdinalT, GlobalOrdinalT, NodeT > VectorType
virtual void initializeData()
Clear out the ghosted vector.
void setOwnedVector(const Teuchos::RCP< const Thyra::VectorBase< double > > &ownedVector)
Set the owned vector (Thyra version)
void useConstantValues(const std::vector< GlobalOrdinalT > &indices, double value)
Teuchos::RCP< const VectorType > getOwnedVector_Tpetra() const
Get the owned vector (Tpetra version)
void print(std::ostream &os) const
Diagnostic function.
Teuchos::RCP< Thyra::VectorBase< double > > getGhostedVector() const
Get the ghosted vector (Thyra version)
Teuchos::RCP< VectorType > getGhostedVector_Tpetra() const
Get the ghosted vector (Tpetra version)
virtual void globalToGhost(int mem)
void setOwnedVector_Tpetra(const Teuchos::RCP< const VectorType > &ownedVector)
Set the owned vector (Tpetra version)
Teuchos::RCP< const Thyra::VectorBase< double > > getOwnedVector() const
Get the owned vector (Thyra version)
std::pair< std::vector< GlobalOrdinalT >, double > FilteredGlobalPair
PHX::View< const LO ** > lids
void initialize(const Teuchos::RCP< const ImportType > &importer, const Teuchos::RCP< const MapType > &ghostedMap, const Teuchos::RCP< const MapType > &ownedMap)