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