From d21f9d6c4b53a6bfe6da3b1cce19c78a919d841d Mon Sep 17 00:00:00 2001
From: Kah Wei <msntkw@msn.com>
Date: Wed, 1 Mar 2023 22:10:09 +0800
Subject: [PATCH] Added functions for adding transitions and parameters to an
 AnimationController

---
 .../src/Animation/SHAnimationController.cpp   | 50 +++++++++++++++++++
 .../src/Animation/SHAnimationController.h     | 16 ++++++
 2 files changed, 66 insertions(+)

diff --git a/SHADE_Engine/src/Animation/SHAnimationController.cpp b/SHADE_Engine/src/Animation/SHAnimationController.cpp
index b3105e64..3a1f581e 100644
--- a/SHADE_Engine/src/Animation/SHAnimationController.cpp
+++ b/SHADE_Engine/src/Animation/SHAnimationController.cpp
@@ -154,6 +154,56 @@ namespace SHADE
     node.Free();
   }
   
+  void SHAnimationController::AddTransition(Handle<Node> source, const Transition& transition)
+  {
+    if (!source)
+    {
+      SHLOG_ERROR("[SHAnimationController] Attempted to add transition from an invalid node.");
+      return;
+    }
+
+    if (!transition.Target)
+    {
+      SHLOG_ERROR("[SHAnimationController] Attempted to add transition to an invalid node.");
+      return;
+    }
+
+    if (transition.Condition != Transition::ConditionType::None && !parameters.contains(transition.ParamName))
+    {
+      SHLOG_ERROR("[SHAnimationController] Attempted to add a conditional transition for an invalid parameter.");
+      return;
+    }
+
+    source->Transitions.emplace_back(transition);
+  }
+  void SHAnimationController::AddParameter(const std::string& name, AnimParam::Type type)
+  {
+    if (name.empty())
+    {
+      SHLOG_ERROR("[SHAnimationController] Attempted to add a parameter with no name.");
+      return;
+    }
+
+    if (parameters.contains(name))
+    {
+      SHLOG_ERROR("[SHAnimationController] Attempted to add a parameter with the same name.");
+      return;
+    }
+
+    // Insert
+    parameters.emplace(name, type);
+  }
+  void SHAnimationController::RemoveParameter(const std::string& name)
+  {
+    if (!parameters.contains(name))
+    {
+      SHLOG_ERROR("[SHAnimationController] Attempted to reemove a parameter that does not exist.");
+      return;
+    }
+
+    parameters.erase(name);
+  }
+
   /*-----------------------------------------------------------------------------------*/
   /* Helper Functions                                                                  */
   /*-----------------------------------------------------------------------------------*/
diff --git a/SHADE_Engine/src/Animation/SHAnimationController.h b/SHADE_Engine/src/Animation/SHAnimationController.h
index 2aa46f3c..90162422 100644
--- a/SHADE_Engine/src/Animation/SHAnimationController.h
+++ b/SHADE_Engine/src/Animation/SHAnimationController.h
@@ -167,8 +167,24 @@ namespace SHADE
     /// </summary>
     /// <param name="node">Node to destroy.</param>
     void DestroyNode(Handle<Node> node);
+    /// <summary>
+    /// Links two nodes together with a Transition. This performs some additional 
+    /// checking to ensure parameters are valid.
+    /// </summary>
+    /// <param name="source">Source node to transition from.</param>
+    /// <param name="transition">Describes the transition to add.</param>
     void AddTransition(Handle<Node> source, const Transition& transition);
+    /// <summary>
+    /// Registers a parameter to the animation controller.
+    /// </summary>
+    /// <param name="name">Name of the parameter.</param>
+    /// <param name="type">Type of the parameter.</param>
     void AddParameter(const std::string& name, AnimParam::Type type);
+    /// <summary>
+    /// Removes a parameter from the animation controller.
+    /// </summary>
+    /// <param name="name">Name of the parameter.</param>
+    void RemoveParameter(const std::string& name);
     
     /*---------------------------------------------------------------------------------*/
     /* Getters                                                                         */