Panzer_ExprEval.hpp

Go to the documentation of this file.

// @HEADER
// *****************************************************************************
//           Panzer: A partial differential equation assembly
//       engine for strongly coupled complex multiphysics systems
//
// Copyright 2011 NTESS and the Panzer contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER

#ifndef PANZER_EXPR_EVAL_HPP
#define PANZER_EXPR_EVAL_HPP

#include <functional>
#include <map>
#include <type_traits>

#include <Teuchos_Reader.hpp>

#include <Kokkos_Core.hpp>

namespace panzer
{

namespace Expr
{

enum class BinaryOpCode {
  OR,
  AND,
  GT,
  LT,
  GEQ,
  LEQ,
  EQ,
  ADD,
  SUB,
  MUL,
  DIV,
  POW,
};

class EvalBase : public Teuchos::Reader {
 public:
  EvalBase();
  using Function = std::function<void(std::string const& name, Teuchos::any&, std::vector<Teuchos::any>& rhs)>;
  void set(std::string const& name, Function const& value);
  template <typename T>
  T const& get(std::string const& name) const {
    auto it = symbol_map.find(name);
    TEUCHOS_TEST_FOR_EXCEPTION(it == symbol_map.end(), std::logic_error,
        "EvalBase::get: \"" << name << "\" not found");
    return Teuchos::any_ref_cast<T>(it->second);
  }
 protected:
  std::map<std::string, Teuchos::any> symbol_map;

  void at_shift(Teuchos::any& result, int token, std::string& text) override;
  void at_reduce(Teuchos::any& result, int prod, std::vector<Teuchos::any>& rhs) override;
  void ternary_op(Teuchos::any& result, Teuchos::any& cond, Teuchos::any& left, Teuchos::any& right);
  void binary_op(BinaryOpCode code, Teuchos::any& result, Teuchos::any& left, Teuchos::any& right);
  void neg_op(Teuchos::any& result, Teuchos::any& right);
 protected:

  virtual void make_constant(Teuchos::any& result, double const& value) = 0;
  virtual void inspect_arg(Teuchos::any const& arg, bool& is_many, bool& is_bool) = 0;
  virtual void single_single_ternary_op(Teuchos::any& result, Teuchos::any& cond, Teuchos::any& left, 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 many_single_ternary_op(Teuchos::any& result, Teuchos::any& cond, 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 single_single_binary_op(BinaryOpCode code, Teuchos::any& result, Teuchos::any& left, Teuchos::any& right) = 0;
  virtual void single_many_binary_op(BinaryOpCode code, Teuchos::any& result, Teuchos::any& left, Teuchos::any& right) = 0;
  virtual void many_single_binary_op(BinaryOpCode code, Teuchos::any& result, 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;
  virtual void many_neg_op(Teuchos::any& result, Teuchos::any& right) = 0;
  virtual void single_neg_op(Teuchos::any& result, Teuchos::any& right) = 0;
};

template <typename DataType, typename NewScalarType>
struct RebindDataType {
  using type = NewScalarType;
};

template <typename NestedDataType, typename NewScalarType>
struct RebindDataType<NestedDataType*, NewScalarType> {
  using type = typename RebindDataType<NestedDataType, NewScalarType>::type *;
};

template <typename NestedDataType, typename NewScalarType>
struct RebindDataType<NestedDataType[], NewScalarType> {
  using type = typename RebindDataType<NestedDataType, NewScalarType>::type [];
};

template <typename NestedDataType, typename NewScalarType, size_t N>
struct RebindDataType<NestedDataType[N], NewScalarType> {
  using type = typename RebindDataType<NestedDataType, NewScalarType>::type [N];
};

template <typename ViewType, typename NewScalarType>
struct RebindViewType;

template <typename DT, typename NewScalarType, typename ... VP>
struct RebindViewType<Kokkos::View<DT, VP ...>, NewScalarType> {
  using type = Kokkos::View<typename RebindDataType<DT, NewScalarType>::type, VP ...>;
};

template <typename DT, typename ... VP>
class Eval : public EvalBase {
 public:
  using original_view_type = Kokkos::View<DT, VP ...>;
  using view_data_type = DT;
  using scalar_type = typename original_view_type::non_const_value_type;
  using const_view_type = Kokkos::View<typename RebindDataType<view_data_type, scalar_type const>::type, VP ...>;
  using const_bool_view_type = Kokkos::View<typename RebindDataType<view_data_type, bool const>::type, VP ...>;
  using single_view_type = Kokkos::View<scalar_type, VP ...>;
  using const_single_view_type = Kokkos::View<scalar_type const, VP ...>;
  using single_bool_view_type = Kokkos::View<bool, VP ...>;
  using const_single_bool_view_type = Kokkos::View<bool const, VP ...>;

  Eval();

  void set(std::string const& name, bool value);
  void set(std::string const& name, scalar_type const& value);
  void set(std::string const& name, const_view_type const& value);
 protected:
  void make_constant(Teuchos::any& result, double const& value) override;
  void inspect_arg(Teuchos::any const& arg, bool& is_many, bool& is_bool) override;
  void single_single_ternary_op(Teuchos::any& result, Teuchos::any& cond, Teuchos::any& left, Teuchos::any& right) override;
  void single_many_ternary_op(Teuchos::any& result, Teuchos::any& cond, Teuchos::any& left, Teuchos::any& right) override;
  void many_single_ternary_op(Teuchos::any& result, Teuchos::any& cond, 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 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_single_binary_op(BinaryOpCode code, Teuchos::any& result, Teuchos::any& left, Teuchos::any& right) override;
  void many_many_binary_op(BinaryOpCode code, Teuchos::any& result, Teuchos::any& left, Teuchos::any& right) override;
  void many_neg_op(Teuchos::any& result, Teuchos::any& right) override;
  void single_neg_op(Teuchos::any& result, Teuchos::any& right) override;
};

template <typename DT, typename ... VP>
void set_cmath_functions(Eval<DT, VP ...>& eval);

}} // end namespace panzer::Expr

#endif // PANZER_EXPR_EVAL_HPP