aboutsummaryrefslogtreecommitdiff
path: root/tests/programs/o3push.zc
diff options
context:
space:
mode:
Diffstat (limited to 'tests/programs/o3push.zc')
-rw-r--r--tests/programs/o3push.zc33
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;