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

#include <vector>
#include <string>

#include "Teuchos_TestForException.hpp"
#include "Phalanx_DataLayout.hpp"

namespace PHAL {

template<typename EvalT, typename Traits>
SaveStateField<EvalT, Traits>::
SaveStateField(const Teuchos::ParameterList& p)
{
  // States Not Saved for Generic Type, only Specializations
  this->setName("Save State Field"+PHX::TypeString<EvalT>::value);
}

// **********************************************************************
template<typename EvalT, typename Traits>
void SaveStateField<EvalT, Traits>::postRegistrationSetup(typename Traits::SetupData d,
                      PHX::FieldManager<Traits>& fm)
{
  // States Not Saved for Generic Type, only Specializations
}

// **********************************************************************
template<typename EvalT, typename Traits>
void SaveStateField<EvalT, Traits>::evaluateFields(typename Traits::EvalData workset)
{
  // States Not Saved for Generic Type, only Specializations
}
// **********************************************************************
// **********************************************************************
template<typename Traits>
SaveStateField<PHAL::AlbanyTraits::Residual, Traits>::
SaveStateField(const Teuchos::ParameterList& p)
{
  fieldName =  p.get<std::string>("Field Name");
  stateName =  p.get<std::string>("State Name");
  PHX::MDField<ScalarT> f(fieldName, p.get<Teuchos::RCP<PHX::DataLayout> >("State Field Layout") );
  field = f;

  savestate_operation = Teuchos::rcp(new PHX::Tag<ScalarT>
    (fieldName, p.get< Teuchos::RCP<PHX::DataLayout> >("Dummy Data Layout")));

  this->addDependentField(field);
  this->addEvaluatedField(*savestate_operation);

  this->setName("Save Field " + fieldName +" to State " + stateName
                + PHX::TypeString<PHAL::AlbanyTraits::Residual>::value);
}

// **********************************************************************
template<typename Traits>
void SaveStateField<PHAL::AlbanyTraits::Residual, Traits>::
postRegistrationSetup(typename Traits::SetupData d,
                      PHX::FieldManager<Traits>& fm)
{
  this->utils.setFieldData(field,fm);
}
// **********************************************************************
template<typename Traits>
void SaveStateField<PHAL::AlbanyTraits::Residual, Traits>::
evaluateFields(typename Traits::EvalData workset)
{
  // Get shards Array (from STK) for this state
  // Need to check if we can just copy full size -- can assume same ordering?
    Albany::StateArray::const_iterator it;
    it = workset.stateArrayPtr->find(stateName);

    TEUCHOS_TEST_FOR_EXCEPTION((it == workset.stateArrayPtr->end()), std::logic_error,
           std::endl << "Error: cannot locate " << stateName << " in PHAL_SaveStateField_Def" << std::endl);

    Albany::MDArray sta = it->second;
    std::vector<int> dims;
    sta.dimensions(dims);
    int size = dims.size();

    switch (size) {
    case 1:
      sta(0) = field(0);
      break;
    case 2:
      for (int cell = 0; cell < dims[0]; ++cell)
  for (int qp = 0; qp < dims[1]; ++qp)
    sta(cell, qp) = field(cell,qp);;
      break;
    case 3:
      for (int cell = 0; cell < dims[0]; ++cell)
  for (int qp = 0; qp < dims[1]; ++qp)
    for (int i = 0; i < dims[2]; ++i)
      sta(cell, qp, i) = field(cell,qp,i);
      break;
    case 4:
      for (int cell = 0; cell < dims[0]; ++cell)
  for (int qp = 0; qp < dims[1]; ++qp)
    for (int i = 0; i < dims[2]; ++i)
      for (int j = 0; j < dims[3]; ++j)
        sta(cell, qp, i, j) = field(cell,qp,i,j);
      break;
    default:
      TEUCHOS_TEST_FOR_EXCEPT_MSG(size<1||size>4,
                          "Unexpected Array dimensions in SaveStateField: " << size);
    }
}

}