blob: 24509cd0c715f9bd3202b423802daba6aa5ba05f [file] [log] [blame]
Nicholas Flintham1e3d3112013-04-10 10:48:38 +01001/*
2 * Copyright (C) 2002 Roman Zippel <zippel@linux-m68k.org>
3 * Released under the terms of the GNU GPL v2.0.
4 */
5
6#include <stdio.h>
7#include <stdlib.h>
8#include <string.h>
9
10#include "lkc.h"
11
12#define DEBUG_EXPR 0
13
14struct expr *expr_alloc_symbol(struct symbol *sym)
15{
16 struct expr *e = calloc(1, sizeof(*e));
17 e->type = E_SYMBOL;
18 e->left.sym = sym;
19 return e;
20}
21
22struct expr *expr_alloc_one(enum expr_type type, struct expr *ce)
23{
24 struct expr *e = calloc(1, sizeof(*e));
25 e->type = type;
26 e->left.expr = ce;
27 return e;
28}
29
30struct expr *expr_alloc_two(enum expr_type type, struct expr *e1, struct expr *e2)
31{
32 struct expr *e = calloc(1, sizeof(*e));
33 e->type = type;
34 e->left.expr = e1;
35 e->right.expr = e2;
36 return e;
37}
38
39struct expr *expr_alloc_comp(enum expr_type type, struct symbol *s1, struct symbol *s2)
40{
41 struct expr *e = calloc(1, sizeof(*e));
42 e->type = type;
43 e->left.sym = s1;
44 e->right.sym = s2;
45 return e;
46}
47
48struct expr *expr_alloc_and(struct expr *e1, struct expr *e2)
49{
50 if (!e1)
51 return e2;
52 return e2 ? expr_alloc_two(E_AND, e1, e2) : e1;
53}
54
55struct expr *expr_alloc_or(struct expr *e1, struct expr *e2)
56{
57 if (!e1)
58 return e2;
59 return e2 ? expr_alloc_two(E_OR, e1, e2) : e1;
60}
61
62struct expr *expr_copy(const struct expr *org)
63{
64 struct expr *e;
65
66 if (!org)
67 return NULL;
68
69 e = malloc(sizeof(*org));
70 memcpy(e, org, sizeof(*org));
71 switch (org->type) {
72 case E_SYMBOL:
73 e->left = org->left;
74 break;
75 case E_NOT:
76 e->left.expr = expr_copy(org->left.expr);
77 break;
78 case E_EQUAL:
79 case E_UNEQUAL:
80 e->left.sym = org->left.sym;
81 e->right.sym = org->right.sym;
82 break;
83 case E_AND:
84 case E_OR:
85 case E_LIST:
86 e->left.expr = expr_copy(org->left.expr);
87 e->right.expr = expr_copy(org->right.expr);
88 break;
89 default:
90 printf("can't copy type %d\n", e->type);
91 free(e);
92 e = NULL;
93 break;
94 }
95
96 return e;
97}
98
99void expr_free(struct expr *e)
100{
101 if (!e)
102 return;
103
104 switch (e->type) {
105 case E_SYMBOL:
106 break;
107 case E_NOT:
108 expr_free(e->left.expr);
109 return;
110 case E_EQUAL:
111 case E_UNEQUAL:
112 break;
113 case E_OR:
114 case E_AND:
115 expr_free(e->left.expr);
116 expr_free(e->right.expr);
117 break;
118 default:
119 printf("how to free type %d?\n", e->type);
120 break;
121 }
122 free(e);
123}
124
125static int trans_count;
126
127#define e1 (*ep1)
128#define e2 (*ep2)
129
130static void __expr_eliminate_eq(enum expr_type type, struct expr **ep1, struct expr **ep2)
131{
132 if (e1->type == type) {
133 __expr_eliminate_eq(type, &e1->left.expr, &e2);
134 __expr_eliminate_eq(type, &e1->right.expr, &e2);
135 return;
136 }
137 if (e2->type == type) {
138 __expr_eliminate_eq(type, &e1, &e2->left.expr);
139 __expr_eliminate_eq(type, &e1, &e2->right.expr);
140 return;
141 }
142 if (e1->type == E_SYMBOL && e2->type == E_SYMBOL &&
143 e1->left.sym == e2->left.sym &&
144 (e1->left.sym == &symbol_yes || e1->left.sym == &symbol_no))
145 return;
146 if (!expr_eq(e1, e2))
147 return;
148 trans_count++;
149 expr_free(e1); expr_free(e2);
150 switch (type) {
151 case E_OR:
152 e1 = expr_alloc_symbol(&symbol_no);
153 e2 = expr_alloc_symbol(&symbol_no);
154 break;
155 case E_AND:
156 e1 = expr_alloc_symbol(&symbol_yes);
157 e2 = expr_alloc_symbol(&symbol_yes);
158 break;
159 default:
160 ;
161 }
162}
163
164void expr_eliminate_eq(struct expr **ep1, struct expr **ep2)
165{
166 if (!e1 || !e2)
167 return;
168 switch (e1->type) {
169 case E_OR:
170 case E_AND:
171 __expr_eliminate_eq(e1->type, ep1, ep2);
172 default:
173 ;
174 }
175 if (e1->type != e2->type) switch (e2->type) {
176 case E_OR:
177 case E_AND:
178 __expr_eliminate_eq(e2->type, ep1, ep2);
179 default:
180 ;
181 }
182 e1 = expr_eliminate_yn(e1);
183 e2 = expr_eliminate_yn(e2);
184}
185
186#undef e1
187#undef e2
188
189int expr_eq(struct expr *e1, struct expr *e2)
190{
191 int res, old_count;
192
193 if (e1->type != e2->type)
194 return 0;
195 switch (e1->type) {
196 case E_EQUAL:
197 case E_UNEQUAL:
198 return e1->left.sym == e2->left.sym && e1->right.sym == e2->right.sym;
199 case E_SYMBOL:
200 return e1->left.sym == e2->left.sym;
201 case E_NOT:
202 return expr_eq(e1->left.expr, e2->left.expr);
203 case E_AND:
204 case E_OR:
205 e1 = expr_copy(e1);
206 e2 = expr_copy(e2);
207 old_count = trans_count;
208 expr_eliminate_eq(&e1, &e2);
209 res = (e1->type == E_SYMBOL && e2->type == E_SYMBOL &&
210 e1->left.sym == e2->left.sym);
211 expr_free(e1);
212 expr_free(e2);
213 trans_count = old_count;
214 return res;
215 case E_LIST:
216 case E_RANGE:
217 case E_NONE:
218 ;
219 }
220
221 if (DEBUG_EXPR) {
222 expr_fprint(e1, stdout);
223 printf(" = ");
224 expr_fprint(e2, stdout);
225 printf(" ?\n");
226 }
227
228 return 0;
229}
230
231struct expr *expr_eliminate_yn(struct expr *e)
232{
233 struct expr *tmp;
234
235 if (e) switch (e->type) {
236 case E_AND:
237 e->left.expr = expr_eliminate_yn(e->left.expr);
238 e->right.expr = expr_eliminate_yn(e->right.expr);
239 if (e->left.expr->type == E_SYMBOL) {
240 if (e->left.expr->left.sym == &symbol_no) {
241 expr_free(e->left.expr);
242 expr_free(e->right.expr);
243 e->type = E_SYMBOL;
244 e->left.sym = &symbol_no;
245 e->right.expr = NULL;
246 return e;
247 } else if (e->left.expr->left.sym == &symbol_yes) {
248 free(e->left.expr);
249 tmp = e->right.expr;
250 *e = *(e->right.expr);
251 free(tmp);
252 return e;
253 }
254 }
255 if (e->right.expr->type == E_SYMBOL) {
256 if (e->right.expr->left.sym == &symbol_no) {
257 expr_free(e->left.expr);
258 expr_free(e->right.expr);
259 e->type = E_SYMBOL;
260 e->left.sym = &symbol_no;
261 e->right.expr = NULL;
262 return e;
263 } else if (e->right.expr->left.sym == &symbol_yes) {
264 free(e->right.expr);
265 tmp = e->left.expr;
266 *e = *(e->left.expr);
267 free(tmp);
268 return e;
269 }
270 }
271 break;
272 case E_OR:
273 e->left.expr = expr_eliminate_yn(e->left.expr);
274 e->right.expr = expr_eliminate_yn(e->right.expr);
275 if (e->left.expr->type == E_SYMBOL) {
276 if (e->left.expr->left.sym == &symbol_no) {
277 free(e->left.expr);
278 tmp = e->right.expr;
279 *e = *(e->right.expr);
280 free(tmp);
281 return e;
282 } else if (e->left.expr->left.sym == &symbol_yes) {
283 expr_free(e->left.expr);
284 expr_free(e->right.expr);
285 e->type = E_SYMBOL;
286 e->left.sym = &symbol_yes;
287 e->right.expr = NULL;
288 return e;
289 }
290 }
291 if (e->right.expr->type == E_SYMBOL) {
292 if (e->right.expr->left.sym == &symbol_no) {
293 free(e->right.expr);
294 tmp = e->left.expr;
295 *e = *(e->left.expr);
296 free(tmp);
297 return e;
298 } else if (e->right.expr->left.sym == &symbol_yes) {
299 expr_free(e->left.expr);
300 expr_free(e->right.expr);
301 e->type = E_SYMBOL;
302 e->left.sym = &symbol_yes;
303 e->right.expr = NULL;
304 return e;
305 }
306 }
307 break;
308 default:
309 ;
310 }
311 return e;
312}
313
314struct expr *expr_trans_bool(struct expr *e)
315{
316 if (!e)
317 return NULL;
318 switch (e->type) {
319 case E_AND:
320 case E_OR:
321 case E_NOT:
322 e->left.expr = expr_trans_bool(e->left.expr);
323 e->right.expr = expr_trans_bool(e->right.expr);
324 break;
325 case E_UNEQUAL:
326
327 if (e->left.sym->type == S_TRISTATE) {
328 if (e->right.sym == &symbol_no) {
329 e->type = E_SYMBOL;
330 e->right.sym = NULL;
331 }
332 }
333 break;
334 default:
335 ;
336 }
337 return e;
338}
339
340static struct expr *expr_join_or(struct expr *e1, struct expr *e2)
341{
342 struct expr *tmp;
343 struct symbol *sym1, *sym2;
344
345 if (expr_eq(e1, e2))
346 return expr_copy(e1);
347 if (e1->type != E_EQUAL && e1->type != E_UNEQUAL && e1->type != E_SYMBOL && e1->type != E_NOT)
348 return NULL;
349 if (e2->type != E_EQUAL && e2->type != E_UNEQUAL && e2->type != E_SYMBOL && e2->type != E_NOT)
350 return NULL;
351 if (e1->type == E_NOT) {
352 tmp = e1->left.expr;
353 if (tmp->type != E_EQUAL && tmp->type != E_UNEQUAL && tmp->type != E_SYMBOL)
354 return NULL;
355 sym1 = tmp->left.sym;
356 } else
357 sym1 = e1->left.sym;
358 if (e2->type == E_NOT) {
359 if (e2->left.expr->type != E_SYMBOL)
360 return NULL;
361 sym2 = e2->left.expr->left.sym;
362 } else
363 sym2 = e2->left.sym;
364 if (sym1 != sym2)
365 return NULL;
366 if (sym1->type != S_BOOLEAN && sym1->type != S_TRISTATE)
367 return NULL;
368 if (sym1->type == S_TRISTATE) {
369 if (e1->type == E_EQUAL && e2->type == E_EQUAL &&
370 ((e1->right.sym == &symbol_yes && e2->right.sym == &symbol_mod) ||
371 (e1->right.sym == &symbol_mod && e2->right.sym == &symbol_yes))) {
372
373 return expr_alloc_comp(E_UNEQUAL, sym1, &symbol_no);
374 }
375 if (e1->type == E_EQUAL && e2->type == E_EQUAL &&
376 ((e1->right.sym == &symbol_yes && e2->right.sym == &symbol_no) ||
377 (e1->right.sym == &symbol_no && e2->right.sym == &symbol_yes))) {
378
379 return expr_alloc_comp(E_UNEQUAL, sym1, &symbol_mod);
380 }
381 if (e1->type == E_EQUAL && e2->type == E_EQUAL &&
382 ((e1->right.sym == &symbol_mod && e2->right.sym == &symbol_no) ||
383 (e1->right.sym == &symbol_no && e2->right.sym == &symbol_mod))) {
384
385 return expr_alloc_comp(E_UNEQUAL, sym1, &symbol_yes);
386 }
387 }
388 if (sym1->type == S_BOOLEAN && sym1 == sym2) {
389 if ((e1->type == E_NOT && e1->left.expr->type == E_SYMBOL && e2->type == E_SYMBOL) ||
390 (e2->type == E_NOT && e2->left.expr->type == E_SYMBOL && e1->type == E_SYMBOL))
391 return expr_alloc_symbol(&symbol_yes);
392 }
393
394 if (DEBUG_EXPR) {
395 printf("optimize (");
396 expr_fprint(e1, stdout);
397 printf(") || (");
398 expr_fprint(e2, stdout);
399 printf(")?\n");
400 }
401 return NULL;
402}
403
404static struct expr *expr_join_and(struct expr *e1, struct expr *e2)
405{
406 struct expr *tmp;
407 struct symbol *sym1, *sym2;
408
409 if (expr_eq(e1, e2))
410 return expr_copy(e1);
411 if (e1->type != E_EQUAL && e1->type != E_UNEQUAL && e1->type != E_SYMBOL && e1->type != E_NOT)
412 return NULL;
413 if (e2->type != E_EQUAL && e2->type != E_UNEQUAL && e2->type != E_SYMBOL && e2->type != E_NOT)
414 return NULL;
415 if (e1->type == E_NOT) {
416 tmp = e1->left.expr;
417 if (tmp->type != E_EQUAL && tmp->type != E_UNEQUAL && tmp->type != E_SYMBOL)
418 return NULL;
419 sym1 = tmp->left.sym;
420 } else
421 sym1 = e1->left.sym;
422 if (e2->type == E_NOT) {
423 if (e2->left.expr->type != E_SYMBOL)
424 return NULL;
425 sym2 = e2->left.expr->left.sym;
426 } else
427 sym2 = e2->left.sym;
428 if (sym1 != sym2)
429 return NULL;
430 if (sym1->type != S_BOOLEAN && sym1->type != S_TRISTATE)
431 return NULL;
432
433 if ((e1->type == E_SYMBOL && e2->type == E_EQUAL && e2->right.sym == &symbol_yes) ||
434 (e2->type == E_SYMBOL && e1->type == E_EQUAL && e1->right.sym == &symbol_yes))
435
436 return expr_alloc_comp(E_EQUAL, sym1, &symbol_yes);
437
438 if ((e1->type == E_SYMBOL && e2->type == E_UNEQUAL && e2->right.sym == &symbol_no) ||
439 (e2->type == E_SYMBOL && e1->type == E_UNEQUAL && e1->right.sym == &symbol_no))
440
441 return expr_alloc_symbol(sym1);
442
443 if ((e1->type == E_SYMBOL && e2->type == E_UNEQUAL && e2->right.sym == &symbol_mod) ||
444 (e2->type == E_SYMBOL && e1->type == E_UNEQUAL && e1->right.sym == &symbol_mod))
445
446 return expr_alloc_comp(E_EQUAL, sym1, &symbol_yes);
447
448 if (sym1->type == S_TRISTATE) {
449 if (e1->type == E_EQUAL && e2->type == E_UNEQUAL) {
450
451 sym2 = e1->right.sym;
452 if ((e2->right.sym->flags & SYMBOL_CONST) && (sym2->flags & SYMBOL_CONST))
453 return sym2 != e2->right.sym ? expr_alloc_comp(E_EQUAL, sym1, sym2)
454 : expr_alloc_symbol(&symbol_no);
455 }
456 if (e1->type == E_UNEQUAL && e2->type == E_EQUAL) {
457
458 sym2 = e2->right.sym;
459 if ((e1->right.sym->flags & SYMBOL_CONST) && (sym2->flags & SYMBOL_CONST))
460 return sym2 != e1->right.sym ? expr_alloc_comp(E_EQUAL, sym1, sym2)
461 : expr_alloc_symbol(&symbol_no);
462 }
463 if (e1->type == E_UNEQUAL && e2->type == E_UNEQUAL &&
464 ((e1->right.sym == &symbol_yes && e2->right.sym == &symbol_no) ||
465 (e1->right.sym == &symbol_no && e2->right.sym == &symbol_yes)))
466
467 return expr_alloc_comp(E_EQUAL, sym1, &symbol_mod);
468
469 if (e1->type == E_UNEQUAL && e2->type == E_UNEQUAL &&
470 ((e1->right.sym == &symbol_yes && e2->right.sym == &symbol_mod) ||
471 (e1->right.sym == &symbol_mod && e2->right.sym == &symbol_yes)))
472
473 return expr_alloc_comp(E_EQUAL, sym1, &symbol_no);
474
475 if (e1->type == E_UNEQUAL && e2->type == E_UNEQUAL &&
476 ((e1->right.sym == &symbol_mod && e2->right.sym == &symbol_no) ||
477 (e1->right.sym == &symbol_no && e2->right.sym == &symbol_mod)))
478
479 return expr_alloc_comp(E_EQUAL, sym1, &symbol_yes);
480
481 if ((e1->type == E_SYMBOL && e2->type == E_EQUAL && e2->right.sym == &symbol_mod) ||
482 (e2->type == E_SYMBOL && e1->type == E_EQUAL && e1->right.sym == &symbol_mod) ||
483 (e1->type == E_SYMBOL && e2->type == E_UNEQUAL && e2->right.sym == &symbol_yes) ||
484 (e2->type == E_SYMBOL && e1->type == E_UNEQUAL && e1->right.sym == &symbol_yes))
485 return NULL;
486 }
487
488 if (DEBUG_EXPR) {
489 printf("optimize (");
490 expr_fprint(e1, stdout);
491 printf(") && (");
492 expr_fprint(e2, stdout);
493 printf(")?\n");
494 }
495 return NULL;
496}
497
498static void expr_eliminate_dups1(enum expr_type type, struct expr **ep1, struct expr **ep2)
499{
500#define e1 (*ep1)
501#define e2 (*ep2)
502 struct expr *tmp;
503
504 if (e1->type == type) {
505 expr_eliminate_dups1(type, &e1->left.expr, &e2);
506 expr_eliminate_dups1(type, &e1->right.expr, &e2);
507 return;
508 }
509 if (e2->type == type) {
510 expr_eliminate_dups1(type, &e1, &e2->left.expr);
511 expr_eliminate_dups1(type, &e1, &e2->right.expr);
512 return;
513 }
514 if (e1 == e2)
515 return;
516
517 switch (e1->type) {
518 case E_OR: case E_AND:
519 expr_eliminate_dups1(e1->type, &e1, &e1);
520 default:
521 ;
522 }
523
524 switch (type) {
525 case E_OR:
526 tmp = expr_join_or(e1, e2);
527 if (tmp) {
528 expr_free(e1); expr_free(e2);
529 e1 = expr_alloc_symbol(&symbol_no);
530 e2 = tmp;
531 trans_count++;
532 }
533 break;
534 case E_AND:
535 tmp = expr_join_and(e1, e2);
536 if (tmp) {
537 expr_free(e1); expr_free(e2);
538 e1 = expr_alloc_symbol(&symbol_yes);
539 e2 = tmp;
540 trans_count++;
541 }
542 break;
543 default:
544 ;
545 }
546#undef e1
547#undef e2
548}
549
550static void expr_eliminate_dups2(enum expr_type type, struct expr **ep1, struct expr **ep2)
551{
552#define e1 (*ep1)
553#define e2 (*ep2)
554 struct expr *tmp, *tmp1, *tmp2;
555
556 if (e1->type == type) {
557 expr_eliminate_dups2(type, &e1->left.expr, &e2);
558 expr_eliminate_dups2(type, &e1->right.expr, &e2);
559 return;
560 }
561 if (e2->type == type) {
562 expr_eliminate_dups2(type, &e1, &e2->left.expr);
563 expr_eliminate_dups2(type, &e1, &e2->right.expr);
564 }
565 if (e1 == e2)
566 return;
567
568 switch (e1->type) {
569 case E_OR:
570 expr_eliminate_dups2(e1->type, &e1, &e1);
571
572 tmp1 = expr_transform(expr_alloc_one(E_NOT, expr_copy(e1)));
573 tmp2 = expr_copy(e2);
574 tmp = expr_extract_eq_and(&tmp1, &tmp2);
575 if (expr_is_yes(tmp1)) {
576 expr_free(e1);
577 e1 = expr_alloc_symbol(&symbol_no);
578 trans_count++;
579 }
580 expr_free(tmp2);
581 expr_free(tmp1);
582 expr_free(tmp);
583 break;
584 case E_AND:
585 expr_eliminate_dups2(e1->type, &e1, &e1);
586
587 tmp1 = expr_transform(expr_alloc_one(E_NOT, expr_copy(e1)));
588 tmp2 = expr_copy(e2);
589 tmp = expr_extract_eq_or(&tmp1, &tmp2);
590 if (expr_is_no(tmp1)) {
591 expr_free(e1);
592 e1 = expr_alloc_symbol(&symbol_yes);
593 trans_count++;
594 }
595 expr_free(tmp2);
596 expr_free(tmp1);
597 expr_free(tmp);
598 break;
599 default:
600 ;
601 }
602#undef e1
603#undef e2
604}
605
606struct expr *expr_eliminate_dups(struct expr *e)
607{
608 int oldcount;
609 if (!e)
610 return e;
611
612 oldcount = trans_count;
613 while (1) {
614 trans_count = 0;
615 switch (e->type) {
616 case E_OR: case E_AND:
617 expr_eliminate_dups1(e->type, &e, &e);
618 expr_eliminate_dups2(e->type, &e, &e);
619 default:
620 ;
621 }
622 if (!trans_count)
623 break;
624 e = expr_eliminate_yn(e);
625 }
626 trans_count = oldcount;
627 return e;
628}
629
630struct expr *expr_transform(struct expr *e)
631{
632 struct expr *tmp;
633
634 if (!e)
635 return NULL;
636 switch (e->type) {
637 case E_EQUAL:
638 case E_UNEQUAL:
639 case E_SYMBOL:
640 case E_LIST:
641 break;
642 default:
643 e->left.expr = expr_transform(e->left.expr);
644 e->right.expr = expr_transform(e->right.expr);
645 }
646
647 switch (e->type) {
648 case E_EQUAL:
649 if (e->left.sym->type != S_BOOLEAN)
650 break;
651 if (e->right.sym == &symbol_no) {
652 e->type = E_NOT;
653 e->left.expr = expr_alloc_symbol(e->left.sym);
654 e->right.sym = NULL;
655 break;
656 }
657 if (e->right.sym == &symbol_mod) {
658 printf("boolean symbol %s tested for 'm'? test forced to 'n'\n", e->left.sym->name);
659 e->type = E_SYMBOL;
660 e->left.sym = &symbol_no;
661 e->right.sym = NULL;
662 break;
663 }
664 if (e->right.sym == &symbol_yes) {
665 e->type = E_SYMBOL;
666 e->right.sym = NULL;
667 break;
668 }
669 break;
670 case E_UNEQUAL:
671 if (e->left.sym->type != S_BOOLEAN)
672 break;
673 if (e->right.sym == &symbol_no) {
674 e->type = E_SYMBOL;
675 e->right.sym = NULL;
676 break;
677 }
678 if (e->right.sym == &symbol_mod) {
679 printf("boolean symbol %s tested for 'm'? test forced to 'y'\n", e->left.sym->name);
680 e->type = E_SYMBOL;
681 e->left.sym = &symbol_yes;
682 e->right.sym = NULL;
683 break;
684 }
685 if (e->right.sym == &symbol_yes) {
686 e->type = E_NOT;
687 e->left.expr = expr_alloc_symbol(e->left.sym);
688 e->right.sym = NULL;
689 break;
690 }
691 break;
692 case E_NOT:
693 switch (e->left.expr->type) {
694 case E_NOT:
695
696 tmp = e->left.expr->left.expr;
697 free(e->left.expr);
698 free(e);
699 e = tmp;
700 e = expr_transform(e);
701 break;
702 case E_EQUAL:
703 case E_UNEQUAL:
704
705 tmp = e->left.expr;
706 free(e);
707 e = tmp;
708 e->type = e->type == E_EQUAL ? E_UNEQUAL : E_EQUAL;
709 break;
710 case E_OR:
711
712 tmp = e->left.expr;
713 e->type = E_AND;
714 e->right.expr = expr_alloc_one(E_NOT, tmp->right.expr);
715 tmp->type = E_NOT;
716 tmp->right.expr = NULL;
717 e = expr_transform(e);
718 break;
719 case E_AND:
720
721 tmp = e->left.expr;
722 e->type = E_OR;
723 e->right.expr = expr_alloc_one(E_NOT, tmp->right.expr);
724 tmp->type = E_NOT;
725 tmp->right.expr = NULL;
726 e = expr_transform(e);
727 break;
728 case E_SYMBOL:
729 if (e->left.expr->left.sym == &symbol_yes) {
730
731 tmp = e->left.expr;
732 free(e);
733 e = tmp;
734 e->type = E_SYMBOL;
735 e->left.sym = &symbol_no;
736 break;
737 }
738 if (e->left.expr->left.sym == &symbol_mod) {
739
740 tmp = e->left.expr;
741 free(e);
742 e = tmp;
743 e->type = E_SYMBOL;
744 e->left.sym = &symbol_mod;
745 break;
746 }
747 if (e->left.expr->left.sym == &symbol_no) {
748
749 tmp = e->left.expr;
750 free(e);
751 e = tmp;
752 e->type = E_SYMBOL;
753 e->left.sym = &symbol_yes;
754 break;
755 }
756 break;
757 default:
758 ;
759 }
760 break;
761 default:
762 ;
763 }
764 return e;
765}
766
767int expr_contains_symbol(struct expr *dep, struct symbol *sym)
768{
769 if (!dep)
770 return 0;
771
772 switch (dep->type) {
773 case E_AND:
774 case E_OR:
775 return expr_contains_symbol(dep->left.expr, sym) ||
776 expr_contains_symbol(dep->right.expr, sym);
777 case E_SYMBOL:
778 return dep->left.sym == sym;
779 case E_EQUAL:
780 case E_UNEQUAL:
781 return dep->left.sym == sym ||
782 dep->right.sym == sym;
783 case E_NOT:
784 return expr_contains_symbol(dep->left.expr, sym);
785 default:
786 ;
787 }
788 return 0;
789}
790
791bool expr_depends_symbol(struct expr *dep, struct symbol *sym)
792{
793 if (!dep)
794 return false;
795
796 switch (dep->type) {
797 case E_AND:
798 return expr_depends_symbol(dep->left.expr, sym) ||
799 expr_depends_symbol(dep->right.expr, sym);
800 case E_SYMBOL:
801 return dep->left.sym == sym;
802 case E_EQUAL:
803 if (dep->left.sym == sym) {
804 if (dep->right.sym == &symbol_yes || dep->right.sym == &symbol_mod)
805 return true;
806 }
807 break;
808 case E_UNEQUAL:
809 if (dep->left.sym == sym) {
810 if (dep->right.sym == &symbol_no)
811 return true;
812 }
813 break;
814 default:
815 ;
816 }
817 return false;
818}
819
820struct expr *expr_extract_eq_and(struct expr **ep1, struct expr **ep2)
821{
822 struct expr *tmp = NULL;
823 expr_extract_eq(E_AND, &tmp, ep1, ep2);
824 if (tmp) {
825 *ep1 = expr_eliminate_yn(*ep1);
826 *ep2 = expr_eliminate_yn(*ep2);
827 }
828 return tmp;
829}
830
831struct expr *expr_extract_eq_or(struct expr **ep1, struct expr **ep2)
832{
833 struct expr *tmp = NULL;
834 expr_extract_eq(E_OR, &tmp, ep1, ep2);
835 if (tmp) {
836 *ep1 = expr_eliminate_yn(*ep1);
837 *ep2 = expr_eliminate_yn(*ep2);
838 }
839 return tmp;
840}
841
842void expr_extract_eq(enum expr_type type, struct expr **ep, struct expr **ep1, struct expr **ep2)
843{
844#define e1 (*ep1)
845#define e2 (*ep2)
846 if (e1->type == type) {
847 expr_extract_eq(type, ep, &e1->left.expr, &e2);
848 expr_extract_eq(type, ep, &e1->right.expr, &e2);
849 return;
850 }
851 if (e2->type == type) {
852 expr_extract_eq(type, ep, ep1, &e2->left.expr);
853 expr_extract_eq(type, ep, ep1, &e2->right.expr);
854 return;
855 }
856 if (expr_eq(e1, e2)) {
857 *ep = *ep ? expr_alloc_two(type, *ep, e1) : e1;
858 expr_free(e2);
859 if (type == E_AND) {
860 e1 = expr_alloc_symbol(&symbol_yes);
861 e2 = expr_alloc_symbol(&symbol_yes);
862 } else if (type == E_OR) {
863 e1 = expr_alloc_symbol(&symbol_no);
864 e2 = expr_alloc_symbol(&symbol_no);
865 }
866 }
867#undef e1
868#undef e2
869}
870
871struct expr *expr_trans_compare(struct expr *e, enum expr_type type, struct symbol *sym)
872{
873 struct expr *e1, *e2;
874
875 if (!e) {
876 e = expr_alloc_symbol(sym);
877 if (type == E_UNEQUAL)
878 e = expr_alloc_one(E_NOT, e);
879 return e;
880 }
881 switch (e->type) {
882 case E_AND:
883 e1 = expr_trans_compare(e->left.expr, E_EQUAL, sym);
884 e2 = expr_trans_compare(e->right.expr, E_EQUAL, sym);
885 if (sym == &symbol_yes)
886 e = expr_alloc_two(E_AND, e1, e2);
887 if (sym == &symbol_no)
888 e = expr_alloc_two(E_OR, e1, e2);
889 if (type == E_UNEQUAL)
890 e = expr_alloc_one(E_NOT, e);
891 return e;
892 case E_OR:
893 e1 = expr_trans_compare(e->left.expr, E_EQUAL, sym);
894 e2 = expr_trans_compare(e->right.expr, E_EQUAL, sym);
895 if (sym == &symbol_yes)
896 e = expr_alloc_two(E_OR, e1, e2);
897 if (sym == &symbol_no)
898 e = expr_alloc_two(E_AND, e1, e2);
899 if (type == E_UNEQUAL)
900 e = expr_alloc_one(E_NOT, e);
901 return e;
902 case E_NOT:
903 return expr_trans_compare(e->left.expr, type == E_EQUAL ? E_UNEQUAL : E_EQUAL, sym);
904 case E_UNEQUAL:
905 case E_EQUAL:
906 if (type == E_EQUAL) {
907 if (sym == &symbol_yes)
908 return expr_copy(e);
909 if (sym == &symbol_mod)
910 return expr_alloc_symbol(&symbol_no);
911 if (sym == &symbol_no)
912 return expr_alloc_one(E_NOT, expr_copy(e));
913 } else {
914 if (sym == &symbol_yes)
915 return expr_alloc_one(E_NOT, expr_copy(e));
916 if (sym == &symbol_mod)
917 return expr_alloc_symbol(&symbol_yes);
918 if (sym == &symbol_no)
919 return expr_copy(e);
920 }
921 break;
922 case E_SYMBOL:
923 return expr_alloc_comp(type, e->left.sym, sym);
924 case E_LIST:
925 case E_RANGE:
926 case E_NONE:
927 ;
928 }
929 return NULL;
930}
931
932tristate expr_calc_value(struct expr *e)
933{
934 tristate val1, val2;
935 const char *str1, *str2;
936
937 if (!e)
938 return yes;
939
940 switch (e->type) {
941 case E_SYMBOL:
942 sym_calc_value(e->left.sym);
943 return e->left.sym->curr.tri;
944 case E_AND:
945 val1 = expr_calc_value(e->left.expr);
946 val2 = expr_calc_value(e->right.expr);
947 return EXPR_AND(val1, val2);
948 case E_OR:
949 val1 = expr_calc_value(e->left.expr);
950 val2 = expr_calc_value(e->right.expr);
951 return EXPR_OR(val1, val2);
952 case E_NOT:
953 val1 = expr_calc_value(e->left.expr);
954 return EXPR_NOT(val1);
955 case E_EQUAL:
956 sym_calc_value(e->left.sym);
957 sym_calc_value(e->right.sym);
958 str1 = sym_get_string_value(e->left.sym);
959 str2 = sym_get_string_value(e->right.sym);
960 return !strcmp(str1, str2) ? yes : no;
961 case E_UNEQUAL:
962 sym_calc_value(e->left.sym);
963 sym_calc_value(e->right.sym);
964 str1 = sym_get_string_value(e->left.sym);
965 str2 = sym_get_string_value(e->right.sym);
966 return !strcmp(str1, str2) ? no : yes;
967 default:
968 printf("expr_calc_value: %d?\n", e->type);
969 return no;
970 }
971}
972
973int expr_compare_type(enum expr_type t1, enum expr_type t2)
974{
975#if 0
976 return 1;
977#else
978 if (t1 == t2)
979 return 0;
980 switch (t1) {
981 case E_EQUAL:
982 case E_UNEQUAL:
983 if (t2 == E_NOT)
984 return 1;
985 case E_NOT:
986 if (t2 == E_AND)
987 return 1;
988 case E_AND:
989 if (t2 == E_OR)
990 return 1;
991 case E_OR:
992 if (t2 == E_LIST)
993 return 1;
994 case E_LIST:
995 if (t2 == 0)
996 return 1;
997 default:
998 return -1;
999 }
1000 printf("[%dgt%d?]", t1, t2);
1001 return 0;
1002#endif
1003}
1004
1005static inline struct expr *
1006expr_get_leftmost_symbol(const struct expr *e)
1007{
1008
1009 if (e == NULL)
1010 return NULL;
1011
1012 while (e->type != E_SYMBOL)
1013 e = e->left.expr;
1014
1015 return expr_copy(e);
1016}
1017
1018struct expr *expr_simplify_unmet_dep(struct expr *e1, struct expr *e2)
1019{
1020 struct expr *ret;
1021
1022 switch (e1->type) {
1023 case E_OR:
1024 return expr_alloc_and(
1025 expr_simplify_unmet_dep(e1->left.expr, e2),
1026 expr_simplify_unmet_dep(e1->right.expr, e2));
1027 case E_AND: {
1028 struct expr *e;
1029 e = expr_alloc_and(expr_copy(e1), expr_copy(e2));
1030 e = expr_eliminate_dups(e);
1031 ret = (!expr_eq(e, e1)) ? e1 : NULL;
1032 expr_free(e);
1033 break;
1034 }
1035 default:
1036 ret = e1;
1037 break;
1038 }
1039
1040 return expr_get_leftmost_symbol(ret);
1041}
1042
1043void expr_print(struct expr *e, void (*fn)(void *, struct symbol *, const char *), void *data, int prevtoken)
1044{
1045 if (!e) {
1046 fn(data, NULL, "y");
1047 return;
1048 }
1049
1050 if (expr_compare_type(prevtoken, e->type) > 0)
1051 fn(data, NULL, "(");
1052 switch (e->type) {
1053 case E_SYMBOL:
1054 if (e->left.sym->name)
1055 fn(data, e->left.sym, e->left.sym->name);
1056 else
1057 fn(data, NULL, "<choice>");
1058 break;
1059 case E_NOT:
1060 fn(data, NULL, "!");
1061 expr_print(e->left.expr, fn, data, E_NOT);
1062 break;
1063 case E_EQUAL:
1064 if (e->left.sym->name)
1065 fn(data, e->left.sym, e->left.sym->name);
1066 else
1067 fn(data, NULL, "<choice>");
1068 fn(data, NULL, "=");
1069 fn(data, e->right.sym, e->right.sym->name);
1070 break;
1071 case E_UNEQUAL:
1072 if (e->left.sym->name)
1073 fn(data, e->left.sym, e->left.sym->name);
1074 else
1075 fn(data, NULL, "<choice>");
1076 fn(data, NULL, "!=");
1077 fn(data, e->right.sym, e->right.sym->name);
1078 break;
1079 case E_OR:
1080 expr_print(e->left.expr, fn, data, E_OR);
1081 fn(data, NULL, " || ");
1082 expr_print(e->right.expr, fn, data, E_OR);
1083 break;
1084 case E_AND:
1085 expr_print(e->left.expr, fn, data, E_AND);
1086 fn(data, NULL, " && ");
1087 expr_print(e->right.expr, fn, data, E_AND);
1088 break;
1089 case E_LIST:
1090 fn(data, e->right.sym, e->right.sym->name);
1091 if (e->left.expr) {
1092 fn(data, NULL, " ^ ");
1093 expr_print(e->left.expr, fn, data, E_LIST);
1094 }
1095 break;
1096 case E_RANGE:
1097 fn(data, NULL, "[");
1098 fn(data, e->left.sym, e->left.sym->name);
1099 fn(data, NULL, " ");
1100 fn(data, e->right.sym, e->right.sym->name);
1101 fn(data, NULL, "]");
1102 break;
1103 default:
1104 {
1105 char buf[32];
1106 sprintf(buf, "<unknown type %d>", e->type);
1107 fn(data, NULL, buf);
1108 break;
1109 }
1110 }
1111 if (expr_compare_type(prevtoken, e->type) > 0)
1112 fn(data, NULL, ")");
1113}
1114
1115static void expr_print_file_helper(void *data, struct symbol *sym, const char *str)
1116{
1117 xfwrite(str, strlen(str), 1, data);
1118}
1119
1120void expr_fprint(struct expr *e, FILE *out)
1121{
1122 expr_print(e, expr_print_file_helper, out, E_NONE);
1123}
1124
1125static void expr_print_gstr_helper(void *data, struct symbol *sym, const char *str)
1126{
1127 struct gstr *gs = (struct gstr*)data;
1128 const char *sym_str = NULL;
1129
1130 if (sym)
1131 sym_str = sym_get_string_value(sym);
1132
1133 if (gs->max_width) {
1134 unsigned extra_length = strlen(str);
1135 const char *last_cr = strrchr(gs->s, '\n');
1136 unsigned last_line_length;
1137
1138 if (sym_str)
1139 extra_length += 4 + strlen(sym_str);
1140
1141 if (!last_cr)
1142 last_cr = gs->s;
1143
1144 last_line_length = strlen(gs->s) - (last_cr - gs->s);
1145
1146 if ((last_line_length + extra_length) > gs->max_width)
1147 str_append(gs, "\\\n");
1148 }
1149
1150 str_append(gs, str);
1151 if (sym && sym->type != S_UNKNOWN)
1152 str_printf(gs, " [=%s]", sym_str);
1153}
1154
1155void expr_gstr_print(struct expr *e, struct gstr *gs)
1156{
1157 expr_print(e, expr_print_gstr_helper, gs, E_NONE);
1158}