7 //     http://www.apache.org/licenses/LICENSE-2.0
24 // Parses 'str' for a 32-bit signed integer.  If successful, writes
32   // Has strtol() consumed all characters in the string?  in ParseInt32()
34     // No - an invalid character was encountered.  in ParseInt32()
35     std::cerr << src_text << " is expected to be a 32-bit integer, "  in ParseInt32()
36               << "but actually has value \"" << str << "\".\n";  in ParseInt32()
49     std::cerr << src_text << " is expected to be a 32-bit integer, "  in ParseInt32()
50               << "but actually has value \"" << str << "\", "  in ParseInt32()
66   // Has strtol() consumed all characters in the string?  in ParseDouble()
68     // No - an invalid character was encountered.  in ParseDouble()
70               << "but actually has value \"" << str << "\".\n";  in ParseDouble()
79 // given flag.  For example, FlagToEnvVar("foo") will return
80 // "BENCHMARK_FOO" in the open-source version.
81 static std::string FlagToEnvVar(const char* flag) {  in FlagToEnvVar()  argument
82   const std::string flag_str(flag);  in FlagToEnvVar()
92 // the given flag; if it's not set, returns default_value.
95 bool BoolFromEnv(const char* flag, bool default_value) {  in BoolFromEnv()  argument
96   const std::string env_var = FlagToEnvVar(flag);  in BoolFromEnv()
102 // Reads and returns a 32-bit integer stored in the environment
103 // variable corresponding to the given flag; if it isn't set or
104 // doesn't represent a valid 32-bit integer, returns default_value.
105 int32_t Int32FromEnv(const char* flag, int32_t default_value) {  in Int32FromEnv()  argument
106   const std::string env_var = FlagToEnvVar(flag);  in Int32FromEnv()
124 // the given flag; if it's not set, returns default_value.
125 const char* StringFromEnv(const char* flag, const char* default_value) {  in StringFromEnv()  argument
126   const std::string env_var = FlagToEnvVar(flag);  in StringFromEnv()
131 // Parses a string as a command line flag.  The string should have
132 // the format "--flag=value".  When def_optional is true, the "=value"
135 // Returns the value of the flag, or nullptr if the parsing failed.
136 const char* ParseFlagValue(const char* str, const char* flag,  in ParseFlagValue()  argument
138   // str and flag must not be nullptr.  in ParseFlagValue()
139   if (str == nullptr || flag == nullptr) return nullptr;  in ParseFlagValue()
141   // The flag must start with "--".  in ParseFlagValue()
142   const std::string flag_str = std::string("--") + std::string(flag);  in ParseFlagValue()
146   // Skips the flag name.  in ParseFlagValue()
153   // flag name, or if def_optional is false, there must be a '=' after  in ParseFlagValue()
154   // the flag name.  in ParseFlagValue()
161 bool ParseBoolFlag(const char* str, const char* flag, bool* value) {  in ParseBoolFlag()  argument
162   // Gets the value of the flag as a string.  in ParseBoolFlag()
163   const char* const value_str = ParseFlagValue(str, flag, true);  in ParseBoolFlag()
173 bool ParseInt32Flag(const char* str, const char* flag, int32_t* value) {  in ParseInt32Flag()  argument
174   // Gets the value of the flag as a string.  in ParseInt32Flag()
175   const char* const value_str = ParseFlagValue(str, flag, false);  in ParseInt32Flag()
180   // Sets *value to the value of the flag.  in ParseInt32Flag()
181   return ParseInt32(std::string("The value of flag --") + flag, value_str,  in ParseInt32Flag()
185 bool ParseDoubleFlag(const char* str, const char* flag, double* value) {  in ParseDoubleFlag()  argument
186   // Gets the value of the flag as a string.  in ParseDoubleFlag()
187   const char* const value_str = ParseFlagValue(str, flag, false);  in ParseDoubleFlag()
192   // Sets *value to the value of the flag.  in ParseDoubleFlag()
193   return ParseDouble(std::string("The value of flag --") + flag, value_str,  in ParseDoubleFlag()
197 bool ParseStringFlag(const char* str, const char* flag, std::string* value) {  in ParseStringFlag()  argument
198   // Gets the value of the flag as a string.  in ParseStringFlag()
199   const char* const value_str = ParseFlagValue(str, flag, false);  in ParseStringFlag()
208 bool IsFlag(const char* str, const char* flag) {  in IsFlag()  argument
209   return (ParseFlagValue(str, flag, true) != nullptr);  in IsFlag()