ConstitutiveModelInterface.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_ConstitutiveModelInterface_hpp)
#define LCM_ConstitutiveModelInterface_hpp

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

#include "models/ConstitutiveModel.hpp"

namespace LCM {

  struct StateVariableRegistrationStruct {
  public:
    std::string name;
    Teuchos::RCP<PHX::DataLayout> data_layout;
    std::string init_type;
    double init_value;
    bool register_old_state;
    bool output_to_exodus;
  };

  template<typename EvalT, typename Traits>
  class ConstitutiveModelInterface : public PHX::EvaluatorWithBaseImpl<Traits>,
                                     public PHX::EvaluatorDerived<EvalT, Traits>  {

  public:

    ConstitutiveModelInterface(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 fillStateVariableStruct(int state_var);

    int getNumStateVars() { return model_->getNumStateVariables(); }

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

    std::string getName() { return sv_struct_.name; }

    Teuchos::RCP<PHX::DataLayout> getLayout() { return sv_struct_.data_layout; }

    std::string getInitType() { return sv_struct_.init_type; }

    double getInitValue() { return sv_struct_.init_value; }

    double getStateFlag() { return sv_struct_.register_old_state; }

    double getOutputFlag() { return sv_struct_.output_to_exodus; }

  private:

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

    std::map<std::string, Teuchos::RCP<PHX::MDField<ScalarT> > > dep_fields_map_;

    std::map<std::string, Teuchos::RCP<PHX::MDField<ScalarT> > > eval_fields_map_;

    Teuchos::RCP<LCM::ConstitutiveModel<EvalT,Traits> > model_;

    StateVariableRegistrationStruct sv_struct_;

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

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

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

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

    bool have_temperature_;

    bool have_damage_;

    bool volume_average_pressure_;
  };

}

#endif