43#ifndef PANZER_EXPR_EVAL_HPP
44#define PANZER_EXPR_EVAL_HPP
54#include <Teuchos_Reader.hpp>
56#include <Kokkos_Core.hpp>
104 using Function = std::function<void(std::string
const& name, Teuchos::any&, std::vector<Teuchos::any>& rhs)>;
108 void set(std::string
const& name,
Function const& value);
112 template <
typename T>
113 T
const&
get(std::string
const& name)
const {
115 TEUCHOS_TEST_FOR_EXCEPTION(it ==
symbol_map.end(), std::logic_error,
116 "EvalBase::get: \"" << name <<
"\" not found");
117 return Teuchos::any_ref_cast<T>(it->second);
126 void at_shift(Teuchos::any&
result,
int token, std::string& text)
override;
128 void at_reduce(Teuchos::any&
result,
int prod, std::vector<Teuchos::any>& rhs)
override;
130 void ternary_op(Teuchos::any&
result, Teuchos::any& cond, Teuchos::any& left, Teuchos::any& right);
150 virtual void inspect_arg(Teuchos::any
const& arg,
bool& is_many,
bool& is_bool) = 0;
165template <
typename DataType,
typename NewScalarType>
171template <
typename NestedDataType,
typename NewScalarType>
176template <
typename NestedDataType,
typename NewScalarType>
181template <
typename NestedDataType,
typename NewScalarType,
size_t N>
187template <
typename ViewType,
typename NewScalarType>
190template <
typename DT,
typename NewScalarType,
typename ... VP>
193 using type = Kokkos::View<typename RebindDataType<DT, NewScalarType>::type, VP ...>;
228template <
typename DT,
typename ... VP>
241 using scalar_type =
typename original_view_type::non_const_value_type;
249 using const_view_type = Kokkos::View<typename RebindDataType<view_data_type, scalar_type const>::type, VP ...>;
276 void set(std::string
const& name,
bool value);
287 void inspect_arg(Teuchos::any
const& arg,
bool& is_many,
bool& is_bool)
override;
317template <
typename DT,
typename ... VP>
PHX::MDField< ScalarT, panzer::Cell, panzer::IP > result
A field that will be used to build up the result of the integral we're performing.
Base class for panzer::Expr::Eval, does everything that is independent of the Kokkos::View template p...
virtual void single_many_binary_op(BinaryOpCode code, Teuchos::any &result, Teuchos::any &left, Teuchos::any &right)=0
virtual void inspect_arg(Teuchos::any const &arg, bool &is_many, bool &is_bool)=0
void neg_op(Teuchos::any &result, Teuchos::any &right)
Executes the only native unary operator in the math language, numeric negation via a minus sign.
void at_shift(Teuchos::any &result, int token, std::string &text) override
Called at every parsed token in the math language.
void at_reduce(Teuchos::any &result, int prod, std::vector< Teuchos::any > &rhs) override
Called at every reduced production in the math language.
virtual void single_single_ternary_op(Teuchos::any &result, Teuchos::any &cond, Teuchos::any &left, Teuchos::any &right)=0
virtual void single_neg_op(Teuchos::any &result, Teuchos::any &right)=0
virtual void single_many_ternary_op(Teuchos::any &result, Teuchos::any &cond, Teuchos::any &left, Teuchos::any &right)=0
virtual void single_single_binary_op(BinaryOpCode code, Teuchos::any &result, Teuchos::any &left, Teuchos::any &right)=0
virtual void many_many_ternary_op(Teuchos::any &result, Teuchos::any &cond, Teuchos::any &left, Teuchos::any &right)=0
virtual void make_constant(Teuchos::any &result, double const &value)=0
void binary_op(BinaryOpCode code, Teuchos::any &result, Teuchos::any &left, Teuchos::any &right)
Executes a binary operator.
virtual void many_single_ternary_op(Teuchos::any &result, Teuchos::any &cond, Teuchos::any &left, Teuchos::any &right)=0
virtual void many_many_binary_op(BinaryOpCode code, Teuchos::any &result, Teuchos::any &left, Teuchos::any &right)=0
T const & get(std::string const &name) const
Get the value of a variable in the symbol map.
std::function< void(std::string const &name, Teuchos::any &, std::vector< Teuchos::any > &rhs)> Function
The type of user-defined functions which are callable in the math language.
std::map< std::string, Teuchos::any > symbol_map
Stores all current symbols including variables and functions.
void ternary_op(Teuchos::any &result, Teuchos::any &cond, Teuchos::any &left, Teuchos::any &right)
Executes the ternary operator, e.g. (a > b) ? a : b.
virtual void many_single_binary_op(BinaryOpCode code, Teuchos::any &result, Teuchos::any &left, Teuchos::any &right)=0
virtual void many_neg_op(Teuchos::any &result, Teuchos::any &right)=0
void set(std::string const &name, Function const &value)
Registers an EvalBase::Function, binding it to a name and making it callable.
Interprets mathematical expressions in a string and evaluates them using Kokkos::View objects as valu...
Kokkos::View< scalar_type, VP ... > single_view_type
One scalar (same for all evaluation points)
Kokkos::View< typename RebindDataType< view_data_type, bool const >::type, VP ... > const_bool_view_type
One boolean for each evaluation point, read-only.
typename original_view_type::non_const_value_type scalar_type
The scalar type.
Kokkos::View< bool const, VP ... > const_single_bool_view_type
One boolean (same for all evaluation points), read-only.
Kokkos::View< DT, VP ... > original_view_type
The corresponding Kokkos::View type, using the same template arguments are were given to Eval.
void many_many_binary_op(BinaryOpCode code, Teuchos::any &result, Teuchos::any &left, Teuchos::any &right) override
void single_single_ternary_op(Teuchos::any &result, Teuchos::any &cond, Teuchos::any &left, Teuchos::any &right) override
void many_neg_op(Teuchos::any &result, Teuchos::any &right) override
DT view_data_type
The data type, including dimension information.
void make_constant(Teuchos::any &result, double const &value) override
void single_single_binary_op(BinaryOpCode code, Teuchos::any &result, Teuchos::any &left, Teuchos::any &right) override
void single_many_binary_op(BinaryOpCode code, Teuchos::any &result, Teuchos::any &left, Teuchos::any &right) override
void many_many_ternary_op(Teuchos::any &result, Teuchos::any &cond, Teuchos::any &left, Teuchos::any &right) override
void set(std::string const &name, bool value)
Assign a boolean value to a variable symbol.
void inspect_arg(Teuchos::any const &arg, bool &is_many, bool &is_bool) override
void many_single_ternary_op(Teuchos::any &result, Teuchos::any &cond, Teuchos::any &left, Teuchos::any &right) override
Kokkos::View< typename RebindDataType< view_data_type, scalar_type const >::type, VP ... > const_view_type
One scalar for each evaluation point, read-only.
void single_neg_op(Teuchos::any &result, Teuchos::any &right) override
Kokkos::View< scalar_type const, VP ... > const_single_view_type
One scalar (same for all evaluation points), read-only.
void single_many_ternary_op(Teuchos::any &result, Teuchos::any &cond, Teuchos::any &left, Teuchos::any &right) override
Kokkos::View< bool, VP ... > single_bool_view_type
One boolean (same for all evaluation points)
void many_single_binary_op(BinaryOpCode code, Teuchos::any &result, Teuchos::any &left, Teuchos::any &right) override
void set_cmath_functions(Eval< DT, VP ... > &eval)
Add support for functions such as sqrt(), sin(), and cos()
BinaryOpCode
Denotes the native binary operators in the Teuchos::MathExpr language.
typename RebindDataType< NestedDataType, NewScalarType >::type * type
typename RebindDataType< NestedDataType, NewScalarType >::type[N] type
typename RebindDataType< NestedDataType, NewScalarType >::type[] type
Rebinds a Kokkos::View data type to use a new scalar type.
NewScalarType type
The new data type, suitable as the first template argument to Kokkos::View.
Kokkos::View< typename RebindDataType< DT, NewScalarType >::type, VP ... > type
The new Kokkos::View type, whose scalar type is now NewScalarType.
Builds on RebindDataType, but acts directly on a Kokkos::View type.