MultiScaleStress.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 MULTISCALESTRESS_HPP
#define MULTISCALESTRESS_HPP

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

namespace LCM {

enum MessageType {STRESS_TENSOR, STRAIN_TENSOR, TANGENT, DIE};


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

  public:

    MultiScaleStressBase(const Teuchos::ParameterList& p);

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

    void evaluateFields(typename Traits::EvalData d);

  protected:

    void calcStress(typename Traits::EvalData workset);

    // Protected function for stress calc, only for RealType
    void mesoBridgeStressRealType(PHX::MDField<RealType, Cell, QuadPoint, Dim, Dim>& stressFieldOut,
                                  PHX::MDField<RealType, Cell, QuadPoint, Dim, Dim>& stressFieldIn,
                                  typename Traits::EvalData workset);

    PHX::MDField<RealType, Cell, QuadPoint, Dim, Dim> stressFieldRealType;


    struct MesoPt {

      int cell;
      int qp;

    };

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

    // Input:
    PHX::MDField<ScalarT, Cell, QuadPoint, Dim, Dim> strain;
    PHX::MDField<ScalarT, Cell, QuadPoint> elasticModulus;
    PHX::MDField<ScalarT, Cell, QuadPoint> poissonsRatio;

    unsigned int numQPs;
    unsigned int numDims;

    int numMesoPEs;
    std::vector<double> exchanged_stresses;
    std::vector<MesoPt> loc_data;
    Teuchos::RCP<MPI_Comm> interCommunicator;


    // Output:
    PHX::MDField<ScalarT, Cell, QuadPoint, Dim, Dim> stress;

    // Link convenience functions

    void sendCellQPData(int cell, int qp, int toProc, MessageType type,
                        PHX::MDField<RealType, Cell, QuadPoint, Dim, Dim>& stressFieldIn);

    void rcvCellQPData(int procIDReached, MessageType type,
                       PHX::MDField<RealType, Cell, QuadPoint, Dim, Dim>& stressFieldOut);

};

// Inherted classes
template<typename EvalT, typename Traits> class MultiScaleStress;

// For all cases except those specialized below, just fall through to base class.
// The base class throws "Not Implemented" for evaluate fields.
template<typename EvalT, typename Traits>
class MultiScaleStress : public MultiScaleStressBase<EvalT, Traits> {
  public:
    MultiScaleStress(Teuchos::ParameterList& p) : MultiScaleStressBase<EvalT, Traits>(p) {};
};


// Template Specialization: Residual Eval calls MultiScale with doubles.
template<typename Traits>
class MultiScaleStress<PHAL::AlbanyTraits::Residual, Traits>
    : public MultiScaleStressBase<PHAL::AlbanyTraits::Residual, Traits> {
  public:
    MultiScaleStress(Teuchos::ParameterList& p) : MultiScaleStressBase<PHAL::AlbanyTraits::Residual, Traits>(p) {};
    void evaluateFields(typename Traits::EvalData d);
};

// Template Specialization: Jacobian Eval does finite difference of MultiScale with doubles.
template<typename Traits>
class MultiScaleStress<PHAL::AlbanyTraits::Jacobian, Traits>
    : public MultiScaleStressBase<PHAL::AlbanyTraits::Jacobian, Traits> {
  public:
    MultiScaleStress(Teuchos::ParameterList& p) : MultiScaleStressBase<PHAL::AlbanyTraits::Jacobian, Traits>(p) {};
    void evaluateFields(typename Traits::EvalData d);
};

// Template Specialization: Tangent Eval does finite difference of MultiScale with doubles.
template<typename Traits>
class MultiScaleStress<PHAL::AlbanyTraits::Tangent, Traits>
    : public MultiScaleStressBase<PHAL::AlbanyTraits::Tangent, Traits> {
  public:
    MultiScaleStress(Teuchos::ParameterList& p) : MultiScaleStressBase<PHAL::AlbanyTraits::Tangent, Traits>(p) {};
    void evaluateFields(typename Traits::EvalData d);
};

}

#endif