Setup to use Callgrind profiler

[stockfish] / src / main.cpp

diff --git a/src/main.cpp b/src/main.cpp
index 89ca9e8ec6fa976d3c7127f78a60a3cbe77eb894..01fb47860b092d43d837099234d56786f3c2164f 100644 (file)
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -7,16 +7,18 @@
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation, either version 3 of the License, or
   (at your option) any later version.
-  
+
   Stockfish is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.
-  
+
   You should have received a copy of the GNU General Public License
   along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
+// To profile with callgrind uncomment following line
+//#define USE_CALLGRIND
 
 ////
 //// Includes
@@ -39,9 +41,13 @@
 #include "uci.h"
 #include "ucioption.h"
 
+#ifdef USE_CALLGRIND
+#include <valgrind/callgrind.h>
+#endif
+
 using std::string;
 
-//// 
+////
 //// Functions
 ////
 
@@ -52,7 +58,6 @@ int main(int argc, char *argv[]) {
   std::cin.rdbuf()->pubsetbuf(NULL, 0);
 
   // Initialization
-
   init_mersenne();
   init_direction_table();
   init_bitboards();
@@ -64,6 +69,10 @@ int main(int argc, char *argv[]) {
   init_bitbases();
   init_threads();
 
+#ifdef USE_CALLGRIND
+  CALLGRIND_START_INSTRUMENTATION;
+#endif
+
   // Make random number generation less deterministic, for book moves
   for (int i = abs(get_system_time() % 10000); i > 0; i--)
       genrand_int32();
@@ -73,16 +82,16 @@ int main(int argc, char *argv[]) {
   {
       if (argc < 4 || argc > 7)
       {
-        std::cout << "Usage: glaurung bench <hash size> <threads> "
+        std::cout << "Usage: stockfish bench <hash size> <threads> "
                   << "[time = 60s] [fen positions file = default] "
-                  << "[time or depth limited = time]"
+                  << "[time, depth or node limited = time]"
                   << std::endl;
         exit(0);
       }
       string time = argc > 4 ? argv[4] : "60";
       string fen = argc > 5 ? argv[5] : "default";
-      string dt = argc > 6 ? argv[6] : "time";
-      benchmark(string(argv[2]) + " " + string(argv[3]) + " " + time + " " + fen + " " + dt);
+      string lim = argc > 6 ? argv[6] : "time";
+      benchmark(string(argv[2]) + " " + string(argv[3]) + " " + time + " " + fen + " " + lim);
       return 0;
   }