renderParam.h

//
// Copyright 2021 Pixar
//
// Licensed under the terms set forth in the LICENSE.txt file available at
// https://openusd.org/license.
//
#ifndef PXR_IMAGING_PLUGIN_HD_ST_RENDER_PARAM_H
#define PXR_IMAGING_PLUGIN_HD_ST_RENDER_PARAM_H
 
#include "pxr/pxr.h"
#include "pxr/imaging/hd/renderDelegate.h"
#include "pxr/imaging/hdSt/api.h"
 
#include <atomic>
#include <shared_mutex>
 
PXR_NAMESPACE_OPEN_SCOPE
 
class HdStRenderParam final : public HdRenderParam
{
public:
    HdStRenderParam();
    ~HdStRenderParam() override;
 
    // ---------------------------------------------------------------------- //
    // ---------------------------------------------------------------------- //
    HDST_API
    void MarkDrawBatchesDirty();
 
    HDST_API
    unsigned int GetDrawBatchesVersion() const;
 
    HDST_API
    void MarkMaterialTagsDirty();
 
    HDST_API
    unsigned int GetMaterialTagsVersion() const;
 
    HDST_API
    void MarkGeomSubsetDrawItemsDirty();
 
    HDST_API
    unsigned int GetGeomSubsetDrawItemsVersion() const;
 
    // ---------------------------------------------------------------------- //
    // ---------------------------------------------------------------------- /
 
    HDST_API
    bool HasMaterialTag(const TfToken &materialTag) const;
 
    HDST_API
    void IncreaseMaterialTagCount(const TfToken &materialTag);
    
    HDST_API
    void DecreaseMaterialTagCount(const TfToken &materialTag);
 
    // ---------------------------------------------------------------------- //
    // ---------------------------------------------------------------------- /
 
    HDST_API
    bool HasAnyRenderTag(const TfTokenVector &renderTags) const;
 
    HDST_API
    void IncreaseRenderTagCount(const TfToken &renderTag);
    
    HDST_API
    void DecreaseRenderTagCount(const TfToken &renderTag);
 
    // ---------------------------------------------------------------------- //
    // ---------------------------------------------------------------------- //
    void SetGarbageCollectionNeeded() {
        _needsGarbageCollection = true;
    }
 
    void ClearGarbageCollectionNeeded() {
        _needsGarbageCollection = false;
    }
 
    bool IsGarbageCollectionNeeded() const {
        return _needsGarbageCollection;
    }
    
private:
    typedef std::unordered_map<TfToken, std::atomic_int, TfHash> _TagToCountMap;
 
    void _AdjustTagCount(
        std::shared_timed_mutex *mutex,
        _TagToCountMap *tagToCountMap,
        const TfToken &tag,
        const int increment);
 
    bool _HasTag(
        std::shared_timed_mutex *mutex,
        const _TagToCountMap *tagToCountMap,
        const TfToken &tag) const;
 
    std::atomic_uint _drawBatchesVersion;
    std::atomic_uint _materialTagsVersion;
    std::atomic_uint _geomSubsetDrawItemsVersion;
    bool _needsGarbageCollection; // Doesn't need to be atomic since parallel
                                  // sync might only set it (and not clear).
 
    mutable std::shared_timed_mutex _materialTagToCountMutex;
    _TagToCountMap _materialTagToCount;
 
    mutable std::shared_timed_mutex _renderTagToCountMutex;
    _TagToCountMap _renderTagToCount;
};
 
PXR_NAMESPACE_CLOSE_SCOPE
 
#endif // PXR_IMAGING_PLUGIN_HD_ST_RENDER_PARAM_H