From 093b5d82fa51c2910b61c4b0b05288a7d3aa6897 Mon Sep 17 00:00:00 2001
From: Rochet2 <tqbattlenet@gmail.com>
Date: Wed, 26 Jul 2017 23:17:49 +0300
Subject: [PATCH] Implement get class mask and get race mask

---
 LuaFunctions.cpp |  2 ++
 UnitMethods.h    | 22 ++++++++++++++++++++++
 2 files changed, 24 insertions(+)

diff --git a/LuaFunctions.cpp b/LuaFunctions.cpp
index 051a5e0..c050eb4 100644
--- a/LuaFunctions.cpp
+++ b/LuaFunctions.cpp
@@ -251,6 +251,8 @@ ElunaRegister<Unit> UnitMethods[] =
     { "GetGender", &LuaUnit::GetGender },
     { "GetRace", &LuaUnit::GetRace },
     { "GetClass", &LuaUnit::GetClass },
+    { "GetRaceMask", &LuaUnit::GetRaceMask },
+    { "GetClassMask", &LuaUnit::GetClassMask },
     { "GetRaceAsString", &LuaUnit::GetRaceAsString },
     { "GetClassAsString", &LuaUnit::GetClassAsString },
     { "GetAura", &LuaUnit::GetAura },
diff --git a/UnitMethods.h b/UnitMethods.h
index f4d8d57..fdcbbe5 100644
--- a/UnitMethods.h
+++ b/UnitMethods.h
@@ -1058,6 +1058,28 @@ namespace LuaUnit
         return 1;
     }
 
+    /**
+    * Returns the race mask
+    *
+    * @return uint32 racemask
+    */
+    int GetRaceMask(lua_State* L, Unit* unit)
+    {
+        Eluna::Push(L, unit->getRaceMask());
+        return 1;
+    }
+
+    /**
+    * Returns the class mask
+    *
+    * @return uint32 classmask
+    */
+    int GetClassMask(lua_State* L, Unit* unit)
+    {
+        Eluna::Push(L, unit->getClassMask());
+        return 1;
+    }
+
     /**
      * Returns the [Unit]'s creature type ID like wolf or humanoid.
      *