MathExpr.hpp

Go to the documentation of this file.

 
#ifndef OPENSWMM_MATHEXPR_HPP
#define OPENSWMM_MATHEXPR_HPP
 
#include <string>
#include <vector>
#include <functional>
 
namespace openswmm {
namespace mathexpr {
 
// ============================================================================
// Token types
// ============================================================================
 
enum class TokenType : int {
    NUMBER   = 0,
    VARIABLE = 1,
    ADD      = 2,
    SUB      = 3,
    MUL      = 4,
    DIV      = 5,
    POW      = 6,
    NEG      = 7,
    LPAREN   = 8,
    RPAREN   = 9,
    FUNC_ABS  = 10,
    FUNC_SGN  = 11,
    FUNC_SQRT = 12,
    FUNC_LOG  = 13,
    FUNC_EXP  = 14,
    FUNC_SIN  = 15,
    FUNC_COS  = 16,
    FUNC_TAN  = 17,
    FUNC_ASIN = 18,
    FUNC_ACOS = 19,
    FUNC_ATAN = 20,
    FUNC_STEP = 21,
    FUNC_MIN  = 22,
    FUNC_MAX  = 23,
    COMMA     = 24
};
 
struct Token {
    TokenType type = TokenType::NUMBER;
    double    value = 0.0;      
    int       var_idx = -1;     
    std::string var_name;       
};
 
// ============================================================================
// Compiled expression (postfix token list)
// ============================================================================
 
struct Expression {
    std::vector<Token> postfix;  
    bool valid = false;
};
 
// ============================================================================
// Parser (Shunting-yard)
// ============================================================================
 
int parse(const std::string& expr_str, Expression& result);
 
// ============================================================================
// Evaluator
// ============================================================================
 
double evaluate(const Expression& expr,
                const std::function<double(const std::string&)>& var_lookup);
 
double evaluate(const Expression& expr, const double* vars, int n_vars);
 
} // namespace mathexpr
} // namespace openswmm
 
#endif // OPENSWMM_MATHEXPR_HPP