1//=====================================================
2// File   :  btl.hh
3// Copyright (C) 2008 Gael Guennebaud <gael.guennebaud@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*/
62class BtlString : public std::string
63{
64public:
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
168class BtlConfig
169{
170public:
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
240protected:
241  std::vector<BtlString> m_selectedActionNames;
242};
243
244#define BTL_MAIN \
245  BtlConfig BtlConfig::Instance
246
247#endif // BTL_HH
248