From 20ab7df19fe6a2a7aa3b2f9c2c52b6cc187e95bd Mon Sep 17 00:00:00 2001
From: Nelsson Huotari <unelsson@gmail.com>
Date: Sun, 7 Apr 2019 21:48:55 +0300
Subject: [PATCH] Fixes, cleanup.

---
 apps/opencs/view/render/terrainselection.cpp   | 4 ++--
 apps/opencs/view/render/terraintexturemode.hpp | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/apps/opencs/view/render/terrainselection.cpp b/apps/opencs/view/render/terrainselection.cpp
index 8fd0d8aed..ab94da354 100644
--- a/apps/opencs/view/render/terrainselection.cpp
+++ b/apps/opencs/view/render/terrainselection.cpp
@@ -110,12 +110,12 @@ void CSVRender::TerrainSelection::toggleSelect(const std::vector<std::pair<int,
 
 void CSVRender::TerrainSelection::activate()
 {
-    mParentNode->addChild(mSelectionNode);
+    if (!mParentNode->containsNode(mSelectionNode)) mParentNode->addChild(mSelectionNode);
 }
 
 void CSVRender::TerrainSelection::deactivate()
 {
-    mParentNode->removeChild(mSelectionNode);
+    if (mParentNode->containsNode(mSelectionNode)) mParentNode->removeChild(mSelectionNode);
 }
 
 void CSVRender::TerrainSelection::update()
diff --git a/apps/opencs/view/render/terraintexturemode.hpp b/apps/opencs/view/render/terraintexturemode.hpp
index 1f18409b0..72e7df9de 100644
--- a/apps/opencs/view/render/terraintexturemode.hpp
+++ b/apps/opencs/view/render/terraintexturemode.hpp
@@ -6,6 +6,8 @@
 #include <string>
 #include <memory>
 
+#include <osg/Group>
+
 #include <QWidget>
 #include <QEvent>
 
@@ -28,8 +30,6 @@ namespace CSVWidget
 
 namespace CSVRender
 {
-    class PagedWorldspaceWidget;
-
     class TerrainTextureMode : public EditMode
     {
         Q_OBJECT