UObjectArchetypeHelper.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "CoreFwd.h"
#include "Misc/NotNull.h"
#include "UObject/Object.h"
#include "UObject/UObjectAnnotation.h"
 
class FObjectArchetypeHelper
{
public:
    class IObjectArchetypePolicy
    {
#if WITH_EDITOR
    public:
        virtual UObject* GetArchetype(const UObject* InObject) const = 0;
#endif
    };
 
#if WITH_EDITOR
private:
    // Restrict usage to FPropertyNode
    friend class FPropertyNode;
#endif
    COREUOBJECT_API static UObject* GetArchetype(const UObject* InObject, const IObjectArchetypePolicy* InPolicy);
};
 
#if WITH_EDITOR
 
struct FCacheArchetype
{
    bool IsDefault() const
    {
        return CachedArchetype == nullptr;
    }
 
    TObjectPtr<UObject> CachedArchetype = nullptr;
};
 
class FEditorCacheArchetypeManager
{
public:
    COREUOBJECT_API static FEditorCacheArchetypeManager& Get();
 
    void CacheArchetype(TNotNull<const UObject*> Object, UObject* Archetype = nullptr)
    {
        FCacheArchetype Annotation = ObjectCachedArchetypeAnnotations.GetAnnotation(Object);
        if (Annotation.CachedArchetype == nullptr)
        {
            Annotation.CachedArchetype = Archetype ? Archetype : Object->GetArchetype();
            ObjectCachedArchetypeAnnotations.AddAnnotation(Object, Annotation);
        }
    }
 
    void ResetCacheArchetype(TNotNull<const UObject*> Object)
    {
        ObjectCachedArchetypeAnnotations.RemoveAnnotation(Object);
    }
 
    UObject* GetCachedArchetype(TNotNull<const UObject*> Object)
    {
        return ObjectCachedArchetypeAnnotations.GetAnnotation(Object).CachedArchetype;
    }
 
private:
    FUObjectAnnotationSparse<FCacheArchetype, true/*bAutoRemove*/> ObjectCachedArchetypeAnnotations;
};
 
#endif // WITH_EDITOR