#include "factor.h"

/* the array is full of undefined data, and must be correctly filled before the
next GC. */
F_ARRAY* allot_array(CELL type, F_FIXNUM capacity)
{
	F_ARRAY *array;

	if(capacity < 0)
		general_error(ERROR_NEGATIVE_ARRAY_SIZE,tag_integer(capacity),true);

	array = allot_object(type,array_size(capacity));
	array->capacity = tag_fixnum(capacity);
	return array;
}

/* make a new array with an initial element */
F_ARRAY* array(CELL type, F_FIXNUM capacity, CELL fill)
{
	int i;
	F_ARRAY* array = allot_array(type, capacity);
	for(i = 0; i < capacity; i++)
		put(AREF(array,i),fill);
	return array;
}

/* push a new array on the stack */
void primitive_array(void)
{
	CELL initial;
	F_FIXNUM size;
	maybe_gc(0);
	initial = dpop();
	size = to_fixnum(dpop());
	dpush(tag_object(array(ARRAY_TYPE,size,initial)));
}

/* push a new tuple on the stack */
void primitive_tuple(void)
{
	F_FIXNUM size = to_fixnum(dpop());
	maybe_gc(array_size(size));
	dpush(tag_object(array(TUPLE_TYPE,size,F)));
}

/* push a new byte on the stack */
void primitive_byte_array(void)
{
	F_FIXNUM size = to_fixnum(dpop());
	maybe_gc(array_size(size));
	F_FIXNUM byte_size = (size + sizeof(CELL) - 1) / sizeof(CELL);
	dpush(tag_object(array(BYTE_ARRAY_TYPE,byte_size,0)));
}

F_ARRAY* resize_array(F_ARRAY* array, F_FIXNUM capacity, CELL fill)
{
	int i;
	F_ARRAY* new_array;
	
	CELL to_copy = array_capacity(array);
	if(capacity < to_copy)
		to_copy = capacity;
	
	new_array = allot_array(untag_header(array->header),capacity);
	
	memcpy(new_array + 1,array + 1,to_copy * CELLS);
	
	for(i = to_copy; i < capacity; i++)
		put(AREF(new_array,i),fill);

	return new_array;
}

void primitive_resize_array(void)
{
	F_ARRAY* array;
	F_FIXNUM capacity = to_fixnum(dpeek2());
	maybe_gc(array_size(capacity));
	array = untag_array(dpop());
	drepl(tag_object(resize_array(array,capacity,F)));
}

void primitive_array_to_tuple(void)
{
	CELL array = dpeek();
	type_check(ARRAY_TYPE,array);
	array = clone(array);
	put(SLOT(UNTAG(array),0),tag_header(TUPLE_TYPE));
	drepl(array);
}

void primitive_tuple_to_array(void)
{
	CELL tuple = dpeek();
	type_check(TUPLE_TYPE,tuple);
	tuple = clone(tuple);
	put(SLOT(UNTAG(tuple),0),tag_header(ARRAY_TYPE));
	drepl(tuple);
}

/* image loading */
void fixup_array(F_ARRAY* array)
{
	int i = 0; CELL capacity = array_capacity(array);
	for(i = 0; i < capacity; i++)
		data_fixup((void*)AREF(array,i));
}

/* GC */
void collect_array(F_ARRAY* array)
{
	int i = 0; CELL capacity = array_capacity(array);
	for(i = 0; i < capacity; i++)
		copy_handle((void*)AREF(array,i));
}