43 #ifndef PANZER_STK_SCATTER_CELL_AVG_VECTOR_IMPL_HPP
44 #define PANZER_STK_SCATTER_CELL_AVG_VECTOR_IMPL_HPP
46 #include "Teuchos_Assert.hpp"
48 #include "Phalanx_config.hpp"
49 #include "Phalanx_Evaluator_Macros.hpp"
50 #include "Phalanx_MDField.hpp"
51 #include "Phalanx_DataLayout.hpp"
52 #include "Phalanx_DataLayout_MDALayout.hpp"
57 #include "Teuchos_FancyOStream.hpp"
58 #include "Teuchos_ArrayRCP.hpp"
60 namespace panzer_stk {
62 template<
typename EvalT,
typename Traits>
72 std::string scatterName = p.
get<std::string>(
"Scatter Name");
77 const std::vector<std::string> & names =
86 for (std::size_t fd = 0; fd < names.size(); ++fd)
94 this->addEvaluatedField(scatterHolder);
96 this->setName(scatterName+
": PanzerSTK::ScatterCellAvgVectors");
100 template<
typename EvalT,
typename Traits>
104 typename Traits::SetupData ,
107 for (std::size_t fd = 0; fd < scatterFields_.size(); ++fd)
109 std::string fieldName = scatterFields_[fd].fieldTag().name();
111 stkFields_[fd] = mesh_->getMetaData()->get_field<
double>(stk::topology::ELEMENT_RANK, fieldName);
116 template<
typename EvalT,
typename Traits>
120 typename Traits::EvalData workset)
125 const std::vector<std::size_t> & localCellIds = this->wda(workset).cell_local_ids;
126 std::string blockId = this->wda(workset).block_id;
127 std::string d_mod[3] = {
"X",
"Y",
"Z"};
130 for(std::size_t fieldIndex = 0; fieldIndex < scatterFields_.size(); fieldIndex++)
133 std::string fieldName = field.fieldTag().name();
134 const int numCells = workset.num_cells;
136 const int numDims = field.extent(2);
138 for (
int dim = 0; dim < numDims; dim++)
144 Kokkos::parallel_for(
"ScatterCellAvgVector",numCells,KOKKOS_LAMBDA(
const int i){
147 average(i,0) += Sacado::scalarValue(
field(i,j,dim));
153 if (!varScaleFactors_.is_null())
155 std::map<std::string,double> *tmp_sfs = varScaleFactors_.get();
156 if(tmp_sfs->find(fieldName) != tmp_sfs->end())
157 scalef = (*tmp_sfs)[fieldName];
160 PHX::Device().fence();
161 mesh_->setCellFieldData(fieldName+d_mod[dim],blockId,localCellIds,average.get_view(),scalef);
T & get(const std::string &name, T def_value)
PHX::MDField< Scalar, T0 > buildStaticArray(const std::string &str, int d0) const
void postRegistrationSetup(typename Traits::SetupData d, PHX::FieldManager< Traits > &fm)
ScatterCellAvgVector(const Teuchos::ParameterList &p)
bool isParameter(const std::string &name) const
TEUCHOS_DEPRECATED RCP< T > rcp(T *p, Dealloc_T dealloc, bool owns_mem)
std::vector< VariableField * > stkFields_
Teuchos::RCP< PHX::DataLayout > dl_vector
Data layout for vector fields.
PHX::MDField< ScalarT, panzer::Cell, panzer::BASIS > field
A field to which we'll contribute, or in which we'll store, the result of computing this integral...
std::vector< PHX::MDField< const ScalarT, panzer::Cell, panzer::Point, panzer::Dim > > scatterFields_
Teuchos::RCP< std::map< std::string, double > > varScaleFactors_
void evaluateFields(typename Traits::EvalData d)