From 875cc9e520d63cdb3fddcbc6b3db7179c7b4c0b0 Mon Sep 17 00:00:00 2001
From: Krisjanis Rijnieks <krisjanis.rijnieks@gmail.com>
Date: Fri, 21 Oct 2016 19:09:26 +0200
Subject: [PATCH] Add TextureMappingMode::setup

---
 src/Application/Modes/TextureMappingMode.cpp | 5 ++++-
 src/Application/Modes/TextureMappingMode.h   | 1 +
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/Application/Modes/TextureMappingMode.cpp b/src/Application/Modes/TextureMappingMode.cpp
index f962359..76cb98b 100644
--- a/src/Application/Modes/TextureMappingMode.cpp
+++ b/src/Application/Modes/TextureMappingMode.cpp
@@ -19,6 +19,10 @@ TextureMappingMode::TextureMappingMode(){
 	_drawMode = 0;
 }
 
+void TextureMappingMode::setup(Application * app){
+	Gui::instance()->getTextureHighlightWidget().setSurfaceManager(app->getSurfaceManager());
+}
+
 void TextureMappingMode::update(Application * app){
 	Gui::instance()->getTextureEditorWidget().update();
 }
@@ -63,7 +67,6 @@ void TextureMappingMode::draw(Application * app){
 	
 	ofPushMatrix();
 	ofTranslate(_canvasTranslate.x, _canvasTranslate.y);
-	Gui::instance()->getTextureHighlightWidget().setSurfaceManager(app->getSurfaceManager());
 	Gui::instance()->getTextureHighlightWidget().draw();
 	Gui::instance()->getTextureEditorWidget().draw();
 	ofPopMatrix();
diff --git a/src/Application/Modes/TextureMappingMode.h b/src/Application/Modes/TextureMappingMode.h
index a48242f..244e1f8 100644
--- a/src/Application/Modes/TextureMappingMode.h
+++ b/src/Application/Modes/TextureMappingMode.h
@@ -24,6 +24,7 @@ class TextureMappingMode : public ApplicationBaseMode {
 	public:
 		static TextureMappingMode * instance();
 	
+		void setup(Application * app);
 		void update(Application * app);
 		void draw(Application * app);
 		void onKeyPressed(Application * app, ofKeyEventArgs & args);