SurfaceBasis.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 SURFACE_BASIS_HPP
#define SURFACE_BASIS_HPP

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

#include "Albany_Layouts.hpp"

#include "Intrepid_CellTools.hpp"
#include "Intrepid_Cubature.hpp"

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

  public:
    typedef typename EvalT::ScalarT ScalarT;
    typedef typename EvalT::MeshScalarT MeshScalarT;
    typedef Intrepid::FieldContainer<ScalarT> SFC;
    typedef Intrepid::FieldContainer<MeshScalarT> MFC;

    SurfaceBasis(const 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);

    void computeReferenceMidplaneCoords(const PHX::MDField<MeshScalarT,Cell,Vertex,Dim> refCoords,
                                        MFC & midplaneCoords);

    void computeCurrentMidplaneCoords(const PHX::MDField<ScalarT,Cell,Vertex,Dim> currentCoords,
                                      SFC & midplaneCoords);

    void computeReferenceBaseVectors(const MFC & midplaneCoords, 
                                     PHX::MDField<MeshScalarT,Cell,QuadPoint,Dim,Dim> basis);

    void computeCurrentBaseVectors(const SFC & midplaneCoords, 
                            PHX::MDField<ScalarT,Cell,QuadPoint,Dim,Dim> basis);

    void computeDualBaseVectors(const MFC & midplaneCoords, 
                                const PHX::MDField<MeshScalarT,Cell,QuadPoint,Dim,Dim> basis, 
                                PHX::MDField<MeshScalarT,Cell,QuadPoint,Dim> normal, 
                                PHX::MDField<MeshScalarT,Cell,QuadPoint,Dim,Dim> dualBasis);

    void computeJacobian(const PHX::MDField<MeshScalarT,Cell,QuadPoint,Dim,Dim> basis,
                         const PHX::MDField<MeshScalarT,Cell,QuadPoint,Dim,Dim> dualBasis,
                         PHX::MDField<MeshScalarT,Cell,QuadPoint> area);

  private:
    unsigned int  numDims, numNodes, numQPs, numPlaneNodes, numPlaneDims;

    bool needCurrentBasis;

    PHX::MDField<MeshScalarT,Cell,Vertex,Dim> referenceCoords;

    Teuchos::RCP<Intrepid::Cubature<RealType> > cubature;
    
    Teuchos::RCP<Intrepid::Basis<RealType, Intrepid::FieldContainer<RealType> > > intrepidBasis;

    Intrepid::FieldContainer<MeshScalarT> refMidplaneCoords;

    Intrepid::FieldContainer<ScalarT> currentMidplaneCoords;

    PHX::MDField<MeshScalarT,Cell,QuadPoint,Dim,Dim> refBasis;

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

    PHX::MDField<MeshScalarT,Cell,QuadPoint,Dim,Dim> refDualBasis;

    PHX::MDField<MeshScalarT,Cell,QuadPoint,Dim> refNormal;

    // if we need to compute the current bases (for mechanics)
    PHX::MDField<ScalarT,Cell,Vertex,Dim> currentCoords;

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

    Intrepid::FieldContainer<RealType> refValues;

    Intrepid::FieldContainer<RealType> refGrads;

    Intrepid::FieldContainer<RealType> refPoints;

    Intrepid::FieldContainer<RealType> refWeights;
  };
}

#endif