TransportResidual.hpp

Go to the documentation of this file.

//*****************************************************************//
//    Albany 2.0:  Copyright 2012 Sandia Corporation               //
//    This Software is released under the BSD license detailed     //
//    in the file "license.txt" in the top-level Albany directory  //
//*****************************************************************//

#if !defined(LCM_TransportResidual_hpp)
#define LCM_TransportResidual_hpp

#include <Phalanx_ConfigDefs.hpp>
#include <Phalanx_Evaluator_WithBaseImpl.hpp>
#include <Phalanx_Evaluator_Derived.hpp>
#include <Phalanx_MDField.hpp>

#include "Albany_Layouts.hpp"

namespace LCM {
  template<typename EvalT, typename Traits>
  class TransportResidual : public PHX::EvaluatorWithBaseImpl<Traits>,
                            public PHX::EvaluatorDerived<EvalT, Traits>  {

  public:

    TransportResidual(Teuchos::ParameterList& p,
                      const Teuchos::RCP<Albany::Layouts>& dl);

    void postRegistrationSetup(typename Traits::SetupData d,
                               PHX::FieldManager<Traits>& vm);

    void evaluateFields(typename Traits::EvalData d);

  private:

    typedef typename EvalT::ScalarT ScalarT;
    typedef typename EvalT::MeshScalarT MeshScalarT;

    PHX::MDField<ScalarT,Cell,QuadPoint> scalar_;

    PHX::MDField<ScalarT,Cell,QuadPoint> scalar_dot_;

    PHX::MDField<ScalarT,Cell,QuadPoint> scalar_grad_;

    PHX::MDField<MeshScalarT,Cell,QuadPoint> weights_;

    PHX::MDField<MeshScalarT,Cell,Node,QuadPoint> w_bf_;

    PHX::MDField<MeshScalarT,Cell,Node,QuadPoint,Dim> w_grad_bf_;

    PHX::MDField<ScalarT,Cell,QuadPoint> source_;

    PHX::MDField<ScalarT,Cell,QuadPoint> transient_coeff_;

    PHX::MDField<ScalarT,Cell,QuadPoint,Dim,Dim> diffusivity_;

    PHX::MDField<ScalarT,Cell,QuadPoint,Dim> convection_vector_;

    PHX::MDField<ScalarT,Cell,QuadPoint> species_coupling_;

    PHX::MDField<ScalarT,Cell,QuadPoint> stabilization_;

    PHX::MDField<ScalarT,Cell,Node> residual_;

    bool have_source_;
    bool have_transient_;
    bool have_diffusion_;
    bool have_convection_;
    bool have_species_coupling_;
    bool have_stabilization_;

    std::size_t num_nodes_, num_pts_, num_dims_;

    std::string scalar_name_;

  };
}

#endif