43 #ifndef PANZER_CONSTANT_FLUX_IMPL_HPP
44 #define PANZER_CONSTANT_FLUX_IMPL_HPP
49 template<
typename EvalT,
typename Traits>
53 flux( p.get<std::string>(
"Flux Field Name"),
54 p.get< Teuchos::
RCP<PHX::DataLayout> >(
"Data Layout") )
59 values.push_back(Teuchos::getValue<double>(i->second));
61 this->addEvaluatedField(
flux);
63 std::string n =
"ConstantFlux: " +
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
ConstIterator end() const
ConstantFlux(const Teuchos::ParameterList &p)
params_t::ConstIterator ConstIterator
ConstIterator begin() const
ParameterList & sublist(const std::string &name, bool mustAlreadyExist=false, const std::string &docString="")
#define TEUCHOS_ASSERT(assertion_test)
PHX::MDField< ScalarT > flux
void evaluateFields(typename Traits::EvalData d)