From a88e4a89e75a2655e4fcda8ee27f0c929f06e62f Mon Sep 17 00:00:00 2001
From: Deukhoofd <deukhoofd@gmail.com>
Date: Mon, 13 Jan 2020 19:55:18 +0100
Subject: [PATCH] Register GrowthRate Library in AngelScript.

---
 .../TypeRegistry/Library/RegisterGrowthRateTypes.cpp | 12 ++++++++++++
 .../TypeRegistry/Library/RegisterGrowthRateTypes.hpp |  1 +
 2 files changed, 13 insertions(+)

diff --git a/src/AngelScript/TypeRegistry/Library/RegisterGrowthRateTypes.cpp b/src/AngelScript/TypeRegistry/Library/RegisterGrowthRateTypes.cpp
index 865e3e6..5ba4c44 100644
--- a/src/AngelScript/TypeRegistry/Library/RegisterGrowthRateTypes.cpp
+++ b/src/AngelScript/TypeRegistry/Library/RegisterGrowthRateTypes.cpp
@@ -1,9 +1,11 @@
 #include "RegisterGrowthRateTypes.hpp"
 #include <Library/GrowthRates/GrowthRate.hpp>
+#include <Library/GrowthRates/GrowthRateLibrary.hpp>
 #include <cassert>
 
 void RegisterGrowthRateTypes::Register(asIScriptEngine* engine) {
     RegisterGrowthRateType(engine);
+    RegisterGrowthRateLibrary(engine);
 }
 void RegisterGrowthRateTypes::RegisterGrowthRateType(asIScriptEngine* engine) {
     [[maybe_unused]] int r = engine->RegisterObjectType("GrowthRate", 0, asOBJ_REF | asOBJ_NOCOUNT);
@@ -15,3 +17,13 @@ void RegisterGrowthRateTypes::RegisterGrowthRateType(asIScriptEngine* engine) {
                                      asMETHOD(CreatureLib::Library::GrowthRate, CalculateExperience), asCALL_THISCALL);
     assert(r >= 0);
 }
+void RegisterGrowthRateTypes::RegisterGrowthRateLibrary(asIScriptEngine* engine) {
+    [[maybe_unused]] int r = engine->RegisterObjectType("GrowthRateLibrary", 0, asOBJ_REF | asOBJ_NOCOUNT);
+    assert(r >= 0);
+    r = engine->RegisterObjectMethod("GrowthRateLibrary", "uint8 CalculateLevel(const string &in growthRate, uint experience) const",
+                                     asMETHOD(CreatureLib::Library::GrowthRateLibrary, CalculateLevel), asCALL_THISCALL);
+    assert(r >= 0);
+    r = engine->RegisterObjectMethod("GrowthRateLibrary", "uint CalculateExperience(const string &in growthRate, uint8 experience) const",
+                                     asMETHOD(CreatureLib::Library::GrowthRateLibrary, CalculateExperience), asCALL_THISCALL);
+    assert(r >= 0);
+}
diff --git a/src/AngelScript/TypeRegistry/Library/RegisterGrowthRateTypes.hpp b/src/AngelScript/TypeRegistry/Library/RegisterGrowthRateTypes.hpp
index 7282701..50509d0 100644
--- a/src/AngelScript/TypeRegistry/Library/RegisterGrowthRateTypes.hpp
+++ b/src/AngelScript/TypeRegistry/Library/RegisterGrowthRateTypes.hpp
@@ -4,6 +4,7 @@
 
 class RegisterGrowthRateTypes {
     static void RegisterGrowthRateType(asIScriptEngine* engine);
+    static void RegisterGrowthRateLibrary(asIScriptEngine* engine);
 public:
     static void Register(asIScriptEngine* engine);
 };