merian: utils: Add formatter for stopwatch

LDAP · Nov 4, 2024 · 2644dc8 · 2644dc8
1 parent 2ff5b8f
commit 2644dc8
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 5 deletions.
diff --git a/include/merian/utils/stopwatch.hpp b/include/merian/utils/stopwatch.hpp
@@ -17,8 +17,12 @@ class Stopwatch {
     double seconds() const;
     std::chrono::nanoseconds duration() const;
 
+    friend std::ostream& operator<<(std::ostream& stream, const Stopwatch& sw);
+
   private:
     chrono_clock::time_point start;
 };
 
+auto format_as(const Stopwatch& sw);
+
 } // namespace merian
diff --git a/include/merian/utils/string.hpp b/include/merian/utils/string.hpp
@@ -9,15 +9,26 @@
 
 namespace merian {
 
-[[nodiscard]] inline std::string format_size(const uint64_t size) {
-    std::vector<const char*> units = {"B", "KB", "MB", "GB", "TB"};
+[[nodiscard]] inline std::string format_size(const uint64_t size_bytes) {
+    const std::vector<const char*> units = {"B", "KB", "MB", "GB", "TB"};
     std::size_t unit_index;
-    if (size == 0) {
+    if (size_bytes == 0) {
         unit_index = 0;
     } else {
-        unit_index = std::min((std::size_t)std::log2(size) / 10, units.size() - 1);
+        unit_index = std::min((std::size_t)std::log2(size_bytes) / 10, units.size() - 1);
     }
-    return fmt::format("{} {}", size / std::pow(1024, unit_index), units[unit_index]);
+    return fmt::format("{} {}", (double)size_bytes / std::pow(1024, unit_index), units[unit_index]);
+}
+
+[[nodiscard]] inline std::string format_duration(const uint64_t duration_ns) {
+    const std::vector<const char*> units = {"ns", "μs", "ms", "s"};
+    std::size_t unit_index;
+    if (duration_ns == 0) {
+        unit_index = 0;
+    } else {
+        unit_index = std::min((std::size_t)(std::log10(duration_ns) / 3), units.size() - 1);
+    }
+    return fmt::format("{} {}", (double)duration_ns / std::pow(1000, unit_index), units[unit_index]);
 }
 
 [[nodiscard]] inline bool ends_with(const std::string& value, const std::string& suffix) {

diff --git a/src/merian/utils/stopwatch.cpp b/src/merian/utils/stopwatch.cpp
@@ -1,5 +1,6 @@
 #include "merian/utils/stopwatch.hpp"
 #include "merian/utils/chrono.hpp"
+#include "merian/utils/string.hpp"
 #include <atomic>
 
 namespace merian {
@@ -37,4 +38,12 @@ std::chrono::nanoseconds Stopwatch::duration() const {
     return end - start;
 }
 
+std::ostream& operator<<(std::ostream& stream, const Stopwatch& sw) {
+    return stream << format_duration(sw.nanos());
+}
+
+auto format_as(const Stopwatch& sw) {
+    return format_duration(sw.nanos());
+}
+
 } // namespace merian