diff options
Diffstat (limited to 'tests/programs/o3push.zc')
| -rw-r--r-- | tests/programs/o3push.zc | 33 |
1 files changed, 18 insertions, 15 deletions
diff --git a/tests/programs/o3push.zc b/tests/programs/o3push.zc index e5f0090..acb9fab 100644 --- a/tests/programs/o3push.zc +++ b/tests/programs/o3push.zc @@ -1,37 +1,40 @@ +namespace; +include rt; + // push in o3 code: the quick path when there is room, dpush when there is not -o3 int fill([]int a, int n) { +o3 int fill(int[?] a, int n) { for (int i = 0; i < n; i++) push(a, i * 3); return len(a); } -o3 int sum([]int a) { +o3 int sum(int[?] a) { int s = 0; for (int i = 0; i < len(a); i++) s += a[i]; return s; } -o3 float fsum([]float a) { +o3 float fsum(float[?] a) { float s = 0.0; for (int i = 0; i < len(a); i++) s += a[i]; return s; } -o3 int ffill([]float a, int n) { +o3 int ffill(float[?] a, int n) { for (int i = 0; i < n; i++) push(a, 0.5 * i); return len(a); } int main() { - []int a = new([]int, 0); - put_n(fill(a, 1000)); put_c(' '); - put_n(sum(a)); put_c(' '); - put_n(a[999]); put_c(' '); - put_n(fill(a, 5)); put_c(' '); - put_n(cap(a) >= 1005); put_c(' '); - []float f = new([]float, 0); + int[?] a = rt.new(int[?], 0); + rt.put_n(fill(a, 1000)); rt.put_c(' '); + rt.put_n(sum(a)); rt.put_c(' '); + rt.put_n(a[999]); rt.put_c(' '); + rt.put_n(fill(a, 5)); rt.put_c(' '); + rt.put_n(cap(a) >= 1005); rt.put_c(' '); + float[?] f = rt.new(float[?], 0); ffill(f, 10); - put_n(len(f)); put_c(' '); put_n(fsum(f) == 22.5); put_c(' '); - put_n(drain(a)); put_c(' '); put_n(len(a)); put_c(10); - free(d*a); free(d*f); + rt.put_n(len(f)); rt.put_c(' '); rt.put_n(fsum(f) == 22.5); rt.put_c(' '); + rt.put_n(drain(a)); rt.put_c(' '); rt.put_n(len(a)); rt.put_c(10); + rt.free(d*a); rt.free(d*f); return 0; } -o3 int drain([]int a) { +o3 int drain(int[?] a) { int s = 0; while (len(a) > 0) s += pop(a); return s; |
