From 19690f030457dadf4976f7403c315b3bbb4a39b8 Mon Sep 17 00:00:00 2001
From: neoraider <devnull@localhost>
Date: Wed, 5 Dec 2007 18:34:01 +0000
Subject: zoomedit: Use toolbox for polygon adder.

---
 EditManager.cpp | 33 ++++++++++++++++++++-------------
 1 file changed, 20 insertions(+), 13 deletions(-)

(limited to 'EditManager.cpp')

diff --git a/EditManager.cpp b/EditManager.cpp
index 5822ffc..37fbcfd 100644
--- a/EditManager.cpp
+++ b/EditManager.cpp
@@ -41,23 +41,30 @@ EditManager::EditManager(Window *window) {
 }
 
 void EditManager::addRoom() {
-  newRoom = Room(idManager.generate("room"));
-  mode = ADD;
-  
-  activeRoom = &newRoom;
-  
-  window->update();
+  if(mode == VIEW) {
+    newRoom = Room(idManager.generate("room"));
+    mode = ADD;
+    
+    activeRoom = &newRoom;
+    
+    window->update();
+  }
 }
 
 void EditManager::finishRoom() {
-  mode = VIEW;
-  
-  if(newRoom.size() > 2 && polygonOk(newRoom)) {
-    window->getLevel().push_back(newRoom);
-    activeRoom = &window->getLevel().back();
+  if(mode == ADD) {
+    mode = VIEW;
+    
+    if(newRoom.size() > 2 && polygonOk(newRoom)) {
+      window->getLevel().push_back(newRoom);
+      activeRoom = &window->getLevel().back();
+    }
+    else {
+      activeRoom = NULL;
+    }
+    
+    window->update();
   }
-  
-  window->update();
 }
 
 void EditManager::addVertex(const Vertex &v) {
-- 
cgit v1.2.3