X-Git-Url: http://gitweb.pgornicz.com/gitweb.cgi?a=blobdiff_plain;f=src%2FentityCreator.cpp;h=191d3960ab350bc7e60809179c18835a6aa75c23;hb=30a93c3992f0e1fc44193dde6d53216d3ae4f4e9;hp=8ed35f4230b0bcb71ae3ba29397bb440075b40ce;hpb=89ca62b2694538f2ca5845b40aed7c2dec5143fe;p=physics.git

diff --git a/src/entityCreator.cpp b/src/entityCreator.cpp
index 8ed35f4..191d396 100644
--- a/src/entityCreator.cpp
+++ b/src/entityCreator.cpp
@@ -1,87 +1,154 @@
+/*
+ *  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 <queue>
 
 #include "entityManager.h"
+#include "input/inputManager.h"
+
 #include "Entities/Ball.h"
+#include "Entities/Polygon.h"
 #include "graphics/colors.h"
 
-/// ***** Private Method Headers *****
-typedef std::queue<Ball*> queBall;
-queBall Balls;
 
 /// ***** Private Variables *****
 
-/// ***** Public Methods *****
-void creator::init()
-{
-    Ball* ball;
-
-    ball = new Ball(Vector2(50, 50), 20, cWhite);
+typedef std::queue<Ball*> queBall;
 
-    ball->applyImpulse(Vector2(0.01,0.01)),
-    Balls.push(ball);
-    manager::add(ball);
+queBall startBalls;
+queBall mouseBalls;
 
-    ball = new Ball(Vector2(100, 50), 20, cBlack);
+/// ***** Private Method Headers *****
 
-    Balls.push(ball);
-    manager::add(ball);
+Ball* addBall(const Vector2& pos,
+              float radius,
+              const float* color,
+              queBall& que);
 
-    ball = new Ball(Vector2(150, 50), 20, cGrey);
+void removeBall(queBall& que);
+void removeAllBalls(queBall& que);
 
-    Balls.push(ball);
-    manager::add(ball);
+/// ***** Initializers/Cleaners *****
 
-    ball = new Ball(Vector2(50, 100), 20, cRed);
+void creator::init()
+{
+    Ball* ball;
+    float startMass = 5;
 
-    Balls.push(ball);
-    manager::add(ball);
+    ball = addBall(Vector2(50, 50), 20, cWhite, startBalls);
+    ball->mass = startMass;
 
-    ball = new Ball(Vector2(100, 100), 20, cGreen);
+    ball = addBall(Vector2(150, 50), 20, cGrey, startBalls);
+    ball->mass = startMass;
 
-    Balls.push(ball);
-    manager::add(ball);
+    ball = addBall(Vector2(50, 100), 20, cRed, startBalls);
+    ball->mass = startMass;
 
-    ball = new Ball(Vector2(150, 100), 20, cBlue);
+    ball = addBall(Vector2(100, 100), 20, cGreen, startBalls);
+    ball->mass = startMass;
 
-    Balls.push(ball);
-    manager::add(ball);
+    ball = addBall(Vector2(150, 100), 20, cBlue, startBalls);
+    ball->mass = startMass;
 
-    ball = new Ball(Vector2(50, 150), 20, cYellow);
+    ball = addBall(Vector2(50, 150), 20, cYellow, startBalls);
+    ball->mass = startMass;
 
-    Balls.push(ball);
-    manager::add(ball);
+    ball = addBall(Vector2(100, 150), 20, cMagenta, startBalls);
+    ball->mass = startMass;
 
-    ball = new Ball(Vector2(100, 150), 20, cMagenta);
+    ball = addBall(Vector2(150, 150), 20, cCyan, startBalls);
+    ball->mass = startMass;
 
-    Balls.push(ball);
-    manager::add(ball);
+    for( int i = 0; i<50; i++)
+    {
+        addBall(Vector2(200+i*2, 200+i*2), 10, cCyan);
+    }
 
-    ball = new Ball(Vector2(150, 150), 20, cCyan);
+    // HACK
+    // add a polygon into the mix (currently not cleaned up)
+    vector<Vector2> points;
+    points.push_back(Vector2(500,500));
+    points.push_back(Vector2(300,500));
+    points.push_back(Vector2(500,300));
 
-    Balls.push(ball);
-    manager::add(ball);
+    manager::add(new Polygon(points, cRed));
 }
 void creator::clean()
 {
-    removeAllBalls();
+    removeAllBalls(startBalls);
+    removeAllBalls(mouseBalls);
 }
 
-void creator::addBall()
-{
+/// ***** Public Methods *****
 
+void creator::addBall(const Vector2& pos, float radius, const float* color)
+{
+    addBall(pos, radius, color, mouseBalls);
 }
 void creator::removeBall()
 {
-
+    removeBall(mouseBalls);
 }
 void creator::removeAllBalls()
 {
-    while(! Balls.empty() )
+    removeAllBalls(mouseBalls);
+}
+
+void creator::handleInput()
+{
+    if(input::mouseLeft())
+        addBall(input::mousePosition(), 10, cCyan);
+
+    if(input::mouseRight() && ! mouseBalls.empty())
+        removeBall();
+}
+
+/// ***** Private Methods *****
+
+Ball* addBall(const Vector2& pos,
+                       float radius,
+                       const float* color,
+                       queBall& que)
+{
+    Ball* ball = new Ball(pos, radius, color);
+
+    que.push(ball);
+    manager::add(ball);
+
+    return ball;
+}
+
+void removeBall(queBall& que)
+{
+    Ball* ball = que.front();
+
+    manager::remove(ball);
+    que.pop();
+
+    delete ball;
+}
+void removeAllBalls(queBall& que)
+{
+    while(! que.empty() )
     {
-        Ball* ball = Balls.front();
-        Balls.pop();
+        Ball* ball = que.front();
+        que.pop();
         manager::remove(ball);
 
         delete ball;