git.sesse.net Git - stockfish/blob - src/thread_win32_osx.h

   1 /*
   2   Stockfish, a UCI chess playing engine derived from Glaurung 2.1
   3   Copyright (C) 2004-2008 Tord Romstad (Glaurung author)
   4   Copyright (C) 2008-2015 Marco Costalba, Joona Kiiski, Tord Romstad
   5   Copyright (C) 2015-2020 Marco Costalba, Joona Kiiski, Gary Linscott, Tord Romstad
   6
   7   Stockfish is free software: you can redistribute it and/or modify
   8   it under the terms of the GNU General Public License as published by
   9   the Free Software Foundation, either version 3 of the License, or
  10   (at your option) any later version.
  11
  12   Stockfish is distributed in the hope that it will be useful,
  13   but WITHOUT ANY WARRANTY; without even the implied warranty of
  14   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15   GNU General Public License for more details.
  16
  17   You should have received a copy of the GNU General Public License
  18   along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19 */
  20
  21 #ifndef THREAD_WIN32_OSX_H_INCLUDED
  22 #define THREAD_WIN32_OSX_H_INCLUDED
  23
  24 #include <thread>
  25
  26 /// On OSX threads other than the main thread are created with a reduced stack
  27 /// size of 512KB by default, this is too low for deep searches, which require
  28 /// somewhat more than 1MB stack, so adjust it to TH_STACK_SIZE.
  29 /// The implementation calls pthread_create() with the stack size parameter
  30 /// equal to the linux 8MB default, on platforms that support it.
  31
  32 #if defined(__APPLE__) || defined(__MINGW32__) || defined(__MINGW64__)
  33
  34 #include <pthread.h>
  35
  36 static const size_t TH_STACK_SIZE = 8 * 1024 * 1024;
  37
  38 template <class T, class P = std::pair<T*, void(T::*)()>>
  39 void* start_routine(void* ptr)
  40 {
  41    P* p = reinterpret_cast<P*>(ptr);
  42    (p->first->*(p->second))(); // Call member function pointer
  43    delete p;
  44    return NULL;
  45 }
  46
  47 class NativeThread {
  48
  49    pthread_t thread;
  50
  51 public:
  52   template<class T, class P = std::pair<T*, void(T::*)()>>
  53   explicit NativeThread(void(T::*fun)(), T* obj) {
  54     pthread_attr_t attr_storage, *attr = &attr_storage;
  55     pthread_attr_init(attr);
  56     pthread_attr_setstacksize(attr, TH_STACK_SIZE);
  57     pthread_create(&thread, attr, start_routine<T>, new P(obj, fun));
  58   }
  59   void join() { pthread_join(thread, NULL); }
  60 };
  61
  62 #else // Default case: use STL classes
  63
  64 typedef std::thread NativeThread;
  65
  66 #endif
  67
  68 #endif // #ifndef THREAD_WIN32_OSX_H_INCLUDED