1 files changed, 9 insertions, 6 deletions
diff --git a/EditManager.h b/EditManager.h
index 6b52389..00729bf 100644
--- a/EditManager.h
+++ b/EditManager.h
@@ -1,6 +1,7 @@
 #ifndef EDITMANAGER_H_
 #define EDITMANAGER_H_
 
+#include "SharedPtr.h"
 #include "Room.h"
 #include "IdManager.h"
 
@@ -22,9 +23,9 @@ class EditManager {
     Window *window;
     
     Room newRoom;
-    Room *activeRoom;
+    LevelObject *selectedObject;
     
-    Room *hoveredRoom;
+    LevelObject *hoveredObject;
     
     Vertex hoveredVertex;
     bool hasHoveredVertex;
@@ -34,6 +35,8 @@ class EditManager {
   public:
     EditManager(Window *window);
     
+    void update();
+    
     void addRoom();
     void finishRoom();
     
@@ -43,12 +46,12 @@ class EditManager {
       return mode;
     }
     
-    Room* getActiveRoom() {
-      return activeRoom;
+    LevelObject* getSelectedObject() {
+      return selectedObject;
     }
     
-    Room* getHoveredRoom() {
-      return hoveredRoom;
+    LevelObject* getHoveredObject() {
+      return hoveredObject;
     }
     
     Vertex* getHoveredVertex();