generic-library/opencv
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

improved some methods in the Directory class

Browse Source
This commit is contained in:
Andrey Morozov 2011-10-04 16:46:55 +00:00
parent d58ff0f478
commit f366553a80
2 changed files with 15 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
modules/contrib/include/opencv2/contrib/contrib.hpp
View File

@ -609,9 +609,9 @@ namespace cv
class CV_EXPORTS Directory class CV_EXPORTS Directory
{ {
public: public:
static std::vector<std::string> GetListFiles ( const string& directoryName, bool addPath = true ); static std::vector<std::string> GetListFiles ( const std::string& path, const std::string & exten = "*", bool addPath = true );
static std::vector<std::string> GetListFilesR ( const string& directoryName, bool addPath = true ); static std::vector<std::string> GetListFilesR ( const std::string& path, const std::string & exten = "*", bool addPath = true );
static std::vector<std::string> GetListFolders( const string& directoryName, bool addPath = true ); static std::vector<std::string> GetListFolders( const std::string& path, const std::string & exten = "*", bool addPath = true );
}; };
} }

24
modules/contrib/src/inputoutput.cpp
View File

@ -10,16 +10,16 @@
namespace cv namespace cv
{ {
std::vector<std::string> Directory::GetListFiles( const string& directoryName, bool addPath ) std::vector<std::string> Directory::GetListFiles( const std::string& path, const std::string & exten, bool addPath )
{ {
std::vector<std::string> list; std::vector<std::string> list;
list.clear(); list.clear();
std::string path = directoryName + "/*"; std::string path_f = path + "/" + exten;
#ifdef WIN32 #ifdef WIN32
WIN32_FIND_DATA FindFileData; WIN32_FIND_DATA FindFileData;
HANDLE hFind; HANDLE hFind;
hFind = FindFirstFile((LPCSTR)path.c_str(), &FindFileData); hFind = FindFirstFile((LPCSTR)path_f.c_str(), &FindFileData);
if (hFind == INVALID_HANDLE_VALUE) if (hFind == INVALID_HANDLE_VALUE)
{ {
return list; return list;
@ -35,7 +35,7 @@ namespace cv
FindFileData.dwFileAttributes == FILE_ATTRIBUTE_READONLY) FindFileData.dwFileAttributes == FILE_ATTRIBUTE_READONLY)
{ {
if (addPath) if (addPath)
list.push_back(directoryName + "/" + FindFileData.cFileName); list.push_back(path + "/" + FindFileData.cFileName);
else else
list.push_back(FindFileData.cFileName); list.push_back(FindFileData.cFileName);
} }
@ -62,16 +62,16 @@ namespace cv
return list; return list;
} }
std::vector<std::string> Directory::GetListFolders( const string& directoryName, bool addPath ) std::vector<std::string> Directory::GetListFolders( const std::string& path, const std::string & exten, bool addPath )
{ {
std::vector<std::string> list; std::vector<std::string> list;
std::string path = directoryName + "/*"; std::string path_f = path + "/" + exten;
list.clear(); list.clear();
#ifdef WIN32 #ifdef WIN32
WIN32_FIND_DATA FindFileData; WIN32_FIND_DATA FindFileData;
HANDLE hFind; HANDLE hFind;
hFind = FindFirstFile((LPCSTR)path.c_str(), &FindFileData); hFind = FindFirstFile((LPCSTR)path_f.c_str(), &FindFileData);
if (hFind == INVALID_HANDLE_VALUE) if (hFind == INVALID_HANDLE_VALUE)
{ {
return list; return list;
@ -85,7 +85,7 @@ namespace cv
strcmp(FindFileData.cFileName, "..") != 0) strcmp(FindFileData.cFileName, "..") != 0)
{ {
if (addPath) if (addPath)
list.push_back(directoryName + "/" + FindFileData.cFileName); list.push_back(path + "/" + FindFileData.cFileName);
else else
list.push_back(FindFileData.cFileName); list.push_back(FindFileData.cFileName);
} }
@ -113,16 +113,16 @@ namespace cv
return list; return list;
} }
std::vector<std::string> Directory::GetListFilesR ( const string& directoryName, bool addPath ) std::vector<std::string> Directory::GetListFilesR ( const std::string& path, const std::string & exten, bool addPath )
{ {
std::vector<std::string> list = Directory::GetListFiles(directoryName, addPath); std::vector<std::string> list = Directory::GetListFiles(path, exten, addPath);
std::vector<std::string> dirs = Directory::GetListFolders(directoryName, addPath); std::vector<std::string> dirs = Directory::GetListFolders(path, exten, addPath);
std::vector<std::string>::const_iterator it; std::vector<std::string>::const_iterator it;
for (it = dirs.begin(); it != dirs.end(); ++it) for (it = dirs.begin(); it != dirs.end(); ++it)
{ {
std::vector<std::string> cl = Directory::GetListFiles(*it, addPath); std::vector<std::string> cl = Directory::GetListFiles(*it, exten, addPath);
list.insert(list.end(), cl.begin(), cl.end()); list.insert(list.end(), cl.begin(), cl.end());
} }