PxCudaContextManager.h

Go to the documentation of this file.

/*
 * Copyright (c) 2008-2015, NVIDIA CORPORATION.  All rights reserved.
 *
 * NVIDIA CORPORATION and its licensors retain all intellectual property
 * and proprietary rights in and to this software, 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.
 */


#ifndef PX_CUDA_CONTEXT_MANAGER_H
#define PX_CUDA_CONTEXT_MANAGER_H

#include "pxtask/PxCudaMemoryManager.h"

/* Forward decl to avoid inclusion of cuda.h */
typedef struct CUctx_st *CUcontext;
typedef struct CUgraphicsResource_st *CUgraphicsResource;

#ifndef PX_DOXYGEN
namespace physx
{
#endif

class PxProfileZoneManager;

struct PxCudaInteropMode
{
    enum Enum
    {
        NO_INTEROP = 0,
        D3D10_INTEROP,
        D3D11_INTEROP,
        OGL_INTEROP,

        COUNT
    };
};


class PxCudaContextManagerDesc
{
public:
    CUcontext            *ctx;

    void                 *graphicsDevice;

#if PX_SUPPORT_GPU_PHYSX

    const char*          appGUID;
#endif

    PxCudaInteropMode::Enum interopMode;


    PxU32       memoryBaseSize[PxCudaBufferMemorySpace::COUNT];

    PxU32       memoryPageSize[PxCudaBufferMemorySpace::COUNT];

    PxU32       maxMemorySize[PxCudaBufferMemorySpace::COUNT];

    PX_INLINE PxCudaContextManagerDesc()
    {
        ctx = NULL;
        interopMode = PxCudaInteropMode::NO_INTEROP;
        graphicsDevice = 0;
#if PX_SUPPORT_GPU_PHYSX
        appGUID  = NULL;
#endif
        for(PxU32 i = 0; i < PxCudaBufferMemorySpace::COUNT; i++)
        {
            memoryBaseSize[i] = 0;
            memoryPageSize[i] = 2 * 1024*1024;
            maxMemorySize[i] = PX_MAX_U32;
        }
    }
};


class PxCudaContextManager
{
public:
    virtual void acquireContext() = 0;

    virtual void releaseContext() = 0;

    virtual PxCudaMemoryManager *getMemoryManager() = 0;

    virtual class PxGpuDispatcher *getGpuDispatcher() = 0;

    virtual bool contextIsValid() const = 0;

    /* Query CUDA context and device properties, without acquiring context */

    virtual bool supportsArchSM10() const = 0;  
    virtual bool supportsArchSM11() const = 0;  
    virtual bool supportsArchSM12() const = 0;  
    virtual bool supportsArchSM13() const = 0;  
    virtual bool supportsArchSM20() const = 0;  
    virtual bool supportsArchSM30() const = 0;  
    virtual bool supportsArchSM35() const = 0;  
    virtual bool supportsArchSM50() const = 0;  
    virtual bool supportsArchSM52() const = 0;  
    virtual bool isIntegrated() const = 0;      
    virtual bool canMapHostMemory() const = 0;  
    virtual int  getDriverVersion() const = 0;  
    virtual size_t getDeviceTotalMemBytes() const = 0; 
    virtual int getMultiprocessorCount() const = 0; 
    virtual unsigned int getClockRate() const = 0; 
    virtual int  getSharedMemPerBlock() const = 0; 
    virtual unsigned int getMaxThreadsPerBlock() const = 0; 
    virtual const char *getDeviceName() const = 0; 
    virtual PxCudaInteropMode::Enum getInteropMode() const = 0; 

    virtual void setUsingConcurrentStreams(bool) = 0; 
    virtual bool getUsingConcurrentStreams() const = 0; 
    /* End query methods that don't require context to be acquired */

    virtual bool registerResourceInCudaGL(CUgraphicsResource &resource, PxU32 buffer) = 0;

    virtual bool registerResourceInCudaD3D(CUgraphicsResource &resource, void *resourcePointer) = 0;

    virtual bool unregisterResourceInCuda(CUgraphicsResource resource) = 0;

    virtual int usingDedicatedPhysXGPU() const = 0;

    virtual void release() = 0;

protected:

    virtual ~PxCudaContextManager() {}
};

class PxScopedCudaLock
{
public:
    PxScopedCudaLock(PxCudaContextManager& ctx) : mCtx(&ctx)
    {
        mCtx->acquireContext();
    }

    ~PxScopedCudaLock()
    {
        mCtx->releaseContext();
    }

protected:

    PxCudaContextManager* mCtx;
};

#if PX_SUPPORT_GPU_PHYSX

int PxGetSuggestedCudaDeviceOrdinal(PxErrorCallback& errc);

PxCudaContextManager* PxCreateCudaContextManager(PxFoundation& foundation, const PxCudaContextManagerDesc& desc, physx::PxProfileZoneManager* mgr);

void* PxLoadPhysxGPUModule(const char* appGUID = NULL);
#endif

#ifndef PX_DOXYGEN
} // end physx namespace
#endif

#endif