TypedElementObjectInterface.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "Elements/Framework/TypedElementHandle.h"
#include "Templates/Casts.h"
#include "Templates/SubclassOf.h"
#include "UObject/Interface.h"
#include "UObject/Object.h"
#include "UObject/ObjectMacros.h"
#include "UObject/UObjectGlobals.h"
 
#include "TypedElementObjectInterface.generated.h"
 
class UClass;
struct FFrame;
 
UINTERFACE(MinimalAPI, BlueprintType, meta = (CannotImplementInterfaceInBlueprint))
class UTypedElementObjectInterface : public UInterface
{
    GENERATED_BODY()
};
 
class ITypedElementObjectInterface
{
    GENERATED_BODY()
 
public:
    TYPEDELEMENTRUNTIME_API virtual UObject* GetObject(const FTypedElementHandle& InElementHandle);
 
    TYPEDELEMENTRUNTIME_API virtual UClass* GetObjectClass(const FTypedElementHandle& InElementHandle);
 
    template <class CastTo>
    CastTo* GetObjectAs(const FTypedElementHandle& InElementHandle)
    {
        return Cast<CastTo>(GetObject(InElementHandle));
    }
 
    template <class CastTo = UObject>
    CastTo* GetObjectAs(const FTypedElementHandle& InElementHandle, TSubclassOf<CastTo> TargetClass)
    {
        if (!TargetClass)
        {
            return nullptr;
        }
 
        UObject* ObjectPtr = GetObject(InElementHandle);
        if (ObjectPtr && ObjectPtr->IsA(TargetClass))
        {
            return (CastTo*)ObjectPtr;
        }
 
        return nullptr;
    }
 
 
    UFUNCTION(BlueprintCallable, Category="TypedElementInterfaces|Object")
    TYPEDELEMENTRUNTIME_API virtual UObject* GetObject(const FScriptTypedElementHandle& InElementHandle);
 
    UFUNCTION(BlueprintCallable, Category = "TypedElementInterfaces|Object")
    TYPEDELEMENTRUNTIME_API virtual UClass* GetObjectClass(const FScriptTypedElementHandle& InElementHandle);
};
 
template <>
struct TTypedElement<ITypedElementObjectInterface> : public TTypedElementBase<ITypedElementObjectInterface>
{
    UObject* GetObject() const { return InterfacePtr->GetObject(*this); }
    UClass* GetObjectClass() const { return InterfacePtr->GetObjectClass(*this); }
 
    template <class CastTo>
    CastTo* GetObjectAs() const { return InterfacePtr->GetObjectAs<CastTo>(*this); }
 
    template <class CastTo>
    CastTo* GetObjectAs(TSubclassOf<CastTo> TargetClass) const { return InterfacePtr->GetObjectAs<CastTo>(*this, TargetClass); }
};