Panzer_FieldLibrary.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_FieldLibrary_hpp__
#define __Panzer_FieldLibrary_hpp__
 
#include "PanzerDiscFE_config.hpp"
 
#include "Teuchos_RCP.hpp"
 
#include "Panzer_BasisIRLayout.hpp"
#include "Panzer_PointRule.hpp"
#include "Panzer_PureBasis.hpp"
 
#include <vector>
 
namespace panzer {
 
class FieldLibraryBase {
public:
   virtual ~FieldLibraryBase() = 0;
 
   virtual Teuchos::RCP<const panzer::PureBasis> lookupBasis(const std::string & fieldName) const = 0;
 
   virtual void uniqueBases(std::vector<Teuchos::RCP<const panzer::PureBasis> > & bases) const = 0;
 
   virtual void basisPairs(std::vector<std::pair<std::string,Teuchos::RCP<const panzer::PureBasis> > > & bases) const = 0;
 
   virtual void print(std::ostream & os) const = 0;
};
 
inline FieldLibraryBase::~FieldLibraryBase() {}
 
class FieldLayoutLibrary : public FieldLibraryBase {
public:
   void addFieldAndLayout(const std::string & fieldName,
                         const Teuchos::RCP<panzer::BasisIRLayout> & basis);   
 
   void uniqueBases(std::vector<Teuchos::RCP<const panzer::PureBasis> > & bases) const;
 
   virtual Teuchos::RCP<const panzer::PureBasis> lookupBasis(const std::string & fieldName) const;
 
   Teuchos::RCP<panzer::BasisIRLayout> lookupLayout(const std::string & fieldName) const;
 
   virtual void print(std::ostream & os) const;
 
   virtual void basisPairs(std::vector<std::pair<std::string,Teuchos::RCP<const panzer::PureBasis> > > & bases) const;
 
private:
 
   std::map<std::string,Teuchos::RCP<panzer::BasisIRLayout> > fieldToLayout_;
   std::map<std::string,Teuchos::RCP<const panzer::PureBasis> > basisNameToPointer_; // to satisfy uniuqeBases interface
 
};
 
class FieldLibrary : public FieldLibraryBase {
public:
 
   virtual Teuchos::RCP<const panzer::PureBasis> lookupBasis(const std::string & fieldName) const;
 
   void uniqueBases(std::vector<Teuchos::RCP<const panzer::PureBasis> > & bases) const;
 
   void addFieldAndBasis(const std::string & fieldName,
                 const Teuchos::RCP<panzer::PureBasis> & basis);   
 
   Teuchos::RCP<const FieldLayoutLibrary> buildFieldLayoutLibrary(panzer::PointRule & ir) const;
 
   virtual void print(std::ostream & os) const;
 
   virtual void basisPairs(std::vector<std::pair<std::string,Teuchos::RCP<const panzer::PureBasis> > > & bases) const;
 
private:
 
   std::map<std::string,Teuchos::RCP<panzer::PureBasis> > fieldToBasis_;
   std::map<std::string,Teuchos::RCP<const panzer::PureBasis> > basisNameToPointer_; // to satisfy uniuqeBases interface
};
 
inline std::ostream & operator<<(std::ostream & os,const FieldLibraryBase & flb)
{
   flb.print(os);
   return os;
}
 
}
 
#endif