NVIDIA(R) PhysX(R) SDK 3.4 API Reference: PxGpuCopyDescQueue.h Source File

NVIDIA PhysX SDK 3.4 API

PxGpuCopyDescQueue.h

Go to the documentation of this file.
00001 // This code contains NVIDIA Confidential Information and is disclosed to you
00002 // under a form of NVIDIA software license agreement provided separately to you.
00003 //
00004 // Notice
00005 // NVIDIA Corporation and its licensors retain all intellectual property and
00006 // proprietary rights in and to this software and related documentation and
00007 // any modifications thereto. Any use, reproduction, disclosure, or
00008 // distribution of this software and related documentation without an express
00009 // license agreement from NVIDIA Corporation is strictly prohibited.
00010 //
00011 // ALL NVIDIA DESIGN SPECIFICATIONS, CODE ARE PROVIDED "AS IS.". NVIDIA MAKES
00012 // NO WARRANTIES, EXPRESSED, IMPLIED, STATUTORY, OR OTHERWISE WITH RESPECT TO
00013 // THE MATERIALS, AND EXPRESSLY DISCLAIMS ALL IMPLIED WARRANTIES OF NONINFRINGEMENT,
00014 // MERCHANTABILITY, AND FITNESS FOR A PARTICULAR PURPOSE.
00015 //
00016 // Information and code furnished is believed to be accurate and reliable.
00017 // However, NVIDIA Corporation assumes no responsibility for the consequences of use of such
00018 // information or for any infringement of patents or other rights of third parties that may
00019 // result from its use. No license is granted by implication or otherwise under any patent
00020 // or patent rights of NVIDIA Corporation. Details are subject to change without notice.
00021 // This code supersedes and replaces all information previously supplied.
00022 // NVIDIA Corporation products are not authorized for use as critical
00023 // components in life support devices or systems without express written approval of
00024 // NVIDIA Corporation.
00025 //
00026 // Copyright (c) 2008-2017 NVIDIA Corporation. All rights reserved.
00027 
00028 #ifndef PXCUDACONTEXTMANAGER_PXGPUCOPYDESCQUEUE_H
00029 #define PXCUDACONTEXTMANAGER_PXGPUCOPYDESCQUEUE_H
00030 
00031 #include "foundation/PxPreprocessor.h"
00032 
00033 #if PX_SUPPORT_GPU_PHYSX
00034 
00035 #include "foundation/PxAssert.h"
00036 #include "task/PxTaskDefine.h"
00037 #include "task/PxGpuDispatcher.h"
00038 #include "cudamanager/PxGpuCopyDesc.h"
00039 #include "cudamanager/PxCudaContextManager.h"
00040 
00041 /* forward decl to avoid including <cuda.h> */
00042 typedef struct CUstream_st* CUstream;
00043 
00044 namespace physx
00045 {
00046 
00047 PX_PUSH_PACK_DEFAULT
00048 
00050 class PxGpuCopyDescQueue
00051 {
00052 public:
00054     PxGpuCopyDescQueue(PxGpuDispatcher& d)
00055         : mDispatcher(d)
00056         , mBuffer(0)
00057         , mStream(0)
00058         , mReserved(0)
00059         , mOccupancy(0)
00060         , mFlushed(0)
00061     {
00062     }
00063 
00065     ~PxGpuCopyDescQueue()
00066     {
00067         if (mBuffer)
00068         {
00069             mDispatcher.getCudaContextManager()->getMemoryManager()->free(PxCudaBufferMemorySpace::T_PINNED_HOST, (size_t) mBuffer);
00070         }
00071     }
00072 
00078     void reset(CUstream stream, uint32_t reserveSize)
00079     {
00080         if (reserveSize > mReserved)
00081         {
00082             if (mBuffer)
00083             {
00084                 mDispatcher.getCudaContextManager()->getMemoryManager()->free(
00085                     PxCudaBufferMemorySpace::T_PINNED_HOST,
00086                     (size_t) mBuffer);
00087                 mReserved = 0;
00088             }
00089             mBuffer = (PxGpuCopyDesc*) mDispatcher.getCudaContextManager()->getMemoryManager()->alloc(
00090                           PxCudaBufferMemorySpace::T_PINNED_HOST,
00091                           reserveSize * sizeof(PxGpuCopyDesc),
00092                           PX_ALLOC_INFO("PxGpuCopyDescQueue", GPU_UTIL));
00093             if (mBuffer)
00094             {
00095                 mReserved = reserveSize;
00096             }
00097         }
00098 
00099         mOccupancy = 0;
00100         mFlushed = 0;
00101         mStream = stream;
00102     }
00103 
00105     void enqueue(PxGpuCopyDesc& desc)
00106     {
00107         PX_ASSERT(desc.isValid());
00108         if (desc.bytes == 0)
00109         {
00110             return;
00111         }
00112 
00113         if (mOccupancy < mReserved)
00114         {
00115             mBuffer[ mOccupancy++ ] = desc;
00116         }
00117         else
00118         {
00119             mDispatcher.launchCopyKernel(&desc, 1, mStream);
00120         }
00121     }
00122 
00124     void flushEnqueued()
00125     {
00126         if (mOccupancy > mFlushed)
00127         {
00128             mDispatcher.launchCopyKernel(mBuffer + mFlushed, mOccupancy - mFlushed, mStream);
00129             mFlushed = mOccupancy;
00130         }
00131     }
00132 
00133 private:
00134     PxGpuDispatcher&    mDispatcher;
00135     PxGpuCopyDesc*  mBuffer;
00136     CUstream        mStream;
00137     uint32_t        mReserved;
00138     uint32_t        mOccupancy;
00139     uint32_t        mFlushed;
00140 
00141     void operator=(const PxGpuCopyDescQueue&); // prevent a warning...
00142 };
00143 
00144 PX_POP_PACK
00145 
00146 } // end physx namespace
00147 
00148 #endif // PX_SUPPORT_GPU_PHYSX
00149 #endif // PXCUDACONTEXTMANAGER_PXGPUCOPYDESCQUEUE_H


Copyright © 2008-2017 NVIDIA Corporation, 2701 San Tomas Expressway, Santa Clara, CA 95050 U.S.A. All rights reserved. www.nvidia.com