MetalTempAllocator.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "MetalRHI.h"
#include "MetalBuffer.h"
#import "MetalThirdParty.h"
#include "Stats/Stats.h"
 
class FMetalCommandBuffer;
class FMetalDevice;
/*
   Simple Temporary allocator that allocates from heaps
 */
class FMetalTempAllocator : public IMetalBufferAllocator
{
public:
    FMetalTempAllocator(FMetalDevice& InDevice, uint32_t InMinAllocationSize, uint32_t InTargetAllocationLimit, uint32_t InAlignment);
    
    FMetalBufferPtr Allocate(const uint32_t Size);
    void Cleanup();
    
    virtual void ReleaseBuffer(FMetalBuffer*) override {}
    
private:
    struct FTempBufferInfo
    {
        MTL::Buffer* Buffer;
        uint32_t Offset;
        uint32_t Size;
    };
    
    FMetalDevice& Device;
    TArray<FTempBufferInfo> Buffers;
    
    FCriticalSection AllocatorLock;
    
    TStatId TotalAllocationStat;
    
    uint32_t TotalAllocated = 0;
    uint32_t MinAllocationSize = 0;
    uint32_t TargetAllocationLimit = 0;
    uint32_t Alignment;
};