primManagingSceneIndexObserver.h

Go to the documentation of this file.

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


#include "pxr/imaging/hdsi/api.h"

#include "pxr/imaging/hd/dataSource.h"
#include "pxr/imaging/hd/sceneIndexObserver.h"

PXR_NAMESPACE_OPEN_SCOPE

#define HDSI_PRIM_MANAGING_SCENE_INDEX_OBSERVER_TOKENS \
    (primFactory)

TF_DECLARE_PUBLIC_TOKENS(HdsiPrimManagingSceneIndexObserverTokens, HDSI_API,
                         HDSI_PRIM_MANAGING_SCENE_INDEX_OBSERVER_TOKENS);

TF_DECLARE_REF_PTRS(HdsiPrimManagingSceneIndexObserver);

class HdsiPrimManagingSceneIndexObserver
    : public HdSceneIndexObserver, public TfRefBase
{
public:
    class PrimBase
    {
    public:
        HDSI_API
        virtual ~PrimBase();
        
        void Dirty(
            const DirtiedPrimEntry &entry,
            const HdsiPrimManagingSceneIndexObserver * observer)
        {
            // NVI so that we can later implement things like a
            // mutex guarding against Dirty from several threads
            // on the same prim.
            _Dirty(entry, observer);
        }

    private:
        virtual void _Dirty(
            const DirtiedPrimEntry &entry,
            const HdsiPrimManagingSceneIndexObserver * observer) = 0;
    };
    using PrimBaseHandle = std::shared_ptr<PrimBase>;

    class PrimFactoryBase 
    {
    public:
        HDSI_API
        virtual ~PrimFactoryBase();
        virtual PrimBaseHandle CreatePrim(
            const AddedPrimEntry &entry,
            const HdsiPrimManagingSceneIndexObserver * observer) = 0;
    };
    using PrimFactoryBaseHandle = std::shared_ptr<PrimFactoryBase>;

    static HdsiPrimManagingSceneIndexObserverRefPtr New(
            HdSceneIndexBaseRefPtr const &sceneIndex,
            HdContainerDataSourceHandle const &inputArgs) {
        return TfCreateRefPtr(
            new HdsiPrimManagingSceneIndexObserver(
                sceneIndex, inputArgs));
    }

    HDSI_API
    ~HdsiPrimManagingSceneIndexObserver() override;

    const HdSceneIndexBaseRefPtr &GetSceneIndex() const {
        return _sceneIndex;
    }

    HDSI_API
    const PrimBaseHandle &GetPrim(const SdfPath &primPath) const;

    template<typename PrimType>
    std::shared_ptr<PrimType> GetTypedPrim(const SdfPath &primPath) const
    {
        return std::dynamic_pointer_cast<PrimType>(GetPrim(primPath));
    }

protected:
    void PrimsAdded(
        const HdSceneIndexBase &sender,
        const AddedPrimEntries &entries) override;

    void PrimsDirtied(
        const HdSceneIndexBase &sender,
        const DirtiedPrimEntries &entries) override;

    void PrimsRemoved(
        const HdSceneIndexBase &sender,
        const RemovedPrimEntries &entries) override;

    void PrimsRenamed(
        const HdSceneIndexBase &sender,
        const RenamedPrimEntries &entries) override;

private:
    HDSI_API
    HdsiPrimManagingSceneIndexObserver(
        HdSceneIndexBaseRefPtr const &sceneIndex,
        HdContainerDataSourceHandle const &inputArgs);

    HdSceneIndexBaseRefPtr const _sceneIndex;
    PrimFactoryBaseHandle const _primFactory;

    // _prims defined after _primFactory so that all prims
    // are destroyed before the handle to _primFactory is
    // released.
    std::map<SdfPath, PrimBaseHandle> _prims;
};

PXR_NAMESPACE_CLOSE_SCOPE

#endif