aboutsummaryrefslogtreecommitdiff
path: root/PhysX_3.4/Source/LowLevelCloth/src/windows/CuDeviceVector.h
diff options
context:
space:
mode:
authorgit perforce import user <a@b>2016-10-25 12:29:14 -0600
committerSheikh Dawood Abdul Ajees <Sheikh Dawood Abdul Ajees>2016-10-25 18:56:37 -0500
commit3dfe2108cfab31ba3ee5527e217d0d8e99a51162 (patch)
treefa6485c169e50d7415a651bf838f5bcd0fd3bfbd /PhysX_3.4/Source/LowLevelCloth/src/windows/CuDeviceVector.h
downloadphysx-3.4-3dfe2108cfab31ba3ee5527e217d0d8e99a51162.tar.xz
physx-3.4-3dfe2108cfab31ba3ee5527e217d0d8e99a51162.zip
Initial commit:
PhysX 3.4.0 Update @ 21294896 APEX 1.4.0 Update @ 21275617 [CL 21300167]
Diffstat (limited to 'PhysX_3.4/Source/LowLevelCloth/src/windows/CuDeviceVector.h')
-rw-r--r--PhysX_3.4/Source/LowLevelCloth/src/windows/CuDeviceVector.h258
1 files changed, 258 insertions, 0 deletions
diff --git a/PhysX_3.4/Source/LowLevelCloth/src/windows/CuDeviceVector.h b/PhysX_3.4/Source/LowLevelCloth/src/windows/CuDeviceVector.h
new file mode 100644
index 00000000..e3997d26
--- /dev/null
+++ b/PhysX_3.4/Source/LowLevelCloth/src/windows/CuDeviceVector.h
@@ -0,0 +1,258 @@
+// This code contains NVIDIA Confidential Information and is disclosed to you
+// under a form of NVIDIA software license agreement provided separately to you.
+//
+// Notice
+// NVIDIA Corporation and its licensors retain all intellectual property and
+// proprietary rights in and to this software and related documentation and
+// any modifications thereto. Any use, reproduction, disclosure, or
+// distribution of this software and related documentation without an express
+// license agreement from NVIDIA Corporation is strictly prohibited.
+//
+// ALL NVIDIA DESIGN SPECIFICATIONS, CODE ARE PROVIDED "AS IS.". NVIDIA MAKES
+// NO WARRANTIES, EXPRESSED, IMPLIED, STATUTORY, OR OTHERWISE WITH RESPECT TO
+// THE MATERIALS, AND EXPRESSLY DISCLAIMS ALL IMPLIED WARRANTIES OF NONINFRINGEMENT,
+// MERCHANTABILITY, AND FITNESS FOR A PARTICULAR PURPOSE.
+//
+// Information and code furnished is believed to be accurate and reliable.
+// However, NVIDIA Corporation assumes no responsibility for the consequences of use of such
+// information or for any infringement of patents or other rights of third parties that may
+// result from its use. No license is granted by implication or otherwise under any patent
+// or patent rights of NVIDIA Corporation. Details are subject to change without notice.
+// This code supersedes and replaces all information previously supplied.
+// NVIDIA Corporation products are not authorized for use as critical
+// components in life support devices or systems without express written approval of
+// NVIDIA Corporation.
+//
+// Copyright (c) 2008-2016 NVIDIA Corporation. All rights reserved.
+// Copyright (c) 2004-2008 AGEIA Technologies, Inc. All rights reserved.
+// Copyright (c) 2001-2004 NovodeX AG. All rights reserved.
+
+#pragma once
+
+#include "foundation/PxMath.h" // for swap
+#include "cudamanager/PxCudaMemoryManager.h"
+#include "cudamanager/PxCudaContextManager.h"
+#include "CuDevicePointer.h"
+#include "PsArray.h"
+#include "PsUtilities.h"
+
+namespace physx
+{
+#if PX_VC
+#pragma warning(push)
+#pragma warning(disable : 4365) // 'action' : conversion from 'type_1' to 'type_2', signed/unsigned mismatch
+#endif
+
+namespace cloth
+{
+
+// STL-style vector that holds POD types in CUDA device memory. The interface
+// is not complete, add whatever you need from the std::vector interface.
+template <typename T>
+class CuDeviceVector
+{
+ public:
+ typedef CuDevicePointer<T> iterator;
+ typedef CuDevicePointer<const T> const_iterator;
+
+ CuDeviceVector(physx::PxCudaContextManager* ctx) : mManager(0)
+ {
+ PX_ASSERT(ctx);
+
+ if(ctx)
+ mManager = ctx->getMemoryManager();
+ }
+
+ CuDeviceVector(const CuDeviceVector& other) : mManager(other.getMemoryManager())
+ {
+ PX_ASSERT(mManager);
+
+ operator=(other);
+ }
+
+ CuDeviceVector(physx::PxCudaContextManager* ctx, const T* first, const T* last) : mManager(0)
+ {
+ PX_ASSERT(ctx);
+
+ if(ctx)
+ {
+ mManager = ctx->getMemoryManager();
+ assign(first, last);
+ }
+ }
+
+ template <typename Alloc>
+ CuDeviceVector(const shdfnd::Array<T, Alloc>& other)
+ {
+ operator=(other);
+ }
+
+ ~CuDeviceVector()
+ {
+ PX_ASSERT(mManager);
+
+ mManager->free(physx::PxCudaBufferMemorySpace::T_GPU, mFirst.dev());
+ }
+
+ CuDeviceVector& operator=(const CuDeviceVector& other)
+ {
+ resize(other.size());
+ checkSuccess(cuMemcpyDtoD(mFirst.dev(), other.mFirst.dev(), other.size() * sizeof(T)));
+ return *this;
+ }
+
+ template <typename Alloc>
+ CuDeviceVector& operator=(const shdfnd::Array<T, Alloc>& other)
+ {
+ const T* first = other.empty() ? 0 : &other.front();
+ assign(first, first + other.size());
+ return *this;
+ }
+
+ bool empty() const
+ {
+ return mLast == mFirst;
+ }
+ size_t size() const
+ {
+ return size_t(mLast - mFirst);
+ }
+ size_t capacity() const
+ {
+ return mEnd - mFirst;
+ }
+
+ iterator begin()
+ {
+ return mFirst;
+ }
+ iterator end()
+ {
+ return mLast;
+ }
+ const_iterator begin() const
+ {
+ return mFirst;
+ }
+ const_iterator end() const
+ {
+ return mLast;
+ }
+
+ void push_back(const T& v)
+ {
+ if(mLast == mEnd)
+ reserve(PxMax<size_t>(1, capacity() * 2));
+
+ *mLast++ = v;
+ }
+
+ void push_back(const T* first, const T* last)
+ {
+ if(mEnd - mLast < last - first)
+ reserve(PxMax<size_t>(2 * capacity(), mLast - mFirst + last - first));
+
+ if(first != last)
+ checkSuccess(cuMemcpyHtoD(mLast.dev(), first, sizeof(T) * (last - first)));
+
+ mLast += last - first;
+ }
+
+ void erase(iterator it)
+ {
+ size_t byteSize = (mLast - it - 1) * sizeof(T);
+ if(byteSize)
+ {
+ CUdeviceptr tmp = 0, dst = it.dev();
+
+ PX_ASSERT(mManager);
+
+ tmp = mManager->alloc(physx::PxCudaBufferMemorySpace::T_GPU, byteSize,
+ PX_ALLOC_INFO("cloth::CuDeviceVector::T_GPU", CLOTH));
+ checkSuccess(cuMemcpyDtoD(tmp, dst + sizeof(T), byteSize));
+ checkSuccess(cuMemcpyDtoD(dst, tmp, byteSize));
+ mManager->free(physx::PxCudaBufferMemorySpace::T_GPU, tmp);
+ }
+ --mLast;
+ }
+
+ void reserve(size_t n)
+ {
+ if(n <= capacity())
+ return;
+
+ CUdeviceptr newFirst = 0, oldFirst = mFirst.dev();
+
+ PX_ASSERT(mManager);
+
+ newFirst = mManager->alloc(physx::PxCudaBufferMemorySpace::T_GPU, sizeof(T) * n,
+ PX_ALLOC_INFO("cloth::CuDeviceVector::T_GPU", CLOTH));
+ checkSuccess(cuMemcpyDtoD(newFirst, oldFirst, sizeof(T) * size()));
+ mManager->free(physx::PxCudaBufferMemorySpace::T_GPU, oldFirst);
+
+ iterator first(reinterpret_cast<T*>(newFirst));
+ mEnd = first + n;
+ mLast = first + size();
+ mFirst = first;
+ }
+
+ void resize(size_t n)
+ {
+ if(capacity() < n)
+ reserve(PxMax(n, capacity() * 2));
+
+ mLast = mFirst + n;
+ }
+
+ void assign(const T* first, const T* last)
+ {
+ size_t n = last - first;
+ resize(n);
+ checkSuccess(cuMemcpyHtoD(mFirst.dev(), first, n * sizeof(T)));
+ }
+
+ void swap(CuDeviceVector& other)
+ {
+ shdfnd::swap(mFirst, other.mFirst);
+ shdfnd::swap(mLast, other.mLast);
+ shdfnd::swap(mEnd, other.mEnd);
+ }
+
+ // match PxArray interface
+ void remove(size_t i)
+ {
+ erase(begin() + i);
+ }
+ void pushBack(const T& v)
+ {
+ push_back(v);
+ }
+
+ physx::PxCudaMemoryManager* getMemoryManager() const
+ {
+ return mManager;
+ }
+
+ private:
+ iterator mFirst, mLast, mEnd;
+ physx::PxCudaMemoryManager* mManager;
+};
+
+} // namespace cloth
+} // namespace physx
+
+#if PX_VC
+#pragma warning(pop)
+#endif
+
+namespace physx
+{
+namespace shdfnd
+{
+template <typename T>
+void swap(physx::cloth::CuDeviceVector<T>& first, physx::cloth::CuDeviceVector<T>& second)
+{
+ first.swap(second);
+}
+}
+}