#include "particle_sys.h"
#include <stdio.h>
#include <unistd.h>
#include "raylib.h"
#include "raymath.h"
#include "constants.h"
static const Vector2 GRAVITY = {0, GRAV_ACCEL};

void simple_particle_system_update(Particle_t* part, void* user_data, float delta_time)
{
    (void)user_data;
    part->rotation += part->angular_vel;

    part->velocity =
        Vector2Add(
            part->velocity,
            Vector2Scale(GRAVITY, delta_time)
        );

    float mag = Vector2Length(part->velocity);
    part->velocity = Vector2Scale(
        Vector2Normalize(part->velocity),
        (mag > PLAYER_MAX_SPEED)? PLAYER_MAX_SPEED:mag
    );
    // 3 dp precision
    if (fabs(part->velocity.x) < 1e-3) part->velocity.x = 0;
    if (fabs(part->velocity.y) < 1e-3) part->velocity.y = 0;

    part->position = Vector2Add(
        part->position,
        Vector2Scale(part->velocity, delta_time)
    );

    // Level boundary collision
    {

        if(
            part->position.x + part->size < 0 || part->position.x - part->size > 1280
            || part->position.y + part->size < 0 || part->position.y - part->size > 640
        )
        {
            part->timer = 0;
        }
    }
}

static bool check_mouse_click(const ParticleEmitter_t* emitter, float delta_time)
{
    return IsMouseButtonDown(MOUSE_RIGHT_BUTTON);
}

int main(void)
{
    InitWindow(1280, 640, "raylib");
    SetTargetFPS(60);
    static ParticleSystem_t part_sys = {0};

    init_particle_system(&part_sys);
    Texture2D tex = LoadTexture("res/bomb.png");
    Sprite_t spr = {
        .texture = &tex,
        .frame_size = (Vector2){tex.width, tex.height},
        .origin = (Vector2){0, 0},
        .anchor = (Vector2){tex.width / 2, tex.height / 2},
        .frame_count = 1,
        .frame_per_row = 1,
        .speed = 0,
        .name = "test_spr"
    };

    const float DT = 1.0f/60.0f;
    float delta_time = 0.0f;

    EmitterConfig_t conf ={
        .one_shot = true,
        .launch_range = {0, 360},
        .speed_range = {400, 2000},
        .angle_range = {0, 360},
        .rotation_range = {-10, 10},
        .particle_lifetime = {30, 110},
        .type = EMITTER_BURST,
    };

    ParticleEmitter_t emitter = {
        .config = &conf,
        .n_particles = MAX_PARTICLES,
        .update_func = &simple_particle_system_update,
        .emitter_update_func = NULL,
        .spr = (tex.width == 0) ? NULL : &spr,
        .user_data = &delta_time,
    };

    EmitterConfig_t conf2 ={
        .one_shot = false,
        .launch_range = {45, 135},
        //.launch_range = {0, 360},
        .speed_range = {300, 800},
        .angle_range = {0, 1},
        .rotation_range = {0, 1},
        .particle_lifetime = {15, 30},
        .initial_spawn_delay = 5,
        .type = EMITTER_STREAM,
    };

    ParticleEmitter_t emitter2 = {
        .config = &conf2,
        .n_particles = MAX_PARTICLES,
        .update_func = &simple_particle_system_update,
        .emitter_update_func = &check_mouse_click,
        .spr = (tex.width == 0) ? NULL : &spr,
        .user_data = NULL,
    };

    bool key_press = false;
    uint8_t key2_press = 0;
    char text_buffer[32];
    EmitterHandle han = 0;
    while(!WindowShouldClose())
    {
        float frame_time = GetFrameTime();
        delta_time = fminf(frame_time, DT);
        Vector2 mouse_pos = GetMousePosition();
        if (IsMouseButtonPressed(MOUSE_LEFT_BUTTON))
        {
            key_press = true;
        }
        else if (key_press && IsMouseButtonReleased(MOUSE_BUTTON_LEFT))
        {
            emitter.position = mouse_pos;
            play_particle_emitter(&part_sys, &emitter);
            key_press = false;
        }

        key2_press <<= 1;
        key2_press |= IsMouseButtonDown(MOUSE_RIGHT_BUTTON)? 1: 0;
        key2_press &= 0b11;

        if (key2_press == 0b01)
        {
            emitter2.position = mouse_pos;
            han = play_particle_emitter(&part_sys, &emitter2);
        }
        else if(key2_press == 0b11)
        {
            update_emitter_handle_position(&part_sys, han, mouse_pos);
        }
        else if (key2_press == 0b10)
        {
            stop_emitter_handle(&part_sys, han);
            han = 0;
        }

        update_particle_system(&part_sys, delta_time);
        sprintf(text_buffer, "free: %u", get_number_of_free_emitter(&part_sys));
        BeginDrawing();
            ClearBackground(RAYWHITE);
            draw_particle_system(&part_sys);
            DrawText(text_buffer, 0, 0, 16, BLACK);
        EndDrawing();
    } 
    UnloadTexture(tex);
    CloseWindow();
    deinit_particle_system(&part_sys);
}