added usages of trim to lists

[physics.git] / src / entityCreator.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 "entityCreator.h"

#include <bear/Queue.h>

#include "entityManager.h"
#include "input/inputManager.h"

#include "Entities/Ball.h"
#include "Entities/Polygon.h"
#include "graphics/colors.h"


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

typedef bear::Queue<Ball*>      queBall;
typedef bear::Queue<Polygon*>   quePoly;

static queBall s_startBalls;
static queBall s_mouseBalls;

static quePoly s_startPolys;

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

static Ball* addBall(const Vector2& vecPos,
                     float fRadius,
                     const float* color,
                     queBall& que);

static Polygon* addPoly(const vector<Vector2>& vecPoints,
                        const float* color,
                        quePoly& que);

static void removeBall(queBall& que);
static void removeAllBalls(queBall& que);

static void removePoly(quePoly& que);
static void removeAllPolys(quePoly& que);

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

void creator::init()
{
    Ball* pBall;
    float fStartMass = 5;

    s_startBalls.init();
    s_mouseBalls.init();

    s_startPolys.init();

    pBall = addBall(Vector2(50, 50), 20, cWhite, s_startBalls);
    pBall->mass = fStartMass;

    pBall = addBall(Vector2(150, 50), 20, cGrey, s_startBalls);
    pBall->mass = fStartMass;

    pBall = addBall(Vector2(50, 100), 20, cRed, s_startBalls);
    pBall->mass = fStartMass;

    pBall = addBall(Vector2(100, 100), 20, cGreen, s_startBalls);
    pBall->mass = fStartMass;

    pBall = addBall(Vector2(150, 100), 20, cBlue, s_startBalls);
    pBall->mass = fStartMass;

    pBall = addBall(Vector2(50, 150), 20, cYellow, s_startBalls);
    pBall->mass = fStartMass;

    pBall = addBall(Vector2(100, 150), 20, cMagenta, s_startBalls);
    pBall->mass = fStartMass;

    pBall = addBall(Vector2(150, 150), 20, cCyan, s_startBalls);
    pBall->mass = fStartMass;


    for( int i = 0; i < 50; i++)
    {
        addBall(Vector2(200 + i * 2, 200 + i * 2), 10, cCyan);
    }


    vector<Vector2> points;
    points.push_back(Vector2(500,500));
    points.push_back(Vector2(300,500));
    points.push_back(Vector2(500,300));

    addPoly(points, cRed, s_startPolys);


    s_startPolys.trim();
    s_startBalls.trim();
}
void creator::clean()
{
    removeAllPolys(s_startPolys);
    s_startPolys.fini();

    removeAllBalls(s_mouseBalls);
    s_mouseBalls.fini();

    removeAllBalls(s_startBalls);
    s_startBalls.fini();
}

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

void creator::addBall(const Vector2& pos, float radius, const float* color)
{
    addBall(pos, radius, color, s_mouseBalls);
}
void creator::removeBall()
{
    removeBall(s_mouseBalls);
}
void creator::removeAllBalls()
{
    removeAllBalls(s_mouseBalls);
}

void creator::handleInput()
{
    if(input::mouseLeft())
        addBall(input::mousePosition(), 10, cCyan);

    if(input::mouseRight() && ! s_mouseBalls.isEmpty())
        removeBall();
}

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

Ball* addBall(const Vector2& vecPos,
              float fRadius,
              const float* color,
              queBall& que)
{
    Ball* pBall = new Ball(vecPos, fRadius, color);

    que.pushBack(pBall);
    manager::add(pBall);

    return pBall;
}
Polygon* addPoly(const vector<Vector2>& vecPoints,
                 const float* color,
                 quePoly& que)
{
    Polygon* pPoly = new Polygon(vecPoints, color);

    que.pushBack(pPoly);
    manager::add(pPoly);

    return pPoly;
}

void removeBall(queBall& que)
{
    Ball* pBall = que.getFront();

    manager::remove(pBall);
    que.popFront();

    delete pBall;
}
void removeAllBalls(queBall& que)
{
    while(! que.isEmpty() )
    {
        removeBall(que);
    }
}

static void removePoly(quePoly& que)
{
    Polygon* pPoly = que.getFront();

    manager::remove(pPoly);
    que.popFront();

    delete pPoly;
}
static void removeAllPolys(quePoly& que)
{
    while(! que.isEmpty() )
    {
        removePoly(que);
    }
}