print function
[calc.git] / calc.c
CommitLineData
ec15bdbc
LM
1/* depend: */
2/* cflags: */
04d68907 3/* linker: debug.o parser.o -lm -lreadline */
ec15bdbc 4
680bf70a 5#include <malloc.h>
01f02c41 6#include <stddef.h>
ec15bdbc 7#include <stdio.h>
04d68907 8#include <stdlib.h>
5898ff24
LM
9#include <unistd.h>
10
11#include <readline/readline.h>
12#include <readline/history.h>
ec15bdbc 13
bc97a989 14#include "debug.h"
bc97a989 15#include "parser.h"
01f02c41 16
ec15bdbc
LM
17/* constants */
18
5898ff24 19#define BUFFER_SIZE 4096
3a7a78ff 20#define HISTORY_LEN 10
5898ff24 21
ec15bdbc
LM
22/* macros */
23
24#define CEIL(x, y) (((x) + (y) - 1) / (y))
25#define MIN(x, y) (((x) < (y)) ? (x) : (y))
26#define MAX(x, y) (((x) > (y)) ? (x) : (y))
27
ec15bdbc
LM
28/* gobal variables */
29
30char *progname = NULL;
5898ff24 31int mode = 1;
4ee92d9d 32int precision = 6;
b9c1c40d 33char **completion_list = NULL;
ec15bdbc
LM
34
35/* help function */
36
01f02c41 37int usage (int ret)
ec15bdbc 38{
04d68907
LM
39 FILE *fid = ret ? stderr : stdout;
40 fprintf (fid, "usage: %s\n", progname);
41 fprintf (fid, " -h : help message\n");
42 fprintf (fid, " -n : no readline mode (%s)\n", mode ? "yes" : "no");
43 fprintf (fid, " -p : precision (%d)\n", precision);
44 fprintf (fid, " -v : verbose level (%d)\n", verbose);
ec15bdbc 45
01f02c41 46 return ret;
ec15bdbc
LM
47}
48
db660e60
LM
49/* completion function */
50
51char *generator (const char *text, int state);
52
53char **completion (const char *text, __attribute__((unused)) int start, __attribute__((unused)) int end)
54{
55 rl_attempted_completion_over = 1;
56 return rl_completion_matches (text, generator);
57}
58
59char *generator (const char *text, int state)
60{
61 static int index, len;
62 char *name;
63
64 if (!state) {
65 index = 0;
66 len = strlen(text);
67 }
68
69 while ((name = completion_list[index++])) {
70 if (strncmp (name, text, len) == 0) {
71 return strdup (name);
72 }
73 }
74
75 return NULL;
76}
77
ec15bdbc
LM
78/* main function */
79
680bf70a 80int main (int argc, char *argv[])
ec15bdbc 81{
5898ff24
LM
82 char *buffer = NULL;
83 char buffer_static[BUFFER_SIZE + 1] = {0};
84 int i = 0, nb = 1;
efdfb543 85 int ret = 0;
01f02c41
LM
86
87 /* program name */
ec15bdbc
LM
88
89 progname = argv[0];
01f02c41
LM
90 while (progname[i] != '\0') {
91 if ((progname[i] == '/') || (progname[i] == '\\')) {
92 progname += i + 1;
93 i = 0;
94 } else {
95 i++;
96 }
97 }
98
99 /* argument processing */
ec15bdbc 100
01f02c41
LM
101 while (argc-- > 1) {
102 char *arg = *(++argv);
103 if (arg[0] != '-') {
b89cbb39
LM
104 VERBOSE (ERROR, fprintf (stderr, "%s: invalid option -- '%s'\n", progname, arg); usage (1));
105 return 1;
01f02c41
LM
106 }
107 char c = arg[1];
ec15bdbc 108 switch (c) {
5898ff24
LM
109 case 'n':
110 mode = 0;
111 buffer = buffer_static;
112 break;
4ee92d9d
LM
113 case 'p':
114 arg = (arg[2]) ? arg + 2 : (--argc > 0) ? *(++argv) : NULL;
115 if (arg == NULL) {
b89cbb39
LM
116 VERBOSE (ERROR, fprintf (stderr, "%s: missing precision\n", progname); usage (1));
117 return 1;
4ee92d9d
LM
118 }
119 precision = atoi (arg);
120 break;
ec15bdbc 121 case 'v':
01f02c41
LM
122 arg = (arg[2]) ? arg + 2 : (--argc > 0) ? *(++argv) : NULL;
123 if (arg == NULL) {
b89cbb39
LM
124 VERBOSE (ERROR, fprintf (stderr, "%s: missing verbose level\n", progname); usage (1));
125 return 1;
01f02c41
LM
126 }
127 verbose = atoi (arg);
ec15bdbc
LM
128 break;
129 case 'h':
ec15bdbc 130 default:
01f02c41 131 return usage (c != 'h');
ec15bdbc
LM
132 }
133 }
680bf70a 134
4ee92d9d 135 /* format */
5075f6ea 136 format[5] = '0' + precision;
ec15bdbc 137
b9c1c40d
LM
138 /* completion list*/
139 completion_list = generate_completion_list ();
db660e60 140 rl_attempted_completion_function = completion;
b9c1c40d 141
ec15bdbc 142 /* read from input stream */
01f02c41 143
5898ff24
LM
144 while (1) {
145 char *line[BUFFER_SIZE] = {0};
680bf70a 146
5898ff24
LM
147 if (mode) {
148 if ((buffer = readline ("<= ")) == NULL) {
149 break;
150 }
ec15bdbc 151
5898ff24
LM
152 /* check empty line */
153 if (strlen (buffer) == 0) {
154 free (buffer);
155 continue;
156 } else if (strcmp (buffer, ".") == 0) {
157 break;
158 }
159 line[0] = buffer;
941d160e 160 nb = 1;
5898ff24
LM
161
162 /* add line into history */
163 add_history (buffer);
3a7a78ff
LM
164 VERBOSE (INFO, fprintf (stdout, "line (%d/%d): '%s'\n",
165 where_history (), history_length, buffer));
166 if (history_length > HISTORY_LEN) {
5898ff24
LM
167 HIST_ENTRY *last = remove_history (0);
168 if (last) {
169 free_history_entry (last);
170 }
ec15bdbc 171 }
5898ff24 172 } else {
3a7a78ff
LM
173 if (read (STDIN_FILENO, buffer, BUFFER_SIZE) == 0) {
174 break;
175 }
176 nb = 0;
177 char *pt = line[nb++] = buffer;
32741902 178 while (*pt != '\0') {
3a7a78ff
LM
179 if (*pt == '\n') {
180 *pt = '\0';
32741902 181 line[nb++] = pt + 1;
3a7a78ff 182 }
32741902 183 pt++;
3a7a78ff
LM
184 }
185 VERBOSE (INFO, fprintf (stdout, "line: '%s'\n", buffer));
ec15bdbc
LM
186 }
187
680bf70a 188 /* look for end of line */
941d160e 189 for (i = 0; i < nb; i++) {
5898ff24
LM
190 if (*line[i] == '\0') {
191 continue;
192 }
45a631a8 193 element_t *element = parser (line[i], NULL, -9);
5898ff24 194 if (element == ERROR_OP) {
32741902 195 VERBOSE (WARNING, fprintf (stdout, "error while parsing: '%s'\n", line[i]); fflush (stdout));
5898ff24 196 ret = 1;
d32ea0b0 197 } else if (element != NULL) {
5898ff24 198 VERBOSE (INFO, print_element (element, 0));
5075f6ea
LM
199 answer = evaluate_element (element, 0);
200 fprintf (stdout, format, answer);
87621fe1 201 fflush (stdout);
5898ff24
LM
202 delelement (element);
203 ret = 0;
204 }
ec15bdbc 205 }
680bf70a 206
5898ff24
LM
207 if (mode) {
208 free (buffer);
209 } else {
210 memset (buffer, 0, BUFFER_SIZE);
211 }
ec15bdbc
LM
212 }
213
b9c1c40d
LM
214 free_completion_list (completion_list);
215
efdfb543 216 return ret;
ec15bdbc
LM
217}
218
219// test: calc.exe -h
220// test: calc.exe -h | awk '/usage:/ { rc=1 } END { exit (1-rc) }'
b449884c 221// test: echo 1 | calc.exe -v3 | grep -q value
ec15bdbc
LM
222// test: calc.exe -_ 2> /dev/null | awk 'END { if (NR == 0) { exit(0) } else exit (1) }'
223// test: calc.exe -_ 2>&1 | awk '/usage:/ { rc=1 } END { exit (1-rc) }'
8a33b741
LM
224// test: calc.exe error 2>&1 | grep -q 'invalid option'
225// test: calc.exe -p 2>&1 | grep -q 'missing precision'
226// test: calc.exe -v 2>&1 | grep -q 'missing verbose'
f2927108
LM
227// test: echo "1 + 2" | calc.exe | grep -q '=> 3'
228// test: echo "1 - 2" | calc.exe | grep -q '=> -1'
229// test: echo "2 * 3" | calc.exe | grep -q '=> 6'
941d160e 230// test: echo "1 / 2" | calc.exe | grep -q '=> 0\.5'
c47a9298 231// test: echo "8 % 3" | calc.exe | grep -q '=> 2'
941d160e 232// test: echo "-9 % 3.1" | calc.exe | grep -q '=> -2\.8'
f2927108 233// test: echo "2 ^ 3" | calc.exe | grep -q '=> 8'
941d160e
LM
234// test: echo "1e-1 + 2.34e3" | calc.exe | grep -q '=> 2340\.1'
235// test: echo "sqrt (2)" | calc.exe | grep -q '=> 1\.41421'
f2927108 236// test: echo "pow (2, 3)" | calc.exe | grep -q '=> 8'
941d160e
LM
237// test: echo "cos (2)" | calc.exe | grep -q '=> -0\.416147'
238// test: echo "sin (2)" | calc.exe | grep -q '=> 0\.909297'
239// test: echo "atan (2)" | calc.exe | grep -q '=> 1\.10715'
240// test: echo "exp (2)" | calc.exe | grep -q '=> 7\.38906'
241// test: echo "log (2)" | calc.exe | grep -q '=> 0\.693147'
f2927108 242// test: echo "2 + 3 - 4" | calc.exe | grep -q '=> 1'
941d160e
LM
243// test: echo "1 + cos (2 - 3)" | calc.exe | grep -q '=> 1\.5403'
244// test: echo "cos (1 / 2) * 3" | calc.exe | grep -q '=> 2\.63275'
f2927108 245// test: echo "1 + 4 * (2 - 3)" | calc.exe | grep -q '=> -3'
941d160e
LM
246// test: echo "(2 - 3) / 4" | calc.exe | grep -q '=> -0\.25'
247// test: echo "pow (8 - 3, 4 / 3)" | calc.exe | grep -q '=> 8\.54988'
f2927108
LM
248// test: echo "1 + -2" | calc.exe | grep -q '=> -1'
249// test: echo "1 - +2" | calc.exe | grep -q '=> -1'
250// test: echo "-1 + +2" | calc.exe | grep -q '=> 1'
251// test: echo "-1+2" | calc.exe | grep -q '=> 1'
252// test: echo "1-2" | calc.exe | grep -q '=> -1'
941d160e
LM
253// test: echo "1 * 2 / 3 + 4" | calc.exe | grep -q '=> 4\.66667'
254// test: echo "2 ^ 3 * 4 + 5" | calc.exe | grep -q '=> 37'
255// test: echo "2 + 3 * 4 ^ 5" | calc.exe | grep -q '=> 3074'
256// test: echo "2 ^ 3 * 4 + cos(5/6)" | calc.exe | grep -q '=> 32\.6724'
257// test: echo "95-6.3*15-1" | calc.exe | grep -q '=> -0.5'
258// test: echo "95 - 6.3 * 15 - 1" | calc.exe | grep -q '=> -0\.5'
259// test: echo "95-6.3+15" | calc.exe | grep -q '=> 103.7'
260// test: echo "-cos (0) + 1" | calc.exe | grep -q '=> -0'
32741902 261// test: echo "-cos(0)+1" | calc.exe | grep -q '=> -0'
89cf0955
LM
262// test: echo "quit" | calc.exe | grep -q 'bye'
263// test: echo "help" | calc.exe | grep -q 'miscellaneous'
031d7bba
LM
264// test: echo "1 + 2 *" | calc.exe | grep -q 'error'
265// test: echo "* 1 - 2" | calc.exe | grep -q 'error'
266// test: echo "2 + * 3" | calc.exe | grep -q 'error'
7fe742c9
LM
267// test: echo "2 + cos(3 *)" | calc.exe | grep -q 'error'
268// test: echo "2 + (foo)" | calc.exe | grep -q 'error'
686c4bc2 269// test: echo "2 + cos (pi +" | calc.exe | grep -q 'error'
3c0db5bc
LM
270// test: echo "2 + cos (pi" | calc.exe | grep -q 'error'
271// test: echo "(2 + " | calc.exe | grep -q 'error'
7fe742c9 272// test: echo "cos (1, 2)" | calc.exe | grep -q 'error'
031d7bba
LM
273// test: echo "sqrt 2" | calc.exe | grep -q 'error'
274// test: echo "pow (2)" | calc.exe | grep -q 'error'
4ee92d9d 275// test: echo "1.23456789" | calc.exe -p 3 | grep -q '1\.234'
8a33b741 276// test: echo . | calc.exe
941d160e 277// test: echo -e '1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1\n1 + 1' | calc.exe -n | grep -q 64
5898ff24 278// test: echo -e '1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1\n1 + 1' | calc.exe -n | grep -q 2
941d160e 279// test: echo -e '1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1\n1 + 1' | calc.exe | grep -q 64
5898ff24 280// test: echo -e '1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1\n1 + 1' | calc.exe | grep -q 2
9e52bbc5
LM
281// test: echo -e '-cos (1)\n1 + 1\n1 - 1\n1 * 1\n1 / 1\n3%2\n2^2\nsqrt (2)\ncos (0)\nsin (0)\ntan (0)\nacos (0)\nasin (0)\natan (0)\nlog (1)\nexp (1)\nabs (-1)\nceil (1.2)\nfloor (-1.2)\nans\ne\npi\nsto (1)\nrcl (2)\ndisp\nhelp\nquit' | calc.exe -v 3 | grep -q bye
282// test: echo -e '1 +\n1 -\n1 * 1\n1 /\n3%\n2^\nsqrt ()\ncos ()\nsin ()\ntan ()\nacos ()\nasin ()\natan ()\nlog ()\nexp ()\nabs ()\nceil ()\nfloor ()\n1 + (\n1+2(\n1+2cos\n1+2pi' | calc.exe | grep -c error | xargs test 21 =
4b16fb43
LM
283// test: echo -e '1 + 1\nans' | calc.exe -p 3 | grep -c 2 | xargs test 2 =
284// test: echo -e 'sin (pi / 2)' | calc.exe -p 4 | grep -q 1
471da7c9 285// test: echo -e 'e ^ 2' | calc.exe | grep -q '7\.38906'
4b16fb43
LM
286// test: echo -e '\n\n\n' | calc.exe | grep -qv 'error'
287// test: echo -e '\n\n\n' | calc.exe -n
6ba1dd0f
LM
288// test: echo -e '1.5\nsto (2)\n3 + rcl(2) * 4\nsto (5)' | calc.exe | grep -q 9
289// test: echo -e '1\nsto (0)\nsto (11)\nrcl (0)\nrcl (11)' | calc.exe | grep -c invalid | xargs test 4 =
3639df66 290// test: echo -e '1\nsto (2)\n3\nsto (5, 7)\nsto(9)\ndisp' | calc.exe | grep -q '0 1 0 0 7 0 0 0 7 0'
7fe742c9 291// test: echo -e '1+1 == 2' | calc.exe | grep -q '=> 1'
ca3e2a2f
LM
292// test: echo -e '1 + 1 == 2 - 0' | calc.exe | grep -q '=> 1'
293// test: echo -e '1 == 1 + 1 == 1' | calc.exe | grep -q '=> 0'
294// test: echo -e '1-cos(3*pi/4) != 0.5^-2' | calc.exe | grep -q '=> 1'
7fe742c9
LM
295// test: echo -e '1 >= 2' | calc.exe | grep -q '=> 0'
296// test: echo -e '2 >= 2' | calc.exe | grep -q '=> 1'
297// test: echo -e '1 <= 2' | calc.exe | grep -q '=> 1'
298// test: echo -e '2 <= 2' | calc.exe | grep -q '=> 1'
299// test: echo -e '1 > 2' | calc.exe | grep -q '=> 0'
300// test: echo -e '2 > 2' | calc.exe | grep -q '=> 0'
301// test: echo -e '1 < 2' | calc.exe | grep -q '=> 1'
302// test: echo -e '2 < 2' | calc.exe | grep -q '=> 0'
303// test: echo -e '1 == 1\n1 != 1\n1 >= 1\n1 <= 1\n1 > 1\n1 < 1\nquit' | calc.exe -v 3 | grep -q bye
ca3e2a2f
LM
304// test: echo -e '(3 == 3) & (2 > 1)' | calc.exe | grep -q '=> 1'
305// test: echo -e '(3 == 4) & (2 > 1)' | calc.exe | grep -q '=> 0'
306// test: echo -e '(3 == 3) & (2 > 2)' | calc.exe | grep -q '=> 0'
307// test: echo -e '(3 == 4) & (2 > 2)' | calc.exe | grep -q '=> 0'
308// test: echo -e '(3 == 3) | (2 > 1)' | calc.exe | grep -q '=> 1'
309// test: echo -e '(3 == 4) | (2 > 1)' | calc.exe | grep -q '=> 1'
310// test: echo -e '(3 == 3) | (2 > 2)' | calc.exe | grep -q '=> 1'
311// test: echo -e '(3 == 4) | (2 > 2)' | calc.exe | grep -q '=> 0'
312// test: echo -e '!(3 == 4)' | calc.exe | grep -q '=> 1'
313// test: echo -e '!(3 == 3)' | calc.exe | grep -q '=> 0'
ce6627f2 314// test: echo -e '1 & 1\n1 | 1\n!1\nquit' | calc.exe -v 3 | grep -qv error
ca3e2a2f
LM
315// test: echo -e '(3 == 3) & (4 > 2)' | calc.exe | grep -q '=> 1'
316// test: echo -e '3 == 3 & 4 > 2' | calc.exe | grep -q '=> 1'
45a631a8
LM
317// test: echo -e '1 + quit' | calc.exe | grep -q error
318// test: echo -e 'cos (quit)' | calc.exe | grep -q error
319// test: echo -e '(quit)' | calc.exe | grep -q error
320// test: echo -e 'cos 3.14\n!\n! 3 4' | calc.exe | grep -c error | xargs test 3 =
7e98ca87 321// test: echo -e 'sto (2, 3)\ncond (rcl (2) > 2, log (64), exp (75 / 10))' | calc.exe | grep -q '=> 4\.15888'
94b4e517 322// test: echo -e 'sto (2, 1)\ncond (rcl (2) > 2, log (64), exp (75 / 10))' | calc.exe | grep -q '=> 1808\.04'
7e98ca87 323// test: echo -e 'sto (2, 1)\ncond (rcl (2) > 2, log (64))' | calc.exe | grep -q '=> 0'
94b4e517 324// test: echo -e 'cond (0, 1, 2)' | calc.exe -v 3 | grep -q Cond
45a631a8 325// test: echo -e 'cond\ncond (\ncond (1 >0,'| calc.exe | grep -c error | xargs test 3 =
7e98ca87 326// test: echo -e 'sto (1, 4)\ninc (1)\ninc (1)\ndec (1)\ninc (1)\nrcl (1) == 6' | calc.exe -v 3 | grep -q '=> 1'
45a631a8
LM
327// test: echo -e 'inc\ninc (\ndec\ndec (' | calc.exe | grep -c error | xargs test 4 =
328// test: echo -e 'inc (11)\ndec (0)' | calc.exe | grep -c invalid | xargs test 2 =
4e3d2e04
LM
329// test: echo -e 'while (inc (1) < 100, sto (2, rcl (1) + rcl (2)))' | calc.exe | grep -q '=> 5050'
330// test: echo -e 'while\nwhile (inc (1) < 3,\nwhile (inc (1) < 100, sto (2, rcl (1) + rcl (2))' | calc.exe | grep -c error | xargs test 3 =
331// test: echo -e 'while (0, 1)' | calc.exe -v 3 | grep -q While
62d56da6
LM
332// test: echo -e '{sto (1, 1 + 1), rcl (1) * 3}' | calc.exe -v 3 | grep -q 'Program'
333// test: echo -e '{sto (1, 1 + 1), rcl (1) * 3}' | calc.exe | grep -q '=> 6'
45a631a8 334// test: echo -e '{\n{}\n{1, 2\n{sto (1, 1 + 1),\npow(2, {sto (1, 1 + 2), 2}, {rcl(2)})\n2 {sto (1, 1 + 1)}' | calc.exe | grep -c error | xargs test 6 =
d32ea0b0 335// test: echo -e '1 }\n1 )\n1 , 2\n ' | calc.exe | grep -c error | xargs test 3 =
db660e60 336// test: echo -e 'si\t\t (pi / 2)' | calc.exe | grep -q '=> 1'
ec15bdbc 337
715580ff
LM
338// Gauss sequence
339// test: echo -e '{sto (1, 0), sto (10, 0), while (inc (10) < 100, {sto (1, rcl (1) + rcl (10)), print (rcl (1))})}' | calc.exe | grep -q '=> 5050'
340
093e9e93 341// Fibonacci sequence
715580ff 342// test: echo -e '{sto (1, 1), sto (2, 1), sto (10, 1), while (inc (10) < 12 - 1, {sto (3, rcl (1) + rcl (2)), sto (1, rcl (2)), print (sto (2, rcl (3)))})}' | calc.exe | grep '=> 144'
093e9e93 343
ec15bdbc 344/* vim: set ts=4 sw=4 et: */