#include "ParticlesRenderer.hpp"

#include <set>

#include "assets/Assets.hpp"
#include "assets/assets_util.hpp"
#include "graphics/core/Shader.hpp"
#include "graphics/core/Texture.hpp"
#include "window/Camera.hpp"
#include "world/Level.hpp"
#include "voxels/Chunks.hpp"
#include "MainBatch.hpp"
#include "settings.hpp"

size_t ParticlesRenderer::visibleParticles = 0;
size_t ParticlesRenderer::aliveEmitters = 0;

ParticlesRenderer::ParticlesRenderer(
    const Assets& assets,
    const Level& level,
    const Chunks& chunks,
    const GraphicsSettings* settings
)
    : chunks(chunks),
      assets(assets),
      settings(settings),
      batch(std::make_unique<MainBatch>(4096)) {
}

ParticlesRenderer::~ParticlesRenderer() = default;

static inline void update_particle(
    Particle& particle, float delta, const Chunks& chunks
) {
    const auto& preset = particle.emitter->preset;
    auto& pos = particle.position;
    auto& vel = particle.velocity;
    auto& angle = particle.angle;

    vel += delta * preset.acceleration;
    if (preset.collision && chunks.isObstacleAt(pos + vel * delta)) {
        vel *= 0.0f;
    }
    pos += vel * delta;
    angle += particle.angularVelocity * delta;
    particle.lifetime -= delta;
}

void ParticlesRenderer::renderParticles(const Camera& camera, float delta) {
    const auto& right = camera.right;
    const auto& up = camera.up;

    bool backlight = settings->backlight.get();

    std::vector<const Texture*> unusedTextures;

    for (auto& [texture, vec] : particles) {
        if (vec.empty()) {
            unusedTextures.push_back(texture);
            continue;
        }
        batch->setTexture(texture);

        visibleParticles += vec.size();

        auto iter = vec.begin();
        while (iter != vec.end()) {
            auto& particle = *iter;
            auto& emitter = *particle.emitter;
            auto& preset = emitter.preset;

            if (!preset.frames.empty()) {
                float time = preset.lifetime - particle.lifetime;
                int framesCount = preset.frames.size();
                int frameid = time / preset.lifetime * framesCount;
                int frameid2 = glm::min(
                    (time + delta) / preset.lifetime * framesCount,
                    framesCount - 1.0f
                );
                if (frameid2 != frameid) {
                    auto tregion = util::get_texture_region(
                        assets, preset.frames.at(frameid2), ""
                    );
                    if (tregion.texture == texture) {
                        particle.region = tregion.region;
                    }
                }
            }
            update_particle(particle, delta, chunks);

            float scale = 1.0f + ((particle.random ^ 2628172) % 1000) *
                0.001f * preset.sizeSpread;

            glm::vec4 light(1, 1, 1, 0);
            if (preset.lighting) {
                light = MainBatch::sampleLight(
                    particle.position,
                    chunks,
                    backlight
                );
                auto size = glm::max(glm::vec3(0.5f), preset.size * scale);
                for (int x = -1; x <= 1; x++) {
                    for (int y = -1; y <= 1; y++) {
                        for (int z = -1; z <= 1; z++) {
                            light = glm::max(
                                light,
                                MainBatch::sampleLight(
                                    particle.position -
                                        size * glm::vec3(x, y, z),
                                    chunks,
                                    backlight
                                )
                            );
                        }
                    }
                }
                light *= 0.9f + (particle.random % 100) * 0.001f;
            }

            glm::vec3 localRight = right;
            glm::vec3 localUp = preset.globalUpVector ? glm::vec3(0, 1, 0) : up;
            float angle = particle.angle;
            if (glm::abs(angle) >= 0.005f) {
                glm::vec3 rotatedRight(glm::cos(angle), -glm::sin(angle), 0.0f);
                glm::vec3 rotatedUp(glm::sin(angle), glm::cos(angle), 0.0f);

                localRight = right * rotatedRight.x + localUp * rotatedRight.y +
                            camera.front * rotatedRight.z;
                localUp = right * rotatedUp.x + localUp * rotatedUp.y +
                        camera.front * rotatedUp.z;
            }
            batch->quad(
                particle.position,
                localRight,
                localUp,
                -camera.front,
                preset.size * scale,
                light,
                glm::vec3(1.0f),
                particle.region,
                preset.lighting ? 0.0f : 1.0f
            );
            if (particle.lifetime <= 0.0f) {
                iter = vec.erase(iter);
                emitter.refCount--;
            } else {
                iter++;
            }
        }
    }
    batch->flush();
    for (const auto& texture : unusedTextures) {
        particles.erase(texture);
    }
}

void ParticlesRenderer::render(const Camera& camera, float delta) {
    batch->begin();
    
    aliveEmitters = emitters.size();
    visibleParticles = 0;

    renderParticles(camera, delta);

    auto iter = emitters.begin();
    while (iter != emitters.end()) {
        auto& emitter = *iter->second;
        if (emitter.isDead() && !emitter.isReferred()) {
            // destruct Emitter only when there is no particles spawned by it
            iter = emitters.erase(iter);
            continue;
        }
        auto texture = emitter.getTexture();
        std::vector<Particle>* vec;
        vec = &particles[texture];
        emitter.update(delta, camera.position, *vec);
        iter++;
    }
}

Emitter* ParticlesRenderer::getEmitter(u64id_t id) const {
    const auto& found = emitters.find(id);
    if (found == emitters.end()) {
        return nullptr;
    }
    return found->second.get();
}

u64id_t ParticlesRenderer::add(std::unique_ptr<Emitter> emitter) {
    u64id_t uid = nextEmitter++;
    emitters[uid] = std::move(emitter);
    return uid;
}