change prompt
[calc.git] / parser.c
index 0caa750c65020c7f490c3da51ed64b7a82e8a0b5..cd0d6bfe6b4e33be2d574408ffe8664405d18b53 100644 (file)
--- a/parser.c
+++ b/parser.c
@@ -1,4 +1,5 @@
 #include <malloc.h>
+#include <math.h>
 #include <stdlib.h>
 
 #include "debug.h"
@@ -44,29 +45,72 @@ element_t *newelement (func_t function, int nbops, int prio)
     return new;
 }
 
+/* desallocate element */
+
+void delelement (element_t *root)
+{
+    int i;
+    if ((root != NULL) && (root != ERROR_OP)) {
+        for (i = 0; i < root->nbops; i++) {
+            if ((root->ops[i] != NULL) && (root->ops[i] != ERROR_OP)) {
+                delelement (root->ops[i]);
+            }
+        }
+        free (root);
+    }
+}
+
 /* functions */
 
-#define NB_OPERATORS 5
+#define NB_OPERATORS 6
 
 keyword_t operators[NB_OPERATORS] = {
     { "+\t", Add, 2, 1, 1},
     { "-\t", Sub, 2, 1, 1},
     { "*",   Mul, 2, 1, 2},
     { "/",   Div, 2, 1, 2},
-    { "^",   Pow, 2, 1, 3}
+    { "%",   Mod, 2, 1, 3},
+    { "^",   Pow, 2, 1, 4}
 };
 
-#define NB_FUNCTIONS 7
+#define NB_FUNCTIONS 9
 keyword_t functions[NB_FUNCTIONS] = {
-    { "sqrt", Sqr, 1, 4, 4},
-    { "pow",  Pow, 2, 3, 4},
-    { "cos",  Cos, 1, 3, 4},
-    { "sin",  Sin, 1, 3, 4},
-    { "atan", Atn, 1, 4, 4},
-    { "exp",  Exp, 1, 3, 4},
-    { "log",  Log, 1, 3, 4}
+    { "sqrt", Sqr, 1, 4, 5},
+    { "pow",  Pow, 2, 3, 5},
+    { "cos",  Cos, 1, 3, 5},
+    { "sin",  Sin, 1, 3, 5},
+    { "atan", Atn, 1, 4, 5},
+    { "exp",  Exp, 1, 3, 5},
+    { "log",  Log, 1, 3, 5},
+    { "quit", Qui, 0, 4, 5},
+    { "help", Hel, 0, 4, 5}
 };
 
+/* subparser function */
+
+element_t *subparser (element_t **proot, char **pstr, func_t func, int nbops, int prio)
+{
+    element_t *new = newelement (func, nbops, prio);
+    if (new == NULL) {
+        return ERROR_OP;
+    }
+    new->ops[0] = *proot;
+    new->ops[1] = parser (*pstr, pstr, new->prio);
+    if (new->ops[1] == ERROR_OP) {
+        delelement (new);
+        *proot = NULL;
+        return ERROR_OP;
+    }
+    *proot = newelement (Val, 1, 5);
+    if (*proot == NULL) {
+        delelement (new);
+        return ERROR_OP;
+    }
+    (*proot)->ops[0] = new;
+
+    return *proot;
+}
+
 /* parser function */
 
 element_t *parser (char *str, char **next, int prio)
