Treatment.hpp

Go to the documentation of this file.

 
#ifndef OPENSWMM_TREATMENT_HPP
#define OPENSWMM_TREATMENT_HPP
 
#include <string>
#include <vector>
 
namespace openswmm {
 
namespace treatment {
 
// ============================================================================
// Expression token types
// ============================================================================
 
enum class TokenType : int {
    NUMBER    = 0,
    VARIABLE  = 1,
    ADD       = 2,
    SUB       = 3,
    MUL       = 4,
    DIV       = 5,
    POW       = 6,
    NEG       = 7,   
    FUNC_EXP  = 8,
    FUNC_LOG  = 9,
    FUNC_SQRT = 10,
    FUNC_MIN  = 11,
    FUNC_MAX  = 12,
    FUNC_ABS  = 13,
    FUNC_SGN  = 14,
    FUNC_STEP = 15,  
    LPAREN    = 16,  
    RPAREN    = 17,  
    COMMA     = 18   
};
 
enum class TreatVar : int {
    C   = 0,   
    R   = 1,   
    DT  = 2,   
    HRT = 3,   
    Q   = 4,   
    V   = 5,   
    D   = 6    
};
 
// ============================================================================
// Expression token
// ============================================================================
 
struct Token {
    TokenType type = TokenType::NUMBER;
    double    value = 0.0;        
    TreatVar  var = TreatVar::C;  
};
 
// ============================================================================
// Treatment expression (parsed, ready to evaluate)
// ============================================================================
 
struct TreatExpr {
    std::vector<Token> tokens;   
    int pollutant_idx = -1;      
    bool is_removal = false;     
};
 
// ============================================================================
// Functions
// ============================================================================
 
int parse(const std::string& expr_str, TreatExpr& result);
 
double evaluate(const TreatExpr& expr, double c, double dt,
                double hrt, double q, double v, double d);
 
double applyTreatment(const TreatExpr& expr, double c_in, double dt,
                      double hrt, double q, double v, double d);
 
} // namespace treatment
} // namespace openswmm
 
#endif // OPENSWMM_TREATMENT_HPP