usbloadergx/source/FileOperations/DirList.h

/****************************************************************************
 * Copyright (C) 2010
 * by Dimok
 *
 * This software is provided 'as-is', without any express or implied
 * warranty. In no event will the authors be held liable for any
 * damages arising from the use of this software.
 *
 * Permission is granted to anyone to use this software for any
 * purpose, including commercial applications, and to alter it and
 * redistribute it freely, subject to the following restrictions:
 *
 * 1. The origin of this software must not be misrepresented; you
 * must not claim that you wrote the original software. If you use
 * this software in a product, an acknowledgment in the product
 * documentation would be appreciated but is not required.
 *
 * 2. Altered source versions must be plainly marked as such, and
 * must not be misrepresented as being the original software.
 *
 * 3. This notice may not be removed or altered from any source
 * distribution.
 *
 * DirList Class
 * for WiiXplorer 2010
 ***************************************************************************/
#ifndef ___DIRLIST_H_
#define ___DIRLIST_H_

#include <vector>
#include <string>
#include <gctypes.h>

typedef struct
{
    char * FilePath;
    u64 FileSize;
    bool isDir;
} FileInfos;

class DirList
{
    public:
        //!Constructor
        //!\param path Path from where to load the filelist of all files
        //!\param filter A fileext that needs to be filtered
        //!\param flags search/filter flags from the enum
        DirList(const char * path, const char *filter = NULL, u32 flags = Files | Dirs);
        //!Destructor
        ~DirList();
        //! Load all the files from a directory
        bool LoadPath(const char * path, const char *filter = NULL, u32 flags = Files | Dirs);
        bool LoadPath(std::string &path, const char *filter = NULL, u32 flags = Files | Dirs);
        //! Get a filename of the list
        //!\param list index
        const char * GetFilename(int index);
        //! Get the a filepath of the list
        //!\param list index
        const char * GetFilepath(int index) { if(!valid(index)) return NULL; return FileInfo[index].FilePath; };
        //! Get the a filesize of the list
        //!\param list index
        u64 GetFilesize(int index) { if(!valid(index)) return 0; return FileInfo[index].FileSize; };
        //! Is index a dir or a file
        //!\param list index
        bool IsDir(int index) { if(!valid(index)) return 0; return FileInfo[index].isDir; };
        //! Get the filecount of the whole list
        int GetFilecount() { return FileInfo.size(); };
        //! Sort list by filepath
        void SortList();
        //! Custom sort command for custom sort functions definitions
        void SortList(bool (*SortFunc)(const FileInfos &a, const FileInfos &b));
		//! Get the index of the specified filename
		int GetFileIndex(const char *filename);
		//! Enum for search/filter flags
		enum
		{
		    Files = 0x01,
		    Dirs = 0x02,
		    CheckSubfolders = 0x08,
		};
    protected:
        //!Add a list entrie
        void AddEntrie(const char * folderpath, const char * filename, u64 filesize, bool isDir);
        //! Clear the list
        void ClearList();
        //! Check if valid pos is requested
        inline bool valid(int pos) { return (pos >= 0 && pos < (int) FileInfo.size()); };

        std::vector<FileInfos> FileInfo;
};

#endif