wip: more thread safety

[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 "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;

SDL_mutex* particleSetLock      = NULL;
SDL_mutex* physicsEntitySetLock = NULL;

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

void manager::init()
{
    particleSetLock = SDL_CreateMutex();
    physicsEntitySetLock = SDL_CreateMutex();

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

    SDL_DestroyMutex(particleSetLock);
    particleSetLock = NULL;

    SDL_DestroyMutex(physicsEntitySetLock);
    physicsEntitySetLock = NULL;
}

/// ***** 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()
{
    SDL_mutexP( particleSetLock );
        // update active Particle*s
        for( setPart::iterator it = active_Particles.begin();
             it != active_Particles.end();
             it++ )
        {
            (*it)->draw();
        }
    SDL_mutexV( particleSetLock );

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

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

void updateParticles(float time_step)
{
    SDL_mutexP( particleSetLock );
        // add new Particle*s to Active
        for( setPart::iterator it = particles_To_Add.begin();
             it != particles_To_Add.end();
             it++ )
        {
            active_Particles.insert(*it);
        }
        particles_To_Add.clear();

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

    // 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)
{
    SDL_mutexP( physicsEntitySetLock );
        // add new PhysicsEntity*s to Active
        for( setPhys::iterator it = physics_To_Add.begin();
             it != physics_To_Add.end();
             it++ )
        {
            active_Physics.insert(*it);
        }
        physics_To_Add.clear();

        // remove dead PhysicsEntity*s from Active
        for( setPhys::iterator it = physics_To_Remove.begin();
             it != physics_To_Remove.end();
             it++ )
        {
            active_Physics.erase(*it);
        }
        physics_To_Remove.clear();
    SDL_mutexV( physicsEntitySetLock );

    // 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);
    }
}