#include "freelist.h"
#include "hexdump.h"
#include <assert.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

typedef struct {
    int x, y, z;
} Vec3;

Vec3 vec3_new(int a, int b, int c) {
    return (Vec3){a, b, c};
}

// Print function
void printvec(const Vec3 *v) {
    printf("Vec3: (%d, %d, %d)\n", v->x, v->y, v->z);
}

#define BUFFER_SIZE (128)

int main() {
    FreeList    fl;
    const char *mem = malloc(BUFFER_SIZE);
    memset((void *)mem, 0, BUFFER_SIZE);

    if (!fl_init(&fl, (uintptr_t)mem, BUFFER_SIZE, sizeof(Vec3))) {
        printf("Allocation error\n");
        exit(EXIT_FAILURE);
    }

    hexdump(mem, BUFFER_SIZE);

    const size_t cap = fl_capacity(&fl);

    printf("Item size: %lu\n", sizeof(Vec3));
    printf("FreeList Capacity: %lu\n", cap);
    printf("Buffer size: %d\n", BUFFER_SIZE);
    printf("Freelist Blocksize: %lu\n", fl.size);
    printf("Space utilization internal: %.2f%%\n", 100.0 * fl_utilization(&fl, sizeof(Vec3)));
    printf("Space utilization external: %.2f%%\n", 100.0 * ((float)fl.size * cap) / BUFFER_SIZE);

    assert(fl_available(&fl) == cap);

    if (!fl_check(&fl))
        printf("Check failed!\n");

    Vec3 *a = fl_alloc(&fl);
    Vec3 *b = fl_alloc(&fl);
    Vec3 *c = fl_alloc(&fl);

    hexdump(mem, BUFFER_SIZE);

    memset(a, 0x23, sizeof(Vec3));
    memset(b, 0x24, sizeof(Vec3));
    memset(c, 0x25, sizeof(Vec3));

    *a = vec3_new(12, 13, 435);
    *b = vec3_new(192, 199, 435);
    *c = vec3_new(432, 11, 435);

    printvec(a);
    printvec(b);
    printvec(c);

    // assert(fl_check(&fl) == cap - 3);
    // assert(fl_capacity(&fl) == fl_available(&fl) + 3);

    printf("Available: %zu of %zu\n", fl_available(&fl), fl_capacity(&fl));

    assert(fl_free(&fl, a));
    assert(fl_free(&fl, b));
    assert(fl_free(&fl, c));

    printf("Available: %zu of %zu\n", fl_available(&fl), fl_capacity(&fl));

    // assert(fl_check(&fl) == fl_capacity(&fl) - fl_allocated(&fl));
    assert(fl_allocated(&fl) == 0);

    /* All memory is free here */

    uintptr_t *ptr_buf = malloc(sizeof(uintptr_t) * fl_capacity(&fl));

    /* Fill it up */
    for (int i = 0; i < fl_capacity(&fl); i++) {
        ptr_buf[i] = (uintptr_t)fl_alloc(&fl);
    }

    /* Write some data to the last element to catch out of bounds writes */
    size_t last_elem = fl_capacity(&fl) - 1;
    *(Vec3 *)ptr_buf[last_elem] = *a;

    assert(0 == fl_available(&fl));
    assert(fl_allocated(&fl) == fl_capacity(&fl));
    assert(fl_check(&fl) == 0);

    /* Return it all */
    for (int i = 0; i < fl_capacity(&fl); i++) {
        fl_free(&fl, (void *)ptr_buf[i]);
    }

    // assert(cap == fl_available(&fl));
    // assert(fl_allocated(&fl) == 0);
    // assert(fl_check(&fl) == cap);

    printf("All tests passed!\n");
    return 0;
}