#include "heap.h"

void * heap_alloc_alligned ( Heap * heap, size_t len, size_t align ) {
  size_t pos = (size_t) heap -> next;
  size_t rem = pos % align;
  if ( rem )
    heap -> next = heap -> next + align - rem;
  if ( heap -> next + len >= heap -> end )
    return NULL;  
  void * ret = heap -> next;
  heap -> next += len;
  return ret;
}

void * heap_alloc ( Heap * heap, size_t len ) {
  size_t pos = (size_t) heap -> next;
  size_t rem = pos % 8;
  if ( rem )
    heap -> next = heap -> next + 8 - rem;
  if ( heap -> next + len >= heap -> end )
    return NULL;  
  void * ret = heap -> next;
  heap -> next += len;
  return ret;
}

void heap_init ( Heap * heap, size_t heap_size ) {
  heap -> begin = (u8*) malloc ( heap_size );
  heap -> next = heap -> begin;
  heap -> end = heap -> begin + heap_size;
}

void heap_destroy ( Heap * heap ) {
  free ( heap -> begin );
}

bool value_to_bool ( Value * value ) {
  if ( value -> kind == VALUE_NULL )
    return false;
  if ( value -> kind == VALUE_BOOLEAN )
    return ((BoolValue*) value ) -> boolean;
  return true;
}

Value * make_int ( Heap * heap, i32 val ) {
  IntValue * value = heap_alloc ( heap, sizeof (IntValue) );
  *value = (IntValue) { .kind = (Value) { .kind = VALUE_INTEGER }, .integer = val };
  return (Value*) value;
}

Value * make_bool ( Heap * heap, bool val ) {
  BoolValue * value = heap_alloc ( heap, sizeof (BoolValue) );
  *value = (BoolValue) { .kind = (Value) { .kind = VALUE_BOOLEAN }, .boolean = val };
  return (Value*) value;
}

Value * make_function ( Heap * heap, AstFunction * function ) {
  FunctionValue * value = heap_alloc ( heap, sizeof (FunctionValue) );
  *value = (FunctionValue) { .kind = (Value) { .kind = VALUE_FUNCTION }, .function = function };
  return (Value*) value;
}

Value * make_cfunction ( Heap * heap, ConstantFunction function ) {
  CFunctionValue * value = heap_alloc ( heap, sizeof (CFunctionValue) );
  *value = (CFunctionValue) { .kind = (Value) { .kind = VALUE_FUNCTION }, .function = function };
  return (Value*) value;
}

Value * make_array ( Heap * heap, size_t len ) {
  ArrayValue * value = heap_alloc ( heap, sizeof (ArrayValue) + len * sizeof (Value*) );
  *value = (ArrayValue) { .kind = (Value) { .kind = VALUE_ARRAY }, .length = len };
  return (Value*) value;
}

Value * make_object ( Heap * heap, size_t member_cnt ) {
  ObjectValue * value = heap_alloc ( heap, sizeof (ObjectValue) + member_cnt * sizeof (SimpleEntry) );
  *value = (ObjectValue) { .kind = (Value) { .kind = VALUE_OBJECT }, .member_cnt = member_cnt };
  return (Value*) value;
}

Value * make_null ( Heap * heap ) {
  NullValue * value = heap_alloc ( heap, sizeof (NullValue) );
  *value = (NullValue) { .kind = (Value) { .kind = VALUE_NULL } };
  return (Value*) value;
}