@@ -98,6 +142,7 @@ element_t *parser (char *str, char **next, int prio)
                     found = 0;
                     new = parser (str + 1, &str, 0);
                     if (new == ERROR_OP) {
+                        delelement (root);
                         return ERROR_OP;
                     }
                     for (i = 0; i < root->nbops; i++) {
@@ -108,16 +153,20 @@ element_t *parser (char *str, char **next, int prio)
                         }
                     }
                     if (!found) {
+                        delelement (new);
+                        delelement (root);
                         return ERROR_OP;
                     }
                 } while (*str == ',');
             } else {
-                root = newelement (Val, 1, 4);
+                root = newelement (Val, 1, 5);
                 if (root == NULL) {
                     return ERROR_OP;
                 }
                 new = parser (str + 1, &str, 0);
                 if ((new == ERROR_OP) || (*str == ',')) {
+                    delelement (new);
+                    delelement (root);
                     return ERROR_OP;
                 }
                 root->ops[0] = new;
@@ -144,26 +193,22 @@ element_t *parser (char *str, char **next, int prio)
                 VERBOSE (DEBUG, PRINTOUT ("start processing operator\n"));
                 if (root) {
                     if ((prio) && (prio > operator->prio)) {
-                        VERBOSE (DEBUG, PRINTOUT ("stop processing operator because operator priority\n"));
+                        VERBOSE (DEBUG, PRINTOUT ("stop because operator priority\n"));
                         *next = str;
                         return root;
                     }
                     str += operator->offset;
                     VERBOSE (INFO, PRINTOUT ("Oper: %d\n", operator->func));
-                    new = newelement (operator->func, operator->nbops, operator->prio);
-                    if (new == NULL) {
-                        return ERROR_OP;
-                    }
-                    new->ops[0] = root;
-                    new->ops[1] = parser (str, &str, new->prio);
-                    if (new->ops[1] == ERROR_OP) {
+                    if (subparser (&root, &str, operator->func, operator->nbops, operator->prio) == ERROR_OP) {
+                        delelement (root);
                         return ERROR_OP;
                     }
-                    root = newelement (Val, 1, 4);
-                    if (root == ERROR_OP) {
+                } else if (*str == '-') {
+                    new = newelement (Sig, 1, 9);
+                    if (new == NULL) {
                         return ERROR_OP;
                     }
-                    root->ops[0] = new;
+                    root = new;
                 } else {
                     return ERROR_OP;
                 }
@@ -189,7 +234,8 @@ element_t *parser (char *str, char **next, int prio)
                         return ERROR_OP;
                     }
                     root = new;
-                 } else {
+                } else {
+                    delelement (root);
                     return ERROR_OP;
                 }
                 str += function->offset;
@@ -208,32 +254,36 @@ element_t *parser (char *str, char **next, int prio)
             (*str == '.') || (*str == '+') || (*str == '-')) {
             VERBOSE (DEBUG, PRINTOUT ("start processing value\n"));
             char *pt;
-            float value = strtof (str, &pt);
+            double value = strtod (str, &pt);
             VERBOSE (INFO, PRINTOUT ("Value: %f\n", value));
             if (str != pt) {
-                new = newelement (Val, 1, 4);
-                if (new == NULL) {
-                    return ERROR_OP;
-                }
-                new->value = value;
                 if (root == NULL) {
+                    new = newelement (Val, 1, 5);
+                    if (new == NULL) {
+                        return ERROR_OP;
+                    }
+                    new->value = value;
                     root = new;
+                    str = pt;
                 } else if (root->func == Val) {
                     if ((*str == '+') || (*str == '-')) {
-                        element_t *add = newelement (Add, 2, 1);
-                        if (add == NULL) {
+                        if ((prio) && (prio > 1)) {
+                            VERBOSE (DEBUG, PRINTOUT ("stop because operator priority\n"));
+                            *next = str;
+                            return root;
+                        }
+                        if (subparser (&root, &str, Add, 2, 1) == ERROR_OP) {
+                            delelement (root);
                             return ERROR_OP;
                         }
-                        add->ops[0] = root;
-                        add->ops[1] = new;
-                        root = add;
                     } else {
+                        delelement (root);
                         return ERROR_OP;
                     }
                 } else {
+                    delelement (root);
                     return ERROR_OP;
                 }
-                str = pt;
                 found = 1;
             }
             VERBOSE (DEBUG, PRINTOUT ("stop processing value\n"));
@@ -242,6 +292,7 @@ element_t *parser (char *str, char **next, int prio)
         /* error */
 
         if (!found) {
+            delelement (root);
             return ERROR_OP;
         }
 
@@ -250,6 +301,7 @@ element_t *parser (char *str, char **next, int prio)
     if (next != NULL) {
         *next = str;
     }
+
     return root;
 }
 
@@ -270,11 +322,12 @@ void print_element (element_t *root, int level)
 
     switch (root->func) {
     case Val: func = "Value"; break;
-    case Set: func = "Set"; break;
+    case Sig: func = "Sign"; break;
     case Add: func = "Addition"; break;
     case Sub: func = "Subtraction"; break;
     case Mul: func = "Multiplication"; break;
     case Div: func = "Division"; break;
+    case Mod: func = "Modulo"; break;
     case Pow: func = "Power"; break;
     case Sqr: func = "Square Root"; break;
     case Cos: func = "Cosine"; break;
@@ -282,6 +335,8 @@ void print_element (element_t *root, int level)
     case Atn: func = "Arc Tangent"; break;
     case Log: func = "Logarithm"; break;
     case Exp: func = "Exponantial"; break;
+    case Qui: func = "Quit"; break;
+    case Hel: func = "Help"; break;
     }
 
     PRINTOUT ("Function: %s\n", func);
@@ -298,4 +353,118 @@ void print_element (element_t *root, int level)
     }
 }
 
