Panzer_AssemblyEngine.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_ASSEMBLY_ENGINE_HPP
#define PANZER_ASSEMBLY_ENGINE_HPP

#include "Teuchos_RCP.hpp"

#include "Panzer_Base.hpp"
#include "Panzer_BC.hpp"
#include "Panzer_Traits.hpp"
#include "Panzer_LinearObjFactory.hpp"
#include "Panzer_LinearObjContainer.hpp"

namespace panzer {
  class FieldManagerBuilder;
  class AssemblyEngineInArgs;
}

namespace panzer {

  template <typename EvalT>
    class AssemblyEngine : public panzer::Base {

  public:    
    struct EvaluationFlags {
      EvaluationFlags(int flags) : value_(flags) {
        TEUCHOS_ASSERT(flags>0 && flags <= EvaluationFlags::All);
      }
      static constexpr int Initialize=1;
      static constexpr int VolumetricFill=2;
      static constexpr int BoundaryFill=4;
      static constexpr int Scatter=8;
      static constexpr int All=15;
      int getValue() const {return value_;}
    protected:
      int value_;
    };

    AssemblyEngine(const Teuchos::RCP<panzer::FieldManagerBuilder>& fmb,
                   const Teuchos::RCP<const panzer::LinearObjFactory<panzer::Traits> > & lof);
    
    void evaluate(const panzer::AssemblyEngineInArgs& input_arguments, const EvaluationFlags flags=EvaluationFlags(EvaluationFlags::All));

    void evaluateVolume(const panzer::AssemblyEngineInArgs& input_arguments);

    Teuchos::RCP<LinearObjContainer> evaluateOnlyDirichletBCs(const panzer::AssemblyEngineInArgs& input_arguments);

    void evaluateNeumannBCs(const panzer::AssemblyEngineInArgs& input_arguments);

    void evaluateInterfaceBCs(const panzer::AssemblyEngineInArgs& input_arguments);

    Teuchos::RCP<LinearObjContainer> evaluateDirichletBCs(const panzer::AssemblyEngineInArgs& input_arguments);

    Teuchos::RCP<panzer::FieldManagerBuilder> getManagerBuilder()
    { return m_field_manager_builder; }
    
  protected:
      
    void evaluateBCs(const panzer::BCType bc_type, 
         const panzer::AssemblyEngineInArgs& input_arguments,
                     const Teuchos::RCP<LinearObjContainer> preEval_loc=Teuchos::null);

  protected:
    
      Teuchos::RCP<panzer::FieldManagerBuilder> m_field_manager_builder;

      Teuchos::RCP<const panzer::LinearObjFactory<panzer::Traits> > m_lin_obj_factory;

      // These members improve performance by ensuring that "buildPrimitiveGhostedLinearObjContainer"
      // is not called uneccessarily often
      bool countersInitialized_;
      Teuchos::RCP<LinearObjContainer> localCounter_;
      Teuchos::RCP<LinearObjContainer> globalCounter_;
      Teuchos::RCP<LinearObjContainer> summedGhostedCounter_;
  };
  
}

// #include "Panzer_AssemblyEngine_impl.hpp"

#endif