Added osgDB::getSortedDirectoryContents and osgDB::FileNameComparator to help with sorting directory contents into alphabetic and numerical order.

This commit is contained in:
Robert Osfield 2012-09-12 16:02:02 +00:00
parent cfe36876d4
commit 5d5cf26138
3 changed files with 62 additions and 0 deletions

View File

@ -81,6 +81,56 @@ extern OSGDB_EXPORT std::string getRealPath(const std::string& path);
/** Splits a path into elements between separators (including Windows' root, if any). */ /** Splits a path into elements between separators (including Windows' root, if any). */
extern OSGDB_EXPORT void getPathElements(const std::string& path, std::vector<std::string> & out_elements); extern OSGDB_EXPORT void getPathElements(const std::string& path, std::vector<std::string> & out_elements);
/** Functor for helping sort filename in alphabetical and numerical order when using in conjunction with std::sort.*/
struct FileNameComparator
{
inline bool operator() (const std::string& lhs, const std::string& rhs) const
{
std::string::size_type size_lhs = lhs.size();
std::string::size_type size_rhs = rhs.size();
std::string::size_type pos_lhs = 0;
std::string::size_type pos_rhs = 0;
while(pos_lhs<size_lhs && pos_rhs<size_rhs)
{
char c_lhs = lhs[pos_rhs];
char c_rhs = rhs[pos_rhs];
bool numeric_lhs = lhs[pos_lhs]>='0' && lhs[pos_lhs]<='9';
bool numeric_rhs = rhs[pos_rhs]>='0' && rhs[pos_rhs]<='9';
if (numeric_lhs && numeric_rhs)
{
std::string::size_type start_lhs = pos_lhs;
++pos_lhs;
while(pos_lhs<size_lhs && (lhs[pos_lhs]>='0' && lhs[pos_lhs]<='9')) ++pos_lhs;
std::string::size_type start_rhs = pos_rhs;
++pos_rhs;
while(pos_rhs<size_rhs && (rhs[pos_rhs]>='0' && rhs[pos_rhs]<='9')) ++pos_rhs;
if (pos_lhs<pos_rhs) return true;
else if (pos_rhs<pos_lhs) return false;
while(start_lhs<pos_lhs && start_rhs<pos_rhs)
{
if (lhs[start_lhs]<rhs[start_rhs]) return true;
if (lhs[start_lhs]>rhs[start_rhs]) return false;
++start_lhs;
++start_rhs;
}
}
else
{
if (c_lhs<c_rhs) return true;
else if (c_rhs<c_lhs) return false;
++pos_lhs;
++pos_rhs;
}
}
return pos_lhs<pos_rhs;
}
};
} }
#endif #endif

View File

@ -71,6 +71,11 @@ typedef std::vector<std::string> DirectoryContents;
* Returns an empty array on any error.*/ * Returns an empty array on any error.*/
extern OSGDB_EXPORT DirectoryContents getDirectoryContents(const std::string& dirName); extern OSGDB_EXPORT DirectoryContents getDirectoryContents(const std::string& dirName);
/** Return the contents of a directory, sorting the names into alphabetic and numberical order.
* Return value will contain filenames only, not absolute paths.
* Returns an empty array on any error.*/
extern OSGDB_EXPORT DirectoryContents getSortedDirectoryContents(const std::string& dirName);
/** Return the list of filenames that match the given filename with wildcards. /** Return the list of filenames that match the given filename with wildcards.
* Will only expand '*', and will not expand wildcards in directory, only in * Will only expand '*', and will not expand wildcards in directory, only in
* filename part of the given filename. * filename part of the given filename.

View File

@ -549,6 +549,13 @@ static void appendInstallationLibraryFilePaths(osgDB::FilePathList& filepath)
#endif // unix getDirectoryContexts #endif // unix getDirectoryContexts
osgDB::DirectoryContents osgDB::getSortedDirectoryContents(const std::string& dirName)
{
osgDB::DirectoryContents filenames = osgDB::getDirectoryContents(dirName);
std::sort(filenames.begin(), filenames.end(), osgDB::FileNameComparator());
return filenames;
}
osgDB::DirectoryContents osgDB::expandWildcardsInFilename(const std::string& filename) osgDB::DirectoryContents osgDB::expandWildcardsInFilename(const std::string& filename)
{ {