pickFromRenderBufferTask.h

//
// Copyright 2019 Pixar
//
// Licensed under the terms set forth in the LICENSE.txt file available at
// https://openusd.org/license.
//
#ifndef PXR_IMAGING_HDX_PICK_FROM_RENDER_BUFFER_TASK_H
#define PXR_IMAGING_HDX_PICK_FROM_RENDER_BUFFER_TASK_H
 
#include "pxr/pxr.h"
#include "pxr/imaging/hdx/api.h"
#include "pxr/imaging/hdx/task.h"
 
#include "pxr/imaging/hd/camera.h"
#include "pxr/imaging/hd/renderBuffer.h"
#include "pxr/imaging/hdx/pickTask.h"
 
PXR_NAMESPACE_OPEN_SCOPE
 
struct HdxPickFromRenderBufferTaskParams
{
    HdxPickFromRenderBufferTaskParams()
        : primIdBufferPath()
        , instanceIdBufferPath()
        , elementIdBufferPath()
        , normalBufferPath()
        , depthBufferPath()
        , cameraId()
        , viewport()
    {}
 
    SdfPath primIdBufferPath;
    SdfPath instanceIdBufferPath;
    SdfPath elementIdBufferPath;
    SdfPath normalBufferPath;
    SdfPath depthBufferPath;
 
    // The id of the camera used to generate the id buffers.
    SdfPath cameraId;
 
    // The framing specifying how the camera frustum in mapped into the
    // render buffers.
    CameraUtilFraming framing;
    // Is application overriding the window policy of the camera.
    std::optional<CameraUtilConformWindowPolicy> overrideWindowPolicy;
 
    // The viewport of the camera used to generate the id buffers.
    // Only used if framing is invalid - for legacy clients.
    GfVec4d viewport;
};
 
class HdxPickFromRenderBufferTask : public HdxTask
{
public:
    using TaskParams = HdxPickFromRenderBufferTaskParams;
 
    HDX_API
    HdxPickFromRenderBufferTask(HdSceneDelegate *delegate, SdfPath const& id);
 
    HDX_API
    ~HdxPickFromRenderBufferTask() override;
 
    bool IsConverged() const override;
 
    HDX_API
    void Prepare(HdTaskContext* ctx,
                 HdRenderIndex* renderIndex) override;
 
    HDX_API
    void Execute(HdTaskContext* ctx) override;
 
protected:
    HDX_API
    void _Sync(HdSceneDelegate* delegate,
               HdTaskContext* ctx,
               HdDirtyBits* dirtyBits) override;
 
private:
    GfMatrix4d _ComputeProjectionMatrix() const;
 
    HdxPickFromRenderBufferTaskParams _params;
    HdxPickTaskContextParams _contextParams;
    // We need to cache a pointer to the render index so Execute() can
    // map prim ID to paths.
    HdRenderIndex *_index;
 
    HdRenderBuffer *_primId;
    HdRenderBuffer *_instanceId;
    HdRenderBuffer *_elementId;
    HdRenderBuffer *_normal;
    HdRenderBuffer *_depth;
    const HdCamera *_camera;
 
    bool _converged;
 
    HdxPickFromRenderBufferTask() = delete;
    HdxPickFromRenderBufferTask(const HdxPickFromRenderBufferTask &) = delete;
    HdxPickFromRenderBufferTask &operator =(
        const HdxPickFromRenderBufferTask &) = delete;
};
 
// VtValue requirements
HDX_API
std::ostream& operator<<(std::ostream& out, const HdxPickFromRenderBufferTaskParams& pv);
HDX_API
bool operator==(const HdxPickFromRenderBufferTaskParams& lhs,
                const HdxPickFromRenderBufferTaskParams& rhs);
HDX_API
bool operator!=(const HdxPickFromRenderBufferTaskParams& lhs,
                const HdxPickFromRenderBufferTaskParams& rhs);
 
PXR_NAMESPACE_CLOSE_SCOPE
 
#endif //PXR_IMAGING_HDX_PICK_FROM_RENDER_BUFFER_TASK_H