X-Git-Url: http://gitweb.pgornicz.com/gitweb.cgi?a=blobdiff_plain;f=src%2FEntities%2FPolygon.h;h=6624c95932c8954a198fd740b8dcdded78ef33d7;hb=7d8057c331d6ed190990b69cf251e062bc2d3126;hp=82ff9d7283cd1ce76705a76e01269ed86b7093db;hpb=aa2791cf43a9ddd3a288e504db08e11d03439653;p=physics.git

diff --git a/src/Entities/Polygon.h b/src/Entities/Polygon.h
index 82ff9d7..6624c95 100644
--- a/src/Entities/Polygon.h
+++ b/src/Entities/Polygon.h
@@ -18,8 +18,10 @@
 #ifndef POLYGON_H
 #define POLYGON_H
 
+#include <bear/Vector2.h>
+using namespace bear;
+
 #include "PhysicsEntity.h"
-#include "Vector2.h"
 
 #include <vector>
 using std::vector;
@@ -30,7 +32,7 @@ using std::vector;
 class Polygon: public PhysicsEntity
 {
     public:
-        Polygon(const vector<Vector2>&);
+        Polygon(const vector<Vector2>&, const float* color);
         virtual ~Polygon();
 
         virtual void draw() const;
@@ -41,7 +43,7 @@ class Polygon: public PhysicsEntity
 
         vector<Vector2> points;
 
-        // color;
+        const float* color;
     private:
         void createBindingBox();
         void centerPosition();