#include "PrecompiledHeader.h"
#include "Physics/PhysicsSystem.h"



void dd::Systems::PhysicsSystem::RegisterComponents(ComponentFactory* cf)
{
    cf->Register<Components::CircleShape>();
}

void dd::Systems::PhysicsSystem::Initialize()
{
    std::random_device rd;
    gen = std::mt19937(rd());

    m_DestructionListener = new DestructionListener(this);
    m_ContactListener = new ContactListener(this);

    m_Gravity = b2Vec2(0.f, -9.82f);
    m_PhysicsWorld = new b2World(m_Gravity);

    m_TimeStep = 1.f/60.f;
    m_VelocityIterations = 6;
    m_PositionIterations = 2;
    m_Accumulator = 0.f;

    m_PhysicsWorld->SetContactListener(m_ContactListener);
    m_PhysicsWorld->SetContactFilter(&m_ParticleContactDisabler);
    m_PhysicsWorld->SetDestructionListener(m_DestructionListener);

    InitializeWater();



    EVENT_SUBSCRIBE_MEMBER(m_SetImpulse, PhysicsSystem::SetImpulse);
    EVENT_SUBSCRIBE_MEMBER(m_ECreateParticleSequence, &PhysicsSystem::CreateParticleSequence);

}

void dd::Systems::PhysicsSystem::InitializeWater()
{
    float radius = 0.13f;
    float gravityScale = 1.0f;

    b2ParticleSystemDef m_ParticleSystemDef;
    m_ParticleSystemDef.radius = radius;
    m_ParticleSystemDef.gravityScale = gravityScale;

    m_WaterParticleSystem = m_PhysicsWorld->CreateParticleSystem(&m_ParticleSystemDef);


}

bool dd::Systems::PhysicsSystem::SetImpulse(const Events::SetImpulse &event)
{
    b2Body* body = m_EntitiesToBodies[event.Entity];

    b2Vec2 impulse;
    impulse.x = event.Impulse.x;
    impulse.y = event.Impulse.y;

    b2Vec2 point;
    point.x = event.Point.x;
    point.y = event.Point.y;


    Impulse i;
    i.Body = body;
    i.Impulse = impulse;
    i.Point = point;

    m_Impulses.push_back(i);

    return true;
}

