From 1fb3ae9368dcb1f540a20cce5a49089643f591f6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?L=C3=A9o=20Lam?= <leo@leolam.fr>
Date: Sun, 9 May 2021 12:34:36 +0200
Subject: [PATCH] ksys: Add BasicProfiler

---
 data/uking_functions.csv                |  6 +--
 src/KingSystem/System/BasicProfiler.cpp | 55 +++++++++++++++++++++++++
 src/KingSystem/System/BasicProfiler.h   | 22 ++++++++++
 src/KingSystem/System/CMakeLists.txt    |  2 +
 4 files changed, 82 insertions(+), 3 deletions(-)
 create mode 100644 src/KingSystem/System/BasicProfiler.cpp
 create mode 100644 src/KingSystem/System/BasicProfiler.h

diff --git a/data/uking_functions.csv b/data/uking_functions.csv
index 40b5161d..ce745913 100644
--- a/data/uking_functions.csv
+++ b/data/uking_functions.csv
@@ -88720,9 +88720,9 @@
 0x00000071010ba1b0,sub_71010BA1B0,56,_ZN4ksys12CameraEditor18SingletonDisposer_D1Ev
 0x00000071010ba1e8,sub_71010BA1E8,92,_ZN4ksys12CameraEditor18SingletonDisposer_D0Ev
 0x00000071010ba244,CameraEditor::createInstance,120,_ZN4ksys12CameraEditor14createInstanceEPN4sead4HeapE
-0x00000071010ba2bc,Profiler::Push,136,
-0x00000071010ba344,Profiler::Pop,596,
-0x00000071010ba598,sub_71010BA598,176,
+0x00000071010ba2bc,Profiler::Push,136,_ZN4ksys13BasicProfiler4pushEPKc
+0x00000071010ba344,Profiler::Pop,596,_ZN4ksys13BasicProfiler3popEPKc
+0x00000071010ba598,sub_71010BA598,176,_GLOBAL__sub_I_BasicProfiler.cpp?
 0x00000071010ba648,sub_71010BA648,212,
 0x00000071010ba71c,sub_71010BA71C,100,
 0x00000071010ba780,sub_71010BA780,108,
diff --git a/src/KingSystem/System/BasicProfiler.cpp b/src/KingSystem/System/BasicProfiler.cpp
new file mode 100644
index 00000000..1ddd728d
--- /dev/null
+++ b/src/KingSystem/System/BasicProfiler.cpp
@@ -0,0 +1,55 @@
+#include "KingSystem/System/BasicProfiler.h"
+#include <container/seadObjList.h>
+#include <devenv/seadEnvUtil.h>
+#include <prim/seadSafeString.h>
+#include <prim/seadStorageFor.h>
+#include <thread/seadAtomic.h>
+#include <time/seadTickTime.h>
+#include "KingSystem/Utils/Debug.h"
+
+namespace ksys {
+
+struct ProfilerRecord {
+    ProfilerRecord() { enter_time.construct(); }
+
+    sead::StorageFor<sead::TickTime, true> enter_time{};
+    const char* description{};
+};
+
+struct ProfilerInitInfo {
+    ProfilerInitInfo() { time.setNow(); }
+    sead::TickTime time;
+};
+
+ProfilerInitInfo sProfilerInitInfo;
+sead::Atomic<bool> sProfilerUnusedBool = true;
+sead::FixedObjList<ProfilerRecord, 10> sProfilerRecords;
+
+void BasicProfiler::push(const char* description) {
+    auto* record = sProfilerRecords.emplaceBack();
+    record->enter_time->setNow();
+    record->description = description;
+}
+
+void BasicProfiler::pop(const char* description) {
+    if (sProfilerRecords.isEmpty())
+        return;
+
+    auto* record = sProfilerRecords.back();
+    if (!record)
+        return;
+
+    if (sead::SafeString(description) != record->description)
+        return;
+
+    if (sProfilerRecords.size() == 1 && sead::EnvUtil::getRomType() != "Show_2017_1st" &&
+        sead::EnvUtil::getRomType() != "RID_Demo") {
+        util::PrintDebug(sead::FormatFixedSafeString<64>(
+            "%s:%.3f秒", record->description,
+            record->enter_time->diffToNow().toMilliSeconds() / 1000.0f));
+    }
+
+    sProfilerRecords.erase(record);
+}
+
+}  // namespace ksys
diff --git a/src/KingSystem/System/BasicProfiler.h b/src/KingSystem/System/BasicProfiler.h
new file mode 100644
index 00000000..db3c8f68
--- /dev/null
+++ b/src/KingSystem/System/BasicProfiler.h
@@ -0,0 +1,22 @@
+#pragma once
+
+namespace ksys {
+
+class BasicProfiler {
+public:
+    class Scope {
+    public:
+        explicit Scope(const char* description) : mDescription(description) { push(description); }
+        ~Scope() { pop(mDescription); }
+        Scope(const Scope&) = delete;
+        auto operator=(const Scope&) = delete;
+
+    private:
+        const char* mDescription;
+    };
+
+    static void push(const char* description);
+    static void pop(const char* description);
+};
+
+}  // namespace ksys
diff --git a/src/KingSystem/System/CMakeLists.txt b/src/KingSystem/System/CMakeLists.txt
index 1fe4bd60..4d178fd4 100644
--- a/src/KingSystem/System/CMakeLists.txt
+++ b/src/KingSystem/System/CMakeLists.txt
@@ -1,6 +1,8 @@
 target_sources(uking PRIVATE
   Account.cpp
   Account.h
+  BasicProfiler.cpp
+  BasicProfiler.h
   CameraEditor.cpp
   CameraEditor.h
   DebugMessage.h