Home | History | Annotate | Download | only in Support 
      1 //===- Support/Chrono.cpp - Utilities for Timing Manipulation ---*- C++ -*-===//
      2 //
      3 //                     The LLVM Compiler Infrastructure
      4 //
      5 // This file is distributed under the University of Illinois Open Source
      6 // License. See LICENSE.TXT for details.
      7 //
      8 //===----------------------------------------------------------------------===//
      9 
     10 #include "llvm/Support/Chrono.h"
     11 #include "llvm/Config/llvm-config.h"
     12 #include "llvm/Support/Format.h"
     13 #include "llvm/Support/raw_ostream.h"
     14 
     15 namespace llvm {
     16 
     17 using namespace sys;
     18 
     19 const char llvm::detail::unit<std::ratio<3600>>::value[] = "h";
     20 const char llvm::detail::unit<std::ratio<60>>::value[] = "m";
     21 const char llvm::detail::unit<std::ratio<1>>::value[] = "s";
     22 const char llvm::detail::unit<std::milli>::value[] = "ms";
     23 const char llvm::detail::unit<std::micro>::value[] = "us";
     24 const char llvm::detail::unit<std::nano>::value[] = "ns";
     25 
     26 static inline struct tm getStructTM(TimePoint<> TP) {
     27   struct tm Storage;
     28   std::time_t OurTime = toTimeT(TP);
     29 
     30 #if defined(LLVM_ON_UNIX)
     31   struct tm *LT = ::localtime_r(&OurTime, &Storage);
     32   assert(LT);
     33   (void)LT;
     34 #endif
     35 #if defined(_WIN32)
     36   int Error = ::localtime_s(&Storage, &OurTime);
     37   assert(!Error);
     38   (void)Error;
     39 #endif
     40 
     41   return Storage;
     42 }
     43 
     44 raw_ostream &operator<<(raw_ostream &OS, TimePoint<> TP) {
     45   struct tm LT = getStructTM(TP);
     46   char Buffer[sizeof("YYYY-MM-DD HH:MM:SS")];
     47   strftime(Buffer, sizeof(Buffer), "%Y-%m-%d %H:%M:%S", &LT);
     48   return OS << Buffer << '.'
     49             << format("%.9lu",
     50                       long((TP.time_since_epoch() % std::chrono::seconds(1))
     51                                .count()));
     52 }
     53 
     54 void format_provider<TimePoint<>>::format(const TimePoint<> &T, raw_ostream &OS,
     55                                           StringRef Style) {
     56   using namespace std::chrono;
     57   TimePoint<seconds> Truncated = time_point_cast<seconds>(T);
     58   auto Fractional = T - Truncated;
     59   struct tm LT = getStructTM(Truncated);
     60   // Handle extensions first. strftime mangles unknown %x on some platforms.
     61   if (Style.empty()) Style = "%Y-%m-%d %H:%M:%S.%N";
     62   std::string Format;
     63   raw_string_ostream FStream(Format);
     64   for (unsigned I = 0; I < Style.size(); ++I) {
     65     if (Style[I] == '%' && Style.size() > I + 1) switch (Style[I + 1]) {
     66         case 'L':  // Milliseconds, from Ruby.
     67           FStream << llvm::format(
     68               "%.3lu", (long)duration_cast<milliseconds>(Fractional).count());
     69           ++I;
     70           continue;
     71         case 'f':  // Microseconds, from Python.
     72           FStream << llvm::format(
     73               "%.6lu", (long)duration_cast<microseconds>(Fractional).count());
     74           ++I;
     75           continue;
     76         case 'N':  // Nanoseconds, from date(1).
     77           FStream << llvm::format(
     78               "%.6lu", (long)duration_cast<nanoseconds>(Fractional).count());
     79           ++I;
     80           continue;
     81         case '%':  // Consume %%, so %%f parses as (%%)f not %(%f)
     82           FStream << "%%";
     83           ++I;
     84           continue;
     85       }
     86     FStream << Style[I];
     87   }
     88   FStream.flush();
     89   char Buffer[256];  // Should be enough for anywhen.
     90   size_t Len = strftime(Buffer, sizeof(Buffer), Format.c_str(), &LT);
     91   OS << (Len ? Buffer : "BAD-DATE-FORMAT");
     92 }
     93 
     94 } // namespace llvm
     95