changeTracker.h

//
// Copyright 2016 Pixar
//
// Licensed under the terms set forth in the LICENSE.txt file available at
// https://openusd.org/license.
//
#ifndef PXR_IMAGING_HD_CHANGE_TRACKER_H
#define PXR_IMAGING_HD_CHANGE_TRACKER_H
 
#include "pxr/pxr.h"
#include "pxr/imaging/hd/api.h"
#include "pxr/imaging/hd/version.h"
#include "pxr/imaging/hd/rprimCollection.h"
#include "pxr/imaging/hd/types.h"
#include "pxr/usd/sdf/path.h"
#include "pxr/base/tf/hashmap.h"
 
#include <tbb/concurrent_hash_map.h>
#include <atomic>
 
PXR_NAMESPACE_OPEN_SCOPE
 
class HdRetainedSceneIndex;
 
class HdChangeTracker 
{
public:
 
    // Common dirty bits for Rprims
    // XXX: Move this to HdRprim
    enum RprimDirtyBits : HdDirtyBits {
        Clean                       = 0,
        InitRepr                    = 1 << 0,
        Varying                     = 1 << 1,
        AllDirty                    = ~Varying,
        DirtyPrimID                 = 1 << 2,
        DirtyExtent                 = 1 << 3,
        DirtyDisplayStyle           = 1 << 4,
        DirtyPoints                 = 1 << 5,
        DirtyPrimvar                = 1 << 6,
        DirtyMaterialId             = 1 << 7,
        DirtyTopology               = 1 << 8,
        DirtyTransform              = 1 << 9,
        DirtyVisibility             = 1 << 10,
        DirtyNormals                = 1 << 11,
        DirtyDoubleSided            = 1 << 12,
        DirtyCullStyle              = 1 << 13,
        DirtySubdivTags             = 1 << 14,
        DirtyWidths                 = 1 << 15,
        DirtyInstancer              = 1 << 16,
        DirtyInstanceIndex          = 1 << 17,
        DirtyRepr                   = 1 << 18,
        DirtyRenderTag              = 1 << 19,
        DirtyComputationPrimvarDesc = 1 << 20,
        DirtyCategories             = 1 << 21,
        DirtyVolumeField            = 1 << 22,
        AllSceneDirtyBits           = ((1<<23) - 1),
 
        NewRepr                     = 1 << 23,
 
        CustomBitsBegin             = 1 << 24,
        CustomBitsEnd               = 1 << 30,
        CustomBitsMask              = 0x7f << 24,
    };
 
    // InstancerDirtybits are a subset of rprim dirty bits right now:
    // DirtyPrimvar, DirtyTransform, DirtyInstanceIndex, DirtyInstancer.
 
    // Dirty bits for Tasks
    // XXX: Move this to HdTask
    enum TaskDirtyBits : HdDirtyBits {
        DirtyParams           = 1 << 2,
        DirtyCollection       = 1 << 3,
        DirtyRenderTags       = 1 << 4,
    };
 
    HD_API
    HdChangeTracker();
    HD_API
    virtual ~HdChangeTracker();
 
    // ---------------------------------------------------------------------- //
    // ---------------------------------------------------------------------- //
 
    HD_API
    void RprimInserted(SdfPath const& id, HdDirtyBits initialDirtyState);
 
    HD_API
    void RprimRemoved(SdfPath const& id);
 
    // ---------------------------------------------------------------------- //
    // ---------------------------------------------------------------------- //
 
    HD_API
    HdDirtyBits GetRprimDirtyBits(SdfPath const& id) const;
 
    HD_API
    void MarkRprimDirty(SdfPath const& id, HdDirtyBits bits=AllDirty);
 
    HD_API
    void MarkRprimClean(SdfPath const& id, HdDirtyBits newBits=Clean);
 
    HD_API
    void MarkPrimvarDirty(SdfPath const& id, TfToken const& name);
 
    HD_API
    void MarkAllRprimsDirty(HdDirtyBits bits);
 
    HD_API
    void ResetVaryingState();
 
    HD_API
    void ResetRprimVaryingState(SdfPath const& id);
 
 
    // ---------------------------------------------------------------------- //
 
    HD_API
    bool IsRprimDirty(SdfPath const& id);
    
    HD_API
    bool IsExtentDirty(SdfPath const& id);
 
    HD_API
    bool IsDisplayStyleDirty(SdfPath const& id);
 
    HD_API
    bool IsPrimvarDirty(SdfPath const& id, TfToken const& name);
 
    HD_API
    bool IsAnyPrimvarDirty(SdfPath const& id);
 
    HD_API
    bool IsTopologyDirty(SdfPath const& id);
 
