Formatter for Langulus::Size

Langulus · Jan 22, 2024 · 9264d2e · 9264d2e
1 parent fe761b2
commit 9264d2e
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/source/Logger.hpp b/source/Logger.hpp
@@ -307,4 +307,27 @@ namespace fmt
       }
    };
 
+   ///                                                                        
+   /// Extend FMT to be capable of logging byte sizes                         
+   ///                                                                        
+   template<>
+   struct fmt::formatter<::Langulus::Size> {
+      template<typename ParseContext>
+      constexpr auto parse(ParseContext& ctx) {
+         return ctx.begin();
+      }
+
+      template<typename FormatContext>
+      auto format(const ::Langulus::Size& bs, FormatContext& ctx) {
+         double f;
+         if      (bs < 1'000LL)                 f = static_cast<float>(bs);
+         else if (bs < 1'000'000LL)             f = bs * 1. / 1000LL;
+         else if (bs < 1'000'000'000LL)         f = bs * 1. / 1000'000LL;
+         else if (bs < 1'000'000'000'000LL)     f = bs * 1. / 1000'000'000LL;
+         else if (bs < 1'000'000'000'000'000LL) f = bs * 1. / 1000'000'000'000LL;
+         else f = bs * 1. / 1000'000'000'000'000LL;
+         return format_to(ctx.out(), "{:.2f} {}", f, bs.GetSuffix());
+      }
+   };
+
 } // namespace fmt