github-mirrors/thirdparty-tinyexpr
1
0
Fork 0
mirror of https://github.com/eledio-devices/thirdparty-tinyexpr.git synced 2025-10-31 00:32:38 +01:00
Code Issues Packages Projects Releases Wiki Activity

Got optimize working. Added TE_FLAG_PURE.

Browse Source
This commit is contained in:
Lewis Van Winkle
2016-02-26 17:35:39 -06:00
parent 7550e4e309
commit 8e7c788313
4 changed files with 108 additions and 67 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
minctest.h
View File

@@ -118,7 +118,8 @@ static int lfails = 0;
/* Assert two floats are equal (Within LTEST_FLOAT_TOLERANCE). */ /* Assert two floats are equal (Within LTEST_FLOAT_TOLERANCE). */
#define lfequal(a, b) do {\ #define lfequal(a, b) do {\
++ltests;\ ++ltests;\
if (fabs((double)(a)-(double)(b)) > LTEST_FLOAT_TOLERANCE) {\ const double __LF_COMPARE = fabs((double)(a)-(double)(b));\
if (__LF_COMPARE > LTEST_FLOAT_TOLERANCE || (__LF_COMPARE != __LF_COMPARE)) {\
++lfails;\ ++lfails;\
printf("%s:%d (%f != %f)\n", __FILE__, __LINE__, (double)(a), (double)(b));\ printf("%s:%d (%f != %f)\n", __FILE__, __LINE__, (double)(a), (double)(b));\
}} while (0) }} while (0)

36
test.c
View File