    HD_API
    bool IsDoubleSidedDirty(SdfPath const& id);
 
    HD_API
    bool IsCullStyleDirty(SdfPath const& id);
 
    HD_API
    bool IsSubdivTagsDirty(SdfPath const& id);
 
    HD_API
    bool IsTransformDirty(SdfPath const& id);
 
    HD_API
    bool IsVisibilityDirty(SdfPath const& id);
    
    HD_API
    bool IsPrimIdDirty(SdfPath const& id);
 
    static bool IsDirty(HdDirtyBits dirtyBits) {
        return (dirtyBits & AllDirty) != 0;
    }
 
    static bool IsClean(HdDirtyBits dirtyBits) {
        return (dirtyBits & AllDirty) == 0;
    }
 
    static bool IsVarying(HdDirtyBits dirtyBits) {
        return (dirtyBits & Varying) != 0;
    }
 
    HD_API
    static bool IsExtentDirty(HdDirtyBits dirtyBits, SdfPath const& id);
 
    HD_API
    static bool IsDisplayStyleDirty(HdDirtyBits dirtyBits, SdfPath const& id);
 
    HD_API
    static bool IsSubdivTagsDirty(HdDirtyBits dirtyBits, SdfPath const& id);
 
    HD_API
    static bool IsPrimvarDirty(HdDirtyBits dirtyBits, SdfPath const& id,
                               TfToken const& name);
 
    HD_API
    static bool IsAnyPrimvarDirty(HdDirtyBits dirtyBits, SdfPath const& id);
 
    HD_API
    static bool IsTopologyDirty(HdDirtyBits dirtyBits, SdfPath const& id);
 
    HD_API
    static bool IsDoubleSidedDirty(HdDirtyBits dirtyBits, SdfPath const& id);
 
    HD_API
    static bool IsCullStyleDirty(HdDirtyBits dirtyBits, SdfPath const& id);
 
    HD_API
    static bool IsTransformDirty(HdDirtyBits dirtyBits, SdfPath const& id);
 
    HD_API
    static bool IsVisibilityDirty(HdDirtyBits dirtyBits, SdfPath const& id);
 
    HD_API
    static bool IsPrimIdDirty(HdDirtyBits dirtyBits, SdfPath const& id);
 
    HD_API
    static bool IsInstancerDirty(HdDirtyBits dirtyBits, SdfPath const& id);
 
    HD_API
    static bool IsInstanceIndexDirty(HdDirtyBits dirtyBits, SdfPath const& id);
 
    HD_API
    static bool IsReprDirty(HdDirtyBits dirtyBits, SdfPath const &id);
 
    // ---------------------------------------------------------------------- //
 
    HD_API
    static void MarkPrimvarDirty(HdDirtyBits *dirtyBits, TfToken const &name);
 
    // ---------------------------------------------------------------------- //
    // ---------------------------------------------------------------------- //
 
    HD_API
    void TaskInserted(SdfPath const& id, HdDirtyBits initialDirtyState);
 
    HD_API
    void TaskRemoved(SdfPath const& id);
 
    HD_API
    void MarkTaskDirty(SdfPath const& id, HdDirtyBits bits=AllDirty);
 
    HD_API
    HdDirtyBits GetTaskDirtyBits(SdfPath const& id);
 
    HD_API
    void MarkTaskClean(SdfPath const& id, HdDirtyBits newBits=Clean);
 
    HD_API
    unsigned GetRenderTagVersion() const;
 
    HD_API
    unsigned GetTaskRenderTagsVersion() const;
 
    // ---------------------------------------------------------------------- //
    // ---------------------------------------------------------------------- //
 
    HD_API
    void InstancerInserted(SdfPath const& id, HdDirtyBits initialDirtyState);
 
    HD_API
    void InstancerRemoved(SdfPath const& id);
 
    HD_API
    HdDirtyBits GetInstancerDirtyBits(SdfPath const& id);
 
    HD_API
    void MarkInstancerDirty(SdfPath const& id, HdDirtyBits bits=AllDirty);
 
    HD_API
    void MarkInstancerClean(SdfPath const& id, HdDirtyBits newBits=Clean);
 
    HD_API
    void AddInstancerRprimDependency(SdfPath const& instancerId,
                                     SdfPath const& rprimId);
 
    HD_API
    void RemoveInstancerRprimDependency(SdfPath const& instancerId,
                                        SdfPath const& rprimId);
 
    HD_API
    void AddInstancerInstancerDependency(SdfPath const& parentInstancerId,
                                         SdfPath const& instancerId);
 
