extComputation.h

//
// Copyright 2017 Pixar
//
// Licensed under the Apache License, Version 2.0 (the "Apache License")
// with the following modification; you may not use this file except in
// compliance with the Apache License and the following modification to it:
// Section 6. Trademarks. is deleted and replaced with:
//
// 6. Trademarks. This License does not grant permission to use the trade
//    names, trademarks, service marks, or product names of the Licensor
//    and its affiliates, except as required to comply with Section 4(c) of
//    the License and to reproduce the content of the NOTICE file.
//
// You may obtain a copy of the Apache License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the Apache License with the above modification is
// distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied. See the Apache License for the specific
// language governing permissions and limitations under the Apache 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"
#include "pxr/imaging/hd/types.h"
#include "pxr/usd/sdf/path.h"
#include "pxr/base/vt/value.h"
#include <vector>

PXR_NAMESPACE_OPEN_SCOPE

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

    HD_API
    virtual ~HdExtComputation() = default;

    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
    virtual void Sync(HdSceneDelegate *sceneDelegate,
                      HdRenderParam   *renderParam,
                      HdDirtyBits     *dirtyBits) override;

    HD_API
    virtual 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; }

    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;
    std::string                            _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