WrappedAttribute.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "CoreMinimal.h"
#include "Templates/UnrealTypeTraits.h"
#include "UObject/Class.h"
 
namespace UE
{
    namespace Anim 
    {
        template<typename InAllocator>
        struct TWrappedAttribute
        {
            TWrappedAttribute() {}
 
            TWrappedAttribute(const UScriptStruct* InStruct)
            {
                Allocate(InStruct);
            }
                        
            template <typename Type>
            inline typename TEnableIf<TIsFundamentalType<Type>::Value, Type*>::Type GetPtr()
            {
                return (Type*)StructMemory.GetData();
            }
 
            template <typename Type>
            inline typename TEnableIf<!TIsFundamentalType<Type>::Value, Type*>::Type GetPtr()
            {
                const UScriptStruct* ScriptStruct = Type::StaticStruct();
                check(ScriptStruct && ScriptStruct->GetStructureSize() == StructMemory.Num());
                return (Type*)StructMemory.GetData();
            }
 
            template<typename Type>
            inline Type& GetRef() { return *GetPtr<Type>(); }
 
            template <typename Type>
            inline typename TEnableIf<TIsFundamentalType<Type>::Value, const Type*>::Type GetPtr() const
            {
                return (const Type*)StructMemory.GetData();
            }
 
            template <typename Type>
            inline typename TEnableIf<!TIsFundamentalType<Type>::Value, const Type*>::Type GetPtr() const
            {
                const UScriptStruct* ScriptStruct = Type::StaticStruct();
                check(ScriptStruct && ScriptStruct->GetStructureSize() == StructMemory.Num());
                return (const Type*)StructMemory.GetData();
            }
 
            template<typename Type>
            inline const Type& GetRef() const { return *GetPtr<Type>(); }
 
            template<typename AttributeType>
            inline void Allocate()
            {
                Allocate(AttributeType::StaticStruct());
            }
 
            inline void Allocate(const UScriptStruct* InStruct)
            {
                check(InStruct);
                if (InStruct)
                {
                    const int32 StructureSize = InStruct->GetStructureSize();
                    ensure(StructureSize > 0);
 
                    StructMemory.SetNum(StructureSize);
                    InStruct->InitializeStruct(GetPtr<void>());
                }
            }
        protected:
            TArray<uint8, InAllocator> StructMemory;
        };
    }
}