1 files changed, 27 insertions, 20 deletions
diff --git a/sigmodr/tree/TreeWidget.cpp b/sigmodr/tree/TreeWidget.cpp
index 85d2973c..dcad6c34 100644
--- a/sigmodr/tree/TreeWidget.cpp
+++ b/sigmodr/tree/TreeWidget.cpp
@@ -77,26 +77,33 @@ const Game* TreeWidget::currentGame() const
 //     model()->setData(index, data.toString(), BaseModel::XmlRole);
 // }
 // 
-// QList<const Game*> TreeWidget::openedGames() const
-// {
-//     return m_games.keys();
-// }
-// 
-// void TreeWidget::addGame(Game* game, const KUrl& url)
-// {
-//     qobject_cast<TreeWidgetModel*>(model())->addGame(game);
-//     m_games[game] = UrlDirty(url, false);
-// }
-// 
-// void TreeWidget::deleteGame(const Game* game)
-// {
-//     if (m_games.contains(game))
-//     {
-//         qobject_cast<TreeWidgetModel*>(model())->deleteGame(game);
-//         m_games.remove(game);
-//     }
-// }
-// 
+QList<const Game*> TreeWidget::openedGames() const
+{
+    return m_games.keys();
+}
+
+void TreeWidget::addGame(Game* game, const KUrl& url)
+{
+    qobject_cast<TreeModel*>(model())->addGame(game);
+    m_games[game] = UrlDirty(url, false);
+}
+
+void TreeWidget::deleteGame(const Game* game)
+{
+    if (m_games.contains(game))
+    {
+        qobject_cast<TreeModel*>(model())->deleteGame(game);
+        m_games.remove(game);
+    }
+}
+
+void TreeWidget::deleteAllGames()
+{
+    const QList<const Game*> games = m_games.keys();
+    foreach (const Game* game, games)
+        deleteGame(game);
+}
+
 bool TreeWidget::isOpen(const KUrl& url) const
 {
     if (url.isEmpty())