Merge branch 'dev' into RZgeometry

1 files changed, 139 insertions, 0 deletions

diff --git a/Source/Parser/wp_parser_c.h b/Source/Parser/wp_parser_c.h
new file mode 100644
index 000000000..d810bd685
--- /dev/null
+++ b/Source/Parser/wp_parser_c.h
@@ -0,0 +1,139 @@
+#ifndef WP_PARSER_C_H_
+#define WP_PARSER_C_H_
+
+#include "wp_parser_y.h"
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+    struct wp_parser* wp_c_parser_new (char const* function_body);
+
+#ifdef __cplusplus
+}
+#endif
+
+#ifdef __cplusplus
+
+#include <set>
+#include <string>
+
+inline
+double
+wp_ast_eval (struct wp_node* node)
+{
+    double result;
+
+    switch (node->type)
+    {
+    case WP_NUMBER:
+        result = ((struct wp_number*)node)->value;
+        break;
+    case WP_SYMBOL:
+        result = *(((struct wp_symbol*)node)->pointer);
+        break;
+    case WP_ADD:
+        result = wp_ast_eval(node->l) + wp_ast_eval(node->r);
+        break;
+    case WP_SUB:
+        result = wp_ast_eval(node->l) - wp_ast_eval(node->r);
+        break;
+    case WP_MUL:
+        result = wp_ast_eval(node->l) * wp_ast_eval(node->r);
+        break;
+    case WP_DIV:
+        result = wp_ast_eval(node->l) / wp_ast_eval(node->r);
+        break;
+    case WP_NEG:
+        result = -wp_ast_eval(node->l);
+        break;
+    case WP_F1:
+        result = wp_call_f1(((struct wp_f1*)node)->ftype,
+                wp_ast_eval(((struct wp_f1*)node)->l));
+        break;
+    case WP_F2:
+        result = wp_call_f2(((struct wp_f2*)node)->ftype,
+                wp_ast_eval(((struct wp_f2*)node)->l),
+                wp_ast_eval(((struct wp_f2*)node)->r));
+        break;
+    case WP_ADD_VP:
+        result = node->lvp.v + *(node->rp);
+        break;
+    case WP_ADD_PP:
+        result = *(node->lvp.p) + *(node->rp);
+        break;
+    case WP_SUB_VP:
+        result = node->lvp.v - *(node->rp);
+        break;
+    case WP_SUB_PP:
+        result = *(node->lvp.p) - *(node->rp);
+        break;
+    case WP_MUL_VP:
+        result = node->lvp.v * *(node->rp);
+        break;
+    case WP_MUL_PP:
+        result = *(node->lvp.p) * *(node->rp);
+        break;
+    case WP_DIV_VP:
+        result = node->lvp.v / *(node->rp);
+        break;
+    case WP_DIV_PP:
+        result = *(node->lvp.p) / *(node->rp);
+        break;
+    case WP_NEG_P:
+        result = -*(node->lvp.p);
+        break;
+    default:
+        yyerror("wp_ast_eval: unknown node type %d\n", node->type);
+    }
+
+    return result;
+}
+
+inline
+void
+wp_ast_get_symbols (struct wp_node* node, std::set<std::string>& symbols)
+{
+    switch (node->type)
+    {
+    case WP_NUMBER:
+        break;
+    case WP_SYMBOL:
+        symbols.emplace(((struct wp_symbol*)node)->name);
+        break;
+    case WP_ADD:
+    case WP_SUB:
+    case WP_MUL:
+    case WP_DIV:
+    case WP_ADD_PP:
+    case WP_SUB_PP:
+    case WP_MUL_PP:
+    case WP_DIV_PP:
+        wp_ast_get_symbols(node->l, symbols);
+        wp_ast_get_symbols(node->r, symbols);
+        break;
+    case WP_NEG:
+    case WP_NEG_P:
+        wp_ast_get_symbols(node->l, symbols);
+        break;
+    case WP_F1:
+        wp_ast_get_symbols(((struct wp_f1*)node)->l, symbols);
+        break;
+    case WP_F2:
+        wp_ast_get_symbols(((struct wp_f2*)node)->l, symbols);
+        wp_ast_get_symbols(((struct wp_f2*)node)->r, symbols);
+        break;
+    case WP_ADD_VP:
+    case WP_SUB_VP:
+    case WP_MUL_VP:
+    case WP_DIV_VP:
+        wp_ast_get_symbols(node->r, symbols);
+        break;
+    default:
+        yyerror("wp_ast_get_symbols: unknown node type %d\n", node->type);
+    }
+}
+
+#endif
+
+#endif