X-Git-Url: http://gitweb.pgornicz.com/gitweb.cgi?a=blobdiff_plain;f=src%2FEntities%2FPolygon.h;h=6b115fcee9f8042f6348299537072b8c4d1da708;hb=f32a9b7c8eab3536ad354f85ee65c41d5b5da006;hp=b8c8493c312dfafdfee191b7f655f3b722c106db;hpb=4a76d2e2cf7874e54a4e6688ebf1fa8ca59ce8c1;p=physics.git

diff --git a/src/Entities/Polygon.h b/src/Entities/Polygon.h
index b8c8493..6b115fc 100644
--- a/src/Entities/Polygon.h
+++ b/src/Entities/Polygon.h
@@ -18,8 +18,9 @@
 #ifndef POLYGON_H
 #define POLYGON_H
 
+#include <pg/Vector2.h>
+
 #include "PhysicsEntity.h"
-#include "Vector2.h"
 
 #include <vector>
 using std::vector;
@@ -30,20 +31,21 @@ 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;
 
-    protected:
+    //protected:
         Vector2 maxP; // stores the max bounding box point
         Vector2 minP; // stores the min bounding box point
 
         vector<Vector2> points;
 
-        // color;
+        const float* color;
     private:
         void createBindingBox();
+        void centerPosition();
 };
 
 #endif // POLYGON_H