| Index: third_party/google_benchmark/src/commandlineflags.cc
|
| diff --git a/third_party/google_benchmark/src/commandlineflags.cc b/third_party/google_benchmark/src/commandlineflags.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..72534e022a8a7411b2250f938934c6c2080cdcd8
|
| --- /dev/null
|
| +++ b/third_party/google_benchmark/src/commandlineflags.cc
|
| @@ -0,0 +1,218 @@
|
| +// Copyright 2015 Google Inc. All rights reserved.
|
| +//
|
| +// Licensed under the Apache License, Version 2.0 (the "License");
|
| +// you may not use this file except in compliance with the License.
|
| +// You may obtain a copy of the License at
|
| +//
|
| +// http://www.apache.org/licenses/LICENSE-2.0
|
| +//
|
| +// Unless required by applicable law or agreed to in writing, software
|
| +// distributed under the License is distributed on an "AS IS" BASIS,
|
| +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
| +// See the License for the specific language governing permissions and
|
| +// limitations under the License.
|
| +
|
| +#include "commandlineflags.h"
|
| +
|
| +#include <cctype>
|
| +#include <cstdlib>
|
| +#include <cstring>
|
| +#include <iostream>
|
| +#include <limits>
|
| +
|
| +namespace benchmark {
|
| +// Parses 'str' for a 32-bit signed integer. If successful, writes
|
| +// the result to *value and returns true; otherwise leaves *value
|
| +// unchanged and returns false.
|
| +bool ParseInt32(const std::string& src_text, const char* str, int32_t* value) {
|
| + // Parses the environment variable as a decimal integer.
|
| + char* end = nullptr;
|
| + const long long_value = strtol(str, &end, 10); // NOLINT
|
| +
|
| + // Has strtol() consumed all characters in the string?
|
| + if (*end != '\0') {
|
| + // No - an invalid character was encountered.
|
| + std::cerr << src_text << " is expected to be a 32-bit integer, "
|
| + << "but actually has value \"" << str << "\".\n";
|
| + return false;
|
| + }
|
| +
|
| + // Is the parsed value in the range of an Int32?
|
| + const int32_t result = static_cast<int32_t>(long_value);
|
| + if (long_value == std::numeric_limits<long>::max() ||
|
| + long_value == std::numeric_limits<long>::min() ||
|
| + // The parsed value overflows as a long. (strtol() returns
|
| + // LONG_MAX or LONG_MIN when the input overflows.)
|
| + result != long_value
|
| + // The parsed value overflows as an Int32.
|
| + ) {
|
| + std::cerr << src_text << " is expected to be a 32-bit integer, "
|
| + << "but actually has value \"" << str << "\", "
|
| + << "which overflows.\n";
|
| + return false;
|
| + }
|
| +
|
| + *value = result;
|
| + return true;
|
| +}
|
| +
|
| +// Parses 'str' for a double. If successful, writes the result to *value and
|
| +// returns true; otherwise leaves *value unchanged and returns false.
|
| +bool ParseDouble(const std::string& src_text, const char* str, double* value) {
|
| + // Parses the environment variable as a decimal integer.
|
| + char* end = nullptr;
|
| + const double double_value = strtod(str, &end); // NOLINT
|
| +
|
| + // Has strtol() consumed all characters in the string?
|
| + if (*end != '\0') {
|
| + // No - an invalid character was encountered.
|
| + std::cerr << src_text << " is expected to be a double, "
|
| + << "but actually has value \"" << str << "\".\n";
|
| + return false;
|
| + }
|
| +
|
| + *value = double_value;
|
| + return true;
|
| +}
|
| +
|
| +// Returns the name of the environment variable corresponding to the
|
| +// given flag. For example, FlagToEnvVar("foo") will return
|
| +// "BENCHMARK_FOO" in the open-source version.
|
| +static std::string FlagToEnvVar(const char* flag) {
|
| + const std::string flag_str(flag);
|
| +
|
| + std::string env_var;
|
| + for (size_t i = 0; i != flag_str.length(); ++i)
|
| + env_var += static_cast<char>(::toupper(flag_str.c_str()[i]));
|
| +
|
| + return "BENCHMARK_" + env_var;
|
| +}
|
| +
|
| +// Reads and returns the Boolean environment variable corresponding to
|
| +// the given flag; if it's not set, returns default_value.
|
| +//
|
| +// The value is considered true iff it's not "0".
|
| +bool BoolFromEnv(const char* flag, bool default_value) {
|
| + const std::string env_var = FlagToEnvVar(flag);
|
| + const char* const string_value = getenv(env_var.c_str());
|
| + return string_value == nullptr ? default_value
|
| + : strcmp(string_value, "0") != 0;
|
| +}
|
| +
|
| +// Reads and returns a 32-bit integer stored in the environment
|
| +// variable corresponding to the given flag; if it isn't set or
|
| +// doesn't represent a valid 32-bit integer, returns default_value.
|
| +int32_t Int32FromEnv(const char* flag, int32_t default_value) {
|
| + const std::string env_var = FlagToEnvVar(flag);
|
| + const char* const string_value = getenv(env_var.c_str());
|
| + if (string_value == nullptr) {
|
| + // The environment variable is not set.
|
| + return default_value;
|
| + }
|
| +
|
| + int32_t result = default_value;
|
| + if (!ParseInt32(std::string("Environment variable ") + env_var, string_value,
|
| + &result)) {
|
| + std::cout << "The default value " << default_value << " is used.\n";
|
| + return default_value;
|
| + }
|
| +
|
| + return result;
|
| +}
|
| +
|
| +// Reads and returns the string environment variable corresponding to
|
| +// the given flag; if it's not set, returns default_value.
|
| +const char* StringFromEnv(const char* flag, const char* default_value) {
|
| + const std::string env_var = FlagToEnvVar(flag);
|
| + const char* const value = getenv(env_var.c_str());
|
| + return value == nullptr ? default_value : value;
|
| +}
|
| +
|
| +// Parses a string as a command line flag. The string should have
|
| +// the format "--flag=value". When def_optional is true, the "=value"
|
| +// part can be omitted.
|
| +//
|
| +// Returns the value of the flag, or nullptr if the parsing failed.
|
| +const char* ParseFlagValue(const char* str, const char* flag,
|
| + bool def_optional) {
|
| + // str and flag must not be nullptr.
|
| + if (str == nullptr || flag == nullptr) return nullptr;
|
| +
|
| + // The flag must start with "--".
|
| + const std::string flag_str = std::string("--") + std::string(flag);
|
| + const size_t flag_len = flag_str.length();
|
| + if (strncmp(str, flag_str.c_str(), flag_len) != 0) return nullptr;
|
| +
|
| + // Skips the flag name.
|
| + const char* flag_end = str + flag_len;
|
| +
|
| + // When def_optional is true, it's OK to not have a "=value" part.
|
| + if (def_optional && (flag_end[0] == '\0')) return flag_end;
|
| +
|
| + // If def_optional is true and there are more characters after the
|
| + // flag name, or if def_optional is false, there must be a '=' after
|
| + // the flag name.
|
| + if (flag_end[0] != '=') return nullptr;
|
| +
|
| + // Returns the string after "=".
|
| + return flag_end + 1;
|
| +}
|
| +
|
| +bool ParseBoolFlag(const char* str, const char* flag, bool* value) {
|
| + // Gets the value of the flag as a string.
|
| + const char* const value_str = ParseFlagValue(str, flag, true);
|
| +
|
| + // Aborts if the parsing failed.
|
| + if (value_str == nullptr) return false;
|
| +
|
| + // Converts the string value to a bool.
|
| + *value = IsTruthyFlagValue(value_str);
|
| + return true;
|
| +}
|
| +
|
| +bool ParseInt32Flag(const char* str, const char* flag, int32_t* value) {
|
| + // Gets the value of the flag as a string.
|
| + const char* const value_str = ParseFlagValue(str, flag, false);
|
| +
|
| + // Aborts if the parsing failed.
|
| + if (value_str == nullptr) return false;
|
| +
|
| + // Sets *value to the value of the flag.
|
| + return ParseInt32(std::string("The value of flag --") + flag, value_str,
|
| + value);
|
| +}
|
| +
|
| +bool ParseDoubleFlag(const char* str, const char* flag, double* value) {
|
| + // Gets the value of the flag as a string.
|
| + const char* const value_str = ParseFlagValue(str, flag, false);
|
| +
|
| + // Aborts if the parsing failed.
|
| + if (value_str == nullptr) return false;
|
| +
|
| + // Sets *value to the value of the flag.
|
| + return ParseDouble(std::string("The value of flag --") + flag, value_str,
|
| + value);
|
| +}
|
| +
|
| +bool ParseStringFlag(const char* str, const char* flag, std::string* value) {
|
| + // Gets the value of the flag as a string.
|
| + const char* const value_str = ParseFlagValue(str, flag, false);
|
| +
|
| + // Aborts if the parsing failed.
|
| + if (value_str == nullptr) return false;
|
| +
|
| + *value = value_str;
|
| + return true;
|
| +}
|
| +
|
| +bool IsFlag(const char* str, const char* flag) {
|
| + return (ParseFlagValue(str, flag, true) != nullptr);
|
| +}
|
| +
|
| +bool IsTruthyFlagValue(const std::string& str) {
|
| + if (str.empty()) return true;
|
| + char ch = str[0];
|
| + return isalnum(ch) &&
|
| + !(ch == '0' || ch == 'f' || ch == 'F' || ch == 'n' || ch == 'N');
|
| +}
|
| +} // end namespace benchmark
|
|
|