1 files changed, 70 insertions, 0 deletions
diff --git a/tests/arr.toc b/tests/arr.toc
new file mode 100644
index 0000000..eee514a
--- /dev/null
+++ b/tests/arr.toc
@@ -0,0 +1,70 @@
+puti ::= fn(x: int) {
+//tcc's giving me "incompatible types for redefinition of 'printf'" for some reason (even though the declarations have the exact same type)
+	 #C("#ifndef __TINYC__
+extern int printf(const char *fmt, ...);
+#endif
+");
+	 #C("printf(\"%ld\\n\", (long)x);");
+};
+putf ::= fn(x: float) {
+	 #C("#ifndef __TINYC__
+extern int printf(const char *fmt, ...);
+#endif
+");
+	 #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 :: Type, a : &Arr(t), x : t) {
+		if a.len >= a.cap {
+		   a.cap = a.cap * 2 + 2;
+		   new_data := new(t, a.cap);
+		   for 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 :: Type, x : t) t {
+	   a : Arr(t);
+	   for i := 1,2..2*x-1 {
+	   		arr_add(t, &a, i);
+	   };
+	   sum := 0 as t;
+	   for i := 0..a.len-1 {
+	   		sum += a.data[i];
+	   };
+	   sum
+};
+
+
+ArrInt ::= Arr(int);
+
+inc ::= fn(t :: Type, x : t) t {
+	x + 1
+};
+
+main ::= fn() {
+	 arr : ArrInt;
+	 farr : Arr(float);
+	 for i := 1..100 {
+	 	   arr_add(int, &arr, inc(int, square(int, i)));
+	 	   arr_add(float, &farr, inc(float, square(float, i as float)));
+	 }
+	 for i := 0..arr.len - 1 {
+	 	  puti(arr.data[i]);
+	 }
+	 for i := 0..farr.len - 1 {
+	 	  putf(farr.data[i]);
+	 }
+};