Improve layout system.

1 files changed, 44 insertions, 0 deletions

diff --git a/src/cru_debug.h b/src/cru_debug.h
new file mode 100644
index 00000000..a9800774
--- /dev/null
+++ b/src/cru_debug.h
@@ -0,0 +1,44 @@
+#pragma once
+
+#include <functional>
+
+#include "base.h"
+#include "format.h"
+
+namespace cru::debug
+{
+    void DebugMessage(const StringView& message);
+
+#ifdef CRU_DEBUG
+    inline void DebugTime(const std::function<void()>& action, const StringView& hint_message)
+    {
+        const auto before = std::chrono::steady_clock::now();
+        action();
+        const auto after = std::chrono::steady_clock::now();
+        const auto duration = std::chrono::duration_cast<std::chrono::milliseconds>(after - before);
+        DebugMessage(Format(L"{}: {}ms.\n", hint_message, duration.count()));
+    }
+
+    template<typename TReturn>
+    TReturn DebugTime(const std::function<TReturn()>& action, const StringView& hint_message)
+    {
+        const auto before = std::chrono::steady_clock::now();
+        auto&& result = action();
+        const auto after = std::chrono::steady_clock::now();
+        const auto duration = std::chrono::duration_cast<std::chrono::milliseconds>(after - before);
+        DebugMessage(Format(L"{}: {}ms.\n", hint_message, duration.count()));
+        return std::move(result);
+    }
+#else
+    inline void DebugTime(Function<void()>&& action, const StringView& hint_message)
+    {
+        action();
+    }
+
+    template<typename TReturn>
+    TReturn DebugTime(Function<TReturn()>&& action, const StringView& hint_message)
+    {
+        return action();
+    }
+#endif
+}