squash! mescc: Fix switch statements' fallthrough --support mes

This commit is contained in:
Janneke Nieuwenhuizen 2023-09-16 08:19:00 +02:00
parent 1ec88d0109
commit ef4f4bb25a
1 changed files with 27 additions and 23 deletions

View File

@ -1791,29 +1791,33 @@
(set! i (1+ i)))
n))))
(define (flatten-cases c)
(define (flatten-case case)
(pmatch case
((case ,test (case . ,body))
(append `((case ,test (expr-stmt))) (flatten-case `(case ,@body))))
((case ,test ,casebody (case . ,body))
(append `((case ,test ,casebody)) (flatten-case `(case ,@body))))
((default (case . ,body))
(append `((default (expr-stmt))) (flatten-case `(case ,@body))))
((default ,defbody (case . ,body))
(append `((default ,defbody)) (flatten-case `(case ,@body))))
((case ,test (default . ,body))
(append `((case ,test (expr-stmt))) (flatten-case `(default ,@body))))
((default ,rest)
(list case))
((case ,test)
(list case))
((case ,test ,expr)
(list case))
(,s (list s))))
(fold (lambda (x acc) (append acc (flatten-case x))) '() c))
(define (flatten-cases cases)
(define (flatten-case o cases)
(let ((c (pmatch o
((case ,test (case . ,body))
(append `((case ,test (expr-stmt)))
(flatten-case `(case ,@body))))
((case ,test ,case-body (case . ,body))
(append `((case ,test ,case-body))
(flatten-case `(case ,@body))))
((default (case . ,body))
(append `((default (expr-stmt)))
(flatten-case `(case ,@body))))
((default ,default-body (case . ,body))
(append `((default ,default-body))
(flatten-case `(case ,@body))))
((case ,test (default . ,body))
(append `((case ,test (expr-stmt)))
(flatten-case `(default ,@body))))
((default ,rest)
(list o))
((case ,test)
(list o))
((case ,test ,expr)
(list o))
(,s (list s)))))
(append c cases)))
(fold flatten-case '() cases))
(let* ((info (append-text info (ast->comment `(switch ,expr (compd-stmt (block-item-list (ellipsis)))))))
(statements (flatten-cases statements))