StaticMeshVertexData.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "CoreMinimal.h"
#include "Containers/DynamicRHIResourceArray.h"
#include "Rendering/StaticMeshVertexDataInterface.h"
 
template<typename VertexDataType>
class TStaticMeshVertexData :
    public FStaticMeshVertexDataInterface
{
    using FVertexResourceArray = TResourceArray<VertexDataType, VERTEXBUFFER_ALIGNMENT>;
    FVertexResourceArray Data;
public:
 
    TStaticMeshVertexData(bool InNeedsCPUAccess=false)
        : Data(InNeedsCPUAccess)
    {
    }
 
    void ResizeBuffer(uint32 NumVertices, EResizeBufferFlags BufferFlags = EResizeBufferFlags::None) override
    {
        if ((uint32)Data.Num() < NumVertices)
        {
            // Enlarge the array.
            if (!EnumHasAnyFlags(BufferFlags, EResizeBufferFlags::AllowSlackOnGrow))
            {
                Data.Reserve(NumVertices);
            }
 
            Data.AddUninitialized(NumVertices - Data.Num());
        }
        else if ((uint32)Data.Num() > NumVertices)
        {
            // Shrink the array.
            EAllowShrinking AllowShinking = EnumHasAnyFlags(BufferFlags, EResizeBufferFlags::AllowSlackOnReduce) ? EAllowShrinking::No : EAllowShrinking::Yes;
            Data.RemoveAt(NumVertices, Data.Num() - NumVertices, AllowShinking);
        }
    }
 
    void Empty(uint32 NumVertices) override
    {
        Data.Empty(NumVertices);
    }
 
    bool IsValidIndex(uint32 Index) override
    {
        return Data.IsValidIndex(Index);
    }
 
    uint32 GetStride() const override
    {
        return sizeof(VertexDataType);
    }
    uint8* GetDataPointer() override
    {
        return (uint8*)Data.GetData();
    }
 
    FResourceArrayInterface* GetResourceArray() override
    {
        return &Data;
    }
 
    const FResourceArrayInterface* GetResourceArray() const
    {
        return &Data;
    }
 
    void Serialize(FArchive& Ar, bool bForcePerElementSerialization = false) override
    {
        Data.FVertexResourceArray::BulkSerialize(Ar, bForcePerElementSerialization);
    }
    void Assign(const TArray<VertexDataType>& Other)
    {
        ResizeBuffer(Other.Num());
        if (Other.Num())
        {
            memcpy(GetDataPointer(), &Other[0], Other.Num() * sizeof(VertexDataType));
        }
    }
 
    SIZE_T GetResourceSize() const override
    {
        return Data.GetAllocatedSize();
    }
 
    virtual int32 Num() const override
    {
        return Data.Num();
    }
 
    bool GetAllowCPUAccess() const override
    {
        return Data.GetAllowCPUAccess();
    }
 
    void OverrideFreezeSizeAndAlignment(int64& Size, int32& Alignment) const override
    {
        Size = sizeof(*this);
    }
};