AnimationDataSource.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "Containers/Map.h"
#include "CoreMinimal.h"
#include "Templates/Casts.h"
#include "UObject/Interface.h"
#include "UObject/NameTypes.h"
#include "UObject/Object.h"
#include "UObject/ObjectMacros.h"
#include "UObject/UObjectGlobals.h"
#include "UObject/WeakObjectPtrTemplates.h"
 
#include "AnimationDataSource.generated.h"
 
class UClass;
 
UCLASS(MinimalAPI)
class UAnimationDataSourceRegistry : public UObject
{
    GENERATED_BODY()
 
public:
 
    ANIMATIONCORE_API bool RegisterDataSource(const FName& InName, UObject* InDataSource);
 
    ANIMATIONCORE_API bool UnregisterDataSource(const FName& InName);
 
    ANIMATIONCORE_API bool ContainsSource(const FName& InName) const;
 
    ANIMATIONCORE_API UObject* RequestSource(const FName& InName, UClass* InExpectedClass) const;
 
    template<class T>
    T* RequestSource(const FName& InName) const
    {
        UObject* DataSource = RequestSource(InName, T::StaticClass());
        if (DataSource == nullptr)
        {
            return nullptr;
        }
        return Cast<T>(DataSource);
    }
 
private:
 
    UPROPERTY(transient)
    TMap<FName, TWeakObjectPtr<UObject>> DataSources;
 
    ANIMATIONCORE_API void ClearInvalidDataSource();
};