// Copyright Epic Games, Inc. All Rights Reserved.
#pragma once

#include <zencore/memory/fmalloc.h>
#include <zencore/memory/memorytrace.h>

namespace zen {

class FTraceMalloc : public FMalloc
{
public:
	FTraceMalloc(FMalloc* InMalloc);
	virtual ~FTraceMalloc();

	virtual void* Malloc(size_t Count, uint32_t Alignment) override;
	virtual void* Realloc(void* Original, size_t Count, uint32_t Alignment) override;
	virtual void  Free(void* Original) override;

	virtual void OnMallocInitialized() override { WrappedMalloc->OnMallocInitialized(); }

	FMalloc* WrappedMalloc;
};

}  // namespace zen