PxGpuTask.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_GPU_TASK_H
#define PX_GPU_TASK_H

#include "pxtask/PxTask.h"
#include "pxtask/PxGpuDispatcher.h"

#ifndef PX_DOXYGEN
namespace physx
{
#endif

PX_PUSH_PACK_DEFAULT

struct PxGpuTaskHint
{
    enum Enum
    {
        HostToDevice,
        Kernel,
        DeviceToHost,

        NUM_GPU_TASK_HINTS
    };
};

class PxGpuTask : public PxTask
{
public:
    PxGpuTask() : mComp(NULL) {}

    virtual bool    launchInstance(CUstream stream, int kernelIndex) = 0;

    virtual PxGpuTaskHint::Enum getTaskHint() const = 0;

    void setCompletionTask(PxBaseTask& task)
    {
        mComp = &task;
    }

    void release()
    {
        if (mComp)
        {
            mComp->removeReference();
            mComp = NULL;
        }
        PxTask::release();
    }

protected:
    PxBaseTask* mComp;
};

PX_POP_PACK

#ifndef PX_DOXYGEN
} // end physx namespace
#endif

#endif