Panzer  Version of the Day
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
Panzer_BlockedTpetraLinearObjFactory.hpp
Go to the documentation of this file.
1 // @HEADER
2 // ***********************************************************************
3 //
4 // Panzer: A partial differential equation assembly
5 // engine for strongly coupled complex multiphysics systems
6 // Copyright (2011) Sandia Corporation
7 //
8 // Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
9 // the U.S. Government retains certain rights in this software.
10 //
11 // Redistribution and use in source and binary forms, with or without
12 // modification, are permitted provided that the following conditions are
13 // met:
14 //
15 // 1. Redistributions of source code must retain the above copyright
16 // notice, this list of conditions and the following disclaimer.
17 //
18 // 2. Redistributions in binary form must reproduce the above copyright
19 // notice, this list of conditions and the following disclaimer in the
20 // documentation and/or other materials provided with the distribution.
21 //
22 // 3. Neither the name of the Corporation nor the names of the
23 // contributors may be used to endorse or promote products derived from
24 // this software without specific prior written permission.
25 //
26 // THIS SOFTWARE IS PROVIDED BY SANDIA CORPORATION "AS IS" AND ANY
27 // EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
28 // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
29 // PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL SANDIA CORPORATION OR THE
30 // CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
31 // EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
32 // PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
33 // PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
34 // LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
35 // NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
36 // SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
37 //
38 // Questions? Contact Roger P. Pawlowski (rppawlo@sandia.gov) and
39 // Eric C. Cyr (eccyr@sandia.gov)
40 // ***********************************************************************
41 // @HEADER
42 
43 #ifndef __Panzer_BlockedTpetraLinearObjFactory_hpp__
44 #define __Panzer_BlockedTpetraLinearObjFactory_hpp__
45 
46 #include <map>
47 
48 // Tpetra includes
49 #include "Tpetra_Map.hpp"
50 #include "Tpetra_CrsGraph.hpp"
51 #include "Tpetra_Import.hpp"
52 #include "Tpetra_Export.hpp"
53 
54 #include "PanzerDiscFE_config.hpp"
62 #include "Panzer_HashUtils.hpp" // for pair_hash
63 
64 #include "Panzer_GatherOrientation.hpp"
70 
71 #include "Thyra_BlockedLinearOpBase.hpp"
72 #include "Thyra_ProductVectorBase.hpp"
73 
74 #include "Teuchos_RCP.hpp"
76 #include "Teuchos_OpaqueWrapper.hpp"
77 
78 namespace panzer {
79 
80 template <typename Traits,typename ScalarT,typename LocalOrdinalT,typename GlobalOrdinalT,typename NodeT=panzer::TpetraNodeType>
82  , public ThyraObjFactory<double> {
83 public:
85  typedef Tpetra::Vector<ScalarT,LocalOrdinalT,GlobalOrdinalT,NodeT> VectorType;
86  typedef Tpetra::CrsMatrix<ScalarT,LocalOrdinalT,GlobalOrdinalT,NodeT> CrsMatrixType;
87  typedef Tpetra::Operator<ScalarT,LocalOrdinalT,GlobalOrdinalT,NodeT> OperatorType;
88  typedef Tpetra::CrsGraph<LocalOrdinalT,GlobalOrdinalT,NodeT> CrsGraphType;
89  typedef Tpetra::Map<LocalOrdinalT,GlobalOrdinalT,NodeT> MapType;
90  typedef Tpetra::Import<LocalOrdinalT,GlobalOrdinalT,NodeT> ImportType;
91  typedef Tpetra::Export<LocalOrdinalT,GlobalOrdinalT,NodeT> ExportType;
92 
93  typedef Thyra::TpetraVector<ScalarT,LocalOrdinalT,GlobalOrdinalT,NodeT> ThyraVector;
94  typedef Thyra::TpetraLinearOp<ScalarT,LocalOrdinalT,GlobalOrdinalT,NodeT> ThyraLinearOp;
95 
96 
99 
105  const std::vector<Teuchos::RCP<const panzer::UniqueGlobalIndexer<LocalOrdinalT,GlobalOrdinalT>>> & gidProviders);
106 
108 
109 /*************** Linear object factory methods *******************/
110 
111  virtual void readVector(const std::string & /* identifier */, LinearObjContainer & /* loc */, int /* id */) const
112  { TEUCHOS_ASSERT(false); }
113 
114  virtual void writeVector(const std::string & /* identifier */, const LinearObjContainer & /* loc */, int /* id */) const
115  { TEUCHOS_ASSERT(false); }
116 
118 
120  { return buildLinearObjContainer(); }
121 
123 
125  { return buildGhostedLinearObjContainer(); }
126 
127  virtual void globalToGhostContainer(const LinearObjContainer & container,
128  LinearObjContainer & ghostContainer,int) const;
129  virtual void ghostToGlobalContainer(const LinearObjContainer & ghostContainer,
130  LinearObjContainer & container,int) const;
131 
138  virtual void adjustForDirichletConditions(const LinearObjContainer & localBCRows,
139  const LinearObjContainer & globalBCRows,
140  LinearObjContainer & ghostedObjs,
141  bool zeroVectorRows=false, bool adjustX = false) const;
142 
146  virtual void applyDirichletBCs(const LinearObjContainer & counter,
147  LinearObjContainer & result) const;
148 
154 
160 
162 
164  template <typename EvalT>
167 
169  template <typename EvalT>
172 
174  template <typename EvalT>
177 
179  template <typename EvalT>
182 
184  template <typename EvalT>
187 
189  template <typename EvalT>
192 
193 /*************** Generic helper functions for container setup *******************/
194 
200  void initializeContainer(int,LinearObjContainer & loc) const;
201 
207  void initializeContainer(int mem,BTLOC & loc) const;
208 
214  void initializeGhostedContainer(int,LinearObjContainer & loc) const;
215 
221  void initializeGhostedContainer(int mem,BTLOC & loc) const;
222 
223 /*************** Thyra based methods *******************/
224 
227 
230 
233 
236 
239 
242 
245 
246  // and now the ghosted versions
247 
250 
253 
256 
259 
262 
263 /*************** Tpetra based methods *******************/
264 
266  virtual Teuchos::RCP<const MapType> getMap(int i) const;
267 
269  virtual Teuchos::RCP<const MapType> getGhostedMap(int i) const;
270 
272  virtual Teuchos::RCP<const CrsGraphType> getGraph(int i,int j) const;
273 
275  virtual Teuchos::RCP<const CrsGraphType> getGhostedGraph(int i,int j) const;
276 
279 
282 
283  Teuchos::RCP<CrsMatrixType> getTpetraMatrix(int i,int j) const;
285 
288 
291 
293  int getBlockRowCount() const;
294 
296  int getBlockColCount() const;
297 
299  void addExcludedPair(int rowBlock,int colBlock);
300 
302  void addExcludedPairs(const std::vector<std::pair<int,int> > & exPairs);
303 
304  virtual void beginFill(LinearObjContainer & loc) const;
305  virtual void endFill(LinearObjContainer & loc) const;
306 
308  { return blockedDOFManager_; }
309 
312  { return blockProvider_; }
313 
316  { return blockProvider_; }
317 
318 protected:
319 /*************** Generic methods/members *******************/
320 
321  // Get the global indexer associated with a particular block
323 
325  void makeRoomForBlocks(std::size_t blockCnt);
326 
329  std::vector<Teuchos::RCP<const UniqueGlobalIndexer<LocalOrdinalT,GlobalOrdinalT> > > gidProviders_;
330 
331  // which block entries are ignored
332  std::unordered_set<std::pair<int,int>,panzer::pair_hash> excludedPairs_;
333 
334 /*************** Thyra based methods/members *******************/
335 
337  const Teuchos::RCP<Thyra::VectorBase<ScalarT> > & out) const;
340  const Teuchos::RCP<Thyra::VectorBase<ScalarT> > & out) const;
341 
344 
347 
348 /*************** Tpetra based methods/members *******************/
349 
350  void adjustForDirichletConditions(const VectorType & local_bcs,
351  const VectorType & global_bcs,
352  const Teuchos::Ptr<VectorType> & f,
353  const Teuchos::Ptr<CrsMatrixType> & A,
354  bool zeroVectorRows) const;
355 
356  void ghostToGlobalTpetraVector(int i,const VectorType & in,VectorType & out) const;
357  void ghostToGlobalTpetraMatrix(int blockRow,const CrsMatrixType & in,CrsMatrixType & out) const;
358  void globalToGhostTpetraVector(int i,const VectorType & in,VectorType & out) const;
359 
360  // get the map from the matrix
361  virtual Teuchos::RCP<const MapType> buildTpetraMap(int i) const;
363 
364  // get the graph of the crs matrix
365  virtual Teuchos::RCP<const CrsGraphType> buildTpetraGraph(int i,int j) const;
367 
368  // storage for Tpetra graphs and maps
370  mutable std::vector<Teuchos::RCP<const MapType> > maps_;
371  mutable std::vector<Teuchos::RCP<const MapType> > ghostedMaps_;
372  mutable std::unordered_map<std::pair<int,int>,Teuchos::RCP<const CrsGraphType>,panzer::pair_hash> graphs_ ;
373  mutable std::unordered_map<std::pair<int,int>,Teuchos::RCP<const CrsGraphType>,panzer::pair_hash> ghostedGraphs_;
374 
375  mutable std::vector<Teuchos::RCP<const ImportType> > importers_;
376  mutable std::vector<Teuchos::RCP<const ExportType> > exporters_;
377 };
378 
379 }
380 
381 #endif
BlockedTpetraLinearObjContainer< ScalarT, LocalOrdinalT, GlobalOrdinalT, NodeT > BTLOC
void initializeContainer(int, LinearObjContainer &loc) const
std::vector< Teuchos::RCP< const ImportType > > importers_
Teuchos::RCP< CrsMatrixType > getTpetraMatrix(int i, int j) const
Thyra::TpetraVector< ScalarT, LocalOrdinalT, GlobalOrdinalT, NodeT > ThyraVector
Pushes residual values into the residual vector for a Newton-based solve.
virtual Teuchos::RCP< LinearObjContainer > buildPrimitiveLinearObjContainer() const
virtual Teuchos::RCP< const ExportType > getGhostedExport(int j) const
get exporter for converting an overalapped object to a &quot;normal&quot; object
Thyra::TpetraLinearOp< ScalarT, LocalOrdinalT, GlobalOrdinalT, NodeT > ThyraLinearOp
virtual void adjustForDirichletConditions(const LinearObjContainer &localBCRows, const LinearObjContainer &globalBCRows, LinearObjContainer &ghostedObjs, bool zeroVectorRows=false, bool adjustX=false) const
Teuchos::RCP< VectorType > getTpetraDomainVector(int i) const
void initializeGhostedContainer(int, LinearObjContainer &loc) const
virtual void readVector(const std::string &, LinearObjContainer &, int) const
virtual Teuchos::RCP< WriteVector_GlobalEvaluationData > buildWriteDomainContainer() const
virtual void ghostToGlobalContainer(const LinearObjContainer &ghostContainer, LinearObjContainer &container, int) const
Teuchos::RCP< const BlockedDOFManager< LocalOrdinalT, GlobalOrdinalT > > blockedDOFManager_
std::vector< Teuchos::RCP< const UniqueGlobalIndexer< LocalOrdinalT, GlobalOrdinalT > > > gidProviders_
Teuchos::RCP< Thyra::VectorBase< ScalarT > > getThyraDomainVector() const
Get a domain vector.
Teuchos::RCP< Thyra::VectorBase< ScalarT > > getGhostedThyraDomainVector() const
Get a domain vector.
std::vector< Teuchos::RCP< const ExportType > > exporters_
Tpetra::Import< LocalOrdinalT, GlobalOrdinalT, NodeT > ImportType
Teuchos::RCP< VectorType > getGhostedTpetraRangeVector(int i) const
virtual Teuchos::RCP< LinearObjContainer > buildLinearObjContainer() const
Teuchos::RCP< Thyra::VectorBase< ScalarT > > getGhostedThyraRangeVector() const
Get a range vector.
void globalToGhostThyraVector(const Teuchos::RCP< const Thyra::VectorBase< ScalarT > > &in, const Teuchos::RCP< Thyra::VectorBase< ScalarT > > &out) const
virtual void applyDirichletBCs(const LinearObjContainer &counter, LinearObjContainer &result) const
virtual Teuchos::RCP< const ImportType > getGhostedImport(int i) const
get importer for converting an overalapped object to a &quot;normal&quot; object
std::vector< Teuchos::RCP< const MapType > > maps_
Teuchos::RCP< const Thyra::VectorSpaceBase< ScalarT > > getThyraRangeSpace() const
Get the range vector space (f)
Teuchos::RCP< panzer::CloneableEvaluator > buildScatter() const
Use preconstructed scatter evaluators.
Teuchos::RCP< const Teuchos::MpiComm< int > > comm_
Tpetra::CrsGraph< LocalOrdinalT, GlobalOrdinalT, NodeT > CrsGraphType
PHX::MDField< ScalarT, panzer::Cell, panzer::IP > result
A field that will be used to build up the result of the integral we&#39;re performing.
Gathers solution values from the Newton solution vector into the nodal fields of the field manager...
Teuchos::RCP< Thyra::ProductVectorSpaceBase< ScalarT > > rangeSpace_
Teuchos::RCP< panzer::CloneableEvaluator > buildGatherDomain() const
Use preconstructed gather evaluators.
Teuchos::RCP< panzer::CloneableEvaluator > buildGather() const
Use preconstructed gather evaluators.
Teuchos::RCP< const Thyra::VectorSpaceBase< ScalarT > > getGhostedThyraDomainSpace() const
Get the domain vector space (x and dxdt)
std::unordered_map< std::pair< int, int >, Teuchos::RCP< const CrsGraphType >, panzer::pair_hash > ghostedGraphs_
void ghostToGlobalTpetraMatrix(int blockRow, const CrsMatrixType &in, CrsMatrixType &out) const
TEUCHOS_DEPRECATED RCP< T > rcp(T *p, Dealloc_T dealloc, bool owns_mem)
Teuchos::RCP< Thyra::ProductVectorSpaceBase< ScalarT > > ghostedDomainSpace_
Tpetra::Export< LocalOrdinalT, GlobalOrdinalT, NodeT > ExportType
virtual Teuchos::RCP< const MapType > buildTpetraMap(int i) const
void globalToGhostTpetraVector(int i, const VectorType &in, VectorType &out) const
virtual Teuchos::RCP< const MapType > buildTpetraGhostedMap(int i) const
std::vector< Teuchos::RCP< const UniqueGlobalIndexer< LocalOrdinalT, GlobalOrdinalT > > > nc2c_vector(const std::vector< Teuchos::RCP< UniqueGlobalIndexer< LocalOrdinalT, GlobalOrdinalT > > > &ugis)
std::unordered_map< std::pair< int, int >, Teuchos::RCP< const CrsGraphType >, panzer::pair_hash > graphs_
void ghostToGlobalThyraMatrix(const Thyra::LinearOpBase< ScalarT > &in, Thyra::LinearOpBase< ScalarT > &out) const
virtual Teuchos::RCP< LinearObjContainer > buildGhostedLinearObjContainer() const
Teuchos::RCP< panzer::CloneableEvaluator > buildGatherOrientation() const
Use preconstructed gather evaluators.
Teuchos::RCP< Thyra::BlockedLinearOpBase< ScalarT > > getGhostedThyraMatrix() const
Get a Thyra operator.
Teuchos::RCP< panzer::CloneableEvaluator > buildScatterDirichlet() const
Use preconstructed dirichlet scatter evaluators.
std::unordered_set< std::pair< int, int >, panzer::pair_hash > excludedPairs_
void makeRoomForBlocks(std::size_t blockCnt)
Allocate the space in the std::vector objects so we can fill with appropriate Tpetra data...
virtual Teuchos::RCP< const MapType > getGhostedMap(int i) const
get the ghosted map from the matrix
Teuchos::RCP< VectorType > getGhostedTpetraDomainVector(int i) const
virtual Teuchos::RCP< LinearObjContainer > buildPrimitiveGhostedLinearObjContainer() const
Tpetra::Vector< ScalarT, LocalOrdinalT, GlobalOrdinalT, NodeT > VectorType
Teuchos::RCP< VectorType > getTpetraRangeVector(int i) const
void ghostToGlobalThyraVector(const Teuchos::RCP< const Thyra::VectorBase< ScalarT > > &in, const Teuchos::RCP< Thyra::VectorBase< ScalarT > > &out) const
virtual Teuchos::RCP< const MapType > getMap(int i) const
get the map from the matrix
Tpetra::CrsMatrix< ScalarT, LocalOrdinalT, GlobalOrdinalT, NodeT > CrsMatrixType
Tpetra::Operator< ScalarT, LocalOrdinalT, GlobalOrdinalT, NodeT > OperatorType
Teuchos::RCP< const UniqueGlobalIndexer< LocalOrdinalT, std::pair< int, GlobalOrdinalT > > > blockProvider_
Teuchos::RCP< Thyra::ProductVectorSpaceBase< ScalarT > > domainSpace_
virtual void endFill(LinearObjContainer &loc) const
void addExcludedPairs(const std::vector< std::pair< int, int > > &exPairs)
exclude a vector of pairs from the matrix
Teuchos::RCP< const Thyra::VectorSpaceBase< ScalarT > > getGhostedThyraRangeSpace() const
Get the range vector space (f)
virtual Teuchos::RCP< const CrsGraphType > getGraph(int i, int j) const
get the graph of the crs matrix
virtual Teuchos::RCP< const CrsGraphType > getGhostedGraph(int i, int j) const
get the ghosted graph of the crs matrix
void addExcludedPair(int rowBlock, int colBlock)
exclude a block pair from the matrix
virtual Teuchos::RCP< const CrsGraphType > buildTpetraGraph(int i, int j) const
virtual void writeVector(const std::string &, const LinearObjContainer &, int) const
virtual void beginFill(LinearObjContainer &loc) const
virtual Teuchos::RCP< const CrsGraphType > buildTpetraGhostedGraph(int i, int j) const
virtual void globalToGhostContainer(const LinearObjContainer &container, LinearObjContainer &ghostContainer, int) const
BlockedTpetraLinearObjFactory(const Teuchos::RCP< const Teuchos::MpiComm< int > > &comm, const Teuchos::RCP< const BlockedDOFManager< LocalOrdinalT, GlobalOrdinalT > > &gidProvider)
Teuchos::RCP< Thyra::ProductVectorSpaceBase< ScalarT > > ghostedRangeSpace_
Teuchos::RCP< panzer::CloneableEvaluator > buildGatherTangent() const
Use preconstructed gather evaluators.
Teuchos::RCP< CrsMatrixType > getGhostedTpetraMatrix(int i, int j) const
Pushes residual values into the residual vector for a Newton-based solve.
void ghostToGlobalTpetraVector(int i, const VectorType &in, VectorType &out) const
Teuchos::RCP< Thyra::VectorBase< ScalarT > > getThyraRangeVector() const
Get a range vector.
virtual Teuchos::RCP< ReadOnlyVector_GlobalEvaluationData > buildReadOnlyDomainContainer() const
Gathers orientations per field from the global indexer and stores them in the field manager...
#define TEUCHOS_ASSERT(assertion_test)
Teuchos::RCP< const panzer::UniqueGlobalIndexerBase > getRangeGlobalIndexer() const
Get the range unique global indexer this factory was created with.
Gathers tangent vectors dx/dp for computing df/dx*dx/dp + df/dp into the nodal fields of the field ma...
std::vector< Teuchos::RCP< const MapType > > ghostedMaps_
Teuchos::RCP< const panzer::UniqueGlobalIndexerBase > getDomainGlobalIndexer() const
Get the domain unique global indexer this factory was created with.
Teuchos::RCP< const panzer::BlockedDOFManager< int, GlobalOrdinalT > > getGlobalIndexer() const
Tpetra::Map< LocalOrdinalT, GlobalOrdinalT, NodeT > MapType
Teuchos::RCP< const Thyra::VectorSpaceBase< ScalarT > > getThyraDomainSpace() const
Get the domain vector space (x and dxdt)
Teuchos::RCP< Thyra::LinearOpBase< ScalarT > > getThyraMatrix() const
Get a Thyra operator.