some fixes on program functions
[calc.git] / parser.c
index a1602a0f3b05360e8a34bfb29b7f25f5582e4b86..4b4fab7678dd65352e28a2922fe2e94a56ba4810 100644 (file)
--- a/parser.c
+++ b/parser.c
@@ -21,6 +21,9 @@ double *storage = NULL;
 char *format = NULL;
 char *minform = NULL;
 
+workspace_t *programs = NULL;
+int nb_programs = 0;
+
 /* compare codes */
 
 int codecmp (char *ref, char *str)
@@ -122,7 +125,7 @@ keyword_t operators[NB_OPERATORS] = {
     { "|",   Or, 2, 1, -2}
 };
 
-#define NB_FUNCTIONS 25
+#define NB_FUNCTIONS 31
 keyword_t functions[NB_FUNCTIONS] = {
     { "sqrt", Sqr, 1, 4, 5},
     { "pow",  Pow, 2, 3, 5},
@@ -143,12 +146,18 @@ keyword_t functions[NB_FUNCTIONS] = {
     { "dec",  Dec, 1, 3, 5},
     { "disp", Disp, 0, 4, 9},
     { "mem",  Mem, 1, 3, 9},
+    { "clr",  Clear, 0, 3, 9},
     { "quit", Quit, 0, 4, 9},
     { "help", Help, 0, 4, 9},
     { "!",    Not, 1, 1, 6},
     { "cond", Cond, 3, 4, 5},
     { "while", While, 2, 5, 5},
-    { "print", Print, 1, 5, 5}
+    { "print", Print, 1, 5, 5},
+    { "prog", Prog, 3, 4, 9},
+    { "call", Call, 10, 4, 5},
+    { "ls",   List, 0, 2, 9},
+    { "edit", Edit, 1, 4, 9},
+    { "del",  Del, 1, 3, 9}
 };
 
 #define NB_CONSTANTS 3
@@ -225,7 +234,7 @@ element_t *parser (char *str, char **next, int prio)
                 return ERROR_OP;
             }
             element_t **prog = NULL;
-            new = newelement (Prog, 0, 5);
+            new = newelement (Code, 0, 5);
             root = new;
             prog = &root;
 
@@ -239,7 +248,7 @@ element_t *parser (char *str, char **next, int prio)
                     delelement (root);
                     return ERROR_OP;
                 }
-                element_t *newprog = newelement (Prog, (*prog)->nbops + 1, 5);
+                element_t *newprog = newelement (Code, (*prog)->nbops + 1, 5);
                 for (i = 0; i < (*prog)->nbops; i++) {
                     newprog->ops[i] = (*prog)->ops[i];
                     (*prog)->ops[i] = NULL;
@@ -507,6 +516,7 @@ void print_element (element_t *root, int level)
     case Dec: func = "Decrease"; break;
     case Disp: func = "Display"; break;
     case Mem: func = "Memory"; break;
+    case Clear: func = "Clear"; break;
     case Quit: func = "Quit"; break;
     case Help: func = "Help"; break;
     case Ans: func = "Ans"; break;
@@ -523,8 +533,13 @@ void print_element (element_t *root, int level)
     case Not: func = "Not"; break;
     case Cond: func = "Condition"; break;
     case While: func = "While"; break;
-    case Prog: func = "Program"; break;
+    case Code: func = "Code"; break;
     case Print: func = "Print"; break;
+    case Prog: func = "Program"; break;
+    case Call: func = "Call"; break;
+    case List: func = "List"; break;
+    case Edit: func = "Edit"; break;
+    case Del: func = "Del"; break;
     }
 
     fprintf (stdout, "Function: %s\n", func);
@@ -631,6 +646,14 @@ void display (void)
     fprintf (stdout, "\n");
 }
 
+void clear ()
+{
+    int i;
+    for (i = 0; i < storage_size; i++) {
+        storage[i] = 0;
+    }
+}
+
 /* While do function */
 
 double while_do (element_t *cond, element_t *action)
@@ -659,7 +682,7 @@ double while_do (element_t *cond, element_t *action)
 
 /* program function */
 
-double program_do (element_t **prog, int nbcalls)
+double execute_code (element_t **prog, int nbcalls)
 {
     double ret = 0;
     int i;
@@ -709,6 +732,130 @@ void quit (void)
     exit (0);
 }
 
