[calc.git] / parser.c

#include <malloc.h>
#include <math.h>
#include <stdlib.h>

#include "debug.h"
#include "fdprintf.h"

#include "parser.h"

/* compare codes */

int codecmp (char *ref, char *str)
{
    int sig;

    while (*ref != '\0') {
        if (*ref == '\t') {
             sig = (*str == '.') ? -1 : ((*str >= '0') && (*str <= '9'));
        } else {
            sig = *str - *ref;
        }
        if (sig != 0) {
            return (sig > 0) ? 1 : -1;
        }
        str++;
        ref++;
    }

    return 0;
}

/* allocate new element */

element_t *newelement (func_t function, int nbops, int prio)
{
    element_t *new = (element_t *) calloc (1, sizeof (element_t));
    if (new == NULL) {
        VERBOSE (ERROR, fdprintf (stdfderr, "can't allocate memory\n"));
        return NULL;
    }
    new->func = function;
    new->nbops = nbops;
    new->prio = prio;

    return new;
}

/* functions */

#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},
    { "%",   Mod, 2, 1, 3},
    { "^",   Pow, 2, 1, 4}
};

#define NB_FUNCTIONS 7
keyword_t functions[NB_FUNCTIONS] = {
    { "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}
};

/* 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) {
        return ERROR_OP;
    }
    *proot = newelement (Val, 1, 4);
    if (*proot == ERROR_OP) {
        return ERROR_OP;
    }
    (*proot)->ops[0] = new;

    return *proot;
}

/* parser function */

element_t *parser (char *str, char **next, int prio)
{
    element_t *root = NULL;
    int i;

    VERBOSE (DEBUG, PRINTOUT ("Starting parsing\n"));

    /* main loop */
    while (*str != '\0') {
        int found = 0;
        element_t *new = NULL;
        VERBOSE (INFO, PRINTOUT ("Processing: %s\n", str));

        /* skip spaces and tabs */

        if ((*str == ' ') || (*str == '\t')) {
            str++;
            continue;
        }

        /* check for open bracket */

        if (*str == '(') {
            VERBOSE (DEBUG, PRINTOUT ("start processing bracket\n"));
            if (root) {
                do {
                    found = 0;
                    new = parser (str + 1, &str, 0);
                    if (new == ERROR_OP) {
                        return ERROR_OP;
                    }
                    for (i = 0; i < root->nbops; i++) {
                        if (root->ops[i] == NULL) {
                            root->ops[i] = new;
                            found = 1;
                            break;
                        }
                    }
                    if (!found) {
                        return ERROR_OP;
                    }
                } while (*str == ',');
            } else {
                root = newelement (Val, 1, 4);
                if (root == NULL) {
                    return ERROR_OP;
                }
                new = parser (str + 1, &str, 0);
                if ((new == ERROR_OP) || (*str == ',')) {
                    return ERROR_OP;
                }
                root->ops[0] = new;
            }
            str++;
            VERBOSE (DEBUG, PRINTOUT ("stop processing bracket\n"));
            continue;
        }

        /* check for closing bracket or koma */

        if ((*str == ')') || (*str == ',')) {
            if (next != NULL) {
                *next = str;
            }
            return root;
        }

        /* look for operators */

        for (i = 0; i < NB_OPERATORS; i++) {
            keyword_t *operator = operators + i;
            if (codecmp (operator->keyword, str) == 0) {
                VERBOSE (DEBUG, PRINTOUT ("start processing operator\n"));
                if (root) {
                    if ((prio) && (prio > operator->prio)) {
                        VERBOSE (DEBUG, PRINTOUT ("stop because operator priority\n"));
                        *next = str;
                        return root;
                    }
                    str += operator->offset;
                    VERBOSE (INFO, PRINTOUT ("Oper: %d\n", operator->func));
                    if (subparser (&root, &str, operator->func, operator->nbops, operator->prio) == ERROR_OP) {
                        return ERROR_OP;
                    }
                } else {
                    return ERROR_OP;
                }
                found = 1;
                VERBOSE (DEBUG, PRINTOUT ("stop processing operator\n"));
                break;
            }
        }
        if (found) {
            continue;
        }

        /* look for functions */

        for (i = 0; i < NB_FUNCTIONS; i++) {
            keyword_t *function = functions + i;
            if (codecmp (function->keyword, str) == 0) {
                VERBOSE (DEBUG, PRINTOUT ("start processing function\n"));
                if (root == NULL) {
                    VERBOSE (INFO, PRINTOUT ("Func: %d\n", function->func));
                    new = newelement (function->func, function->nbops, function->prio);
                    if (new == NULL) {
                        return ERROR_OP;
                    }
                    root = new;
                 } else {
                    return ERROR_OP;
                }
                str += function->offset;
                found = 1;
                VERBOSE (DEBUG, PRINTOUT ("stop processing function\n"));
                break;
            }
        }
        if (found) {
            continue;
        }

        /* look for number */

        if (((*str >= '0') && (*str <= '9')) ||
            (*str == '.') || (*str == '+') || (*str == '-')) {
            VERBOSE (DEBUG, PRINTOUT ("start processing value\n"));
            char *pt;
            float value = strtof (str, &pt);
            VERBOSE (INFO, PRINTOUT ("Value: %f\n", value));
            if (str != pt) {
                if (root == NULL) {
                    new = newelement (Val, 1, 4);
                    if (new == NULL) {
                        return ERROR_OP;
                    }
                    new->value = value;
                    root = new;
                    str = pt;
                } else if (root->func == Val) {
                    if ((*str == '+') || (*str == '-')) {
                        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) {
                            return ERROR_OP;
                        }
                    } else {
                        return ERROR_OP;
                    }
                } else {
                    return ERROR_OP;
                }
                found = 1;
            }
            VERBOSE (DEBUG, PRINTOUT ("stop processing value\n"));
        }

        /* error */

        if (!found) {
            return ERROR_OP;
        }

    }

    if (next != NULL) {
        *next = str;
    }
    return root;
}

