SuperMarioOdysseyOnline/include/sead/heap/seadExpHeap.h
2022-06-16 14:33:18 -07:00

110 lines
3.5 KiB
C++

#pragma once
#include "heap/seadHeap.h"
#include "heap/seadMemBlock.h"
#include "prim/seadSizedEnum.h"
namespace sead
{
class ExpHeap : public Heap
{
SEAD_RTTI_OVERRIDE(ExpHeap, Heap)
public:
enum class AllocMode
{
FirstFit = 0,
BestFit = 1,
};
enum class FindFreeBlockMode
{
Auto = 0,
FromFreeList = 1,
ByIteratingMemBlock = 2,
};
// FIXME: incomplete
enum class FindMode
{
};
static ExpHeap* create(size_t size, const SafeString& name, Heap* parent, s32 alignment,
HeapDirection direction, bool);
static ExpHeap* create(void* address, size_t size, const SafeString& name, bool);
static ExpHeap* tryCreate(size_t size, const SafeString& name, Heap* parent, s32 alignment,
HeapDirection direction, bool);
static ExpHeap* tryCreate(void* address, size_t size, const SafeString& name, bool);
static size_t getManagementAreaSize(s32);
void destroy() override;
size_t adjust() override;
void* tryAlloc(size_t size, s32 alignment) override;
void free(void* ptr) override;
void* resizeFront(void* p_void, size_t size) override;
void* resizeBack(void* p_void, size_t size) override;
void* tryRealloc(void* ptr, size_t size, s32 alignment) override;
void freeAll() override;
uintptr_t getStartAddress() const override;
uintptr_t getEndAddress() const override;
size_t getSize() const override;
size_t getFreeSize() const override;
size_t getMaxAllocatableSize(int alignment) const override;
bool isInclude(const void* p_void) const override;
bool isEmpty() const override;
bool isFreeable() const override;
bool isResizable() const override;
bool isAdjustable() const override;
void dump() const override;
void dumpYAML(WriteStream& stream, int i) const override;
void genInformation_(hostio::Context* context) override;
virtual s32 destroyAndGetAllocatableSize(s32);
virtual void setFindFreeBlockMode(FindFreeBlockMode mode);
AllocMode getAllocMode() const { return mAllocMode; }
void setAllocMode(AllocMode mode) { mAllocMode = mode; }
void dumpFreeList() const;
void dumpUseList() const;
void checkFreeList() const;
bool tryCheckFreeList() const;
void checkUseList() const;
bool tryCheckUseList() const;
protected:
ExpHeap(const SafeString& name, Heap* parent, void* address, size_t size,
HeapDirection direction, bool);
~ExpHeap() override;
static void doCreate(ExpHeap*, Heap*);
static void createMaxSizeFreeMemBlock_(ExpHeap*);
MemBlock* findFreeMemBlockFromHead_(size_t, FindMode) const;
MemBlock* findFreeMemBlockFromHead_(size_t, s32, FindMode) const;
MemBlock* findFreeMemBlockFromTail_(size_t, FindMode) const;
MemBlock* findFreeMemBlockFromTail_(size_t, s32, FindMode) const;
MemBlock* findLastMemBlockIfFree_();
MemBlock* findFirstMemBlockIfFree_();
void pushToUseList_(MemBlock*);
void pushToFreeList_(MemBlock*);
size_t adjustBack_();
size_t adjustFront_();
MemBlock* allocFromHead_(size_t);
MemBlock* allocFromHead_(size_t, s32);
MemBlock* allocFromTail_(size_t);
MemBlock* allocFromTail_(size_t, s32);
static s32 compareMemBlockAddr_(const MemBlock*, const MemBlock*);
SizedEnum<AllocMode, u8> mAllocMode;
SizedEnum<FindFreeBlockMode, u8> mFindFreeBlockMode;
MemBlockList mFreeList;
MemBlockList mUseList;
};
} // namespace sead