extComputation.h

//
// Copyright 2017 Pixar
//
// Licensed under the terms set forth in the LICENSE.txt file available at
// https://openusd.org/license.
//
#ifndef PXR_IMAGING_HD_EXT_COMPUTATION_H
#define PXR_IMAGING_HD_EXT_COMPUTATION_H

#include "pxr/pxr.h"
#include "pxr/imaging/hd/api.h"
#include "pxr/imaging/hd/sceneDelegate.h"
#include "pxr/imaging/hd/sprim.h"

PXR_NAMESPACE_OPEN_SCOPE

class HdExtComputation : public HdSprim
{
public:
    HD_API
    HdExtComputation(SdfPath const &id);

    HD_API
    ~HdExtComputation() override;

    enum DirtyBits : HdDirtyBits {
        Clean                 = 0,
        DirtyInputDesc        = 1 << 0,  
        DirtyOutputDesc       = 1 << 1,  
        DirtyElementCount     = 1 << 2,  
        DirtySceneInput       = 1 << 3,  
        DirtyCompInput        = 1 << 4,  
        DirtyKernel           = 1 << 5,  

        DirtyDispatchCount    = 1 << 6,  

        AllDirty              = (DirtyInputDesc
                                |DirtyOutputDesc
                                |DirtyElementCount
                                |DirtySceneInput
                                |DirtyCompInput
                                |DirtyKernel
                                |DirtyDispatchCount)
    };

    HD_API
    void Sync(HdSceneDelegate *sceneDelegate,
              HdRenderParam   *renderParam,
              HdDirtyBits     *dirtyBits) override;

    HD_API
    HdDirtyBits GetInitialDirtyBitsMask() const override;

    HD_API
    size_t GetDispatchCount() const;

    HD_API
    size_t GetElementCount() const { return _elementCount; }

    HD_API
    TfTokenVector const & GetSceneInputNames() const {
        return _sceneInputNames;
    }

    HD_API
    TfTokenVector GetOutputNames() const;

    HD_API
    HdExtComputationInputDescriptorVector const &
    GetComputationInputs() const {
        return _computationInputs;
    }

    HD_API
    HdExtComputationOutputDescriptorVector const &
    GetComputationOutputs() const {
        return _computationOutputs;
    }

    HD_API
    const std::string& GetGpuKernelSource() const {
        return _gpuKernelSource.GetString();
    }

    HD_API
    bool IsInputAggregation() const;

protected:
    HD_API
    void
    _Sync(HdSceneDelegate *sceneDelegate,
          HdRenderParam   *renderParam,
          HdDirtyBits     *dirtyBits);

    HD_API
    static bool _IsEnabledSharedExtComputationData();

private:
    size_t                                 _dispatchCount;
    size_t                                 _elementCount;
    TfTokenVector                          _sceneInputNames;
    HdExtComputationInputDescriptorVector  _computationInputs;
    HdExtComputationOutputDescriptorVector _computationOutputs;
    TfToken                                _gpuKernelSource;

    // No default construction or copying
    HdExtComputation() = delete;
    HdExtComputation(const HdExtComputation &) = delete;
    HdExtComputation &operator =(const HdExtComputation &) = delete;

};

PXR_NAMESPACE_CLOSE_SCOPE

#endif // PXR_IMAGING_HD_EXT_COMPUTATION_H