aboutsummaryrefslogtreecommitdiff
path: root/lib/control_flow.nom
blob: c4d6900a7c845bcea4330f321619b8b9b2f4261d (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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
require "lib/metaprogramming.nom"
require "lib/operators.nom"
require "lib/utils.nom"

# Conditionals
compile [if %condition %if_body] to code: ".."
    |if \(%condition as lua) then;
    |    \(%if_body as lua statements)
    |end;

compile [if %condition %if_body else %else_body] to code: ".."
    |if \(%condition as lua) then;
    |    \(%if_body as lua statements)
    |else;
    |    \(%else_body as lua statements)
    |end;

# Return
compile [return] to code: "do; return; end;"
compile [return %return-value] to code: "do; return \(%return-value as lua); end;"

# GOTOs
compile [-> %label] to code: ".."
    |::label_\(nomsu "var_to_lua_identifier" [%label])::;
compile [go to %label] to code: ".."
    |goto label_\(nomsu "var_to_lua_identifier" [%label]);

# Loop control flow
compile [stop, stop loop, break] to code: "break"
compile [stop for, stop for-loop, break for] to code: "goto break_for"
compile [stop repeat, stop repeat-loop, break repeat] to code: "goto break_repeat"
compile [stop %var, break %var] to code: ".."
    |goto break_\(nomsu "var_to_lua_identifier" [%var]);

compile [continue, continue loop] to code: "continue"
compile [continue for, continue for-loop] to code: "goto continue_for"
compile [continue repeat, continue repeat-loop] to code: "goto continue_repeat"
compile [continue %var, go to next %var, on to the next %var] to code: ".."
    |goto continue_\(nomsu "var_to_lua_identifier" [%var]);

# While loops
compile [repeat while %condition %body] to code: ".."
    |do;
    |    while \(%condition as lua) do;
    |        \(%body as lua statements)
    |        ::continue_repeat::;
    |    end;
    |    ::break_repeat::;
    |end;
parse [repeat %body] as: repeat while (true) %body
parse [repeat until %condition %body] as: repeat while (not %condition) %body

# Numeric range for loops
compile [..]
    for %var from %start to %stop by %step %body
    for %var from %start to %stop via %step %body
..to code: ".."
    |do;
    |    for i=\(%start as lua),\(%stop as lua),\(%step as lua) do;
    # This trashes the loop variables, just like in Python.
    |        \(%var as lua) = i;
    |        \(%body as lua statements)
    |        ::continue_for::;
    |        ::continue_\(nomsu "var_to_lua_identifier" [%var])::;
    |    end;
    |    ::break_for::;
    |    ::break_\(nomsu "var_to_lua_identifier" [%var])::;
    |end;
parse [for %var from %start to %stop %body] as: for %var from %start to %stop via 1 %body
parse [..]
    for all %start to %stop by %step %body
    for all %start to %stop via %step %body
..as: for % from %start to %stop via %step %body
parse [for all %start to %stop %body] as: for all %start to %stop via 1 %body

compile [for %var in %iterable %body] to code:
    ".."
        |do;
        |    for i,value in ipairs(\(%iterable as lua)) do;
        # This trashes the loop variables, just like in Python.
        |        \(%var as lua) = value;
        |        \(%body as lua statements)
        |        ::continue_for::;
        |        ::continue_\(nomsu "var_to_lua_identifier" [%var])::;
        |    end;
        |    ::break_for::;
        |    ::break_\(nomsu "var_to_lua_identifier" [%var])::;
        |end;
parse [for all %iterable %body] as: for % in %iterable %body


# Switch statement/multi-branch if
compile [when %body] to code:
    %result = "do;\n"
    %fallthroughs = []
    for %func-call in (%body's "value"):
        assert ((%func-call's "type") == "FunctionCall") ".."
            |Invalid format for 'when' statement. Only '*' blocks are allowed.
        %tokens = (%func-call's "value")
        %star = (%tokens -> 1)
        assert (lua expr "vars.star and vars.star.type == 'Word' and vars.star.value == '*'") ".."
            |Invalid format for 'when' statement. Lines must begin with '*'

        %condition = (%tokens -> 2)
        assert %condition ".."
            |Invalid format for 'when' statement. Lines must begin with '*' and have a condition or the word "else"

        %action = (%tokens -> 3)
        if (%action == (nil)):
            lua block "table.insert(vars.fallthroughs, vars.condition)"
            go to next %func-call

        if (lua expr "vars.condition.type == 'Word' and vars.condition.value == 'else'"):
            %result join= ".."
                |
                |do;
        ..else:
            %condition = (%condition as lua)
            for all %fallthroughs:
                %condition join= " or \(% as lua)"
            %result join= ".."
                |
                |if \(%condition) then;
        %result join= ".."
            |
            |    \(%action as lua statements)
            |    goto finished_when;
            |end;

        %fallthroughs = []

    %result join= "\n::finished_when::;\nend;"
    %result

# Switch statement
compile [when %branch-value == ? %body] to code:
    %result = "do;\nlocal branch_value = \(%branch-value as lua)"
    %fallthroughs = []
    for %func-call in (%body's "value"):
        assert ((%func-call's "type") == "FunctionCall") ".."
            |Invalid format for 'when' statement. Only '*' blocks are allowed.
        %tokens = (%func-call's "value")
        %star = (%tokens -> 1)
        assert (lua expr "vars.star and vars.star.type == 'Word' and vars.star.value == '*'") ".."
            |Invalid format for 'when' statement. Lines must begin with '*'

        %condition = (%tokens -> 2)
        assert %condition ".."
            |Invalid format for 'when' statement. Lines must begin with '*' and have a condition or the word "else"

        %action = (%tokens -> 3)
        if (%action == (nil)):
            lua block "table.insert(vars.fallthroughs, vars.condition)"
            go to next %func-call

        if (lua expr "vars.condition.type == 'Word' and vars.condition.value == 'else'"):
            %result join= ".."
                |
                |do;
        ..else:
            %condition = "branch_value == (\(%condition as lua))"
            for all %fallthroughs:
                %condition join= " or (branch_value == \(% as lua))"
            %result join= ".."
                |
                |if \(%condition) then;
        %result join= ".."
            |
            |    \(%action as lua statements)
            |    goto finished_when;
            |end;

        %fallthroughs = []

    %result join= "\n::finished_when::;\nend;"
    %result