43 #ifndef PANZER_CONSTANT_FLUX_IMPL_HPP 44 #define PANZER_CONSTANT_FLUX_IMPL_HPP 49 template<
typename EvalT,
typename Traits>
52 const Teuchos::ParameterList& p) :
53 flux( p.get<
std::string>(
"Flux Field Name"),
54 p.get<
Teuchos::RCP<
PHX::DataLayout> >(
"Data Layout") )
56 const Teuchos::ParameterList& flux_values = p.sublist(
"Flux Values");
58 for (Teuchos::ParameterList::ConstIterator i = flux_values.begin(); i != flux_values.end(); ++i)
59 values.push_back(Teuchos::getValue<double>(i->second));
61 this->addEvaluatedField(
flux);
63 std::string n =
"Constant: " +
flux.fieldTag().name();
68 template<
typename EvalT,
typename Traits>
76 this->utils.setFieldData(flux,fm);
78 TEUCHOS_ASSERT(static_cast<std::size_t>(flux.extent(2)) == values.size());
80 auto flux_v = flux.get_static_view();
82 for (
int dim = 0; dim < flux_v.extent_int(2); ++dim) {
83 auto val = values[dim];
84 Kokkos::parallel_for (
"ConstantFlux", flux.extent_int(0), KOKKOS_LAMBDA(
const int cell) {
85 for (
int ip = 0; ip < flux_v.extent_int(1); ++ip)
86 flux_v(cell,ip,dim) = val;
93 template<
typename EvalT,
typename Traits>
void postRegistrationSetup(typename Traits::SetupData d, PHX::FieldManager< Traits > &fm)
std::vector< ScalarT > values
ConstantFlux(const Teuchos::ParameterList &p)
PHX::MDField< ScalarT > flux
void evaluateFields(typename Traits::EvalData d)