// Copyright Epic Games, Inc. All Rights Reserved.

#pragma once

#include "CoreMinimal.h"
#include "UObject/Object.h"
#include "Serialization/ArchiveUObject.h"

/**
 * Archive for counting memory usage.
 */
class FArchiveCountMem: public FArchiveUObject
{
public:
    FArchiveCountMem(UObject* Src, bool bFilterEditorOnly = false)
        : Num(0), Max(0)
    {
        ArIsCountingMemory   = true;
        ArIsFilterEditorOnly = bFilterEditorOnly;
        if (Src)
        {
            Src->Serialize(*this);
        }
    }
    SIZE_T GetNum()
    {
        return Num;
    }
    SIZE_T GetMax()
    {
        return Max;
    }
    void CountBytes(SIZE_T InNum, SIZE_T InMax)
    {
        Num += InNum;
        Max += InMax;
    }
    /**
     * Returns the name of the Archive.  Useful for getting the name of the package a struct or object
     * is in when a loading error occurs.
     *
     * This is overridden for the specific Archive Types
     **/
    virtual FString GetArchiveName() const { return TEXT("FArchiveCountMem"); }

protected:
    SIZE_T Num, Max;
};