blob: 56fffc97ca383dcf824a95a92772c4e92eed77b4 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
(ns cljcc.analyze.label-loops
(:require [cljcc.parser :as p]
[cljcc.exception :as exc]
[cljcc.analyze.resolve :as r]
[cljcc.schema :as s]
[cljcc.util :as util]
[malli.dev.pretty :as pretty]))
(defn- unique-identifier [identifier]
(util/create-identifier! identifier))
(defn- annotate-label [m label]
(assoc m :label label))
(defn- label-statement
([s]
(label-statement s nil))
([{:keys [statement-type] :as s} current-label]
(condp = statement-type
:break (if (nil? current-label)
(exc/analyzer-error "break statement outside of loop" s)
(p/break-statement-node current-label))
:continue (if (nil? current-label)
(exc/analyzer-error "continue statement outside of loop" s)
(p/continue-statement-node current-label))
:while (let [new-label (unique-identifier "while_label")
l-body (label-statement (:body s) new-label)
l-while (p/while-statement-node (:condition s) l-body)]
(annotate-label l-while new-label))
:do-while (let [new-label (unique-identifier "do_while_label")
l-body (label-statement (:body s) new-label)
l-do-while (p/do-while-statement-node (:condition s) l-body)]
(annotate-label l-do-while new-label))
:for (let [new-label (unique-identifier "for_label")
l-body (label-statement (:body s) new-label)
l-for (p/for-statement-node (:init s) (:condition s) (:post s) l-body)]
(annotate-label l-for new-label))
:if (if (:else-statement s)
(p/if-statement-node (:condition s)
(label-statement (:then-statement s) current-label)
(label-statement (:else-statement s) current-label))
(p/if-statement-node (:condition s)
(label-statement (:then-statement s) current-label)))
:compound (let [update-block-f (fn [item]
(if (= (:type item) :statement)
(label-statement item current-label)
item))
new-block (mapv update-block-f (:block s))]
(p/compound-statement-node new-block))
:return s
:expression s
:empty s
(exc/analyzer-error "invalid statement reached during loop labelling." s))))
(defn- label-loop-function-body [fn-declaration]
(let [statement? (fn [x] (= :statement (:type x)))
labelled-body (mapv (fn [block-item]
(if (statement? block-item)
(label-statement block-item)
block-item))
(:body fn-declaration))]
(assoc fn-declaration :body labelled-body)))
(defn label-loops
"Annotates labels on looping constructs.
Parameter:
program: List of declarations / blocks"
[program]
(let [fn-declaration? (fn [x] (= :function (:declaration-type x)))]
(mapv (fn [block]
(if (fn-declaration? block)
(label-loop-function-body block)
block))
program)))
(comment
(-> "./test-programs/example.c"
slurp
p/parse-from-src
r/resolve-program)
(-> "./test-programs/example.c"
slurp
p/parse-from-src
r/resolve-program
label-loops)
(pretty/explain
s/Program
(-> "./test-programs/example.c"
slurp
p/parse-from-src
r/resolve-program))
(pretty/explain
s/Program
(-> "./test-programs/example.c"
slurp
p/parse-from-src
r/resolve-program
label-loops))
())
|
