struct-ast.toy
2.48 KB
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
# RUN: toyc-ch7 %s -emit=ast 2>&1 | FileCheck %s
struct Struct {
var a;
var b;
}
# User defined generic function may operate on struct types as well.
def multiply_transpose(Struct value) {
# We can access the elements of a struct via the '.' operator.
return transpose(value.a) * transpose(value.b);
}
def main() {
# We initialize struct values using a composite initializer.
Struct value = {[[1, 2, 3], [4, 5, 6]], [[1, 2, 3], [4, 5, 6]]};
# We pass these arguments to functions like we do with variables.
var c = multiply_transpose(value);
print(c);
}
# CHECK: Module:
# CHECK-NEXT: Struct: Struct @{{.*}}struct-ast.toy:3:1
# CHECK-NEXT: Variables: [
# CHECK-NEXT: VarDecl a<> @{{.*}}struct-ast.toy:4:3
# CHECK-NEXT: VarDecl b<> @{{.*}}struct-ast.toy:5:3
# CHECK-NEXT: ]
# CHECK-NEXT:Function
# CHECK-NEXT: Proto 'multiply_transpose' @{{.*}}struct-ast.toy:9:1'
# CHECK-NEXT: Params: [value]
# CHECK-NEXT: Block {
# CHECK-NEXT: Return
# CHECK-NEXT: BinOp: * @{{.*}}struct-ast.toy:11:31
# CHECK-NEXT: Call 'transpose' [ @{{.*}}struct-ast.toy:11:10
# CHECK-NEXT: BinOp: . @{{.*}}struct-ast.toy:11:26
# CHECK-NEXT: var: value @{{.*}}struct-ast.toy:11:20
# CHECK-NEXT: var: a @{{.*}}struct-ast.toy:11:26
# CHECK-NEXT: ]
# CHECK-NEXT: Call 'transpose' [ @{{.*}}struct-ast.toy:11:31
# CHECK-NEXT: BinOp: . @{{.*}}struct-ast.toy:11:47
# CHECK-NEXT: var: value @{{.*}}struct-ast.toy:11:41
# CHECK-NEXT: var: b @{{.*}}struct-ast.toy:11:47
# CHECK-NEXT: ]
# CHECK-NEXT: }
# CHECK-NEXT:Function
# CHECK-NEXT: Proto 'main' @{{.*}}struct-ast.toy:14:1'
# CHECK-NEXT: Params: []
# CHECK-NEXT: Block {
# CHECK-NEXT: VarDecl value<Struct> @{{.*}}struct-ast.toy:16:3
# CHECK-NEXT: Struct Literal: Literal: <2, 3>[ <3>[ 1.000000e+00, 2.000000e+00, 3.000000e+00], <3>[ 4.000000e+00, 5.000000e+00, 6.000000e+00]] @{{.*}}struct-ast.toy:16:19
# CHECK-NEXT: Literal: <2, 3>[ <3>[ 1.000000e+00, 2.000000e+00, 3.000000e+00], <3>[ 4.000000e+00, 5.000000e+00, 6.000000e+00]] @{{.*}}struct-ast.toy:16:43
# CHECK-NEXT: @{{.*}}struct-ast.toy:16:18
# CHECK-NEXT: VarDecl c<> @{{.*}}struct-ast.toy:19:3
# CHECK-NEXT: Call 'multiply_transpose' [ @{{.*}}struct-ast.toy:19:11
# CHECK-NEXT: var: value @{{.*}}struct-ast.toy:19:30
# CHECK-NEXT: ]
# CHECK-NEXT: Print [ @{{.*}}struct-ast.toy:20:3
# CHECK-NEXT: var: c @{{.*}}struct-ast.toy:20:9
# CHECK-NEXT: ]
# CHECK-NEXT: }