/* print element tree */

void print_element (element_t *root, int level)
{
    char *func = NULL;
    int i;

    if ((root == NULL) || (root == ERROR_OP)) {
        return;
    }

    for (i = 0; i < level; i++) {
        PRINTOUT (" ");
    }

    switch (root->func) {
    case Val: func = "Value"; 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;
    case Sin: func = "Sine"; break;
    case Atn: func = "Arc Tangent"; break;
    case Log: func = "Logarithm"; break;
    case Exp: func = "Exponantial"; break;
    }

    PRINTOUT ("Function: %s\n", func);

    if ((root->func == Val) && (root->ops[0] == NULL)) {
        for (i = 0; i < level; i++) {
            PRINTOUT (" ");
        }
        PRINTOUT ("value: %f\n", root->value);
    } else {
        for (i = 0; i < root->nbops; i++) {
            print_element (root->ops[i], level + 1);
        }
    }
}

/* evaluate element tree */

double evaluate_element (element_t *root)
{
    double op0 = 0, op1 = 0;

    if ((root == NULL) || (root == ERROR_OP)) {
        VERBOSE (WARNING, PRINTOUT ("error while evaluating\n"));
        return 0;
    }

    switch (root->func) {
    case Val:
        if (root->ops[0]) {
            return evaluate_element (root->ops[0]);
        } else {
            return 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]);
        } 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]);
        } else {
            VERBOSE (WARNING, PRINTOUT ("error while evaluating (op[0])\n"));
            return 0;
        }
    }

    switch (root->func) {
    case Add: return op0 + op1;
    case Sub: return op0 - op1;
    case Mul: return op0 * op1;
    case Div: return 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);
    default: break;
    }

    return 0;
}

/* vim: set ts=4 sw=4 et: */
Commit	Line	Data
bc97a989	1	#include <malloc.h>
f2927108	2	#include <math.h>
bc97a989 LM	3	#include <stdlib.h>
	4
	5	#include "debug.h"
	6	#include "fdprintf.h"
	7
	8	#include "parser.h"
	9
	10	/* compare codes */
	11
	12	int codecmp (char ref, char str)
	13	{
	14	int sig;
	15
	16	while (*ref != '\0') {
0b9cc9b0 LM	17	if (*ref == '\t') {
	18	sig = (str == '.') ? -1 : ((str >= '0') && (*str <= '9'));
	19	} else {
	20	sig = str - ref;
	21	}
bc97a989 LM	22	if (sig != 0) {
	23	return (sig > 0) ? 1 : -1;
	24	}
0b9cc9b0 LM	25	str++;
0b9cc9b0 LM	26	ref++;
bc97a989 LM	27	}
	28
	29	return 0;
	30	}
	31
	32	/* allocate new element */
	33
