| 1 | // Copyright 2015 Google Inc. All rights reserved. |
| 2 | // |
| 3 | // Licensed under the Apache License, Version 2.0 (the "License"); |
| 4 | // you may not use this file except in compliance with the License. |
| 5 | // You may obtain a copy of the License at |
| 6 | // |
| 7 | // http://www.apache.org/licenses/LICENSE-2.0 |
| 8 | // |
| 9 | // Unless required by applicable law or agreed to in writing, software |
| 10 | // distributed under the License is distributed on an "AS IS" BASIS, |
| 11 | // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 12 | // See the License for the specific language governing permissions and |
| 13 | // limitations under the License. |
| 14 | |
| 15 | #include <algorithm> |
| 16 | #include <cstdint> |
| 17 | #include <cstdio> |
| 18 | #include <cstring> |
| 19 | #include <iostream> |
| 20 | #include <string> |
| 21 | #include <tuple> |
| 22 | #include <vector> |
| 23 | |
| 24 | #include "benchmark/benchmark.h" |
| 25 | #include "check.h" |
| 26 | #include "colorprint.h" |
| 27 | #include "commandlineflags.h" |
| 28 | #include "complexity.h" |
| 29 | #include "counter.h" |
| 30 | #include "internal_macros.h" |
| 31 | #include "string_util.h" |
| 32 | #include "timers.h" |
| 33 | |
| 34 | namespace benchmark { |
| 35 | |
| 36 | BENCHMARK_EXPORT |
| 37 | bool ConsoleReporter::ReportContext(const Context& context) { |
| 38 | name_field_width_ = context.name_field_width; |
| 39 | printed_header_ = false; |
| 40 | prev_counters_.clear(); |
| 41 | |
| 42 | PrintBasicContext(out: &GetErrorStream(), context); |
| 43 | |
| 44 | #ifdef BENCHMARK_OS_WINDOWS |
| 45 | if ((output_options_ & OO_Color)) { |
| 46 | auto stdOutBuf = std::cout.rdbuf(); |
| 47 | auto outStreamBuf = GetOutputStream().rdbuf(); |
| 48 | if (stdOutBuf != outStreamBuf) { |
| 49 | GetErrorStream() |
| 50 | << "Color printing is only supported for stdout on windows." |
| 51 | " Disabling color printing\n" ; |
| 52 | output_options_ = static_cast<OutputOptions>(output_options_ & ~OO_Color); |
| 53 | } |
| 54 | } |
| 55 | #endif |
| 56 | |
| 57 | return true; |
| 58 | } |
| 59 | |
| 60 | BENCHMARK_EXPORT |
| 61 | void ConsoleReporter::(const Run& run) { |
| 62 | std::string str = |
| 63 | FormatString(msg: "%-*s %13s %15s %12s" , static_cast<int>(name_field_width_), |
| 64 | "Benchmark" , "Time" , "CPU" , "Iterations" ); |
| 65 | if (!run.counters.empty()) { |
| 66 | if (output_options_ & OO_Tabular) { |
| 67 | for (auto const& c : run.counters) { |
| 68 | str += FormatString(msg: " %10s" , c.first.c_str()); |
| 69 | } |
| 70 | } else { |
| 71 | str += " UserCounters..." ; |
| 72 | } |
| 73 | } |
| 74 | std::string line = std::string(str.length(), '-'); |
| 75 | GetOutputStream() << line << "\n" << str << "\n" << line << "\n" ; |
| 76 | } |
| 77 | |
| 78 | BENCHMARK_EXPORT |
| 79 | void ConsoleReporter::ReportRuns(const std::vector<Run>& reports) { |
| 80 | for (const auto& run : reports) { |
| 81 | // print the header: |
| 82 | // --- if none was printed yet |
| 83 | bool = !printed_header_; |
| 84 | // --- or if the format is tabular and this run |
| 85 | // has different fields from the prev header |
| 86 | print_header |= (output_options_ & OO_Tabular) && |
| 87 | (!internal::SameNames(l: run.counters, r: prev_counters_)); |
| 88 | if (print_header) { |
| 89 | printed_header_ = true; |
| 90 | prev_counters_ = run.counters; |
| 91 | PrintHeader(run); |
| 92 | } |
| 93 | // As an alternative to printing the headers like this, we could sort |
| 94 | // the benchmarks by header and then print. But this would require |
| 95 | // waiting for the full results before printing, or printing twice. |
| 96 | PrintRunData(report: run); |
| 97 | } |
| 98 | } |
| 99 | |
| 100 | static void IgnoreColorPrint(std::ostream& out, LogColor, const char* fmt, |
| 101 | ...) { |
| 102 | va_list args; |
| 103 | va_start(args, fmt); |
| 104 | out << FormatString(msg: fmt, args); |
| 105 | va_end(args); |
| 106 | } |
| 107 | |
| 108 | static std::string FormatTime(double time) { |
| 109 | // For the time columns of the console printer 13 digits are reserved. One of |
| 110 | // them is a space and max two of them are the time unit (e.g ns). That puts |
| 111 | // us at 10 digits usable for the number. |
| 112 | // Align decimal places... |
| 113 | if (time < 1.0) { |
| 114 | return FormatString(msg: "%10.3f" , time); |
| 115 | } |
| 116 | if (time < 10.0) { |
| 117 | return FormatString(msg: "%10.2f" , time); |
| 118 | } |
| 119 | if (time < 100.0) { |
| 120 | return FormatString(msg: "%10.1f" , time); |
| 121 | } |
| 122 | // Assuming the time is at max 9.9999e+99 and we have 10 digits for the |
| 123 | // number, we get 10-1(.)-1(e)-1(sign)-2(exponent) = 5 digits to print. |
| 124 | if (time > 9999999999 /*max 10 digit number*/) { |
| 125 | return FormatString(msg: "%1.4e" , time); |
| 126 | } |
| 127 | return FormatString(msg: "%10.0f" , time); |
| 128 | } |
| 129 | |
| 130 | BENCHMARK_EXPORT |
| 131 | void ConsoleReporter::PrintRunData(const Run& result) { |
| 132 | typedef void(PrinterFn)(std::ostream&, LogColor, const char*, ...); |
| 133 | auto& Out = GetOutputStream(); |
| 134 | PrinterFn* printer = (output_options_ & OO_Color) |
| 135 | ? static_cast<PrinterFn*>(ColorPrintf) |
| 136 | : IgnoreColorPrint; |
| 137 | auto name_color = |
| 138 | (result.report_big_o || result.report_rms) ? COLOR_BLUE : COLOR_GREEN; |
| 139 | printer(Out, name_color, "%-*s " , name_field_width_, |
| 140 | result.benchmark_name().c_str()); |
| 141 | |
| 142 | if (internal::SkippedWithError == result.skipped) { |
| 143 | printer(Out, COLOR_RED, "ERROR OCCURRED: \'%s\'" , |
| 144 | result.skip_message.c_str()); |
| 145 | printer(Out, COLOR_DEFAULT, "\n" ); |
| 146 | return; |
| 147 | } else if (internal::SkippedWithMessage == result.skipped) { |
| 148 | printer(Out, COLOR_WHITE, "SKIPPED: \'%s\'" , result.skip_message.c_str()); |
| 149 | printer(Out, COLOR_DEFAULT, "\n" ); |
| 150 | return; |
| 151 | } |
| 152 | |
| 153 | const double real_time = result.GetAdjustedRealTime(); |
| 154 | const double cpu_time = result.GetAdjustedCPUTime(); |
| 155 | const std::string real_time_str = FormatTime(time: real_time); |
| 156 | const std::string cpu_time_str = FormatTime(time: cpu_time); |
| 157 | |
| 158 | if (result.report_big_o) { |
| 159 | std::string big_o = GetBigOString(complexity: result.complexity); |
| 160 | printer(Out, COLOR_YELLOW, "%10.2f %-4s %10.2f %-4s " , real_time, |
| 161 | big_o.c_str(), cpu_time, big_o.c_str()); |
| 162 | } else if (result.report_rms) { |
| 163 | printer(Out, COLOR_YELLOW, "%10.0f %-4s %10.0f %-4s " , real_time * 100, "%" , |
| 164 | cpu_time * 100, "%" ); |
| 165 | } else if (result.run_type != Run::RT_Aggregate || |
| 166 | result.aggregate_unit == StatisticUnit::kTime) { |
| 167 | const char* timeLabel = GetTimeUnitString(unit: result.time_unit); |
| 168 | printer(Out, COLOR_YELLOW, "%s %-4s %s %-4s " , real_time_str.c_str(), |
| 169 | timeLabel, cpu_time_str.c_str(), timeLabel); |
| 170 | } else { |
| 171 | assert(result.aggregate_unit == StatisticUnit::kPercentage); |
| 172 | printer(Out, COLOR_YELLOW, "%10.2f %-4s %10.2f %-4s " , |
| 173 | (100. * result.real_accumulated_time), "%" , |
| 174 | (100. * result.cpu_accumulated_time), "%" ); |
| 175 | } |
| 176 | |
| 177 | if (!result.report_big_o && !result.report_rms) { |
| 178 | printer(Out, COLOR_CYAN, "%10lld" , result.iterations); |
| 179 | } |
| 180 | |
| 181 | for (auto& c : result.counters) { |
| 182 | const std::size_t cNameLen = |
| 183 | std::max(a: std::string::size_type(10), b: c.first.length()); |
| 184 | std::string s; |
| 185 | const char* unit = "" ; |
| 186 | if (result.run_type == Run::RT_Aggregate && |
| 187 | result.aggregate_unit == StatisticUnit::kPercentage) { |
| 188 | s = StrFormat(format: "%.2f" , 100. * c.second.value); |
| 189 | unit = "%" ; |
| 190 | } else { |
| 191 | s = HumanReadableNumber(n: c.second.value, one_k: c.second.oneK); |
| 192 | if (c.second.flags & Counter::kIsRate) |
| 193 | unit = (c.second.flags & Counter::kInvert) ? "s" : "/s" ; |
| 194 | } |
| 195 | if (output_options_ & OO_Tabular) { |
| 196 | printer(Out, COLOR_DEFAULT, " %*s%s" , cNameLen - strlen(s: unit), s.c_str(), |
| 197 | unit); |
| 198 | } else { |
| 199 | printer(Out, COLOR_DEFAULT, " %s=%s%s" , c.first.c_str(), s.c_str(), unit); |
| 200 | } |
| 201 | } |
| 202 | |
| 203 | if (!result.report_label.empty()) { |
| 204 | printer(Out, COLOR_DEFAULT, " %s" , result.report_label.c_str()); |
| 205 | } |
| 206 | |
| 207 | printer(Out, COLOR_DEFAULT, "\n" ); |
| 208 | } |
| 209 | |
| 210 | } // end namespace benchmark |
| 211 | |