|| || || || || || || ||

---

DirectoryInfo.h

Go to the documentation of this file.

#ifndef Impala_Util_DirectoryInfo_h
#define Impala_Util_DirectoryInfo_h

#include <vector>
#include <string>
#include "Util/FileInfo.h"
#ifndef unix
#include <windows.h>
#endif


namespace Impala
{
namespace Util
{

class DirectoryInfo
{
public:
    DirectoryInfo(const std::string& name)
        : mName(name), mShortName(name), mFilter("*")
    {
    }

    DirectoryInfo(const std::string& name, const std::string& shortname)
        : mName(name), mShortName(shortname), mFilter("*")
    {

    }

    void
    SetFilter(const std::string& filter)
    {
        mFilter = filter;
    }

    void
    Scan();

    void
    Clear()
    {
        unsigned int i;
        for (i=0 ; i<mSubDirs.size() ; i++)
        {
            delete mSubDirs[i];
        }
        mSubDirs.clear();
        for (i=0 ; i<mFiles.size() ; i++)
        {
            delete mFiles[i];
        }
        mFiles.clear();
    }

    size_t
    GetFileCount()
    {
        return mFiles.size();
    }

    Util::FileInfo*
    GetFile(unsigned int index)
    {
        if(index > mFiles.size())
            throw std::logic_error("[DirectoryInfo::GetFile] index out of bounds");
        return mFiles[index];
    }

    size_t
    GetSubDirCount()
    {
        return mSubDirs.size();
    }

    DirectoryInfo*
    GetSubDir(unsigned int index)
    {
        if (index > mSubDirs.size())
            throw std::logic_error("[DirectoryInfo::GetSubDir] index out of bounds");
        return mSubDirs[index];
    }

    std::string GetName()
    {
        return mName;
    }

    std::string GetShortName()
    {
        return mShortName;
    }

    static void
    SetAutoScanSubDirs(bool value)
    {
        mAutoScanSubDirs = value;
    }

    void
    Print(std::ostream& os) const
    {
        os << "directory \"" << mName << "\"\n";
        for (unsigned int i=0 ; i<mFiles.size() ; ++i)
            os << "- " << mFiles[i]->mName << "\n";
    }

protected:

    std::vector<DirectoryInfo*> mSubDirs;
    std::vector<Util::FileInfo*> mFiles;
    std::string mName;
    std::string mShortName;
    std::string mFilter;
#ifndef unix
    WIN32_FIND_DATA mInfo;

    void CopyWinFindData(WIN32_FIND_DATA* dst, const WIN32_FIND_DATA* src) const;
#endif
    static bool mAutoScanSubDirs;
};

inline std::ostream&
operator<< (std::ostream& os, const DirectoryInfo& d)
{
    d.Print(os);
    return os;
}

} // namespace Util
} // namespace Impala

#endif Impala_Util_DirectoryInfo_h

---