void dd::Systems::PhysicsSystem::Update(double dt) {

    m_Accumulator += dt;
    while (m_Accumulator >= m_TimeStep) {

        for (auto i : m_EntitiesToBodies) {
            EntityID entity = i.first;
            b2Body *body = i.second;

            auto transformComponent = m_World->GetComponent<Components::Transform>(entity);
            if (!transformComponent) {
                continue;
                LOG_ERROR("RigidBody with no TransformComponent");
            }
            auto physicsComponent = m_World->GetComponent<Components::Physics>(entity);
            if (!physicsComponent) {
                continue;
                LOG_ERROR("RigidBody with no PhysicsComponent");
            }

            if (body == nullptr) {
                LOG_ERROR("This body should not exist");
                continue;
            }

            if (m_World->GetEntityParent(entity) == 0) { //TODO: Make this work with childs too
                b2Vec2 position;
                position.x = transformComponent->Position.x;
                position.y = transformComponent->Position.y;
                float angle = glm::eulerAngles(transformComponent->Orientation).z;
                body->SetTransform(position, angle);
                body->SetLinearVelocity(b2Vec2(transformComponent->Velocity.x, transformComponent->Velocity.y));
                body->SetGravityScale(physicsComponent->GravityScale);

                b2Filter filter;
                filter.categoryBits = physicsComponent->Category;
                filter.maskBits = physicsComponent->Mask;
                body->GetFixtureList()->SetFilterData(filter);
            }
        }

        UpdateParticleEmitters(dt);

        for (auto i : m_Impulses) {
            i.Body->ApplyLinearImpulse(i.Impulse, i.Point, true);
        }
        m_Impulses.clear();

        m_PhysicsWorld->Step(m_TimeStep, m_VelocityIterations, m_PositionIterations);


        for (auto i : m_EntitiesToBodies) {
            EntityID entity = i.first;
            b2Body *body = i.second;

            if (body == nullptr) {
                LOG_ERROR("This body should not exist");
                continue;
            }

            auto transformComponent = m_World->GetComponent<Components::Transform>(entity);
            if (!transformComponent)
                continue;
            auto parent = m_World->GetEntityParent(entity);
            if (parent == 0) {

                auto physicsComponent = m_World->GetComponent<Components::Physics>(entity);
                if (physicsComponent->Calculate) { //TODO: REPLACE THIS WITH PARTICLE COLLISION FILTERS
                    transformComponent->Position.x =
                            transformComponent->Position.x + (transformComponent->Velocity.x * m_TimeStep);
                    transformComponent->Position.y =
                            transformComponent->Position.y + (transformComponent->Velocity.y * m_TimeStep);
                }
                else {
                    b2Vec2 position = body->GetPosition();
                    transformComponent->Position.x = position.x;
                    transformComponent->Position.y = position.y;

                    float angle = body->GetAngle();
                    transformComponent->Orientation = glm::quat(glm::vec3(0, 0, angle));

                    b2Vec2 velocity = body->GetLinearVelocity();
                    transformComponent->Velocity.x = velocity.x;
                    transformComponent->Velocity.y = velocity.y;
                }
            }
        }



        //water
        b2Vec2 *positionBuffer = m_WaterParticleSystem->GetPositionBuffer();
        for (auto i : m_EntitiesToWaterParticleHandle){
            EntityID entity = i.first;
            b2ParticleHandle *particleH = i.second;


            b2Vec2 positionB2 = positionBuffer[particleH->GetIndex()];
            glm::vec2 position = glm::vec2(positionB2.x, positionB2.y);

            EntityID entityParent = m_World->GetEntityParent(entity);
            glm::vec3 parentTransform = glm::vec3(0.f);
            if(entityParent) {
                auto tp = m_World->GetComponent<Components::Transform>(entityParent);
                parentTransform = tp->Position;
            }
            auto transform = m_World->GetComponent<Components::Transform>(entity);

            transform->Position = glm::vec3(position.x, position.y, -10);
        }

        //Normal particles
        for ( int e = 0; e < m_EntitiesToParticleHandle.size(); e++) {
            b2Vec2 *positionBuffer = m_ParticleEmitters.ParticleSystem[e]->GetPositionBuffer();
            if(!positionBuffer) {
                break;
            }
            for (auto i : m_EntitiesToParticleHandle[e]){
                EntityID entity = i.first;
                b2ParticleHandle *particleH = i.second;

                b2Vec2 positionB2 = positionBuffer[particleH->GetIndex()];
                glm::vec2 position = glm::vec2(positionB2.x, positionB2.y);

                EntityID entityParent = m_World->GetEntityParent(entity);
                glm::vec3 parentTransform = glm::vec3(0.f);
                if(entityParent) {
                    auto tp = m_World->GetComponent<Components::Transform>(entityParent);
                    parentTransform = tp->Position;
                }
                auto transform = m_World->GetComponent<Components::Transform>(entity);

                transform->Position = glm::vec3(position.x, position.y, transform->Position.z);
            }
        }

        m_Accumulator -= dt;
    }
}


