#include "master.hpp"

namespace factor
{

/* make a new array with an initial element */
array *allot_array(cell capacity, cell fill_)
{
	gc_root<object> fill(fill_);
	gc_root<array> new_array(allot_array_internal<array>(capacity));

	if(fill.value() == tag_fixnum(0))
		memset(new_array->data(),'\0',capacity * sizeof(cell));
	else
	{
		/* No need for write barrier here. Either the object is in
		the nursery, or it was allocated directly in tenured space
		and the write barrier is already hit for us in that case. */
		cell i;
		for(i = 0; i < capacity; i++)
			new_array->data()[i] = fill.value();
	}
	return new_array.untagged();
}

/* push a new array on the stack */
PRIMITIVE(array)
{
	cell initial = dpop();
	cell size = unbox_array_size();
	dpush(tag<array>(allot_array(size,initial)));
}

cell allot_array_1(cell obj_)
{
	gc_root<object> obj(obj_);
	gc_root<array> a(allot_array_internal<array>(1));
	set_array_nth(a.untagged(),0,obj.value());
	return a.value();
}

cell allot_array_2(cell v1_, cell v2_)
{
	gc_root<object> v1(v1_);
	gc_root<object> v2(v2_);
	gc_root<array> a(allot_array_internal<array>(2));
	set_array_nth(a.untagged(),0,v1.value());
	set_array_nth(a.untagged(),1,v2.value());
	return a.value();
}

cell allot_array_4(cell v1_, cell v2_, cell v3_, cell v4_)
{
	gc_root<object> v1(v1_);
	gc_root<object> v2(v2_);
	gc_root<object> v3(v3_);
	gc_root<object> v4(v4_);
	gc_root<array> a(allot_array_internal<array>(4));
	set_array_nth(a.untagged(),0,v1.value());
	set_array_nth(a.untagged(),1,v2.value());
	set_array_nth(a.untagged(),2,v3.value());
	set_array_nth(a.untagged(),3,v4.value());
	return a.value();
}

PRIMITIVE(resize_array)
{
	array* a = untag_check<array>(dpop());
	cell capacity = unbox_array_size();
	dpush(tag<array>(reallot_array(a,capacity)));
}

void growable_array::add(cell elt_)
{
	gc_root<object> elt(elt_);
	if(count == array_capacity(elements.untagged()))
		elements = reallot_array(elements.untagged(),count * 2);

	set_array_nth(elements.untagged(),count++,elt.value());
}

void growable_array::trim()
{
	elements = reallot_array(elements.untagged(),count);
}

}