gravity well now follows the mouse

[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 "Entities/Entity.h"
#include "Entities/Particle.h"
#include "Entities/PhysicsEntity.h"

#include "collisionHandler.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;

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

void manager::init()
{
    collision::init();
}
void manager::clean()
{
    collision::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;
        }
    }

#ifdef WARNINGS
    cerr << "ENTITY TYPE NOT SUPPORTED BY addEntity()!!" << endl;
#endif
}
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;
        }
    }

#ifdef WARNINGS
    cerr << "ENTITY TYPE NOT SUPPORTED BY deleteEntity()!!" << endl;
#endif
}

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

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

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

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

void updateParticles(float time_step)
{
    // 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();

    // 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)
{
    // 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();

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

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