11cda8d7	34	element_t *newelement (func_t function, int nbops, int prio)
bc97a989 LM	35	{
	36	element_t new = (element_t ) calloc (1, sizeof (element_t));
	37	if (new == NULL) {
	38	VERBOSE (ERROR, fdprintf (stdfderr, "can't allocate memory\n"));
	39	return NULL;
	40	}
	41	new->func = function;
	42	new->nbops = nbops;
11cda8d7	43	new->prio = prio;
bc97a989 LM	44
	45	return new;
	46	}
	47
	48	/* functions */
	49
c47a9298	50	#define NB_OPERATORS 6
bc97a989 LM	51
bc97a989 LM	52	keyword_t operators[NB_OPERATORS] = {
11cda8d7 LM	53	{ "+\t", Add, 2, 1, 1},
	54	{ "-\t", Sub, 2, 1, 1},
	55	{ "*", Mul, 2, 1, 2},
	56	{ "/", Div, 2, 1, 2},
c47a9298 LM	57	{ "%", Mod, 2, 1, 3},
c47a9298 LM	58	{ "^", Pow, 2, 1, 4}
bc97a989 LM	59	};
	60
	61	#define NB_FUNCTIONS 7
	62	keyword_t functions[NB_FUNCTIONS] = {
c47a9298 LM	63	{ "sqrt", Sqr, 1, 4, 5},
	64	{ "pow", Pow, 2, 3, 5},
	65	{ "cos", Cos, 1, 3, 5},
	66	{ "sin", Sin, 1, 3, 5},
	67	{ "atan", Atn, 1, 4, 5},
	68	{ "exp", Exp, 1, 3, 5},
	69	{ "log", Log, 1, 3, 5}
bc97a989 LM	70	};
bc97a989 LM	71
d2ff8478 LM	72	/* subparser function */
	73
	74	element_t subparser (element_t proot, char *pstr, func_t func, int nbops, int prio)
	75	{
	76	element_t *new = newelement (func, nbops, prio);
	77	if (new == NULL) {
	78	return ERROR_OP;
	79	}
	80	new->ops[0] = *proot;
	81	new->ops[1] = parser (*pstr, pstr, new->prio);
	82	if (new->ops[1] == ERROR_OP) {
	83	return ERROR_OP;
	84	}
	85	*proot = newelement (Val, 1, 4);
	86	if (*proot == ERROR_OP) {
	87	return ERROR_OP;
	88	}
	89	(*proot)->ops[0] = new;
	90
	91	return *proot;
	92	}
	93
