#include "std/mem.toc";
#include "std/io.toc";

// 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 := news(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]);
	 }
};