IpplExpressions.h

Go to the documentation of this file.

//
// File IpplExpressions.h
//   Expression Templates classes.
//
#ifndef IPPL_EXPRESSIONS_H
#define IPPL_EXPRESSIONS_H
 
#include <type_traits>
 
namespace ippl {
    namespace detail {

        template <typename E, size_t N = sizeof(E)>
        struct Expression {
            constexpr static unsigned dim = E::dim;

            KOKKOS_INLINE_FUNCTION auto operator[](size_t i) const {
                return static_cast<const E&>(*this)[i];
            }
        };

        template <typename E, size_t N = sizeof(E)>
        struct CapturedExpression {
            constexpr static unsigned dim = E::dim;
 
            template <typename... Args>
            KOKKOS_INLINE_FUNCTION auto operator()(Args... args) const {
                static_assert(sizeof...(Args) == dim || dim == 0);
                return reinterpret_cast<const E&>(*this)(args...);
            }
 
            char buffer[N];
        };

        template <typename T>
        struct Scalar : public Expression<Scalar<T>, sizeof(T)> {
            typedef T value_type;
            constexpr static unsigned dim = 0;
 
            KOKKOS_FUNCTION
            Scalar(value_type val)
                : val_m(val) {}

            KOKKOS_INLINE_FUNCTION value_type operator[](size_t /*i*/) const { return val_m; }

            template <typename... Args>
            KOKKOS_INLINE_FUNCTION auto operator()(Args... /*args*/) const {
                return val_m;
            }
 
        private:
            value_type val_m;
        };
 
        template <typename T>
        struct isExpression : std::false_type {};
 
        template <typename T>
        struct isExpression<Scalar<T>> : std::true_type {};
 
    }  // namespace detail
}  // namespace ippl
 
#include "Expression/IpplOperations.h"
 
#endif