PHAL_Dirichlet.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  //
//*****************************************************************//

#ifndef PHAL_DIRICHLET_HPP
#define PHAL_DIRICHLET_HPP

#include "Phalanx_ConfigDefs.hpp"
#include "Phalanx_Evaluator_WithBaseImpl.hpp"
#include "Phalanx_Evaluator_Derived.hpp"
#include "Phalanx_MDField.hpp"

#include "Teuchos_ParameterList.hpp"
#include "Epetra_Vector.h"

#include "Sacado_ParameterAccessor.hpp"
#include "PHAL_AlbanyTraits.hpp"

namespace PHAL {
// **************************************************************
// Generic Template Impelementation for constructor and PostReg
// **************************************************************

template<typename EvalT, typename Traits>
class DirichletBase
  : public PHX::EvaluatorWithBaseImpl<Traits>,
    public PHX::EvaluatorDerived<EvalT, Traits>,
    public Sacado::ParameterAccessor<EvalT, SPL_Traits>  
   {
  
private:

  //typedef typename Traits::Residual::ScalarT ScalarT;
  typedef typename EvalT::ScalarT ScalarT;

public:
  
  DirichletBase(Teuchos::ParameterList& p);
  
  void postRegistrationSetup(typename Traits::SetupData d,
                             PHX::FieldManager<Traits>& vm);
  
  // This function will be overloaded with template specialized code
  void evaluateFields(typename Traits::EvalData d)=0;
  
  virtual ScalarT& getValue(const std::string &n) { return value; }

protected:
  const int offset;
  ScalarT value;
  std::string nodeSetID;
};

// **************************************************************
// **************************************************************
// * Specializations
// **************************************************************
// **************************************************************

template<typename EvalT, typename Traits> class Dirichlet;

// **************************************************************
// Residual 
// **************************************************************
template<typename Traits>
class Dirichlet<PHAL::AlbanyTraits::Residual,Traits>
   : public DirichletBase<PHAL::AlbanyTraits::Residual, Traits> {
public:
  Dirichlet(Teuchos::ParameterList& p);
  void evaluateFields(typename Traits::EvalData d);
};

// **************************************************************
// Jacobian
// **************************************************************
template<typename Traits>
class Dirichlet<PHAL::AlbanyTraits::Jacobian,Traits>
   : public DirichletBase<PHAL::AlbanyTraits::Jacobian, Traits> {
public:
  Dirichlet(Teuchos::ParameterList& p);
  void evaluateFields(typename Traits::EvalData d);
};

// **************************************************************
// Tangent
// **************************************************************
template<typename Traits>
class Dirichlet<PHAL::AlbanyTraits::Tangent,Traits>
   : public DirichletBase<PHAL::AlbanyTraits::Tangent, Traits> {
public:
  Dirichlet(Teuchos::ParameterList& p);
  void evaluateFields(typename Traits::EvalData d);
};

// **************************************************************
// Stochastic Galerkin Residual 
// **************************************************************
#ifdef ALBANY_SG_MP
template<typename Traits>
class Dirichlet<PHAL::AlbanyTraits::SGResidual,Traits>
   : public DirichletBase<PHAL::AlbanyTraits::SGResidual, Traits> {
public:
  Dirichlet(Teuchos::ParameterList& p);
  void evaluateFields(typename Traits::EvalData d);
};

// **************************************************************
// Stochastic Galerkin Jacobian
// **************************************************************
template<typename Traits>
class Dirichlet<PHAL::AlbanyTraits::SGJacobian,Traits>
   : public DirichletBase<PHAL::AlbanyTraits::SGJacobian, Traits> {
public:
  Dirichlet(Teuchos::ParameterList& p);
  void evaluateFields(typename Traits::EvalData d);
};

// **************************************************************
// Stochastic Galerkin Tangent
// **************************************************************
template<typename Traits>
class Dirichlet<PHAL::AlbanyTraits::SGTangent,Traits>
   : public DirichletBase<PHAL::AlbanyTraits::SGTangent, Traits> {
public:
  Dirichlet(Teuchos::ParameterList& p);
  void evaluateFields(typename Traits::EvalData d);
};

// **************************************************************
// Multi-point Residual 
// **************************************************************
template<typename Traits>
class Dirichlet<PHAL::AlbanyTraits::MPResidual,Traits>
   : public DirichletBase<PHAL::AlbanyTraits::MPResidual, Traits> {
public:
  Dirichlet(Teuchos::ParameterList& p);
  void evaluateFields(typename Traits::EvalData d);
};

// **************************************************************
// Multi-point Jacobian
// **************************************************************
template<typename Traits>
class Dirichlet<PHAL::AlbanyTraits::MPJacobian,Traits>
   : public DirichletBase<PHAL::AlbanyTraits::MPJacobian, Traits> {
public:
  Dirichlet(Teuchos::ParameterList& p);
  void evaluateFields(typename Traits::EvalData d);
};

// **************************************************************
// Multi-point Tangent
// **************************************************************
template<typename Traits>
class Dirichlet<PHAL::AlbanyTraits::MPTangent,Traits>
   : public DirichletBase<PHAL::AlbanyTraits::MPTangent, Traits> {
public:
  Dirichlet(Teuchos::ParameterList& p);
  void evaluateFields(typename Traits::EvalData d);
};
#endif //ALBANY_SG_MP

// **************************************************************
// **************************************************************
// Evaluator to aggregate all Dirichlet BCs into one "field"
// **************************************************************
template<typename EvalT, typename Traits>
class DirichletAggregator
  : public PHX::EvaluatorWithBaseImpl<Traits>,
    public PHX::EvaluatorDerived<EvalT, Traits>
{
private:

  typedef typename EvalT::ScalarT ScalarT;

public:
  
  DirichletAggregator(Teuchos::ParameterList& p);
  
  void postRegistrationSetup(typename Traits::SetupData d,
                             PHX::FieldManager<Traits>& vm) {};
  
  // This function will be overloaded with template specialized code
  void evaluateFields(typename Traits::EvalData d) {};
};
}

#endif