#include "./common.h"
#include "r3d.h"
#include "raylib.h"
#include "raymath.h"

/* === Resources === */

static R3D_Mesh plane = { 0 };
static R3D_Mesh sphere = { 0 };
static R3D_Material material = { 0 };
static Camera3D camera = { 0 };

static Matrix* transforms = NULL;

static R3D_Light lights[100] = { 0 };

/* === Example === */

const char* Init(void)
{
    R3D_Init(GetScreenWidth(), GetScreenHeight(), 0);
    SetTargetFPS(60);

    plane = R3D_GenMeshPlane(1000, 1000, 1, 1, true);
    sphere = R3D_GenMeshSphere(0.35f, 16, 16, true);
    material = R3D_GetDefaultMaterial();

    camera = (Camera3D) {
        .position = (Vector3) { 0, 2, 2 },
        .target = (Vector3) { 0, 0, 0 },
        .up = (Vector3) { 0, 1, 0 },
        .fovy = 60,
    };

    transforms = RL_MALLOC(100 * 100 * sizeof(Matrix));

    for (int x = -50; x < 50; x++) {
        for (int z = -50; z < 50; z++) {
            int index = (z + 50) * 100 + (x + 50);
            transforms[index] = MatrixTranslate(x, 0, z);
        }
    }

    for (int x = -5; x < 5; x++) {
        for (int z = -5; z < 5; z++) {
            int index = (z + 5) * 10 + (x + 5);
            lights[index] = R3D_CreateLight(R3D_LIGHT_OMNI);
            R3D_SetLightPosition(lights[index], (Vector3) { x * 10, 10, z * 10 });
            R3D_SetLightColor(lights[index], ColorFromHSV((float)index / 100 * 360, 1.0f, 1.0f));
            R3D_SetLightRange(lights[index], 20.0f);
            R3D_SetLightActive(lights[index], true);
        }
    }

    return "[r3d] - Many lights example";
}

void Update(float delta)
{
    UpdateCamera(&camera, CAMERA_ORBITAL);
}

void Draw(void)
{
    R3D_Begin(camera);
        R3D_DrawMesh(&plane, &material, MatrixTranslate(0, -0.5f, 0));
        R3D_DrawMeshInstanced(&sphere, &material, transforms, 100 * 100);
    R3D_End();

    if (IsKeyDown(KEY_SPACE)) {
        BeginMode3D(camera);
        for (int i = 0; i < 100; i++) {
            R3D_DrawLightShape(lights[i]);
        }
        EndMode3D();
    }

    DrawFPS(10, 10);
    DrawText("Press SPACE to show the lights", 10, GetScreenHeight() - 34, 24, BLACK);
}

void Close(void)
{
    R3D_UnloadMesh(&plane);
    R3D_UnloadMesh(&sphere);
    R3D_Close();
}