X-Git-Url: http://gitweb.pgornicz.com/gitweb.cgi?a=blobdiff_plain;f=src%2FEffects%2FScreen.cpp;h=61108c42b01341553e2e21d208c38d8b82a6cab9;hb=46e842c7cc2e977f9bdd9fb4480dac26163fa25b;hp=44cce1879eff8d5d8d0f7f5803872431c2b25436;hpb=e68f847b245153427266841ae724d602ca434c29;p=physics.git

diff --git a/src/Effects/Screen.cpp b/src/Effects/Screen.cpp
index 44cce18..61108c4 100644
--- a/src/Effects/Screen.cpp
+++ b/src/Effects/Screen.cpp
@@ -34,6 +34,34 @@ Screen::~Screen()
 
 /// ***** Public Class Methods *****
 
+Vector2 Screen::positionDelta(const PhysicsEntity* e, float time_step) const
+{
+    const Vector2& pos = e->positionRaw();
+    const Vector2& velo = e->velocityRaw();
+
+    Vector2 acc(0,0);
+
+    float radius = 0;
+    const Ball* b = dynamic_cast<const Ball*>(e);
+    if( b != NULL )
+        radius = b->getRadius();
+
+
+    if(pos.y > 600-radius)
+        acc.y += 600-radius - pos.y;
+
+    if(pos.y < 0+radius)
+        acc.y += 0+radius - pos.y;
+
+    if(pos.x > 800-radius)
+        acc.x += 800-radius - pos.x;
+
+    if(pos.x < 0+radius)
+        acc.x += 0+radius - pos.x;
+
+    return acc;
+}
+
 Vector2 Screen::velocityDelta(const PhysicsEntity* e, float time_step) const
 {
     const Vector2& pos = e->positionRaw();
@@ -41,6 +69,7 @@ Vector2 Screen::velocityDelta(const PhysicsEntity* e, float time_step) const
 
     Vector2 acc(0,0);
 
+    float CoR = e->CoR;
     float radius = 0;
     const Ball* b = dynamic_cast<const Ball*>(e);
     if( b != NULL )
@@ -48,16 +77,16 @@ Vector2 Screen::velocityDelta(const PhysicsEntity* e, float time_step) const
 
 
     if(pos.y > 600-radius && velo.y > 0)
-        acc.y += velo.y * -2;
+        acc.y += velo.y * -(1 + CoR);
 
     if(pos.y < 0+radius && velo.y < 0)
-        acc.y += velo.y * -2;
+        acc.y += velo.y * -(1 + CoR);
 
     if(pos.x > 800-radius && velo.x > 0)
-        acc.x += velo.x * -2;
+        acc.x += velo.x * -(1 + CoR);
 
     if(pos.x < 0+radius && velo.x < 0)
-        acc.x += velo.x * -2;
+        acc.x += velo.x * -(1 + CoR);
 
     return acc;
 }