MemoryUtil.hpp

#pragma once
 
#include <revolution.h>
#include <revolution/mem.h>
#include <cstring>
#include <size_t.h>
#include "JSystem/JKernel/JKRHeap.hpp"
#include "Game/SingletonHolder.hpp"
#include "Game/System/HeapMemoryWatcher.hpp"
 
class JKRHeap;
class JKRExpHeap;
class JKRSolidHeap;
 
namespace MR {
    void becomeCurrentHeap(JKRHeap *);
    JKRHeap* getCurrentHeap();
    bool isEqualCurrentHeap(JKRHeap *);
 
    void copyMemory(void *, const void *, u32);
    void fillMemory(void *, u8, u32);
    void zeroMemory(void *, u32);
    s32 calcCheckSum(const void *, u32);
 
    void* allocFromWPadHeap(u32);
    u8 freeFromWPadHeap(void *);
 
    JKRExpHeap* getStationedHeapNapa();
    JKRExpHeap* getStationedHeapGDDR3();
    JKRSolidHeap* getSceneHeapNapa();
    JKRSolidHeap* getSceneHeapGDDR3();
 
    JKRHeap* getAproposHeapForSceneArchive(f32);
 
    void adjustHeapSize(JKRExpHeap *, const char *);
 
    inline JKRSolidHeap* getAudHeap() {
        return SingletonHolder<HeapMemoryWatcher>::get()->mAudSystemHeap;
    }
 
    class CurrentHeapRestorer {
    public:
        CurrentHeapRestorer(JKRHeap *);
        ~CurrentHeapRestorer();
 
        JKRHeap* _0;
    };
 
    class NewDeleteAllocator {
    public:
        static void* alloc(MEMAllocator *, u32);
        static void free(MEMAllocator *, void *);
 
        static MEMAllocatorFunc sAllocatorFunc;
 
        static MEMAllocator sAllocator;
    };
 
    template<int T>
    class JKRHeapAllocator {
    public:
        static void* alloc(MEMAllocator *pAllocator, u32 size) {
            return JKRHeapAllocator<T>::sHeap->alloc(size, 0);
        }
 
        static void free(MEMAllocator *pAllocator, void *pData) {
            JKRHeapAllocator<T>::sHeap->free(pData);
        }
 
        static JKRHeap* sHeap;
    };
};