@@ -239,13 +239,15 @@ void test_variables() {
lok(expr3); lok(expr3);
lok(!err); lok(!err);
te_expr *expr4 = te_compile("test", lookup, 3, &err); te_expr *expr4 = te_compile("test+5", lookup, 3, &err);
lok(expr4); lok(expr4);
lok(!err); lok(!err);
for (y = 2; y < 3; ++y) { for (y = 2; y < 3; ++y) {
for (x = 0; x < 5; ++x) { for (x = 0; x < 5; ++x) {
double ev = te_eval(expr1); double ev;
ev = te_eval(expr1);
lfequal(ev, cos(x) + sin(y)); lfequal(ev, cos(x) + sin(y));
ev = te_eval(expr2); ev = te_eval(expr2);
@@ -256,7 +258,7 @@ void test_variables() {
test = x; test = x;
ev = te_eval(expr4); ev = te_eval(expr4);
lfequal(ev, x); lfequal(ev, x+5);
} }
} }
@@ -287,6 +289,7 @@ void test_variables() {
#define cross_check(a, b) do {\ #define cross_check(a, b) do {\
if ((b)!=(b)) break;\
expr = te_compile((a), lookup, 2, &err);\ expr = te_compile((a), lookup, 2, &err);\
lfequal(te_eval(expr), (b));\ lfequal(te_eval(expr), (b));\
lok(!err);\ lok(!err);\
@@ -486,6 +489,32 @@ void test_closure() {
} }
} }
void test_optimize() {
test_case cases[] = {
{"5+5", 10},
{"pow(2,2)", 4},
{"sqrt 100", 10},
};
int i;
for (i = 0; i < sizeof(cases) / sizeof(test_case); ++i) {
const char *expr = cases[i].expr;
const double answer = cases[i].answer;
int err;
te_expr *ex = te_compile(expr, 0, 0, &err);
lok(ex);
/* The answer should be know without
* even running eval. */
lfequal(ex->value, answer);
lfequal(te_eval(ex), answer);
te_free(ex);
}
}
int main(int argc, char *argv[]) int main(int argc, char *argv[])
{ {
@@ -496,6 +525,7 @@ int main(int argc, char *argv[])
lrun("Functions", test_functions); lrun("Functions", test_functions);
lrun("Dynamic", test_dynamic); lrun("Dynamic", test_dynamic);
lrun("Closure", test_closure); lrun("Closure", test_closure);
lrun("Optimize", test_optimize);
lresults(); lresults();
return lfails != 0; return lfails != 0;

126
tinyexpr.c
View File

@@ -51,12 +51,12 @@ typedef struct state {
} state; } state;
#define TYPE_MASK(TYPE) ((TYPE)&0x0000001F)
#define IS_FUNCTION(TYPE) ((TYPE) >= TE_FUNCTION0 && (TYPE) <= TE_FUNCTION7) #define IS_PURE(TYPE) (((TYPE) & TE_FLAG_PURE) != 0)
#define IS_CLOSURE(TYPE) ((TYPE) >= TE_CLOSURE0 && (TYPE) <= TE_CLOSURE7) #define IS_FUNCTION(TYPE) (((TYPE) & TE_FUNCTION0) != 0)
#define ARITY_CLO(TYPE) ( IS_CLOSURE(TYPE) ? ((TYPE)-TE_CLOSURE0) : 0 ) #define IS_CLOSURE(TYPE) (((TYPE) & TE_CLOSURE0) != 0)
#define ARITY(TYPE) ( IS_FUNCTION(TYPE) ? ((TYPE)-TE_FUNCTION0) : ARITY_CLO(TYPE) ) #define ARITY(TYPE) ( ((TYPE) & (TE_FUNCTION0 | TE_CLOSURE0)) ? ((TYPE) & 0x00000007) : 0 )
#define NEW_EXPR(type, ...) new_expr((type), (const te_expr*[]){__VA_ARGS__}) #define NEW_EXPR(type, ...) new_expr((type), (const te_expr*[]){__VA_ARGS__})
static te_expr *new_expr(const int type, const te_expr *parameters[]) { static te_expr *new_expr(const int type, const te_expr *parameters[]) {
@@ -73,9 +73,9 @@ static te_expr *new_expr(const int type, const te_expr *parameters[]) {
} }
void te_free(te_expr *n) { void te_free_parameters(te_expr *n) {
if (!n) return; if (!n) return;
switch (n->type) { switch (TYPE_MASK(n->type)) {
case TE_FUNCTION7: case TE_CLOSURE7: te_free(n->parameters[6]); case TE_FUNCTION7: case TE_CLOSURE7: te_free(n->parameters[6]);
case TE_FUNCTION6: case TE_CLOSURE6: te_free(n->parameters[5]); case TE_FUNCTION6: case TE_CLOSURE6: te_free(n->parameters[5]);
case TE_FUNCTION5: case TE_CLOSURE5: te_free(n->parameters[4]); case TE_FUNCTION5: case TE_CLOSURE5: te_free(n->parameters[4]);
@@ -84,6 +84,12 @@ void te_free(te_expr *n) {
case TE_FUNCTION2: case TE_CLOSURE2: te_free(n->parameters[1]); case TE_FUNCTION2: case TE_CLOSURE2: te_free(n->parameters[1]);
case TE_FUNCTION1: case TE_CLOSURE1: te_free(n->parameters[0]); case TE_FUNCTION1: case TE_CLOSURE1: te_free(n->parameters[0]);
} }
}
void te_free(te_expr *n) {
if (!n) return;
te_free_parameters(n);
free(n); free(n);
} }
@@ -93,26 +99,26 @@ static const double e = 2.71828182845904523536;
static const te_variable functions[] = { static const te_variable functions[] = {
/* must be in alphabetical order */ /* must be in alphabetical order */
{"abs", fabs, TE_FUNCTION1}, {"abs", fabs, TE_FUNCTION1 | TE_FLAG_PURE},
{"acos", acos, TE_FUNCTION1}, {"acos", acos, TE_FUNCTION1 | TE_FLAG_PURE},
{"asin", asin, TE_FUNCTION1}, {"asin", asin, TE_FUNCTION1 | TE_FLAG_PURE},
{"atan", atan, TE_FUNCTION1}, {"atan", atan, TE_FUNCTION1 | TE_FLAG_PURE},
{"atan2", atan2, TE_FUNCTION2}, {"atan2", atan2, TE_FUNCTION2 | TE_FLAG_PURE},
{"ceil", ceil, TE_FUNCTION1}, {"ceil", ceil, TE_FUNCTION1 | TE_FLAG_PURE},
{"cos", cos, TE_FUNCTION1}, {"cos", cos, TE_FUNCTION1 | TE_FLAG_PURE},
{"cosh", cosh, TE_FUNCTION1}, {"cosh", cosh, TE_FUNCTION1 | TE_FLAG_PURE},
{"e", &e, TE_VARIABLE}, {"e", &e, TE_VARIABLE},
{"exp", exp, TE_FUNCTION1}, {"exp", exp, TE_FUNCTION1 | TE_FLAG_PURE},
{"floor", floor, TE_FUNCTION1}, {"floor", floor, TE_FUNCTION1 | TE_FLAG_PURE},
{"ln", log, TE_FUNCTION1}, {"ln", log, TE_FUNCTION1 | TE_FLAG_PURE},
{"log", log10, TE_FUNCTION1}, {"log", log10, TE_FUNCTION1 | TE_FLAG_PURE},
{"pi", &pi, TE_VARIABLE}, {"pi", &pi, TE_VARIABLE},
{"pow", pow, TE_FUNCTION2}, {"pow", pow, TE_FUNCTION2 | TE_FLAG_PURE},
{"sin", sin, TE_FUNCTION1}, {"sin", sin, TE_FUNCTION1 | TE_FLAG_PURE},
{"sinh", sinh, TE_FUNCTION1}, {"sinh", sinh, TE_FUNCTION1 | TE_FLAG_PURE},
{"sqrt", sqrt, TE_FUNCTION1}, {"sqrt", sqrt, TE_FUNCTION1 | TE_FLAG_PURE},
{"tan", tan, TE_FUNCTION1}, {"tan", tan, TE_FUNCTION1 | TE_FLAG_PURE},
{"tanh", tanh, TE_FUNCTION1}, {"tanh", tanh, TE_FUNCTION1 | TE_FLAG_PURE},
{0} {0}
}; };
@@ -185,7 +191,7 @@ void next_token(state *s) {
if (!var) { if (!var) {
s->type = TOK_ERROR; s->type = TOK_ERROR;
} else { } else {
switch(var->type) switch(TYPE_MASK(var->type))
{ {
case TE_VARIABLE: case TE_VARIABLE:
s->type = TOK_VARIABLE; s->type = TOK_VARIABLE;
@@ -234,7 +240,7 @@ static te_expr *base(state *s) {
te_expr *ret; te_expr *ret;
int arity; int arity;
switch (s->type) { switch (TYPE_MASK(s->type)) {
case TOK_NUMBER: case TOK_NUMBER:
ret = new_expr(TE_CONSTANT, 0); ret = new_expr(TE_CONSTANT, 0);
ret->value = s->value; ret->value = s->value;
@@ -337,7 +343,7 @@ static te_expr *power(state *s) {
if (sign == 1) { if (sign == 1) {
ret = base(s); ret = base(s);
} else { } else {
ret = NEW_EXPR(TE_FUNCTION1, base(s)); ret = NEW_EXPR(TE_FUNCTION1 | TE_FLAG_PURE, base(s));
ret->function = negate; ret->function = negate;
} }
@@ -352,7 +358,7 @@ static te_expr *factor(state *s) {
while (s->type == TOK_INFIX && (s->function == pow)) { while (s->type == TOK_INFIX && (s->function == pow)) {
te_fun2 t = s->function; te_fun2 t = s->function;
next_token(s); next_token(s);
ret = NEW_EXPR(TE_FUNCTION2, ret, power(s)); ret = NEW_EXPR(TE_FUNCTION2 | TE_FLAG_PURE, ret, power(s));
ret->function = t; ret->function = t;
} }
@@ -367,7 +373,7 @@ static te_expr *term(state *s) {
while (s->type == TOK_INFIX && (s->function == mul || s->function == divide || s->function == fmod)) { while (s->type == TOK_INFIX && (s->function == mul || s->function == divide || s->function == fmod)) {
te_fun2 t = s->function; te_fun2 t = s->function;
next_token(s); next_token(s);
ret = NEW_EXPR(TE_FUNCTION2, ret, factor(s)); ret = NEW_EXPR(TE_FUNCTION2 | TE_FLAG_PURE, ret, factor(s));
ret->function = t; ret->function = t;
} }
@@ -382,7 +388,7 @@ static te_expr *expr(state *s) {
while (s->type == TOK_INFIX && (s->function == add || s->function == sub)) { while (s->type == TOK_INFIX && (s->function == add || s->function == sub)) {
te_fun2 t = s->function; te_fun2 t = s->function;
next_token(s); next_token(s);
ret = NEW_EXPR(TE_FUNCTION2, ret, term(s)); ret = NEW_EXPR(TE_FUNCTION2 | TE_FLAG_PURE, ret, term(s));
ret->function = t; ret->function = t;
} }
@@ -396,7 +402,7 @@ static te_expr *list(state *s) {
while (s->type == TOK_SEP) { while (s->type == TOK_SEP) {
next_token(s); next_token(s);
ret = NEW_EXPR(TE_FUNCTION2, ret, expr(s)); ret = NEW_EXPR(TE_FUNCTION2 | TE_FLAG_PURE, ret, expr(s));
ret->function = comma; ret->function = comma;
} }
@@ -404,13 +410,14 @@ static te_expr *list(state *s) {
} }
double te_eval(const te_expr *n) {
if (!n) return 0.0/0.0;
#define TE_FUN(...) ((double(*)(__VA_ARGS__))n->function) #define TE_FUN(...) ((double(*)(__VA_ARGS__))n->function)
#define M(e) te_eval(n->parameters[e]) #define M(e) te_eval(n->parameters[e])
switch(n->type) {
double te_eval(const te_expr *n) {
if (!n) return 0.0/0.0;
switch(TYPE_MASK(n->type)) {
case TE_CONSTANT: return n->value; case TE_CONSTANT: return n->value;
case TE_VARIABLE: return *n->bound; case TE_VARIABLE: return *n->bound;
@@ -445,38 +452,34 @@ double te_eval(const te_expr *n) {
default: return 0.0/0.0; default: return 0.0/0.0;
} }
#undef TE_FUN
#undef M
} }
#undef TE_FUN
#undef M
static void optimize(te_expr *n) { static void optimize(te_expr *n) {
/* Evaluates as much as possible. */ /* Evaluates as much as possible. */
if (n->type == TE_CONSTANT) return;
if (n->type == TE_VARIABLE) return;
/* /* Only optimize out functions flagged as pure. */
if (n->bound) return; if (IS_PURE(n->type)) {
const int arity = ARITY(n->type);
if (n->left) optimize(n->left); int known = 1;
if (n->right) optimize(n->right); int i;
for (i = 0; i < arity; ++i) {
if (n->left && n->right) optimize(n->parameters[i]);
{ if (((te_expr*)(n->parameters[i]))->type != TE_CONSTANT) {
if (n->left->left == 0 && n->left->right == 0 && n->right->left == 0 && n->right->right == 0 && n->right->bound == 0 && n->left->bound == 0) known = 0;
{ }
const double r = n->f2(n->left->value, n->right->value);
free(n->left); free(n->right);
n->left = 0; n->right = 0;
n->value = r;
} }
} else if (n->left && !n->right) { if (known) {
if (n->left->left == 0 && n->left->right == 0 && n->left->bound == 0) { const double value = te_eval(n);
const double r = n->f1(n->left->value); te_free_parameters(n);
free(n->left); n->type = TE_CONSTANT;
n->left = 0; n->value = value;
n->value = r;
} }
} }
*/
} }
@@ -520,11 +523,14 @@ static void pn (const te_expr *n, int depth) {
int i, arity; int i, arity;
printf("%*s", depth, ""); printf("%*s", depth, "");
switch(n->type) { switch(TYPE_MASK(n->type)) {
case TE_CONSTANT: printf("%f\n", n->value); break; case TE_CONSTANT: printf("%f\n", n->value); break;
case TE_VARIABLE: printf("bound %p\n", n->bound); break; case TE_VARIABLE: printf("bound %p\n", n->bound); break;
case TE_FUNCTION0: case TE_FUNCTION1: case TE_FUNCTION2: case TE_FUNCTION3: case TE_FUNCTION0: case TE_FUNCTION1: case TE_FUNCTION2: case TE_FUNCTION3:
case TE_FUNCTION4: case TE_FUNCTION5: case TE_FUNCTION6: case TE_FUNCTION7: case TE_FUNCTION4: case TE_FUNCTION5: case TE_FUNCTION6: case TE_FUNCTION7:
case TE_CLOSURE0: case TE_CLOSURE1: case TE_CLOSURE2: case TE_CLOSURE3:
case TE_CLOSURE4: case TE_CLOSURE5: case TE_CLOSURE6: case TE_CLOSURE7:
arity = ARITY(n->type); arity = ARITY(n->type);
printf("f%d", arity); printf("f%d", arity);
for(i = 0; i < arity; i++) { for(i = 0; i < arity; i++) {

10
tinyexpr.h
View File

@@ -41,10 +41,14 @@ typedef struct te_expr {
enum { enum {
TE_VARIABLE = 0, TE_VARIABLE = 0,
TE_FUNCTION0 = 2, TE_FUNCTION1, TE_FUNCTION2, TE_FUNCTION3,
TE_FUNCTION0 = 8, TE_FUNCTION1, TE_FUNCTION2, TE_FUNCTION3,
TE_FUNCTION4, TE_FUNCTION5, TE_FUNCTION6, TE_FUNCTION7, TE_FUNCTION4, TE_FUNCTION5, TE_FUNCTION6, TE_FUNCTION7,
TE_CLOSURE0, TE_CLOSURE1, TE_CLOSURE2, TE_CLOSURE3,
TE_CLOSURE4, TE_CLOSURE5, TE_CLOSURE6, TE_CLOSURE7 TE_CLOSURE0 = 16, TE_CLOSURE1, TE_CLOSURE2, TE_CLOSURE3,
TE_CLOSURE4, TE_CLOSURE5, TE_CLOSURE6, TE_CLOSURE7,
TE_FLAG_PURE = 32
}; };
typedef struct te_variable { typedef struct te_variable {