From 93fa8d3e3bbf10246725bb9db7764380e22c5ac8 Mon Sep 17 00:00:00 2001
From: dfighter1985 <dfighter1985@localhost>
Date: Sat, 26 Jul 2014 02:11:02 +0200
Subject: [PATCH] A little refactoring.

---
 .../tile_editor/tile_editor_main_window.cpp   | 33 +++++++++++--------
 .../tile_editor/tile_editor_main_window.h     |  1 +
 2 files changed, 21 insertions(+), 13 deletions(-)

diff --git a/code/studio/src/plugins/tile_editor/tile_editor_main_window.cpp b/code/studio/src/plugins/tile_editor/tile_editor_main_window.cpp
index 25406e389..4884ea42b 100644
--- a/code/studio/src/plugins/tile_editor/tile_editor_main_window.cpp
+++ b/code/studio/src/plugins/tile_editor/tile_editor_main_window.cpp
@@ -573,13 +573,7 @@ void TileEditorMainWindow::onActionAddTile(int tabId)
 
 	QModelIndex rootIdx = model->index( tabId, 0, m_ui->tileSetLV->currentIndex());
 
-	QListView *lv = NULL;
-
-	switch( tabId )
-	{
-	case TAB_128: lv = m_ui->listView128; break;
-	case TAB_256: lv = m_ui->listView256; break;
-	}
+	QListView *lv = getListViewByTab( tabId );
 
 	lv->reset();
 	lv->setRootIndex( rootIdx );
@@ -588,12 +582,7 @@ void TileEditorMainWindow::onActionAddTile(int tabId)
 
 void TileEditorMainWindow::onActionDeleteTile( int tabId )
 {
-	QListView *lv = NULL;
-	switch( tabId )
-	{
-	case TAB_128: lv = m_ui->listView128; break;
-	case TAB_256: lv = m_ui->listView256; break;
-	}
+	QListView *lv = getListViewByTab( tabId );
 
 	QModelIndex idx = lv->currentIndex();
 	if( !idx.isValid() )
@@ -671,3 +660,21 @@ void TileEditorMainWindow::changeActiveTileSet(const QModelIndex &newIndex, cons
 	//m_ui->listViewTransition->reset();
 	//m_ui->listViewDisplacement->reset();
 }
+
+
+QListView* TileEditorMainWindow::getListViewByTab( int tab ) const
+{
+	QListView *lv = NULL;
+
+	switch( tab )
+	{
+	case TAB_128: lv = m_ui->listView128; break;
+	case TAB_256: lv = m_ui->listView256; break;
+	case TAB_TRANSITION: lv = m_ui->listViewTransition; break;
+	case TAB_DISPLACEMENT: lv = m_ui->listViewDisplacement; break;
+	}
+
+	return lv;
+}
+
+
diff --git a/code/studio/src/plugins/tile_editor/tile_editor_main_window.h b/code/studio/src/plugins/tile_editor/tile_editor_main_window.h
index f839e7c32..2eae0b89f 100644
--- a/code/studio/src/plugins/tile_editor/tile_editor_main_window.h
+++ b/code/studio/src/plugins/tile_editor/tile_editor_main_window.h
@@ -71,6 +71,7 @@ private:
 	void onActionAddTile(int tabId);
 	void onActionDeleteTile(int tabId);
 	TileModel* createTileModel();
+	QListView* getListViewByTab( int tab ) const;
 
 	Ui::TileEditorMainWindow *m_ui;
 	QUndoStack *m_undoStack;