    HD_API
    void RemoveInstancerInstancerDependency(SdfPath const& parentInstancerId,
                                            SdfPath const& instancerId);
 
    // ---------------------------------------------------------------------- //
    // ---------------------------------------------------------------------- //
 
    HD_API
    void SprimInserted(SdfPath const& id, HdDirtyBits initialDirtyState);
 
    HD_API
    void SprimRemoved(SdfPath const& id);
 
    HD_API
    HdDirtyBits GetSprimDirtyBits(SdfPath const& id);
 
    HD_API
    void MarkSprimDirty(SdfPath const& id, HdDirtyBits bits);
 
    HD_API
    void MarkSprimClean(SdfPath const& id, HdDirtyBits newBits=Clean);
 
    HD_API
    void AddInstancerSprimDependency(SdfPath const& instancerId,
                                     SdfPath const& sprimId);
 
    HD_API
    void RemoveInstancerSprimDependency(SdfPath const& instancerId,
                                        SdfPath const& sprimId);
 
    HD_API
    void AddSprimSprimDependency(SdfPath const& parentSprimId,
                                 SdfPath const& sprimId);
 
    HD_API
    void RemoveSprimSprimDependency(SdfPath const& parentSprimId,
                                    SdfPath const& sprimId);
 
    HD_API
    void AddSprimRprimDependency(SdfPath const& sprimId,
                                 SdfPath const& rprimId);
 
    HD_API
    void RemoveSprimRprimDependency(SdfPath const& sprimId,
                                    SdfPath const& rprimId);
 
    HD_API
    void RemoveSprimFromSprimSprimDependencies(SdfPath const& sprimId);
 
    // ---------------------------------------------------------------------- //
    // ---------------------------------------------------------------------- //
 
    HD_API
    void BprimInserted(SdfPath const& id, HdDirtyBits initialDirtyState);
 
    HD_API
    void BprimRemoved(SdfPath const& id);
 
    HD_API
    HdDirtyBits GetBprimDirtyBits(SdfPath const& id);
 
    HD_API
    void MarkBprimDirty(SdfPath const& id, HdDirtyBits bits);
 
    HD_API
    void MarkBprimClean(SdfPath const& id, HdDirtyBits newBits=Clean);
 
    // ---------------------------------------------------------------------- //
    // ---------------------------------------------------------------------- //
 
    HD_API
    void AddCollection(TfToken const& collectionName);
 
    HD_API
    void MarkCollectionDirty(TfToken const& collectionName);
 
    HD_API
    unsigned GetCollectionVersion(TfToken const& collectionName) const;
 
    HD_API
    unsigned GetVisibilityChangeCount() const;
 
    HD_API
    unsigned GetInstanceIndicesChangeCount() const;
 
    unsigned GetVaryingStateVersion() const {
        return _varyingStateVersion;
    }
 
    // ---------------------------------------------------------------------- //
    // ---------------------------------------------------------------------- //
 
    unsigned GetRprimIndexVersion() const {
        return _rprimIndexVersion;
    }
 
    unsigned GetSprimIndexVersion() const {
        return _sprimIndexVersion;
    }
 
    unsigned GetBprimIndexVersion() const {
        return _bprimIndexVersion;
    }
 
    unsigned GetInstancerIndexVersion() const {
        return _instancerIndexVersion;
    }
 
 
    unsigned GetSceneStateVersion() const {
        return _sceneStateVersion;
    }
 
    // ---------------------------------------------------------------------- //
    // ---------------------------------------------------------------------- //
 
    HD_API
    void AddState(TfToken const& name);
 
    HD_API
    void MarkStateDirty(TfToken const& name);
 
    HD_API
    unsigned GetStateVersion(TfToken const &name) const;
 
    // ---------------------------------------------------------------------- //
    // ---------------------------------------------------------------------- //
    HD_API
    static std::string StringifyDirtyBits(HdDirtyBits dirtyBits);
 
    HD_API
    static void DumpDirtyBits(HdDirtyBits dirtyBits);
 
 
private:
 
    // Don't allow copies
    HdChangeTracker(const HdChangeTracker &) = delete;
    HdChangeTracker &operator=(const HdChangeTracker &) = delete;
 
 
    static void _LogCacheAccess(TfToken const& cacheName,
                                SdfPath const& id, bool hit);
 
    typedef TfHashMap<SdfPath, HdDirtyBits, SdfPath::Hash> _IDStateMap;
    typedef TfHashMap<TfToken, int, TfToken::HashFunctor> _CollectionStateMap;
    typedef TfHashMap<TfToken, unsigned, TfToken::HashFunctor> _GeneralStateMap;
 
