UDir.h

Go to the documentation of this file.

//    Copyright (C) 2010-2014 Dylan Blair
//
//    email: dblair@alumni.cs.utexas.edu
//
//    This library is free software; you can redistribute it and/or
//    modify it under the terms of the GNU Lesser General Public
//    License as published by the Free Software Foundation; either
//    version 2.1 of the License, or (at your option) any later version.
//
//    This library is distributed in the hope that it will be useful,
//    but WITHOUT ANY WARRANTY; without even the implied warranty of
//    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
//    Lesser General Public License for more details.
//
//    You should have received a copy of the GNU Lesser General Public
//    License along with this library; if not, write to the Free Software
//    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

#ifndef UDIR_H
#define UDIR_H

#include <list>
#include <string>

#ifdef _WIN32
#ifndef WIN32_LEAN_AND_MEAN
#define WIN32_LEAN_AND_MEAN
#endif
#include <windows.h>

namespace OpenSkyNet {
    namespace Utils {
        inline void GetFileNamesInDir(const std::string& dir_, std::list<std::string>& fileNames_) {
            fileNames_.clear();

            WIN32_FIND_DATA findData;
            HANDLE hFind = FindFirstFile((dir_ + "*").c_str(), &findData);
            if (hFind != INVALID_HANDLE_VALUE) {
                std::string fileName(findData.cFileName);
                if (fileName != "." && fileName != "..")
                    fileNames_.push_back(fileName);
                while (FindNextFile(hFind, &findData)) {
                    std::string fileName(findData.cFileName);
                    if (fileName != "." && fileName != "..")
                        fileNames_.push_back(fileName);
                }
                FindClose(hFind);
            }
        }
    }
}
#else
#include <dirent.h>

namespace OpenSkyNet {
    namespace Utils {
        inline void GetFileNamesInDir(const std::string& dir_, std::list<std::string>& fileNames_) {
            fileNames_.clear();

            dirent** nameList;
            int numNames = scandir(dir_.c_str(), &nameList, 0, alphasort);
            if (numNames > 0) {
                for (int i = 0; i < numNames; ++i) {
                    std::string fileName(nameList[i]->d_name);
                    if (fileName != "." && fileName != "..")
                        fileNames_.push_back(fileName);
                    free(nameList[i]);
                }
                free(nameList);
            }
        }
    }
}
#endif

#endif //UDIR_H