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

---

DirImageFileNames.h

Go to the documentation of this file.

//345678901234567890123456789012345678901234567890123456789012345678901234567890
// Obtain image filenames in given directory
//
// Author: Richard van Balen

#ifndef Impala_Util_DirImageFileNames_h
#define Impala_Util_DirImageFileNames_h

#include <string>
#include <vector>
#include "Basis/FileName.h"

#ifndef WIN32
#include <dirent.h>
#endif

namespace Impala {
namespace Util {

/*
std::string DirFindRaw(std::string dirName)
{
    WIN32_FIND_DATA FindFileData; 
    HANDLE          hFind = INVALID_HANDLE_VALUE; 

    dirName += "/*raw";

    hFind = FindFirstFile(dirName.c_str(), &FindFileData);
        if(hFind == INVALID_HANDLE_VALUE)
        return "";
    return FindFileData.cFileName;
}
*/

#ifdef WIN32

void DirImageFileNames(std::string dirName, std::vector<std::string*>& fNames)
{
    WIN32_FIND_DATA FindFileData; 
    HANDLE          hFind = INVALID_HANDLE_VALUE; 
    std::string*    fileName;

    fNames.clear();
    dirName += "/*";

    hFind = FindFirstFile(dirName.c_str(), &FindFileData);
        if(hFind == INVALID_HANDLE_VALUE)
        return;

    do {
        std::string ext = FileNameExt(FindFileData.cFileName,true);
        if (ext == "jpg" || ext == "jpeg" || ext == "png" || ext == "raw")
        {
            fileName = new std::string(FindFileData.cFileName);
            fNames.push_back(fileName);
        }
        } while(FindNextFile(hFind, &FindFileData) != 0);

    FindClose(hFind);
}

#else

void DirImageFileNames(std::string dirName, std::vector<std::string*>& fNames)
{
    DIR *dp;
    struct dirent *dirp;
    std::string* fileNameP;

    fNames.clear();

    if((dp  = opendir(dirName.c_str())) == NULL) {
        return;
    }


    while ((dirp = readdir(dp)) != NULL) {
        std::string fileName = std::string(dirp->d_name);
        std::string ext = FileNameExt(fileName,true);
        if (ext == "jpg" || ext == "jpeg" || ext == "png" || ext == "raw")
        {
            fileNameP = new std::string(fileName);
            fNames.push_back(fileNameP);
        }
    }
    closedir(dp);
    return;
}

#endif

} // namespace Util
} // namespace Impala
#endif

---