    struct _PathHashCompare {
        static bool     equal(const SdfPath& a, const SdfPath& b)
                        { return a == b; }
 
        static size_t   hash(const SdfPath& path)
                        { return hash_value(path); }
    };
    typedef tbb::concurrent_hash_map<SdfPath, SdfPathSet, _PathHashCompare>
        _DependencyMap;
 
    // Core dirty state.
    _IDStateMap _rprimState;
    _IDStateMap _instancerState;
    _IDStateMap _taskState;
    _IDStateMap _sprimState;
    _IDStateMap _bprimState;
    _GeneralStateMap _generalState;
 
    // Collection versions / state.
    _CollectionStateMap _collectionState;
 
    // Provides reverse-association between instancers and the child
    // instancers/prims that use them.
    _DependencyMap _instancerRprimDependencies;
    _DependencyMap _instancerSprimDependencies;
    _DependencyMap _instancerInstancerDependencies;
 
    // Provides forward and reverse-association between sprims and the child
    // sprims that reference them. For example, a light prim (child) who needs 
    // to know when its light filter (parent) is modified.
    // Maps parent sprim to child sprim.
    _DependencyMap _sprimSprimTargetDependencies;
    // Maps child sprim to parent sprim.
    _DependencyMap _sprimSprimSourceDependencies;
 
    _DependencyMap _sprimRprimTargetDependencies;
    
    // Dependency map helpers
    void _AddDependency(_DependencyMap &depMap,
        SdfPath const& parent, SdfPath const& child);
    void _RemoveDependency(_DependencyMap &depMap,
        SdfPath const& parent, SdfPath const& child);
 
    // Typically the Rprims that get marked dirty per update iteration end up
    // being a stable set of objects; to leverage this fact, we require the
    // delegate notify the change tracker when that state changes, which bumps
    // the varyingStateVersion, which triggers downstream invalidation.
    unsigned _varyingStateVersion;
 
    // Tracks changes (insertions/removals) of prims in the render index.
    // This is used to indicating that cached gather operations need to be
    // re-evaluated, such as dirty lists or batch building.
    unsigned _rprimIndexVersion;
    unsigned _sprimIndexVersion;
    unsigned _bprimIndexVersion;
    unsigned _instancerIndexVersion;
 
    // The following tracks any changes of state.  As a result it is very broad.
    // The use case to detect, when no changes have been made, as to
    // avoid the need to sync or reset progressive renderers.
    unsigned _sceneStateVersion;
 
    // Used to detect that visibility changed somewhere in the render index.
    unsigned _visChangeCount;
 
    // Used to detect that instance indices changed somewhere in the render index.
    unsigned _instanceIndicesChangeCount;
 
    // Used to detect changes to the render tag opinion of rprims.
    unsigned _rprimRenderTagVersion;
 
    // Used to detect changes to the render tags opinion of tasks.
    unsigned _taskRenderTagsVersion;
 
    // Allow HdRenderIndex to provide a scene index to forward dirty
    // information. This is necessary to accommodate legacy HdSceneDelegate
    // based applications that rely on the HdChangeTracker for invalidating
    // state on Hydra prims.
    friend class HdRenderIndex;
    friend class HdDirtyList;
    // Does not take ownership. The HdRenderIndex manages the lifetime of this
    // scene index.
    HdRetainedSceneIndex * _emulationSceneIndex;
    void _SetTargetSceneIndex(HdRetainedSceneIndex *emulationSceneIndex);
 
    // Private methods which implement the behaviors of their public
    // equivalents. The public versions check to see if legacy emulation is
    // active. If so, they dirty the HdRetainedSceneIndex member instead
    // of directly acting. If legacy render delegate emulation is active, these
    // will eventually make their way back to the private methods via
    // HdSceneIndexAdapterSceneDelegate. This prevents dirtying cycles while
    // allowing single HdRenderIndex/HdChangeTracker instances to be used for
    // both ends of emulation.
    friend class HdSceneIndexAdapterSceneDelegate;
    void _MarkRprimDirty(SdfPath const& id, HdDirtyBits bits=AllDirty);
    void _MarkSprimDirty(SdfPath const& id, HdDirtyBits bits=AllDirty);
    void _MarkBprimDirty(SdfPath const& id, HdDirtyBits bits=AllDirty);
    void _MarkInstancerDirty(SdfPath const& id, HdDirtyBits bits=AllDirty);
    void _MarkTaskDirty(SdfPath const& id, HdDirtyBits bits=AllDirty);
};
 
 
PXR_NAMESPACE_CLOSE_SCOPE
 
#endif //PXR_IMAGING_HD_CHANGE_TRACKER_H