1 files changed, 38 insertions, 0 deletions
diff --git a/pokescripting/PokemodWrapper.cpp b/pokescripting/PokemodWrapper.cpp
index 65befc51..149cd283 100644
--- a/pokescripting/PokemodWrapper.cpp
+++ b/pokescripting/PokemodWrapper.cpp
@@ -171,6 +171,44 @@ Pokescripting::WeatherWrapper* Pokescripting::PokemodWrapper::weather(const int
     return WeatherWrapper::create(m_pokemod->weatherById(id), this);
 }
 
+Pokemod::Stat Pokescripting::PokemodWrapper::stat(const QString& name) const
+{
+    if (name == "HP")
+        return Pokemod::ST_HP;
+    else if (name == "Attack")
+        return Pokemod::ST_Attack;
+    else if (name == "Defense")
+        return Pokemod::ST_Defense;
+    else if (name == "Speed")
+        return Pokemod::ST_Speed;
+    else if (name == "Special")
+        return Pokemod::ST_Special;
+    else if (name == "Special Attack")
+        return Pokemod::ST_SpecialAttack;
+    else if (name == "Special Defense")
+        return Pokemod::ST_SpecialAttack;
+    else if (name == "Accuracy")
+        return Pokemod::ST_Accuracy;
+    else if (name == "Evasion")
+        return Pokemod::ST_Evasion;
+    return Pokemod::ST_Invalid;
+}
+
+Pokemod::Direction Pokescripting::PokemodWrapper::direction(const QString& name) const
+{
+    if (name == "Up")
+        return Pokemod::D_Up;
+    else if (name == "Down")
+        return Pokemod::D_Down;
+    else if (name == "Left")
+        return Pokemod::D_Left;
+    else if (name == "Right")
+        return Pokemod::D_Right;
+    else if (name == "None")
+        return Pokemod::D_None;
+    return Pokemod::D_Invalid;
+}
+
 QString Pokescripting::PokemodWrapper::title() const
 {
     return m_pokemod->title();