IlmThreadSemaphore.h 3.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127
  1. ///////////////////////////////////////////////////////////////////////////
  2. //
  3. // Copyright (c) 2005-2012, Industrial Light & Magic, a division of Lucas
  4. // Digital Ltd. LLC
  5. //
  6. // All rights reserved.
  7. //
  8. // Redistribution and use in source and binary forms, with or without
  9. // modification, are permitted provided that the following conditions are
  10. // met:
  11. // * Redistributions of source code must retain the above copyright
  12. // notice, this list of conditions and the following disclaimer.
  13. // * Redistributions in binary form must reproduce the above
  14. // copyright notice, this list of conditions and the following disclaimer
  15. // in the documentation and/or other materials provided with the
  16. // distribution.
  17. // * Neither the name of Industrial Light & Magic nor the names of
  18. // its contributors may be used to endorse or promote products derived
  19. // from this software without specific prior written permission.
  20. //
  21. // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  22. // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  23. // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  24. // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  25. // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  26. // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  27. // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  28. // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  29. // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  30. // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  31. // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  32. //
  33. ///////////////////////////////////////////////////////////////////////////
  34. #ifndef INCLUDED_ILM_THREAD_SEMAPHORE_H
  35. #define INCLUDED_ILM_THREAD_SEMAPHORE_H
  36. //-----------------------------------------------------------------------------
  37. //
  38. // class Semaphore -- a wrapper class for
  39. // system-dependent counting semaphores
  40. //
  41. //-----------------------------------------------------------------------------
  42. #include "IlmBaseConfig.h"
  43. #include "IlmThreadExport.h"
  44. #include "IlmThreadNamespace.h"
  45. #if defined _WIN32 || defined _WIN64
  46. # ifdef NOMINMAX
  47. # undef NOMINMAX
  48. # endif
  49. # define NOMINMAX
  50. # include <windows.h>
  51. #elif HAVE_POSIX_SEMAPHORES
  52. # include <semaphore.h>
  53. #else
  54. # ifdef ILMBASE_FORCE_CXX03
  55. # if HAVE_PTHREAD
  56. # include <pthread.h>
  57. # endif
  58. # else
  59. # include <mutex>
  60. # include <condition_variable>
  61. # endif
  62. #endif
  63. ILMTHREAD_INTERNAL_NAMESPACE_HEADER_ENTER
  64. class ILMTHREAD_EXPORT Semaphore
  65. {
  66. public:
  67. Semaphore (unsigned int value = 0);
  68. virtual ~Semaphore();
  69. void wait();
  70. bool tryWait();
  71. void post();
  72. int value() const;
  73. private:
  74. #if defined _WIN32 || defined _WIN64
  75. mutable HANDLE _semaphore;
  76. #elif HAVE_POSIX_SEMAPHORES
  77. mutable sem_t _semaphore;
  78. #else
  79. //
  80. // If the platform has Posix threads but no semapohores,
  81. // then we implement them ourselves using condition variables
  82. //
  83. struct sema_t
  84. {
  85. unsigned int count;
  86. unsigned long numWaiting;
  87. # if ILMBASE_FORCE_CXX03
  88. # if HAVE_PTHREAD
  89. pthread_mutex_t mutex;
  90. pthread_cond_t nonZero;
  91. # else
  92. # error unhandled legacy setup
  93. # endif
  94. # else
  95. std::mutex mutex;
  96. std::condition_variable nonZero;
  97. # endif
  98. };
  99. mutable sema_t _semaphore;
  100. #endif
  101. void operator = (const Semaphore& s); // not implemented
  102. Semaphore (const Semaphore& s); // not implemented
  103. };
  104. ILMTHREAD_INTERNAL_NAMESPACE_HEADER_EXIT
  105. #endif // INCLUDED_ILM_THREAD_SEMAPHORE_H