Expression.hpp

Go to the documentation of this file.

#ifndef OASIS_EXPRESSION_HPP
#define OASIS_EXPRESSION_HPP
 
#include <any>
#include <cstdint>
#include <memory>
#include <optional>
#include <vector>
 
#include "Concepts.hpp"
 
namespace Oasis {
 
class Visitor;
 
enum class ExpressionType {
    None,
    Real,
    Imaginary,
    Variable,
    Add,
    Subtract,
    Multiply,
    Divide,
    Exponent,
    Log,
    Integral,
    Limit,
    Derivative,
    Negate,
    Sqrt,
    Matrix,
    Pi,
    EulerNumber,
    Magnitude
};
 
enum ExpressionCategory : uint32_t {
    None = 0,
    Associative = 1,
    Commutative = 1 << 1,
    BinExp = 1 << 2,
    UnExp = 1 << 3,
};
 
class Expression {
public:
    [[nodiscard]] virtual auto Copy() const -> std::unique_ptr<Expression> = 0;
 
    [[nodiscard]] virtual auto Differentiate(const Expression&) const -> std::unique_ptr<Expression>;
 
    [[nodiscard]] virtual auto Equals(const Expression& other) const -> bool = 0;
 
    auto FindZeros() const -> std::vector<std::unique_ptr<Expression>>;
 
    [[nodiscard]] virtual auto GetCategory() const -> uint32_t;
 
    [[nodiscard]] virtual auto GetType() const -> ExpressionType;
 
    [[nodiscard]] virtual auto Generalize() const -> std::unique_ptr<Expression>;
 
    [[nodiscard]] virtual auto Integrate(const Expression&) const -> std::unique_ptr<Expression>;
 
    [[nodiscard]] virtual auto IntegrateWithBounds(const Expression&, const Expression&, const Expression&) -> std::unique_ptr<Expression>;
 
    template <IExpression T>
    [[nodiscard]] bool Is() const
    {
        return GetType() == T::GetStaticType();
    }
 
    template <template <typename> typename T>
        requires(DerivedFromUnaryExpression<T<Expression>> && !DerivedFromBinaryExpression<T<Expression>>)
    [[nodiscard]] bool Is() const
    {
        return GetType() == T<Expression>::GetStaticType();
    }
 
    template <template <typename, typename> typename T>
        requires DerivedFromBinaryExpression<T<Expression, Expression>>
    [[nodiscard]] bool Is() const
    {
        return GetType() == T<Expression, Expression>::GetStaticType();
    }
 
    [[nodiscard]] virtual auto Simplify() const -> std::unique_ptr<Expression>;
 
    [[nodiscard]] virtual auto StructurallyEquivalent(const Expression& other) const -> bool = 0;
 
    [[nodiscard]] virtual auto Substitute(const Expression& var, const Expression& val) -> std::unique_ptr<Expression> = 0;
 
    template <IVisitor T>
    std::optional<typename T::RetT> Accept(T& visitor) const;
 
    virtual ~Expression() = default;
 
protected:
    virtual std::any AcceptInternal(Visitor& visitor) const = 0;
};
 
template <IVisitor T>
std::optional<typename T::RetT> Expression::Accept(T& visitor) const
{
    try {
        return std::any_cast<typename T::RetT>(this->AcceptInternal(visitor));
    } catch (std::bad_any_cast&) {
        return std::nullopt;
    }
}
 
#define EXPRESSION_TYPE(type)                       \
    auto GetType() const -> ExpressionType override \
    {                                               \
        return ExpressionType::type;                \
    }                                               \
                                                    \
    static auto GetStaticType() -> ExpressionType   \
    {                                               \
        return ExpressionType::type;                \
    }
 
#define EXPRESSION_CATEGORY(category)                     \
    auto GetCategory() const -> uint32_t override         \
    {                                                     \
        return category;                                  \
    }                                                     \
                                                          \
    constexpr static auto GetStaticCategory() -> uint32_t \
    {                                                     \
        return category;                                  \
    }
 
} // namespace Oasis
 
std::unique_ptr<Oasis::Expression> operator+(const std::unique_ptr<Oasis::Expression>& lhs, const std::unique_ptr<Oasis::Expression>& rhs);
std::unique_ptr<Oasis::Expression> operator-(const std::unique_ptr<Oasis::Expression>& lhs, const std::unique_ptr<Oasis::Expression>& rhs);
std::unique_ptr<Oasis::Expression> operator*(const std::unique_ptr<Oasis::Expression>& lhs, const std::unique_ptr<Oasis::Expression>& rhs);
std::unique_ptr<Oasis::Expression> operator/(const std::unique_ptr<Oasis::Expression>& lhs, const std::unique_ptr<Oasis::Expression>& rhs);
 
#endif // OASIS_EXPRESSION_HPP