1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107
|
#include "multidircollection.hpp"
#include <boost/filesystem.hpp>
#include <components/debug/debuglog.hpp>
namespace Files
{
struct NameEqual
{
bool mStrict;
NameEqual (bool strict) : mStrict (strict) {}
bool operator() (const std::string& left, const std::string& right) const
{
if (mStrict)
return left==right;
std::size_t len = left.length();
if (len!=right.length())
return false;
for (std::size_t i=0; i<len; ++i)
{
char l = Misc::StringUtils::toLower (left[i]);
char r = Misc::StringUtils::toLower (right[i]);
if (l!=r)
return false;
}
return true;
}
};
MultiDirCollection::MultiDirCollection(const Files::PathContainer& directories,
const std::string& extension, bool foldCase)
: mFiles (NameLess (!foldCase))
{
NameEqual equal (!foldCase);
for (PathContainer::const_iterator iter = directories.begin();
iter!=directories.end(); ++iter)
{
if (!boost::filesystem::is_directory(*iter))
{
Log(Debug::Info) << "Skipping invalid directory: " << (*iter).string();
continue;
}
for (boost::filesystem::directory_iterator dirIter(*iter);
dirIter != boost::filesystem::directory_iterator(); ++dirIter)
{
boost::filesystem::path path = *dirIter;
if (!equal (extension, path.extension().string()))
continue;
std::string filename = path.filename().string();
TIter result = mFiles.find (filename);
if (result==mFiles.end())
{
mFiles.insert (std::make_pair (filename, path));
}
else if (result->first==filename)
{
mFiles[filename] = path;
}
else
{
// handle case folding
mFiles.erase (result->first);
mFiles.insert (std::make_pair (filename, path));
}
}
}
}
boost::filesystem::path MultiDirCollection::getPath (const std::string& file) const
{
TIter iter = mFiles.find (file);
if (iter==mFiles.end())
throw std::runtime_error ("file " + file + " not found");
return iter->second;
}
bool MultiDirCollection::doesExist (const std::string& file) const
{
return mFiles.find (file)!=mFiles.end();
}
MultiDirCollection::TIter MultiDirCollection::begin() const
{
return mFiles.begin();
}
MultiDirCollection::TIter MultiDirCollection::end() const
{
return mFiles.end();
}
}
|