From 12f45fc073fcecee42f794f65be99b6e50331605 Mon Sep 17 00:00:00 2001
From: korenkonder <admin@mysterious-home.net>
Date: Tue, 5 Mar 2024 13:22:57 +0300
Subject: [PATCH] `aet_manager`: added `get_set`

---
 src/CRE/auth_2d.cpp | 8 +++++++-
 src/CRE/auth_2d.hpp | 2 ++
 src/CRE/sprite.hpp  | 1 +
 3 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/src/CRE/auth_2d.cpp b/src/CRE/auth_2d.cpp
index fc6b885c..43b36cf7 100644
--- a/src/CRE/auth_2d.cpp
+++ b/src/CRE/auth_2d.cpp
@@ -6,7 +6,6 @@
 #include <map>
 #include <vector>
 #include "auth_2d.hpp"
-#include "../KKdLib/aet.hpp"
 #include "../KKdLib/str_utils.hpp"
 #include "data.hpp"
 #include "file_handler.hpp"
@@ -393,6 +392,13 @@ uint32_t aet_manager_get_scenes_count(uint32_t set_id, const aet_database* aet_d
     return aet_manager->GetSetScenesCount(aet_db->get_aet_set_by_id(set_id)->index);
 }
 
+::aet_set* aet_manager_get_set(uint32_t set_id, const aet_database* aet_db) {
+    AetSet* set = aet_manager->GetSet(aet_db->get_aet_set_by_id(set_id)->index);
+    if (set)
+        return set->aet_set;
+    return 0;
+}
+
 bool aet_manager_get_set_ready(uint32_t set_id, const aet_database* aet_db) {
     return aet_manager->GetSetReady(aet_db->get_aet_set_by_id(set_id)->index);
 };
diff --git a/src/CRE/auth_2d.hpp b/src/CRE/auth_2d.hpp
index f747a53f..65cf243c 100644
--- a/src/CRE/auth_2d.hpp
+++ b/src/CRE/auth_2d.hpp
@@ -7,6 +7,7 @@
 
 #include "../KKdLib/default.hpp"
 #include "../KKdLib/database/aet.hpp"
+#include "../KKdLib/aet.hpp"
 #include "../KKdLib/vec.hpp"
 #include "color.hpp"
 #include "frame_rate_control.hpp"
@@ -127,6 +128,7 @@ extern float_t aet_manager_get_scene_layer_start_time(uint32_t aet_id,
 extern const char* aet_manager_get_scene_name(uint32_t aet_id, const aet_database* aet_db);
 extern float_t aet_manager_get_scene_start_time(uint32_t aet_id, const aet_database* aet_db);
 extern uint32_t aet_manager_get_scenes_count(uint32_t set_id, const aet_database* aet_db);
+extern ::aet_set* aet_manager_get_set(uint32_t set_id, const aet_database* aet_db);
 extern bool aet_manager_get_set_ready(uint32_t set_id, const aet_database* aet_db);
 extern void aet_manager_init_aet_layout(AetComp* composition,
     AetArgs args, const aet_database* aet_db);
diff --git a/src/CRE/sprite.hpp b/src/CRE/sprite.hpp
index 5a3ae1fc..06122922 100644
--- a/src/CRE/sprite.hpp
+++ b/src/CRE/sprite.hpp
@@ -11,6 +11,7 @@
 #include "../KKdLib/database/sprite.hpp"
 #include "../KKdLib/mat.hpp"
 #include "../KKdLib/rectangle.hpp"
+#include "../KKdLib/spr.hpp"
 #include "../KKdLib/vec.hpp"
 #include "color.hpp"
 #include "render_context.hpp"