added better thread safty ... still broke

[physics.git] / src / entityManager.cpp

/*
 *  Copyright (C) 2008 Patrik Gornicz, Gornicz_P (at) hotmail (dot) com.
 *
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include "entityManager.h"
#include "debug.h"

#include <set>
#include <SDL/SDL.h>

#include "locks/Mutex.h"
#include "locks/Autolock.h"

#include "Entities/Entity.h"
#include "Entities/Particle.h"
#include "Entities/PhysicsEntity.h"

#include "collisionManager.h"
#include "effectManager.h"

/// ***** Private Method Headers *****

void updateParticles(float);
void updatePhysics(float);

/// ***** Private Variables *****

typedef std::set<Particle*> setPart;
setPart particles_To_Add;
setPart active_Particles;
setPart particles_To_Remove;

typedef std::set<PhysicsEntity*> setPhys;
setPhys physics_To_Add;
setPhys active_Physics;
setPhys physics_To_Remove;

Mutex particleSetLock;
Mutex physicsEntitySetLock;

/// ***** Initializers/Cleaners *****

void manager::init()
{
    particleSetLock.init();
    physicsEntitySetLock.init();

    collision::init();
}
void manager::clean()
{
    collision::clean();

    physicsEntitySetLock.clean();
    particleSetLock.clean();
}

/// ***** Public Methods *****

void manager::add(Entity* e)
{
    {
        Particle* p = dynamic_cast<Particle*>(e);
        if( p != 0 )
        {
            particles_To_Add.insert(p);
            return;
        }
    }

    {
        PhysicsEntity* p = dynamic_cast<PhysicsEntity*>(e);
        if( p != 0 )
        {
            physics_To_Add.insert(p);
            return;
        }
    }

    DPF(0, "ENTITY TYPE NOT SUPPORTED BY addEntity()!!");
}
void manager::remove(Entity* e)
{
    {
        Particle* p = dynamic_cast<Particle*>(e);
        if( p != 0 )
        {
            particles_To_Remove.insert(p);
            return;
        }
    }

    {
        PhysicsEntity* p = dynamic_cast<PhysicsEntity*>(e);
        if( p != 0 )
        {
            physics_To_Remove.insert(p);
            return;
        }
    }

    DPF(0, "ENTITY TYPE NOT SUPPORTED BY deleteEntity()!!");
}

void manager::handleInput()
{
    effect::handleInput();
}
void manager::update(float time_step)
{
    effect::update(time_step);

    updateParticles(time_step);
    updatePhysics(time_step);
}
void manager::draw()
{
    {
        Autolock lock( particleSetLock );
        DPF(0, "Particle Draw Start");

        // remove dead Particle*s from Active
        for( setPart::iterator it = particles_To_Remove.begin();
             it != particles_To_Remove.end();
             it++ )
        {
            DPF(0, "Particle Delete");
            active_Particles.erase(*it);
        }
        particles_To_Remove.clear();

        // update active Particle*s
        for( setPart::iterator it = active_Particles.begin();
             it != active_Particles.end();
             it++ )
        {
            (*it)->draw();
        }

        DPF(0, "Particle Draw End");
    }

    {
        Autolock lock( physicsEntitySetLock );
        DPF(0, "Physics Draw Start");

        // remove dead PhysicsEntity*s from Active
        for( setPhys::iterator it = physics_To_Remove.begin();
             it != physics_To_Remove.end();
             it++ )
        {
            DPF(0, "Physics Delete");
            active_Physics.erase(*it);
        }
        physics_To_Remove.clear();

        // update active PhysicsEntity*s
        for( setPhys::iterator it = active_Physics.begin();
             it != active_Physics.end();
             it++ )
        {
            (*it)->draw();
        }

        DPF(0, "Physics Draw End");
    }
}

/// ***** Private Methods *****

void updateParticles(float time_step)
{
    {
        Autolock lock( particleSetLock );

        // add new Particle*s to Active
        for( setPart::iterator it = particles_To_Add.begin();
             it != particles_To_Add.end();
             it++ )
        {
            DPF(0, "Particle Insert");
            active_Particles.insert(*it);
        }

        // remove dead Particle*s from Active
        for( setPart::iterator it = particles_To_Remove.begin();
             it != particles_To_Remove.end();
             it++ )
        {
            DPF(0, "Particle Delete");
            active_Particles.erase(*it);
        }
        particles_To_Remove.clear();
    }

    particles_To_Add.clear();

    // update active Particle*s
    for( setPart::iterator it = active_Particles.begin();
         it != active_Particles.end();
         it++ )
    {
        (*it)->update(time_step);
    }
}
void updatePhysics(float time_step)
{
    {
        Autolock lock( physicsEntitySetLock );

        // add new PhysicsEntity*s to Active
        for( setPhys::iterator it = physics_To_Add.begin();
             it != physics_To_Add.end();
             it++ )
        {
            DPF(0, "Physics Insert");
            active_Physics.insert(*it);
        }

        // remove dead PhysicsEntity*s from Active
        for( setPhys::iterator it = physics_To_Remove.begin();
             it != physics_To_Remove.end();
             it++ )
        {
            DPF(0, "Physics Delete");
            active_Physics.erase(*it);
        }
        physics_To_Remove.clear();
    }

    physics_To_Add.clear();

    // apply collision math
    collision::update(active_Physics);

    // update active PhysicsEntity*s
    for( setPhys::iterator it = active_Physics.begin();
         it != active_Physics.end();
         it++ )
    {
        (*it)->update(time_step);
    }
}