void dd::Systems::PhysicsSystem::UpdateEntity(double dt, EntityID entity, EntityID parent)
{
    auto particle = m_World->GetComponent<Components::Particle>(entity);
    auto pTemplate = m_World->GetComponent<Components::Template>(entity);
    if (particle && !pTemplate) {
        particle->LifeTime -= dt;
        if (particle->LifeTime <= 0) {
            for (int i = 0; i < m_EntitiesToParticleHandle.size(); i++) {
                b2ParticleHandle* handle = m_EntitiesToParticleHandle[i][entity];
                std::unordered_map<EntityID, const b2ParticleHandle*>::iterator iter1;
                std::unordered_map<const b2ParticleHandle*,EntityID>::iterator iter2;
                iter1 = m_EntitiesToParticleHandle[i].find(entity);
                iter2 = m_ParticleHandleToEntities[i].find(handle);
                if (iter1 != m_EntitiesToParticleHandle[i].end()) {
                    m_EntitiesToParticleHandle[i].erase(iter1);
                }
                if (iter2 != m_ParticleHandleToEntities[i].end()) {
                    m_ParticleHandleToEntities[i].erase(iter2);
                }
                m_World->RemoveEntity(entity);
            }

        }
    }
}

void dd::Systems::PhysicsSystem::OnEntityCommit(EntityID entity)
{
    auto physicsComponent = m_World->GetComponent<Components::Physics>(entity);
    auto waterComponent = m_World->GetComponent<Components::WaterVolume>(entity);
    auto particleEmitterComponent = m_World->GetComponent<Components::ParticleEmitter>(entity);

    if (physicsComponent && waterComponent) {
        LOG_ERROR("Entity has both water and physics component, this is illegal. The police has been alerted.");
        return;
    }
    if (physicsComponent && particleEmitterComponent) {
        LOG_ERROR("Entity has both particle and physics component, this is illegal. The police has been alerted.");
        return;
    }

    if (physicsComponent) {
        CreateBody(entity);
    } else if (waterComponent) {
        CreateParticleGroup(entity);
    } else if (particleEmitterComponent) {
        CreateParticleEmitter(entity);
    }
}


void dd::Systems::PhysicsSystem::OnEntityRemoved(EntityID entity)
{
    auto physics = m_World->GetComponent<Components::Physics>(entity);
    if (physics == nullptr) {
        return;
    }

    b2Body* body = m_EntitiesToBodies[entity];

    if (body != nullptr) {
        m_EntitiesToBodies.erase(entity);
        m_BodiesToEntities.erase(body);

        m_PhysicsWorld->DestroyBody(body);
    }

}

void dd::Systems::PhysicsSystem::CreateBody(EntityID entity)
{
    auto physicsComponent = m_World->GetComponent<Components::Physics>(entity);
    if(!physicsComponent){
        LOG_ERROR("No PhysicsComponent in CreateBody");
        return;
    }


    auto transformComponent = m_World->GetComponent<Components::Transform>(entity);
    if(!transformComponent) {
        LOG_ERROR("No TransformComponent in CreateBody");
        return;
    }

    auto absoluteTransform = m_World->GetSystem<Systems::TransformSystem>()->AbsoluteTransform(entity);

    b2BodyDef bodyDef;
    bodyDef.position.Set(absoluteTransform.Position.x, absoluteTransform.Position.y);
    bodyDef.angle = -glm::eulerAngles(absoluteTransform.Orientation).z;

    if (physicsComponent->Static) {
        bodyDef.type = b2_staticBody;
    } else {
        bodyDef.type = b2_dynamicBody;

    }

    b2Body* body = m_PhysicsWorld->CreateBody(&bodyDef);

    b2Shape* pShape;

    auto boxComponent = m_World->GetComponent<Components::RectangleShape>(entity);
    if (boxComponent) {
        b2PolygonShape* bShape = new b2PolygonShape();
        bShape->SetAsBox(absoluteTransform.Scale.x/2, absoluteTransform.Scale.y/2);
        pShape = bShape;
    } else {
        auto circleComponent = m_World->GetComponent<Components::CircleShape>(entity);
        if (circleComponent) {
            pShape = new b2CircleShape();
            pShape->m_radius = absoluteTransform.Scale.x;


            if (absoluteTransform.Scale.x != absoluteTransform.Scale.y &&  absoluteTransform.Scale.y != absoluteTransform.Scale.z) {
                LOG_WARNING("Circles has to be of uniform scale.");
            }
            pShape->m_radius = absoluteTransform.Scale.x/2;

        }
    }

    b2FixtureDef fixtureDef;
    fixtureDef.shape = pShape;
    fixtureDef.filter.categoryBits = physicsComponent->Category;
    fixtureDef.filter.maskBits = physicsComponent->Mask;


    if(physicsComponent->Static) {
        body->CreateFixture(&fixtureDef); //Density kanske ska vara 0 p� statiska kroppar
    }
    else {
        fixtureDef.shape = pShape;
        fixtureDef.density = 10.f;
        fixtureDef.restitution = 1.0f;
        fixtureDef.friction = 0.0f;
        body->CreateFixture(&fixtureDef);

    }

    delete pShape;


    body->SetGravityScale(physicsComponent->GravityScale);
    m_EntitiesToBodies.insert(std::make_pair(entity, body));
    m_BodiesToEntities.insert(std::make_pair(body, entity));
}