bc97a989 LM	94	/* parser function */
bc97a989 LM	95
ef37d966	96	element_t parser (char str, char **next, int prio)
0b489a77	97	{
bc97a989	98	element_t *root = NULL;
85b4a72c	99	int i;
bc97a989	100
49223129 LM	101	VERBOSE (DEBUG, PRINTOUT ("Starting parsing\n"));
49223129 LM	102
bc97a989 LM	103	/* main loop */
	104	while (*str != '\0') {
	105	int found = 0;
	106	element_t *new = NULL;
49223129	107	VERBOSE (INFO, PRINTOUT ("Processing: %s\n", str));
bc97a989 LM	108
	109	/* skip spaces and tabs */
	110
	111	if ((str == ' ') \|\| (str == '\t')) {
	112	str++;
	113	continue;
	114	}
	115
85b4a72c	116	/* check for open bracket */
0b489a77	117
0b489a77	118	if (*str == '(') {
49223129	119	VERBOSE (DEBUG, PRINTOUT ("start processing bracket\n"));
0b489a77	120	if (root) {
85b4a72c LM	121	do {
85b4a72c LM	122	found = 0;
ef37d966	123	new = parser (str + 1, &str, 0);
85b4a72c LM	124	if (new == ERROR_OP) {
	125	return ERROR_OP;
	126	}
	127	for (i = 0; i < root->nbops; i++) {
	128	if (root->ops[i] == NULL) {
0b489a77 LM	129	root->ops[i] = new;
	130	found = 1;
	131	break;
	132	}
	133	}
85b4a72c LM	134	if (!found) {
	135	return ERROR_OP;
	136	}
	137	} while (*str == ',');
	138	} else {
11cda8d7	139	root = newelement (Val, 1, 4);
85b4a72c	140	if (root == NULL) {
0b489a77 LM	141	return ERROR_OP;
0b489a77 LM	142	}
ef37d966	143	new = parser (str + 1, &str, 0);
85b4a72c	144	if ((new == ERROR_OP) \|\| (*str == ',')) {
efdfb543 LM	145	return ERROR_OP;
efdfb543 LM	146	}
85b4a72c	147	root->ops[0] = new;
0b489a77	148	}
85b4a72c	149	str++;
49223129	150	VERBOSE (DEBUG, PRINTOUT ("stop processing bracket\n"));
efdfb543	151	continue;
49223129	152	}
85b4a72c LM	153
	154	/* check for closing bracket or koma */
	155
	156	if ((str == ')') \|\| (str == ',')) {
49223129	157	if (next != NULL) {
85b4a72c	158	*next = str;
49223129 LM	159	}
49223129 LM	160	return root;
0b489a77 LM	161	}
0b489a77 LM	162
bc97a989 LM	163	/* look for operators */
	164
	165	for (i = 0; i < NB_OPERATORS; i++) {
	166	keyword_t *operator = operators + i;
	167	if (codecmp (operator->keyword, str) == 0) {
49223129	168	VERBOSE (DEBUG, PRINTOUT ("start processing operator\n"));
11cda8d7	169	if (root) {
ef37d966	170	if ((prio) && (prio > operator->prio)) {
c47a9298	171	VERBOSE (DEBUG, PRINTOUT ("stop because operator priority\n"));
ef37d966 LM	172	*next = str;
	173	return root;
	174	}
	175	str += operator->offset;
49223129	176	VERBOSE (INFO, PRINTOUT ("Oper: %d\n", operator->func));
d2ff8478	177	if (subparser (&root, &str, operator->func, operator->nbops, operator->prio) == ERROR_OP) {
ef37d966	178	return ERROR_OP;
11cda8d7	179	}
49223129	180	} else {
0b489a77	181	return ERROR_OP;
bc97a989	182	}
bc97a989	183	found = 1;
49223129	184	VERBOSE (DEBUG, PRINTOUT ("stop processing operator\n"));
bc97a989 LM	185	break;
	186	}
	187	}
	188	if (found) {
bc97a989 LM	189	continue;
bc97a989 LM	190	}
49223129	191
bc97a989 LM	192	/* look for functions */
	193
	194	for (i = 0; i < NB_FUNCTIONS; i++) {
	195	keyword_t *function = functions + i;
	196	if (codecmp (function->keyword, str) == 0) {
49223129	197	VERBOSE (DEBUG, PRINTOUT ("start processing function\n"));
bc97a989	198	if (root == NULL) {
49223129	199	VERBOSE (INFO, PRINTOUT ("Func: %d\n", function->func));
11cda8d7	200	new = newelement (function->func, function->nbops, function->prio);
bc97a989	201	if (new == NULL) {
0b489a77	202	return ERROR_OP;
bc97a989 LM	203	}
	204	root = new;
	205	} else {
0b489a77	206	return ERROR_OP;
bc97a989 LM	207	}
	208	str += function->offset;
	209	found = 1;
49223129	210	VERBOSE (DEBUG, PRINTOUT ("stop processing function\n"));
bc97a989 LM	211	break;
	212	}
	213	}
	214	if (found) {
bc97a989 LM	215	continue;
	216	}
	217
bc97a989 LM	218	/* look for number */
bc97a989 LM	219
0b9cc9b0 LM	220	if (((str >= '0') && (str <= '9')) \|\|
0b9cc9b0 LM	221	(str == '.') \|\| (str == '+') \|\| (*str == '-')) {
49223129	222	VERBOSE (DEBUG, PRINTOUT ("start processing value\n"));
bc97a989 LM	223	char *pt;
bc97a989 LM	224	float value = strtof (str, &pt);
49223129	225	VERBOSE (INFO, PRINTOUT ("Value: %f\n", value));
bc97a989	226	if (str != pt) {
bc97a989	227	if (root == NULL) {
c47a9298 LM	228	new = newelement (Val, 1, 4);
	229	if (new == NULL) {
	230	return ERROR_OP;
	231	}
	232	new->value = value;
bc97a989	233	root = new;
c47a9298	234	str = pt;
0b9cc9b0 LM	235	} else if (root->func == Val) {
0b9cc9b0 LM	236	if ((str == '+') \|\| (str == '-')) {
c47a9298 LM	237	if ((prio) && (prio > 1)) {
	238	VERBOSE (DEBUG, PRINTOUT ("stop because operator priority\n"));
	239	*next = str;
	240	return root;
	241	}
d2ff8478	242	if (subparser (&root, &str, Add, 2, 1) == ERROR_OP) {
0b489a77 LM	243	return ERROR_OP;
0b489a77 LM	244	}
0b9cc9b0	245	} else {
0b489a77	246	return ERROR_OP;
bc97a989	247	}
0b9cc9b0 LM	248	} else {
0b9cc9b0 LM	249	return ERROR_OP;
bc97a989	250	}
bc97a989 LM	251	found = 1;
bc97a989 LM	252	}
49223129	253	VERBOSE (DEBUG, PRINTOUT ("stop processing value\n"));
bc97a989 LM	254	}
	255
	256	/* error */
	257
	258	if (!found) {
0b489a77	259	return ERROR_OP;
bc97a989 LM	260	}
	261
	262	}
	263
49223129 LM	264	if (next != NULL) {
	265	*next = str;
	266	}
bc97a989 LM	267	return root;
	268	}
	269
	270	/* print element tree */
	271
	272	void print_element (element_t *root, int level)
	273	{
	274	char *func = NULL;
	275	int i;
	276
49223129	277	if ((root == NULL) \|\| (root == ERROR_OP)) {
bc97a989	278	return;
49223129 LM	279	}
49223129 LM	280
bc97a989 LM	281	for (i = 0; i < level; i++) {
	282	PRINTOUT (" ");
	283	}
	284
	285	switch (root->func) {
	286	case Val: func = "Value"; break;
	287	case Add: func = "Addition"; break;
	288	case Sub: func = "Subtraction"; break;
	289	case Mul: func = "Multiplication"; break;
	290	case Div: func = "Division"; break;
c47a9298	291	case Mod: func = "Modulo"; break;
bc97a989 LM	292	case Pow: func = "Power"; break;
	293	case Sqr: func = "Square Root"; break;
	294	case Cos: func = "Cosine"; break;
	295	case Sin: func = "Sine"; break;
	296	case Atn: func = "Arc Tangent"; break;
	297	case Log: func = "Logarithm"; break;
	298	case Exp: func = "Exponantial"; break;
	299	}
	300
	301	PRINTOUT ("Function: %s\n", func);
	302
85b4a72c	303	if ((root->func == Val) && (root->ops[0] == NULL)) {
bc97a989 LM	304	for (i = 0; i < level; i++) {
	305	PRINTOUT (" ");
	306	}
	307	PRINTOUT ("value: %f\n", root->value);
	308	} else {
	309	for (i = 0; i < root->nbops; i++) {
	310	print_element (root->ops[i], level + 1);
	311	}
	312	}
	313	}
	314
