ContextObjectStore.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "Containers/Array.h"
#include "Templates/Casts.h"
#include "UObject/Object.h"
#include "UObject/ObjectMacros.h"
#include "UObject/ObjectPtr.h"
#include "UObject/UObjectGlobals.h"
 
#include "ContextObjectStore.generated.h"
 
class UClass;
 
UCLASS(Transient, MinimalAPI)
class UContextObjectStore : public UObject
{
    GENERATED_BODY()
 
public:
    template <typename TObjectType>
    TObjectType* FindContext() const
    {
        for (UObject* ContextObject : ContextObjects)
        {
            if (TObjectType* CastedObject = Cast<TObjectType>(ContextObject))
            {
                return CastedObject;
            }
        }
 
        if (UContextObjectStore* ParentStore = Cast<UContextObjectStore>(GetOuter()))
        {
            return ParentStore->FindContext<TObjectType>();
        }
 
        return nullptr;
    }
 
    INTERACTIVETOOLSFRAMEWORK_API UObject* FindContextByClass(UClass* InClass) const;
 
    INTERACTIVETOOLSFRAMEWORK_API bool AddContextObject(UObject* InContextObject);
 
    INTERACTIVETOOLSFRAMEWORK_API bool RemoveContextObject(UObject* InContextObject);
 
    INTERACTIVETOOLSFRAMEWORK_API bool RemoveContextObjectsOfType(const UClass* InClass);
 
    template <class TObjectType>
    bool RemoveContextObjectsOfType()
    {
        return RemoveContextObjectsOfType(TObjectType::StaticClass());
    }
 
    INTERACTIVETOOLSFRAMEWORK_API void Shutdown();
 
protected:
    UPROPERTY()
    TArray<TObjectPtr<UObject>> ContextObjects;
};