void dd::Systems::PhysicsSystem::CreateParticleGroup(EntityID e)
{
    auto transform = m_World->GetComponent<Components::Transform>(e);
    if (!transform) {
        LOG_ERROR("No Transform component in CreateParticleGroup");
    }
    b2ParticleGroupDef pd;
    b2PolygonShape shape;
    auto water = m_World->GetComponent<Components::WaterVolume>(e);
    if (water) {

        shape.SetAsBox(transform->Scale.x/2.f, transform->Scale.y/2.f);
        pd.shape = &shape;
        pd.flags = b2_tensileParticle;
        pd.position.Set(transform->Position.x, transform->Position.y);
        t_ParticleGroup.push_back(m_WaterParticleSystem->CreateParticleGroup(pd));
        b2Vec2* t_ParticlePositions = m_WaterParticleSystem->GetPositionBuffer();
        for(int i = 0; i < m_WaterParticleSystem->GetParticleCount(); i++){
            {
                auto t_waterparticle = m_World->CreateEntity(e);
                auto transformChild = m_World->AddComponent<Components::Transform>(t_waterparticle);

                transformChild->Position = glm::vec3(t_ParticlePositions[i].x - transform->Position.x, t_ParticlePositions[i].y - transform->Position.y, -9.5f);
                transformChild->Scale = glm::vec3(m_WaterParticleSystem->GetRadius())/transform->Scale;
                m_World->CommitEntity(t_waterparticle);

                m_EntitiesToWaterParticleHandle.insert(std::make_pair(t_waterparticle, m_WaterParticleSystem->GetParticleHandleFromIndex(i)));
                m_WaterParticleHandleToEntities.insert(std::make_pair(m_WaterParticleSystem->GetParticleHandleFromIndex(i), t_waterparticle));
            }

        }

    }
}

