diff options
author | Leo Tenenbaum <pommicket@gmail.com> | 2019-12-15 17:38:12 -0500 |
---|---|---|
committer | Leo Tenenbaum <pommicket@gmail.com> | 2019-12-15 17:38:12 -0500 |
commit | 388723942772aba7431cb80f106bf783c0145bf3 (patch) | |
tree | 722b2acee6dd5bfe6bc4f5dccfb0e76c5609b879 /tests/arr2/arr2.toc | |
parent | 054fec911b18dc0c8ec0180858b0bd00f60be90b (diff) |
got everything working for new test: arr2
Diffstat (limited to 'tests/arr2/arr2.toc')
-rw-r--r-- | tests/arr2/arr2.toc | 63 |
1 files changed, 63 insertions, 0 deletions
diff --git a/tests/arr2/arr2.toc b/tests/arr2/arr2.toc new file mode 100644 index 0000000..ea9ea79 --- /dev/null +++ b/tests/arr2/arr2.toc @@ -0,0 +1,63 @@ +puti ::= fn(x: int) { + #C("printf(\"%ld\\n\", (long)x); +"); +}; +putf ::= fn(x: float) { + #C("printf(\"%f\\n\", (double)x); +"); +}; + +// it would be nice if Arr.data.len == Arr.len (: but this will require some C code... +Arr ::= fn (t :: Type) Type { + struct { + data : []t; + len, cap : int; + } +}; + +arr_add ::= fn(t ::=, a : &Arr(t), x : t) { + if a.len >= a.cap { + a.cap = a.cap * 2 + 2; + new_data := new(t, a.cap); + each i := 0..a.len-1 { + new_data[i] = a.data[i]; + } + a.data = new_data; + } + a.data[a.len] = x; + a.len += 1; +}; + +square ::= fn(t ::=, x : t) t { + a : Arr(t); + each i := 1,2..2*x-1 { + arr_add(&a, i); + }; + sum := 0 as t; + each i := 0..a.len-1 { + sum += a.data[i]; + }; + sum +}; + + +// ArrInt ::= Arr(int); + +inc ::= fn(t ::=, x : t) t { + x + 1 +}; + +main ::= fn() { + arr : Arr(int); + farr : Arr(float); + each i := 1..100 { + arr_add(&arr, inc(square(i))); + arr_add(&farr, inc(square(i as float))); + } + each i := 0..arr.len - 1 { + puti(arr.data[i]); + } + each i := 0..farr.len - 1 { + putf(farr.data[i]); + } +}; |