1 files changed, 15 insertions, 2 deletions
diff --git a/EditManager.h b/EditManager.h
index 44e1e11..bd33c9d 100644
--- a/EditManager.h
+++ b/EditManager.h
@@ -16,6 +16,7 @@ class EditManager {
     Window *window;
     
     LevelObject *hoveredObject;
+    LevelObject *highlightedObject;
     LevelObject *selectedObject;
     
     Vertex hoveredVertex;
@@ -32,6 +33,18 @@ class EditManager {
       return hoveredObject;
     }
     
+    void setHighlightedObject(LevelObject *object) {
+      highlightedObject = object;
+    }
+    
+    LevelObject* getHighlightedObject() {
+      return highlightedObject;
+    }
+    
+    void highlightHoveredObject() {
+      highlightedObject = hoveredObject;
+    }
+    
     void setSelectedObject(LevelObject *object) {
       selectedObject = object;
     }
@@ -42,8 +55,8 @@ class EditManager {
     
     bool addRoom(const Room &room);
     
-    Vertex* getHoveredVertex();
-    void setHoveredVertex(Vertex *v);
+    const Vertex* getHoveredVertex() const;
+    void setHoveredVertex(const Vertex *v);
     
     bool vertexOk(const Vertex &v, const Room *newRoom = NULL) const;
     bool polygonOk(const Polygon &polygon) const;