+/* program function */
+
+void prog (int id, int nbmems, element_t *root)
+{
+    int i, n = -1;
+
+    if (programs == NULL) {
+
+        /* initial memory allocation */
+        programs = (workspace_t *) calloc (1, sizeof (workspace_t));
+        if (programs == NULL) {
+            VERBOSE (ERROR, fprintf (stderr, "can't allocate memory\n"));
+            exit (1);
+        }
+        nb_programs = 1;
+        n = 0;
+
+    } else {
+
+        /* look for existing program */
+        for (i = 0; i < nb_programs; i++) {
+            if ((programs + i)->id == id) {
+                n = i;
+                break;
+            }
+        }
+        if (n == -1) {
+
+            /* new program */
+            n = nb_programs++;
+            programs = (workspace_t *) realloc (programs, nb_programs * sizeof (workspace_t));
+        } else {
+
+            /* clean old program */
+            if ((programs + n)->storage) {
+                free ((programs + n)->storage);
+            }
+            if ((programs + n)->root) {
+                delelement ((programs + n)->root);
+            }
+        }
+    }
+
+    /* set program */
+    (programs + n)->id = id;
+    (programs + n)->answer = 0;
+    (programs + n)->storage = (double *) calloc (nbmems, sizeof (double));
+    if ((programs + n)->storage == NULL) {
+        VERBOSE (ERROR, fprintf (stderr, "can't allocate memory\n"));
+        exit (1);
+    }
+    (programs + n)->storage_size = nbmems;
+    (programs + n)->root = dupelement (root);
+}
+
+double call (int id, int nbops, element_t **ops)
+{
+    workspace_t tmp = {0};
+    int i, n = -1;
+    double ret = 0;
+
+    if (programs) {
+
+        /* look for program */
+        for (i = 0; i < nb_programs; i++) {
+            if ((programs + i)->id == id) {
+                n = i;
+                break;
+            }
+        }
+        if (n == -1) {
+            VERBOSE (WARNING, fprintf (stdout, "error unknown program (%d)\n", id));
+            return 0;
+        }
+
+        /* store context */
+        tmp.answer = answer;
+        tmp.storage = storage;
+        tmp.storage_size = storage_size;
+
+        /* change context */
+        answer = 0;
+        storage = (programs + n)->storage;
+        storage_size = (programs + n)->storage_size;
+        if (nbops > storage_size) {
+            storage = realloc (storage, nbops * sizeof (double));
+            storage_size = nbops;
+        }
+        for (i = 0; i < nbops; i++) {
+            double val = evaluate_element (ops[i], 0);
+            store (i + 1, val);
+        }
+
+        /* evaluate program */
+        element_t *elements = dupelement ((programs + n)->root);
+        ret = evaluate_element (elements, 0);
+
+        /* restore context */
+        answer = tmp.answer;
+        storage = tmp.storage;
+        storage_size = tmp.storage_size;
+    }
+
+    return ret;
+}
+
+void list ()
+{
+    int i;
+    if (programs != NULL) {
+        fprintf (stdout, "programs:");
+        for (i = 0; i < nb_programs; i++) {
+            fprintf (stdout, " %d", (programs + i)->id);
+        }
+        fprintf (stdout, "\n");
+    }
+}
+
+void edit (int id)
+{ }
+
+void del (int id)
+{ }
+
 /* help message */
 
 void help (void)
@@ -745,6 +892,7 @@ double evaluate_element (element_t *root, char mask)
 {
     double op0 = 0, op1 = 0;
     char nextmask = mask;
+    int i, nb;
 
     if ((root == NULL) || (root == ERROR_OP)) {
         VERBOSE (WARNING, fprintf (stdout, "error while evaluating\n"));
@@ -791,6 +939,7 @@ double evaluate_element (element_t *root, char mask)
     case Lt:
     case And:
     case Or:
+    case Prog:
         if (root->ops[1]) {
             op1 = evaluate_element (root->ops[1], nextmask);
         } else if (root->func != Store) {
@@ -816,6 +965,9 @@ double evaluate_element (element_t *root, char mask)
     case Not:
     case Mem:
     case Cond:
+    case Call:
+    case Edit:
+    case Del:
         if (root->ops[0]) {
             op0 = evaluate_element (root->ops[0], 0);
         } else {
@@ -824,12 +976,14 @@ double evaluate_element (element_t *root, char mask)
         }
         break;
     case Disp:
+    case Clear:
     case Quit:
     case Help:
     case Ans:
     case Pi:
     case E:
-    case Prog:
+    case Code:
+    case List:
         break;
     case While:
         if (root->ops[0] == NULL) {
@@ -869,6 +1023,7 @@ double evaluate_element (element_t *root, char mask)
     case Dec: return decrease ((int)op0);
     case Disp: display (); break;
     case Mem: memory ((int)op0); break;
+    case Clear: clear (); break;
     case Quit: quit (); break;
     case Help: help (); break;
     case Ans: return answer;
@@ -892,8 +1047,19 @@ double evaluate_element (element_t *root, char mask)
             return 0;
         }
     case While: return while_do (root->ops[0], root->ops[1]);
-    case Prog: return program_do (root->ops, root->nbops);
+    case Code: return execute_code (root->ops, root->nbops);
     case Print: return print (op0);
+    case Prog: prog ((int)op0, (int)op1, root->ops[2]); break;
+    case Call:
+        for (i = 1, nb =0; i < root->nbops; i++) {
+            if (root->ops[i]) {
+                nb++;
+            }
+        }
+        return call ((int)op0, nb, root->ops + 1);
+    case List: list (); break;
+    case Edit: edit ((int)op0); break;
+    case Del: del ((int)op0); break;
     }
 
     return 0;