void dd::Systems::PhysicsSystem::UpdateParticleEmitters(double dt)
{
    std::vector<EntityID> emittersToDelete;
    int pSizeTest = 0;
    for (int i = 0; i < m_ParticleEmitters.ParticleSystem.size(); i++) {
        auto e = m_ParticleEmitters.ParticleEmitter[i];
        auto pt = m_ParticleEmitters.ParticleTemplate[i];
        auto ps = m_ParticleEmitters.ParticleSystem[i];
        pSizeTest += ps->GetParticleCount();


        auto emitter = m_World->GetComponent<Components::ParticleEmitter>(e);
        auto particleTemplate = m_World->GetComponent<Components::Particle>(pt);
        if(!emitter) {
            LOG_INFO("ParticleEmitter Component is nil");
            continue;
        }
        if(!particleTemplate) {
            LOG_INFO("ParticleTemplate Component is nil.");
            continue;
        }
        auto TempTransform = m_World->GetComponent<Components::Transform>(e);
        TempTransform->Orientation = glm::rotate(glm::quat(), 3.f, glm::vec3(0.f, 0.f, -1.f));

        emitter->TimeSinceLastSpawn += dt;
        emitter->LifeTime -= dt;

        if (emitter->LifeTime <= 0 || emitter->NumberOfTicks <= 0) {
            emittersToDelete.push_back(e);
            continue;
        }
        if(emitter->TimeSinceLastSpawn < emitter->SpawnRate || emitter->NumberOfTicks < 1) {
            continue;
        }
        emitter->NumberOfTicks--;
        emitter->TimeSinceLastSpawn -= emitter->SpawnRate;
        auto templateTransform = m_World->GetComponent<Components::Transform>(pt);
        b2ParticleDef particleDef;
        for ( int j = 0; j < emitter->ParticlesPerTick; j++) {
            particleDef.flags = particleTemplate->Flags;
            //particleDef.color TODO: Implement this if we want color mixing and shit.
            particleDef.lifetime = particleTemplate->LifeTime;
            float eAngle = emitter->EmittingAngle;
            float halfSpread = emitter->Spread / 2;

            std::uniform_real_distribution<float> dis(eAngle - halfSpread, eAngle + halfSpread);
            float pAngle = dis(gen);
            glm::vec2 unitVec = glm::normalize(glm::vec2(glm::cos(pAngle), glm::sin(pAngle)));
            glm::vec2 vel = unitVec * emitter->Speed;
            particleDef.velocity = b2Vec2(vel.x, vel.y);
            particleDef.position = b2Vec2(templateTransform->Position.x, templateTransform->Position.y);


            auto particle = m_World->CloneEntity(pt, 0);
            m_World->RemoveComponent<Components::Template>(particle);
            auto transform2 = m_World->GetComponent<Components::Transform>(particle);

            std::uniform_real_distribution<float> dist(0, 1);
            float zDistribution = dist(gen);
            transform2->Position = glm::vec3(transform2->Position.x, transform2->Position.y, -9.5f + zDistribution);
            transform2->Scale = glm::vec3(particleTemplate->Radius * 2.f, particleTemplate->Radius * 2.f, 1);


            auto b2Particle = ps->CreateParticle(particleDef);

            auto b2ParticleHandle = ps->GetParticleHandleFromIndex(b2Particle);
            m_EntitiesToParticleHandle[i].insert(std::make_pair(particle, b2ParticleHandle));
            m_ParticleHandleToEntities[i].insert(std::make_pair(b2ParticleHandle, particle));
        }
    }

    for (int i = 0; i < emittersToDelete.size(); i++)
    {
        EntityID ent = emittersToDelete[i];
        int key;
        for(key = 0; key < m_ParticleEmitters.ParticleEmitter.size(); key++) {
            if (m_ParticleEmitters.ParticleEmitter[key] == ent) {
                break;
            }
        }

        if (m_ParticleEmitters.ParticleSystem[key]->GetParticleCount() == 0 && key <= m_ParticleEmitters.ParticleSystem.size()) {
            m_World->RemoveEntity(ent);
            m_PhysicsWorld->DestroyParticleSystem(m_ParticleEmitters.ParticleSystem[key]);
            m_ParticleEmitters.ParticleSystem.erase(m_ParticleEmitters.ParticleSystem.begin() + key);
            m_ParticleEmitters.ParticleEmitter.erase(m_ParticleEmitters.ParticleEmitter.begin() + key);
            m_ParticleEmitters.ParticleTemplate.erase(m_ParticleEmitters.ParticleTemplate.begin() + key);
            m_EntitiesToParticleHandle.erase(m_EntitiesToParticleHandle.begin() + key);
            m_ParticleHandleToEntities.erase(m_ParticleHandleToEntities.begin() + key);
        }
    }

    int stp = 0;
    for (auto ts : m_EntitiesToParticleHandle)
    {
        stp += ts.size();
    }
}

