d_res.hpp

#pragma once
#include <types.h>
#include <lib/rvl/arc/ARC.h>
#include <lib/egg/heap/eggHeap.hpp>
#include <lib/egg/heap/eggFrmHeap.hpp>
#include <lib/egg/archive.hpp>
#include <game/mLib/m_dvd.h>
 
class dRes_c {
public:
 
    class callback_c {
    public:
        virtual void init(const char *name) = 0;
 
        virtual void *execute(void *data, u32 folderSig, const char* path) = 0;
    };
 
    class searchCallback_c {
    public:
        searchCallback_c(callback_c *callback, u8 **files, int numFiles, int fileIdx, u32 folderSig) :
            mpCallback(callback),
            mpFiles(files),
            mNumFiles(numFiles),
            mFileIdx(fileIdx),
            mFolderSig(folderSig) {
        }
 
        static void callback(void *cbInfo, void *file, const ARCDirEntry *dirEntry, const char *path);
 
    private:
        void callback(void *file, const ARCDirEntry *dirEntry, const char *path);
 
        callback_c *mpCallback; 
        u8 **mpFiles; 
        int mNumFiles; 
        unsigned int mFileIdx; 
        u32 mFolderSig; 
    };
 
    class info_c {
    public:
 
        enum LOAD_STATUS_e {
            LOAD_ERROR = -1, 
            LOAD_SUCCESS = 0, 
            LOAD_IN_PROGRESS = 1, 
        };
 
        info_c(); 
        ~info_c(); 
 
        bool cleanup();
 
        bool set(const char *arcName, const char *containingFolder, u8 allocDirection, EGG::Heap *heap);
 
        LOAD_STATUS_e setRes(callback_c *callback);
 
        const char *getName() const { return mName; }
        int getRefCount() const { return mRefCount; }
 
        mDvd_mountMemArchive_c *getDvdCmd() const { return mpDvdCmd; }
 
        EGG::Archive *getArchive() const { return mpArchive; }
 
        void incRefCount() { mRefCount++; } 
        void decRefCount() { mRefCount--; } 
 
    private:
 
        LOAD_STATUS_e loadRes(callback_c *callback);
 
        char mName[0x20]; 
        u16 mRefCount; 
        mDvd_mountMemArchive_c *mpDvdCmd; 
        EGG::Archive *mpArchive; 
        EGG::Heap *mpMountHeap; 
        void *mpArcBinary; 
        int mArchiveSize; 
 
        EGG::FrmHeap *mDataHeap;
 
        u8 **mpFiles; 
    };
 
    dRes_c(); 
    ~dRes_c(); 
 
    bool init(u16 maxCount, callback_c *callback);
 
    bool setRes(const char *arcName, const char *containingFolder, u8 allocDir, EGG::Heap *heap);
 
    bool deleteRes(const char *arcName);
 
    void *getRes(const char *arcName, const char *resPath) const;
 
    void *getRes(const char *arcName, const char *resPath, unsigned long *size) const;
 
    void *getRes(const char *arcName, const char *resPath, unsigned long *size, int *compressionType) const;
 
    void *getResSilently(const char *arcName, const char *resPath) const;
 
    void *getResSilently(const char *arcName, const char *resPath, unsigned long *size) const;
 
    bool syncAllRes();
 
private:
    info_c *getResInfo(const char *arcName) const;
 
    info_c *getResInfoLoaded(const char *arcName) const;
 
    info_c *newResInfo();
 
public:
    static void copyRes(const void *from, void *to, int size);
 
    static void copyRes(const void *from, void *to, int size, int compressionType);
 
private:
    info_c *mpArcInfos; 
    u16 mNumArcs; 
    callback_c *mpCallback; 
 
    static void (*mSetCallback)(const char *arcName, EGG::Heap *heap);
};