Add the ATB arena to the plugin

1 files changed, 14 insertions, 5 deletions

diff --git a/sigencore/plugins/arenas/SigenArenas.cpp b/sigencore/plugins/arenas/SigenArenas.cpp
index d7fe9faa..42c581ab 100644
--- a/sigencore/plugins/arenas/SigenArenas.cpp
+++ b/sigencore/plugins/arenas/SigenArenas.cpp
@@ -20,6 +20,7 @@
 
 // Arena includes
 #include "standard/TurnArena.h"
+#include "atb/ATBArena.h"
 
 // Sigencore includes
 #include <sigencore/Arena.h>
@@ -55,14 +56,15 @@ QString SigenArenas::pluginDescription() const
 
 QStringList SigenArenas::arenas() const
 {
-    // TODO
-    return QStringList() << TurnArena::name();
+    return QStringList() << TurnArena::name() << ATBArena::name();
 }
 
 QString SigenArenas::description(const QString& name) const
 {
     if (name == TurnArena::name())
         return TurnArena::description();
+    if (name == ATBArena::name())
+        return ATBArena::description();
     return "(Unknown arena)";
 }
 
@@ -70,7 +72,8 @@ QIcon SigenArenas::icon(const QString& name)
 {
     if (name == TurnArena::name())
         return TurnArena::icon();
-    // TODO
+    if (name == ATBArena::name())
+        return ATBArena::icon();
     return QIcon();
 }
 
@@ -78,7 +81,8 @@ Arena* SigenArenas::createArena(const QString& name, GameWrapper* game, Config*
 {
     if (name == TurnArena::name())
         return new TurnArena(game, parent);
-    // TODO
+    if (name == ATBArena::name())
+        return new ATBArena(game, parent);
     return NULL;
 }
 
@@ -90,5 +94,10 @@ void SigenArenas::cleanupArena(Arena* arena)
         delete turnArena;
         return;
     }
-    // TODO
+    ATBArena* atbArena = qobject_cast<ATBArena*>(arena);
+    if (atbArena)
+    {
+        delete atbArena;
+        return;
+    }
 }