bool dd::Systems::PhysicsSystem::CreateParticleSequence(const Events::CreateParticleSequence &event)
{
    //Creating Emitter
    auto emitter = m_World->CreateEntity();
    auto emitterTransform = m_World->AddComponent<Components::Transform>(emitter);
    auto particleEmitter= m_World->AddComponent<Components::ParticleEmitter>(emitter);

    emitterTransform->Position = event.Position;
    particleEmitter->GravityScale = event.GravityScale;
    particleEmitter->SpawnRate = event.SpawnRate;
    particleEmitter->NumberOfTicks = event.NumberOfTicks;
    particleEmitter->Speed = event.Speed;
    particleEmitter->ParticlesPerTick = event.ParticlesPerTick;
    particleEmitter->Spread = event.Spread;
    particleEmitter->EmittingAngle = event.EmittingAngle;
    particleEmitter->LifeTime = event.EmitterLifeTime;

    {
        //Creating Particle Template
        auto particle = m_World->CreateEntity(emitter);
        auto particleTransform = m_World->AddComponent<Components::Transform>(particle);
        auto sprite = m_World->AddComponent<Components::Sprite>(particle);
        auto particleComponent = m_World->AddComponent<Components::Particle>(particle);
        m_World->AddComponent<Components::Template>(particle);

        particleTransform->Position = emitterTransform->Position;
        sprite->SpriteFile = event.SpriteFile;
        particleComponent->LifeTime = event.ParticleLifeTime;
        particleComponent->Flags = event.Flags;
        particleComponent->Radius = event.Radius;
    }
    m_World->CommitEntity(emitter);
    return true;
}

void dd::Systems::PhysicsSystem::CreateParticleEmitter(EntityID entity)
{
    auto childEntities = m_World->GetEntityChildren(entity);
    if (childEntities.empty()) {
        LOG_ERROR("Particle Emitter does not have any child. Please adopt one or this will not work.");
        return;
    }
    int pf = 0;
    dd::Components::Particle* particle;
    EntityID childEntity;
    for ( auto c : childEntities ) {
        auto p = m_World->GetComponent<Components::Particle>(c);
        auto particleTemplate = m_World->GetComponent<Components::Template>(c);
        if (!p) {
            continue;
            LOG_WARNING("ParticleEmitter's Child is not a particle.");
        }
        if(!particleTemplate) {
            LOG_ERROR("ParticleEmitter's particleChild is not a template.");
            continue;
        }
        if (pf != 0) {
            LOG_WARNING("ParticleEmitter has more than one child that is a particleTemplate, only the first one is used.");
            break;
        }
        childEntity = c;
        particle = p;
        pf++;
    }
    //TODO: Skicka med fler flaggor till particlesystemet;

    m_ParticleEmitters.ParticleSystem.push_back(CreateParticleSystem(particle->Radius, 0.f, 0));
    m_ParticleEmitters.ParticleEmitter.push_back(entity);
    m_ParticleEmitters.ParticleTemplate.push_back(childEntity);
}


b2ParticleSystem* dd::Systems::PhysicsSystem::CreateParticleSystem(float radius, float gravityScale, int maxCount)
{
    std::unordered_map<EntityID, const b2ParticleHandle*> m1;
    std::unordered_map<const b2ParticleHandle*, EntityID> m2;
    m_EntitiesToParticleHandle.push_back(m1);
    m_ParticleHandleToEntities.push_back(m2);

    b2ParticleSystemDef m_ParticleSystemDef;
    m_ParticleSystemDef.maxCount = maxCount;
    m_ParticleSystemDef.radius = radius;
    m_ParticleSystemDef.gravityScale = gravityScale;
    m_ParticleSystemDef.destroyByAge = true;

    return m_PhysicsWorld->CreateParticleSystem(&m_ParticleSystemDef);
}

dd::Systems::PhysicsSystem::~PhysicsSystem()
{
    if (m_ContactListener != nullptr) {
        delete m_ContactListener;
        m_ContactListener = nullptr;
    }
}