1 //===================================================== 2 // File : btl.hh 3 // Copyright (C) 2008 Gael Guennebaud <gael.guennebaud (at) inria.fr> 4 //===================================================== 5 // 6 // This program is free software; you can redistribute it and/or 7 // modify it under the terms of the GNU General Public License 8 // as published by the Free Software Foundation; either version 2 9 // of the License, or (at your option) any later version. 10 // 11 // This program is distributed in the hope that it will be useful, 12 // but WITHOUT ANY WARRANTY; without even the implied warranty of 13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 14 // GNU General Public License for more details. 15 // You should have received a copy of the GNU General Public License 16 // along with this program; if not, write to the Free Software 17 // Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. 18 // 19 #ifndef BTL_HH 20 #define BTL_HH 21 22 #include "bench_parameter.hh" 23 #include <iostream> 24 #include <algorithm> 25 #include <vector> 26 #include <string> 27 #include "utilities.h" 28 29 #if (defined __GNUC__) 30 #define BTL_ALWAYS_INLINE __attribute__((always_inline)) inline 31 #else 32 #define BTL_ALWAYS_INLINE inline 33 #endif 34 35 #if (defined __GNUC__) 36 #define BTL_DONT_INLINE __attribute__((noinline)) 37 #else 38 #define BTL_DONT_INLINE 39 #endif 40 41 #if (defined __GNUC__) 42 #define BTL_ASM_COMMENT(X) asm("#" X) 43 #else 44 #define BTL_ASM_COMMENT(X) 45 #endif 46 47 #if (defined __GNUC__) && (!defined __INTEL_COMPILER) && !defined(__arm__) && !defined(__powerpc__) 48 #define BTL_DISABLE_SSE_EXCEPTIONS() { \ 49 int aux; \ 50 asm( \ 51 "stmxcsr %[aux] \n\t" \ 52 "orl $32832, %[aux] \n\t" \ 53 "ldmxcsr %[aux] \n\t" \ 54 : : [aux] "m" (aux)); \ 55 } 56 #else 57 #define BTL_DISABLE_SSE_EXCEPTIONS() 58 #endif 59 60 /** Enhanced std::string 61 */ 62 class BtlString : public std::string 63 { 64 public: 65 BtlString() : std::string() {} 66 BtlString(const BtlString& str) : std::string(static_cast<const std::string&>(str)) {} 67 BtlString(const std::string& str) : std::string(str) {} 68 BtlString(const char* str) : std::string(str) {} 69 70 operator const char* () const { return c_str(); } 71 72 void trim( bool left = true, bool right = true ) 73 { 74 int lspaces, rspaces, len = length(), i; 75 lspaces = rspaces = 0; 76 77 if ( left ) 78 for (i=0; i<len && (at(i)==' '||at(i)=='\t'||at(i)=='\r'||at(i)=='\n'); ++lspaces,++i); 79 80 if ( right && lspaces < len ) 81 for(i=len-1; i>=0 && (at(i)==' '||at(i)=='\t'||at(i)=='\r'||at(i)=='\n'); rspaces++,i--); 82 83 *this = substr(lspaces, len-lspaces-rspaces); 84 } 85 86 std::vector<BtlString> split( const BtlString& delims = "\t\n ") const 87 { 88 std::vector<BtlString> ret; 89 unsigned int numSplits = 0; 90 size_t start, pos; 91 start = 0; 92 do 93 { 94 pos = find_first_of(delims, start); 95 if (pos == start) 96 { 97 ret.push_back(""); 98 start = pos + 1; 99 } 100 else if (pos == npos) 101 ret.push_back( substr(start) ); 102 else 103 { 104 ret.push_back( substr(start, pos - start) ); 105 start = pos + 1; 106 } 107 //start = find_first_not_of(delims, start); 108 ++numSplits; 109 } while (pos != npos); 110 return ret; 111 } 112 113 bool endsWith(const BtlString& str) const 114 { 115 if(str.size()>this->size()) 116 return false; 117 return this->substr(this->size()-str.size(),str.size()) == str; 118 } 119 bool contains(const BtlString& str) const 120 { 121 return this->find(str)<this->size(); 122 } 123 bool beginsWith(const BtlString& str) const 124 { 125 if(str.size()>this->size()) 126 return false; 127 return this->substr(0,str.size()) == str; 128 } 129 130 BtlString toLowerCase( void ) 131 { 132 std::transform(begin(), end(), begin(), static_cast<int(*)(int)>(::tolower) ); 133 return *this; 134 } 135 BtlString toUpperCase( void ) 136 { 137 std::transform(begin(), end(), begin(), static_cast<int(*)(int)>(::toupper) ); 138 return *this; 139 } 140 141 /** Case insensitive comparison. 142 */ 143 bool isEquiv(const BtlString& str) const 144 { 145 BtlString str0 = *this; 146 str0.toLowerCase(); 147 BtlString str1 = str; 148 str1.toLowerCase(); 149 return str0 == str1; 150 } 151 152 /** Decompose the current string as a path and a file. 153 For instance: "dir1/dir2/file.ext" leads to path="dir1/dir2/" and filename="file.ext" 154 */ 155 void decomposePathAndFile(BtlString& path, BtlString& filename) const 156 { 157 std::vector<BtlString> elements = this->split("/\\"); 158 path = ""; 159 filename = elements.back(); 160 elements.pop_back(); 161 if (this->at(0)=='/') 162 path = "/"; 163 for (unsigned int i=0 ; i<elements.size() ; ++i) 164 path += elements[i] + "/"; 165 } 166 }; 167 168 class BtlConfig 169 { 170 public: 171 BtlConfig() 172 : overwriteResults(false), checkResults(true), realclock(false), tries(DEFAULT_NB_TRIES) 173 { 174 char * _config; 175 _config = getenv ("BTL_CONFIG"); 176 if (_config!=NULL) 177 { 178 std::vector<BtlString> config = BtlString(_config).split(" \t\n"); 179 for (int i = 0; i<config.size(); i++) 180 { 181 if (config[i].beginsWith("-a")) 182 { 183 if (i+1==config.size()) 184 { 185 std::cerr << "error processing option: " << config[i] << "\n"; 186 exit(2); 187 } 188 Instance.m_selectedActionNames = config[i+1].split(":"); 189 190 i += 1; 191 } 192 else if (config[i].beginsWith("-t")) 193 { 194 if (i+1==config.size()) 195 { 196 std::cerr << "error processing option: " << config[i] << "\n"; 197 exit(2); 198 } 199 Instance.tries = atoi(config[i+1].c_str()); 200 201 i += 1; 202 } 203 else if (config[i].beginsWith("--overwrite")) 204 { 205 Instance.overwriteResults = true; 206 } 207 else if (config[i].beginsWith("--nocheck")) 208 { 209 Instance.checkResults = false; 210 } 211 else if (config[i].beginsWith("--real")) 212 { 213 Instance.realclock = true; 214 } 215 } 216 } 217 218 BTL_DISABLE_SSE_EXCEPTIONS(); 219 } 220 221 BTL_DONT_INLINE static bool skipAction(const std::string& _name) 222 { 223 if (Instance.m_selectedActionNames.empty()) 224 return false; 225 226 BtlString name(_name); 227 for (int i=0; i<Instance.m_selectedActionNames.size(); ++i) 228 if (name.contains(Instance.m_selectedActionNames[i])) 229 return false; 230 231 return true; 232 } 233 234 static BtlConfig Instance; 235 bool overwriteResults; 236 bool checkResults; 237 bool realclock; 238 int tries; 239 240 protected: 241 std::vector<BtlString> m_selectedActionNames; 242 }; 243 244 #define BTL_MAIN \ 245 BtlConfig BtlConfig::Instance 246 247 #endif // BTL_HH 248
