[FIX] Refactored the apply/discard buttons out of the widgets

git-svn-id: https://pokegen.svn.sourceforge.net/svnroot/pokegen/trunk@137 6ecfd1a5-f3ed-3746-8530-beee90d26b22

1 files changed, 5 insertions, 2 deletions

diff --git a/pokemodr/PokemodTree.cpp b/pokemodr/PokemodTree.cpp
index ad9c93e2..1812e593 100644
--- a/pokemodr/PokemodTree.cpp
+++ b/pokemodr/PokemodTree.cpp
@@ -21,6 +21,9 @@
 // Model includes
 #include "models/BaseModel.h"
 
+// PokeModr includes
+#include "ObjectUI.h"
+
 // KDE includes
 #include <KUrl>
 
@@ -38,9 +41,9 @@ QString PokemodTree::description(const QModelIndex& index)
     return QString("%1: %2").arg(model()->data(index, BaseModel::TypeRole).toString()).arg(model()->data(index, Qt::DisplayRole).toString());
 }
 
-QWidget* PokemodTree::editorWidget(const QModelIndex& index)
+ObjectUI* PokemodTree::editorWidget(const QModelIndex& index)
 {
-    return model()->data(index, BaseModel::WidgetRole).value<QWidget*>();
+    return static_cast<ObjectUI*>(model()->data(index, BaseModel::WidgetRole).value<QWidget*>());
 }
 
 KMenu* PokemodTree::contextMenu(const QModelIndex& index)