cleaned how balls are added

[physics.git] / src / entityCreator.cpp

diff --git a/src/entityCreator.cpp b/src/entityCreator.cpp
index eddfb3d..b0c034c 100644 (file)
--- a/src/entityCreator.cpp
+++ b/src/entityCreator.cpp
@@ -20,55 +20,102 @@
 #include <queue>
 
 #include "entityManager.h"
+#include "input/inputManager.h"
+
 #include "Entities/Ball.h"
 #include "graphics/colors.h"
 
+
 /// ***** Private Variables *****
 
 typedef std::queue<Ball*> queBall;
-queBall Balls;
+
+queBall startBalls;
+queBall mouseBalls;
+
+/// ***** Private Method Headers *****
+
+Ball* addBall(const Vector2& pos,
+              float radius,
+              const float* color,
+              queBall& que);
+
+void removeBall(queBall& que);
+void removeAllBalls(queBall& que);
 
 /// ***** Initializers/Cleaners *****
 
 void creator::init()
 {
-    addBall(Vector2(50, 50), 20, cWhite);
-    addBall(Vector2(150, 50), 20, cGrey);
-    addBall(Vector2(50, 100), 20, cRed);
-    addBall(Vector2(100, 100), 20, cGreen);
-    addBall(Vector2(150, 100), 20, cBlue);
-    addBall(Vector2(50, 150), 20, cYellow);
-    addBall(Vector2(100, 150), 20, cMagenta);
-    addBall(Vector2(150, 150), 20, cCyan);
+    addBall(Vector2(50, 50), 20, cWhite, startBalls);
+    addBall(Vector2(150, 50), 20, cGrey, startBalls);
+    addBall(Vector2(50, 100), 20, cRed, startBalls);
+    addBall(Vector2(100, 100), 20, cGreen, startBalls);
+    addBall(Vector2(150, 100), 20, cBlue, startBalls);
+    addBall(Vector2(50, 150), 20, cYellow, startBalls);
+    addBall(Vector2(100, 150), 20, cMagenta, startBalls);
+    addBall(Vector2(150, 150), 20, cCyan, startBalls);
 }
 void creator::clean()
 {
-    removeAllBalls();
+    removeAllBalls(startBalls);
+    removeAllBalls(mouseBalls);
 }
 
 /// ***** 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()
+{
+    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);
 
-    Balls.push(ball);
+    que.push(ball);
     manager::add(ball);
+
+    return ball;
 }
-void creator::removeBall()
+
+void removeBall(queBall& que)
 {
-    Ball* ball = Balls.front();
+    Ball* ball = que.front();
 
-    Balls.pop();
+    manager::remove(ball);
+    que.pop();
 
     delete ball;
 }
-void creator::removeAllBalls()
+void removeAllBalls(queBall& que)
 {
-    while(! Balls.empty() )
+    while(! que.empty() )
     {
-        Ball* ball = Balls.front();
-        Balls.pop();
+        Ball* ball = que.front();
+        que.pop();
         manager::remove(ball);
 
         delete ball;