From 124da7fdcc2dabecd99034d31e9cd32bea0e9da9 Mon Sep 17 00:00:00 2001
From: Laurent Mazet <mazet@softndesign.org>
Date: Wed, 25 Jan 2023 08:11:08 +0100
Subject: [PATCH] partial program feature

---
 parser.c | 74 ++++++++++++++++++++++++++++++++++++++++++++++++++++++--
 parser.h |  2 +-
 2 files changed, 73 insertions(+), 3 deletions(-)

diff --git a/parser.c b/parser.c
index 7a9f988..798abb8 100644
--- a/parser.c
+++ b/parser.c
@@ -200,6 +200,60 @@ element_t *parser (char *str, char **next, int prio)
             continue;
         }
 
+        /* check for open brace */
+
+        if (*str == '{') {
+            VERBOSE (DEBUG, fprintf (stdout, "start processing brace\n"));
+            element_t *prog = newelement (Prog, 0, 5);
+            if (prog == NULL) {
+                delelement (root);
+                return ERROR_OP;
+            }
+            if (root == NULL) {
+                root = prog;
+            } else {
+                for (i = 0; i < root->nbops; i++) {
+                    if (root->ops[i] == NULL) {
+                        root->ops[i] = prog;
+                        found = 1;
+                    }
+                }
+                if (!found) {
+                    delelement (prog);
+                    delelement (root);
+                    return ERROR_OP;
+                }
+            }
+
+            do {
+                found = 0;
+                new = parser (str + 1, &str, 0);
+                if ((new == NULL) || ((new != ERROR_OP) && (new->prio == 9))) {
+                    delelement (new);
+                    new = ERROR_OP;
+                }
+                if ((new == NULL) || (new == ERROR_OP)) {
+                    delelement (root);
+                return ERROR_OP;
+                }
+                element_t *prognew = newelement (Prog, prog->nbops + 1, 5);
+                for (i = 0; i < prog->nbops; i++) {
+                    prognew->ops[i] = prog->ops[i];
+                }
+                prog->ops[prog->nbops] = new;
+                delelement (prog);
+                prog = prognew;
+            } while (*str == ',');
+
+            if (*str != '}') {
+                delelement (root);
+                return ERROR_OP;
+            }
+            str++;
+            VERBOSE (DEBUG, fprintf (stdout, "stop processing brace\n"));
+            continue;
+        }
+
         /* check for open bracket */
 
         if (*str == '(') {
@@ -255,9 +309,9 @@ element_t *parser (char *str, char **next, int prio)
             continue;
         }
 
-        /* check for closing bracket or koma */
+        /* check for closing bracket, closing brace or koma */
 
-        if ((*str == ')') || (*str == ',')) {
+        if ((*str == ')') || (*str == '}') || (*str == ',')) {
             if (next != NULL) {
                 *next = str;
             }
@@ -472,6 +526,7 @@ 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;
     }
 
     fprintf (stdout, "Function: %s\n", func);
@@ -569,6 +624,19 @@ double while_do (element_t *cond, element_t *action)
     return ret;
 }
 
+/* program function */
+
+double program_do (element_t **prog, int nbcalls)
+{
+    double ret = 0;
+    int i;
+    for (i = 0; i < nbcalls; i++) {
+        ret = evaluate_element (prog[i], 0);
+        prog[i] = NULL;
+    }
+    return ret;
+}
+
 /* quit function */
 
 void quit (void)
@@ -686,6 +754,7 @@ double evaluate_element (element_t *root, char mask)
     case Ans:
     case Pi:
     case E:
+    case Prog:
         break;
     case While:
         if (root->ops[0] == NULL) {
@@ -738,6 +807,7 @@ 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);
     }
 
     return 0;
diff --git a/parser.h b/parser.h
index 67dd9ab..734d818 100644
--- a/parser.h
+++ b/parser.h
@@ -19,7 +19,7 @@ typedef enum {
     Ans, E, Pi,
     Equal, Diff, Ge, Le, Gt, Lt,
     And, Or, Not,
-    Cond, While
+    Cond, While, Prog
 } func_t;
 
 /* keyword type */
-- 
2.30.2