threading.h 3.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124
  1. /**
  2. * Copyright (c) 2016 Tino Reichardt
  3. * All rights reserved.
  4. *
  5. * This source code is licensed under both the BSD-style license (found in the
  6. * LICENSE file in the root directory of this source tree) and the GPLv2 (found
  7. * in the COPYING file in the root directory of this source tree).
  8. *
  9. * You can contact the author at:
  10. * - zstdmt source repository: https://github.com/mcmilk/zstdmt
  11. */
  12. #ifndef THREADING_H_938743
  13. #define THREADING_H_938743
  14. #if defined (__cplusplus)
  15. extern "C" {
  16. #endif
  17. #if defined(ZSTD_MULTITHREAD) && defined(_WIN32)
  18. /**
  19. * Windows minimalist Pthread Wrapper, based on :
  20. * http://www.cse.wustl.edu/~schmidt/win32-cv-1.html
  21. */
  22. #ifdef WINVER
  23. # undef WINVER
  24. #endif
  25. #define WINVER 0x0600
  26. #ifdef _WIN32_WINNT
  27. # undef _WIN32_WINNT
  28. #endif
  29. #define _WIN32_WINNT 0x0600
  30. #ifndef WIN32_LEAN_AND_MEAN
  31. # define WIN32_LEAN_AND_MEAN
  32. #endif
  33. #undef ERROR /* reported already defined on VS 2015 (Rich Geldreich) */
  34. #include <windows.h>
  35. #undef ERROR
  36. #define ERROR(name) ZSTD_ERROR(name)
  37. /* mutex */
  38. #define ZSTD_pthread_mutex_t CRITICAL_SECTION
  39. #define ZSTD_pthread_mutex_init(a, b) (InitializeCriticalSection((a)), 0)
  40. #define ZSTD_pthread_mutex_destroy(a) DeleteCriticalSection((a))
  41. #define ZSTD_pthread_mutex_lock(a) EnterCriticalSection((a))
  42. #define ZSTD_pthread_mutex_unlock(a) LeaveCriticalSection((a))
  43. /* condition variable */
  44. #define ZSTD_pthread_cond_t CONDITION_VARIABLE
  45. #define ZSTD_pthread_cond_init(a, b) (InitializeConditionVariable((a)), 0)
  46. #define ZSTD_pthread_cond_destroy(a) /* No delete */
  47. #define ZSTD_pthread_cond_wait(a, b) SleepConditionVariableCS((a), (b), INFINITE)
  48. #define ZSTD_pthread_cond_signal(a) WakeConditionVariable((a))
  49. #define ZSTD_pthread_cond_broadcast(a) WakeAllConditionVariable((a))
  50. /* ZSTD_pthread_create() and ZSTD_pthread_join() */
  51. typedef struct {
  52. HANDLE handle;
  53. void* (*start_routine)(void*);
  54. void* arg;
  55. } ZSTD_pthread_t;
  56. int ZSTD_pthread_create(ZSTD_pthread_t* thread, const void* unused,
  57. void* (*start_routine) (void*), void* arg);
  58. int ZSTD_pthread_join(ZSTD_pthread_t thread, void** value_ptr);
  59. /**
  60. * add here more wrappers as required
  61. */
  62. #elif defined(ZSTD_MULTITHREAD) /* posix assumed ; need a better detection method */
  63. /* === POSIX Systems === */
  64. # include <pthread.h>
  65. #define ZSTD_pthread_mutex_t pthread_mutex_t
  66. #define ZSTD_pthread_mutex_init(a, b) pthread_mutex_init((a), (b))
  67. #define ZSTD_pthread_mutex_destroy(a) pthread_mutex_destroy((a))
  68. #define ZSTD_pthread_mutex_lock(a) pthread_mutex_lock((a))
  69. #define ZSTD_pthread_mutex_unlock(a) pthread_mutex_unlock((a))
  70. #define ZSTD_pthread_cond_t pthread_cond_t
  71. #define ZSTD_pthread_cond_init(a, b) pthread_cond_init((a), (b))
  72. #define ZSTD_pthread_cond_destroy(a) pthread_cond_destroy((a))
  73. #define ZSTD_pthread_cond_wait(a, b) pthread_cond_wait((a), (b))
  74. #define ZSTD_pthread_cond_signal(a) pthread_cond_signal((a))
  75. #define ZSTD_pthread_cond_broadcast(a) pthread_cond_broadcast((a))
  76. #define ZSTD_pthread_t pthread_t
  77. #define ZSTD_pthread_create(a, b, c, d) pthread_create((a), (b), (c), (d))
  78. #define ZSTD_pthread_join(a, b) pthread_join((a),(b))
  79. #else /* ZSTD_MULTITHREAD not defined */
  80. /* No multithreading support */
  81. typedef int ZSTD_pthread_mutex_t;
  82. #define ZSTD_pthread_mutex_init(a, b) ((void)a, 0)
  83. #define ZSTD_pthread_mutex_destroy(a)
  84. #define ZSTD_pthread_mutex_lock(a)
  85. #define ZSTD_pthread_mutex_unlock(a)
  86. typedef int ZSTD_pthread_cond_t;
  87. #define ZSTD_pthread_cond_init(a, b) ((void)a, 0)
  88. #define ZSTD_pthread_cond_destroy(a)
  89. #define ZSTD_pthread_cond_wait(a, b)
  90. #define ZSTD_pthread_cond_signal(a)
  91. #define ZSTD_pthread_cond_broadcast(a)
  92. /* do not use ZSTD_pthread_t */
  93. #endif /* ZSTD_MULTITHREAD */
  94. #if defined (__cplusplus)
  95. }
  96. #endif
  97. #endif /* THREADING_H_938743 */