source/common/threading.cpp

   1 /*****************************************************************************
   2  * Copyright (C) 2013 x265 project
   3  *
   4  * Authors: Steve Borho <steve@borho.org>
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (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  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02111, USA.
  19  *
  20  * This program is also available under a commercial proprietary license.
  21  * For more information, contact us at license @ x265.com
  22  *****************************************************************************/
  23
  24 #include "threading.h"
  25
  26 namespace x265 {
  27 // x265 private namespace
  28
  29 /* C shim for forced stack alignment */
  30 static void stackAlignMain(Thread *instance)
  31 {
  32     instance->threadMain();
  33 }
  34
  35 #if _WIN32
  36
  37 static DWORD WINAPI ThreadShim(Thread *instance)
  38 {
  39     // defer processing to the virtual function implemented in the derived class
  40     x265_stack_align(stackAlignMain, instance);
  41
  42     return 0;
  43 }
  44
  45 bool Thread::start()
  46 {
  47     DWORD threadId;
  48
  49     thread = CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)ThreadShim, this, 0, &threadId);
  50
  51     return threadId > 0;
  52 }
  53
  54 void Thread::stop()
  55 {
  56     if (thread)
  57         WaitForSingleObject(thread, INFINITE);
  58 }
  59
  60 Thread::~Thread()
  61 {
  62     if (thread)
  63         CloseHandle(thread);
  64 }
  65
  66 #else /* POSIX / pthreads */
  67
  68 static void *ThreadShim(void *opaque)
  69 {
  70     // defer processing to the virtual function implemented in the derived class
  71     Thread *instance = reinterpret_cast<Thread *>(opaque);
  72
  73     x265_stack_align(stackAlignMain, instance);
  74
  75     return NULL;
  76 }
  77
  78 bool Thread::start()
  79 {
  80     if (pthread_create(&thread, NULL, ThreadShim, this))
  81     {
  82         thread = 0;
  83         return false;
  84     }
  85
  86     return true;
  87 }
  88
  89 void Thread::stop()
  90 {
  91     if (thread)
  92         pthread_join(thread, NULL);
  93 }
  94
  95 Thread::~Thread() {}
  96
  97 #endif // if _WIN32
  98
  99 Thread::Thread()
 100 {
 101     thread = 0;
 102 }
 103
 104 }