f2927108 LM	315	/* evaluate element tree */
	316
	317	double evaluate_element (element_t *root)
	318	{
	319	double op0 = 0, op1 = 0;
	320
	321	if ((root == NULL) \|\| (root == ERROR_OP)) {
	322	VERBOSE (WARNING, PRINTOUT ("error while evaluating\n"));
	323	return 0;
	324	}
	325
	326	switch (root->func) {
	327	case Val:
	328	if (root->ops[0]) {
	329	return evaluate_element (root->ops[0]);
	330	} else {
	331	return root->value;
	332	}
	333	break;
	334	case Add:
	335	case Sub:
	336	case Mul:
	337	case Div:
c47a9298	338	case Mod:
f2927108 LM	339	case Pow:
	340	if (root->ops[1]) {
	341	op1 = evaluate_element (root->ops[1]);
	342	} else {
	343	VERBOSE (WARNING, PRINTOUT ("error while evaluating (op[1])\n"));
	344	return 0;
	345	}
	346	/* fallthrough */
	347	case Sqr:
	348	case Cos:
	349	case Sin:
	350	case Atn:
	351	case Log:
	352	case Exp:
	353	if (root->ops[0]) {
	354	op0 = evaluate_element (root->ops[0]);
	355	} else {
	356	VERBOSE (WARNING, PRINTOUT ("error while evaluating (op[0])\n"));
	357	return 0;
	358	}
	359	}
	360
	361	switch (root->func) {
	362	case Add: return op0 + op1;
	363	case Sub: return op0 - op1;
	364	case Mul: return op0 * op1;
	365	case Div: return op0 / op1;
c47a9298	366	case Mod: return fmod (op0, op1);
f2927108 LM	367	case Pow: return pow (op0, op1);
	368	case Sqr: return sqrt (op0);
	369	case Cos: return cos (op0);
	370	case Sin: return sin (op0);
	371	case Atn: return atan (op0);
	372	case Log: return log (op0);
	373	case Exp: return exp (op0);
	374	default: break;
	375	}
	376
	377	return 0;
	378	}
	379
bc97a989	380	/* vim: set ts=4 sw=4 et: */