+/* quit function */
+
+void quit (void)
+{
+    PRINTOUT ("bye\n");
+    exit (0);
+}
+
+/* help message */
+
+void help (void)
+{
+    PRINTOUT ("calc is a simple calculator\n\n");
+    PRINTOUT ("supported operators:\n");
+    PRINTOUT (" + - * / %% ^\n\n");
+    PRINTOUT ("supported functions:\n");
+    PRINTOUT (" pow sqrt cos sin atan log exp\n\n");
+    PRINTOUT ("miscellaneous functions:\n");
+    PRINTOUT (" quit help\n");
+}
+
+/* evaluate element tree */
+
+#define MASK_SUB 0x1
+#define MASK_DIV 0x2
+
+double evaluate_element (element_t *root, char mask)
+{
+    double op0 = 0, op1 = 0;
+    char nextmask = mask;
+
+    if ((root == NULL) || (root == ERROR_OP)) {
+        VERBOSE (WARNING, PRINTOUT ("error while evaluating\n"));
+        return 0;
+    }
+
+    /* mask to manage sub operator sub and div */
+    switch (root->func) {
+    case Add:
+        nextmask &= ~MASK_SUB;
+        nextmask &= ~MASK_DIV;
+        break;
+    case Sub:
+        nextmask |= MASK_SUB;
+        nextmask &= ~MASK_DIV;
+        break;
+    case Mul:
+        nextmask &= ~MASK_DIV;
+        break;
+    case Div:
+        nextmask |= MASK_DIV;
+        break;
+    default:
+        nextmask = mask;
+    }
+
+    switch (root->func) {
+    case Val:
+    case Sig:
+        op0 = (root->ops[0]) ? evaluate_element (root->ops[0], nextmask) : root->value;
+        break;
+    case Add:
+    case Sub:
+    case Mul:
+    case Div:
+    case Mod:
+    case Pow:
+        if (root->ops[1]) {
+            op1 = evaluate_element (root->ops[1], nextmask);
+        } else {
+            VERBOSE (WARNING, PRINTOUT ("error while evaluating (op[1])\n"));
+            return 0;
+        }
+        /* fallthrough */
+    case Sqr:
+    case Cos:
+    case Sin:
+    case Atn:
+    case Log:
+    case Exp:
+        if (root->ops[0]) {
+            op0 = evaluate_element (root->ops[0], 0);
+        } else {
+            VERBOSE (WARNING, PRINTOUT ("error while evaluating (op[0])\n"));
+            return 0;
+        }
+        break;
+    case Qui:
+    case Hel:
+        break;
+    }
+
+    switch (root->func) {
+    case Val: return op0;
+    case Sig: return -op0;
+    case Add: return ((mask & MASK_SUB) == 0) ? op0 + op1 : op0 - op1;
+    case Sub: return ((mask & MASK_SUB) == 0) ? op0 - op1 : op0 + op1;
+    case Mul: return ((mask & MASK_DIV) == 0) ? op0 * op1 : op0 / op1;
+    case Div: return ((mask & MASK_DIV) == 0) ? op0 / op1 : op0 * op1;
+    case Mod: return fmod (op0, op1);
+    case Pow: return pow (op0, op1);
+    case Sqr: return sqrt (op0);
+    case Cos: return cos (op0);
+    case Sin: return sin (op0);
+    case Atn: return atan (op0);
+    case Log: return log (op0);
+    case Exp: return exp (op0);
+    case Qui: quit (); break;
+    case Hel: help (); break;
+    }
+
+    return 0;
+}
+
 /* vim: set ts=4 sw=4 et: */