mathexpr.hh revision 11531
111527Sdavid.guillen@arm.com/* 211527Sdavid.guillen@arm.com * Copyright (c) 2016 ARM Limited 311527Sdavid.guillen@arm.com * All rights reserved 411527Sdavid.guillen@arm.com * 511527Sdavid.guillen@arm.com * The license below extends only to copyright in the software and shall 611527Sdavid.guillen@arm.com * not be construed as granting a license to any other intellectual 711527Sdavid.guillen@arm.com * property including but not limited to intellectual property relating 811527Sdavid.guillen@arm.com * to a hardware implementation of the functionality of the software 911527Sdavid.guillen@arm.com * licensed hereunder. You may use the software subject to the license 1011527Sdavid.guillen@arm.com * terms below provided that you ensure that this notice is replicated 1111527Sdavid.guillen@arm.com * unmodified and in its entirety in all distributions of the software, 1211527Sdavid.guillen@arm.com * modified or unmodified, in source code or in binary form. 1311527Sdavid.guillen@arm.com * 1411527Sdavid.guillen@arm.com * Redistribution and use in source and binary forms, with or without 1511527Sdavid.guillen@arm.com * modification, are permitted provided that the following conditions are 1611527Sdavid.guillen@arm.com * met: redistributions of source code must retain the above copyright 1711527Sdavid.guillen@arm.com * notice, this list of conditions and the following disclaimer; 1811527Sdavid.guillen@arm.com * redistributions in binary form must reproduce the above copyright 1911527Sdavid.guillen@arm.com * notice, this list of conditions and the following disclaimer in the 2011527Sdavid.guillen@arm.com * documentation and/or other materials provided with the distribution; 2111527Sdavid.guillen@arm.com * neither the name of the copyright holders nor the names of its 2211527Sdavid.guillen@arm.com * contributors may be used to endorse or promote products derived from 2311527Sdavid.guillen@arm.com * this software without specific prior written permission. 2411527Sdavid.guillen@arm.com * 2511527Sdavid.guillen@arm.com * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 2611527Sdavid.guillen@arm.com * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 2711527Sdavid.guillen@arm.com * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 2811527Sdavid.guillen@arm.com * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 2911527Sdavid.guillen@arm.com * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 3011527Sdavid.guillen@arm.com * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 3111527Sdavid.guillen@arm.com * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 3211527Sdavid.guillen@arm.com * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 3311527Sdavid.guillen@arm.com * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 3411527Sdavid.guillen@arm.com * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 3511527Sdavid.guillen@arm.com * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 3611527Sdavid.guillen@arm.com * 3711527Sdavid.guillen@arm.com * Authors: David Guillen Fandos 3811527Sdavid.guillen@arm.com */ 3911527Sdavid.guillen@arm.com 4011527Sdavid.guillen@arm.com#ifndef __SIM_MATHEXPR_HH__ 4111527Sdavid.guillen@arm.com#define __SIM_MATHEXPR_HH__ 4211527Sdavid.guillen@arm.com 4311527Sdavid.guillen@arm.com#include <algorithm> 4411531Ssergei.trofimov@arm.com#include <functional> 4511527Sdavid.guillen@arm.com#include <string> 4611527Sdavid.guillen@arm.com 4711527Sdavid.guillen@arm.comclass MathExpr { 4811527Sdavid.guillen@arm.com public: 4911527Sdavid.guillen@arm.com 5011527Sdavid.guillen@arm.com MathExpr(std::string expr); 5111527Sdavid.guillen@arm.com 5211527Sdavid.guillen@arm.com typedef std::function<double(std::string)> EvalCallback; 5311527Sdavid.guillen@arm.com 5411527Sdavid.guillen@arm.com /** 5511527Sdavid.guillen@arm.com * Prints an ASCII representation of the expression tree 5611527Sdavid.guillen@arm.com * 5711527Sdavid.guillen@arm.com * @return A string containing the ASCII representation of the expression 5811527Sdavid.guillen@arm.com */ 5911527Sdavid.guillen@arm.com std::string toStr() const { return toStr(root, ""); } 6011527Sdavid.guillen@arm.com 6111527Sdavid.guillen@arm.com /** 6211527Sdavid.guillen@arm.com * Evaluates the expression 6311527Sdavid.guillen@arm.com * 6411527Sdavid.guillen@arm.com * @param fn A callback funcion to evaluate variables 6511527Sdavid.guillen@arm.com * 6611527Sdavid.guillen@arm.com * @return The value for this expression 6711527Sdavid.guillen@arm.com */ 6811527Sdavid.guillen@arm.com double eval(EvalCallback fn) const { return eval(root, fn); } 6911527Sdavid.guillen@arm.com 7011527Sdavid.guillen@arm.com private: 7111527Sdavid.guillen@arm.com enum Operator { 7211527Sdavid.guillen@arm.com bAdd, bSub, bMul, bDiv, bPow, uNeg, sValue, sVariable, nInvalid 7311527Sdavid.guillen@arm.com }; 7411527Sdavid.guillen@arm.com 7511527Sdavid.guillen@arm.com // Match operators 7611527Sdavid.guillen@arm.com const int MAX_PRIO = 4; 7711527Sdavid.guillen@arm.com typedef double (*binOp)(double, double); 7811527Sdavid.guillen@arm.com struct OpSearch { 7911527Sdavid.guillen@arm.com bool binary; 8011527Sdavid.guillen@arm.com Operator op; 8111527Sdavid.guillen@arm.com int priority; 8211527Sdavid.guillen@arm.com char c; 8311527Sdavid.guillen@arm.com binOp fn; 8411527Sdavid.guillen@arm.com }; 8511527Sdavid.guillen@arm.com 8611527Sdavid.guillen@arm.com /** Operator list */ 8711527Sdavid.guillen@arm.com std::array<OpSearch, uNeg + 1> ops; 8811527Sdavid.guillen@arm.com 8911527Sdavid.guillen@arm.com class Node { 9011527Sdavid.guillen@arm.com public: 9111527Sdavid.guillen@arm.com Node() : op(nInvalid), l(0), r(0), value(0) {} 9211527Sdavid.guillen@arm.com std::string toStr() const { 9311527Sdavid.guillen@arm.com const char opStr[] = {'+', '-', '*', '/', '^', '-'}; 9411527Sdavid.guillen@arm.com switch (op) { 9511527Sdavid.guillen@arm.com case nInvalid: 9611527Sdavid.guillen@arm.com return "INVALID"; 9711527Sdavid.guillen@arm.com case sVariable: 9811527Sdavid.guillen@arm.com return variable; 9911527Sdavid.guillen@arm.com case sValue: 10011527Sdavid.guillen@arm.com return std::to_string(value); 10111527Sdavid.guillen@arm.com default: 10211527Sdavid.guillen@arm.com return std::string(1, opStr[op]); 10311527Sdavid.guillen@arm.com }; 10411527Sdavid.guillen@arm.com } 10511527Sdavid.guillen@arm.com 10611527Sdavid.guillen@arm.com Operator op; 10711527Sdavid.guillen@arm.com Node *l, *r; 10811527Sdavid.guillen@arm.com double value; 10911527Sdavid.guillen@arm.com std::string variable; 11011527Sdavid.guillen@arm.com }; 11111527Sdavid.guillen@arm.com 11211527Sdavid.guillen@arm.com /** Root node */ 11311527Sdavid.guillen@arm.com Node * root; 11411527Sdavid.guillen@arm.com 11511527Sdavid.guillen@arm.com /** Parse and create nodes from string */ 11611527Sdavid.guillen@arm.com Node *parse(std::string expr); 11711527Sdavid.guillen@arm.com 11811527Sdavid.guillen@arm.com /** Print tree as string */ 11911527Sdavid.guillen@arm.com std::string toStr(Node *n, std::string prefix) const; 12011527Sdavid.guillen@arm.com 12111527Sdavid.guillen@arm.com /** Eval a node */ 12211527Sdavid.guillen@arm.com double eval(const Node *n, EvalCallback fn) const; 12311527Sdavid.guillen@arm.com}; 12411527Sdavid.guillen@arm.com 12511527Sdavid.guillen@arm.com#endif 12611527Sdavid.guillen@arm.com 12711527Sdavid.guillen@arm.com 128