hkThreadLocalData.h
上传用户:yisoukefu
上传日期:2020-08-09
资源大小:39506k
文件大小:4k
源码类别:

其他游戏

开发平台:

Visual C++

  1. /* 
  2.  * 
  3.  * Confidential Information of Telekinesys Research Limited (t/a Havok). Not for disclosure or distribution without Havok's
  4.  * prior written consent. This software contains code, techniques and know-how which is confidential and proprietary to Havok.
  5.  * Level 2 and Level 3 source code contains trade secrets of Havok. Havok Software (C) Copyright 1999-2009 Telekinesys Research Limited t/a Havok. All Rights Reserved. Use of this software is subject to the terms of an end user license agreement.
  6.  * 
  7.  */
  8. #ifndef HKBASE_HK_THREAD_LOCAL_POINTER_H
  9. #define HKBASE_HK_THREAD_LOCAL_POINTER_H
  10. #include <Common/Base/Config/hkConfigThread.h>
  11. //HK_THREAD_LOCAL - declare a thread local variable
  12. //HK_THREAD_LOCAL_SET(var,value) - assign to tls var
  13. //HK_THREAD_LOCAL_GET(var) - get from tls var
  14. #if (HK_CONFIG_THREAD == HK_CONFIG_MULTI_THREADED) && !defined(HK_PLATFORM_SPU)
  15. #if defined(HK_PLATFORM_PS3_PPU) || defined(HK_PLATFORM_UNIX)
  16. # define HK_THREAD_LOCAL(TYPE) __thread TYPE
  17. # elif defined(HK_PLATFORM_WIN32)
  18. extern "C" {
  19. __declspec(dllimport) unsigned long _stdcall TlsAlloc(void);
  20. __declspec(dllimport) void* _stdcall TlsGetValue( unsigned long dwTlsIndex );
  21. __declspec(dllimport) int _stdcall TlsSetValue( unsigned long dwTlsIndex, void* lpTlsValue);
  22. __declspec(dllimport) int _stdcall TlsFree( unsigned long dwTlsIndex );
  23. }
  24. /// A platform independent wrapper for thread local storage
  25. /// We assume we always just store a pointer (or data the same size or smaller than a pointer) 
  26. /// ie,  sizeof(T) <= sizeof(char*)
  27. template<typename T>
  28. class hkThreadLocalData
  29. {
  30. public:
  31. hkThreadLocalData() { m_slotID =  TlsAlloc(); }
  32. ~hkThreadLocalData() { TlsFree(m_slotID); }
  33. HK_FORCE_INLINE T getData() const
  34. {
  35. return (T)(hkUlong)TlsGetValue(m_slotID) ;
  36. }
  37. HK_FORCE_INLINE void setData(T p) 
  38. {
  39. hkUlong v = hkUlong(p);
  40. TlsSetValue( m_slotID, (void*)v );
  41. }
  42. unsigned long m_slotID; 
  43. };
  44. # define HK_THREAD_LOCAL(TYPE) hkThreadLocalData<TYPE>
  45. # define HK_THREAD_LOCAL_SET(var,value) var.setData(value)
  46. # define HK_THREAD_LOCAL_GET(var) var.getData()
  47. # elif defined(HK_PLATFORM_MACPPC) || defined(HK_PLATFORM_MAC386) 
  48. # include <pthread.h>
  49. template < typename T > 
  50. class hkThreadLocalData
  51. {
  52. public:
  53. hkThreadLocalData() { pthread_key_create(&m_key, HK_NULL); }
  54. ~hkThreadLocalData() { pthread_key_delete(m_key); }
  55. HK_FORCE_INLINE T getData() 
  56. {
  57. return (T) pthread_getspecific(m_key);
  58. }
  59. HK_FORCE_INLINE void setData(T p) 
  60. {
  61. pthread_setspecific(m_key, (void*)p );
  62. }
  63. protected:
  64. pthread_key_t m_key; 
  65. };
  66. # define HK_THREAD_LOCAL(TYPE) hkThreadLocalData<TYPE>
  67. # define HK_THREAD_LOCAL_SET(var,value) var.setData(value)
  68. # define HK_THREAD_LOCAL_GET(var) var.getData()
  69. # else
  70. # error fixme
  71. # endif
  72. #else
  73. # define HK_THREAD_LOCAL(TYPE) TYPE
  74. #endif
  75. #ifndef HK_THREAD_LOCAL_SET
  76. # define HK_THREAD_LOCAL_SET(var,value) var = value
  77. #endif
  78. #ifndef HK_THREAD_LOCAL_GET
  79. # define HK_THREAD_LOCAL_GET(var) var
  80. #endif
  81. #endif // HKBASE_HK_THREAD_LOCAL_POINTER_H
  82. /*
  83. * Havok SDK - NO SOURCE PC DOWNLOAD, BUILD(#20090216)
  84. * Confidential Information of Havok.  (C) Copyright 1999-2009
  85. * Telekinesys Research Limited t/a Havok. All Rights Reserved. The Havok
  86. * Logo, and the Havok buzzsaw logo are trademarks of Havok.  Title, ownership
  87. * rights, and intellectual property rights in the Havok software remain in
  88. * Havok and/or its suppliers.
  89. * Use of this software for evaluation purposes is subject to and indicates
  90. * acceptance of the End User licence Agreement for this product. A copy of
  91. * the license is included with this software and is also available at www.havok.com/tryhavok.
  92. */